diff --git a/python/tvm/autotvm/measure/measure_methods.py b/python/tvm/autotvm/measure/measure_methods.py index bddb42034a50..6533e75eef93 100644 --- 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