try adding dummy knob

neo-ai · Jan 11, 2021 · 1fed883 · 1fed883
1 parent 990c5b9
commit 1fed883
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/topi/cuda/scatter.py b/python/tvm/topi/cuda/scatter.py
@@ -461,7 +461,8 @@ def update_func(dst_ptr, dst_index, update):
     out_shape = data.shape
     out_buf = tvm.tir.decl_buffer(out_shape, data.dtype, "out_buf")
 
-    cfg.add_flop(1)
+    cfg.define_knob("dummy", [1])
+    cfg.add_flop(1) # dummy value to satisfy AutoTVM
 
     out = te.extern(
         [out_shape],
@@ -592,6 +593,7 @@ def scatter_via_sort(cfg, data, indices, updates, axis=0):
     assert axis == 0 and len(data.shape) == 1, "sorting based scatter only supported for 1d input"
     assert is_thrust_available(), "Thrust is required for this op"
 
+    cfg.define_knob("dummy", [1])
     cfg.add_flop(1)
 
     out_shape = data.shape