Skip to content

T
tic
Unverified Commit 96873076 by JishinMaster Committed by GitHub
Options

Corrected TVM autotuning on GPU (#5432) 

Added missing "tir" in tvm.tir.analysis.verify_gpu_code(f, kwargs)
parent 9a989e57
Showing with 1 additions and 1 deletions
python/tvm/autotvm/measure/measure_methods.py
...@@ -615,7 +615,7 @@ def gpu_verify_pass(**kwargs): ...@@ -615,7 +615,7 @@ def gpu_verify_pass(**kwargs):
This pass will check memory usage and number of threads per block. This pass will check memory usage and number of threads per block.
""" """
def verify_pass(f, *_): def verify_pass(f, *_):
valid = tvm.analysis.verify_gpu_code(f, kwargs) valid = tvm.tir.analysis.verify_gpu_code(f, kwargs)
if not valid: if not valid:
raise InstantiationError("Skipped because of invalid gpu kernel") raise InstantiationError("Skipped because of invalid gpu kernel")
return f return f
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment