Skip to content

Commit 9687307

Browse files
authored
Corrected TVM autotuning on GPU (#5432)
Added missing "tir" in tvm.tir.analysis.verify_gpu_code(f, kwargs)
1 parent 9a989e5 commit 9687307

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tvm/autotvm/measure/measure_methods.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -615,7 +615,7 @@ def gpu_verify_pass(**kwargs):
615615
This pass will check memory usage and number of threads per block.
616616
"""
617617
def verify_pass(f, *_):
618-
valid = tvm.analysis.verify_gpu_code(f, kwargs)
618+
valid = tvm.tir.analysis.verify_gpu_code(f, kwargs)
619619
if not valid:
620620
raise InstantiationError("Skipped because of invalid gpu kernel")
621621
return f

0 commit comments

Comments
 (0)