From 96f43c262d9d2031efbe20645cc1c35f525f0d19 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Mon, 6 Sep 2021 20:27:20 -0400 Subject: [PATCH] Add assertion --- python/tvm/auto_scheduler/measure.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/tvm/auto_scheduler/measure.py b/python/tvm/auto_scheduler/measure.py index 560d4920480bb..5e42dd090f2f5 100644 --- a/python/tvm/auto_scheduler/measure.py +++ b/python/tvm/auto_scheduler/measure.py @@ -689,6 +689,9 @@ def local_builder_build(inputs, timeout, n_parallel, build_func="default", verbo res : List[BuildResult] The build results of these MeasureInputs. """ + assert build_func == BuildFunc.name, ( + "BuildFunc.name: " + BuildFunc.name + ", but args is: " + build_func + ) executor = PopenPoolExecutor( n_parallel, timeout, reset_global_scope, (AutotvmGlobalScope.current,) )