[Autotvm] Fix autotvm customized template (apache#5034)

* init * fix template * tweak naming
tqchen · Mar 12, 2020 · 40f7116 · 40f7116
1 parent 38da791
commit 40f7116
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/autotvm/tune_relay_vta.py b/tutorials/autotvm/tune_relay_vta.py
@@ -310,7 +310,7 @@ def my_clip(x, a_min, a_max):
     # init autotvm env to register VTA operator
     TaskExtractEnv()
 
-    @autotvm.register_customized_task("conv2d_packed.vta")
+    @autotvm.template("conv2d_packed.vta")
     def _topi_nn_conv2d(*args, **kwargs):
         assert not kwargs, "Do not support kwargs in template function call"
         A, W = args[:2]