[AutoScheduler] Fix custom build func in PopenWorker

apache · Sep 7, 2021 · 3e1b60b · 3e1b60b
1 parent 9b034d7
commit 3e1b60b
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/python/tvm/auto_scheduler/measure.py b/python/tvm/auto_scheduler/measure.py
@@ -661,10 +661,6 @@ def local_build_worker(args):
         The build result of this Builder thread.
     """
     inp, build_func, verbose = args
-    assert build_func == BuildFunc.name, (
-        "BuildFunc.name: " + BuildFunc.name + ", but args is: " + build_func
-    )
-    build_func = BuildFunc.build_func
 
     return _local_build_worker(inp, build_func, verbose)
 
@@ -701,7 +697,7 @@ def local_builder_build(inputs, timeout, n_parallel, build_func="default", verbo
         [
             (
                 i.serialize(),
-                build_func,
+                BuildFunc.build_func,
                 verbose,
             )
             for i in inputs