Corrected TVM autotuning on GPU (apache#5432)

Added missing "tir" in tvm.tir.analysis.verify_gpu_code(f, kwargs)
trevor-m · Jun 8, 2020 · a69ec9f · a69ec9f
1 parent 5b73599
commit a69ec9f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/autotvm/measure/measure_methods.py b/python/tvm/autotvm/measure/measure_methods.py
@@ -615,7 +615,7 @@ def gpu_verify_pass(**kwargs):
     This pass will check memory usage and number of threads per block.
     """
     def verify_pass(f, *_):
-        valid = tvm.analysis.verify_gpu_code(f, kwargs)
+        valid = tvm.tir.analysis.verify_gpu_code(f, kwargs)
         if not valid:
             raise InstantiationError("Skipped because of invalid gpu kernel")
         return f