skip random_fill when a tuning workload is from scatter

This reverts commit 1fed883.

python/tvm/autotvm/measure/measure_methods.py

@@ -561,9 +561,9 @@ def run_through_rpc(
                 "Please make sure USE_RANDOM is ON in the config.cmake " "on the remote devices"
             )
         args = [nd.empty(x[0], dtype=x[1], ctx=ctx) for x in build_result.arg_info]
-        for arg in args:
-            random_fill(arg)
-        ctx.sync()
+        if "scatter" not in measure_input.task.name:
+            for arg in args:
+                random_fill(arg)
 
         costs = time_f(*args).results
 

python/tvm/topi/cuda/scatter.py

@@ -461,8 +461,7 @@ 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.define_knob("dummy", [1])
-    cfg.add_flop(1) # dummy value to satisfy AutoTVM
+    cfg.add_flop(1)  # A dummy value to satisfy AutoTVM
 
     out = te.extern(
         [out_shape],
@@ -593,8 +592,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)
+    cfg.add_flop(1)  # A dummy value to satisfy AutoTVM
 
     out_shape = data.shape
     out_buf = tvm.tir.decl_buffer(out_shape, data.dtype, "out_buf")

tutorials/autotvm/test_scatter.py

@@ -89,7 +89,7 @@ def tune_tasks(
 def tune_and_evaluate(tuning_opt):
     # extract workloads from relay program
     print("Extract tasks...")
-    size = (5000, )
+    size = (10000, )
     dshape = ishape = size
     axis = 0
     mod, params = simple_mod(size, size, axis)