From a69ec9f144a3fc15cdc413202604e0c537a5a7f9 Mon Sep 17 00:00:00 2001 From: JishinMaster Date: Fri, 24 Apr 2020 18:50:01 +0200 Subject: [PATCH] Corrected TVM autotuning on GPU (#5432) Added missing "tir" in tvm.tir.analysis.verify_gpu_code(f, kwargs) --- python/tvm/autotvm/measure/measure_methods.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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