Update python/tvm/auto_scheduler/measure.py

apache · Sep 22, 2021 · 80ee67c · 80ee67c
1 parent 80df07a
commit 80ee67c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/auto_scheduler/measure.py b/python/tvm/auto_scheduler/measure.py
@@ -920,7 +920,7 @@ def _timed_eval_func(
                     random_fill(empty_array)
                     loc_args.append(empty_array)
                 else:
-                    loc_args.append(ndarray.array(arg))
+                    loc_args.append(ndarray.array(args[idx], dev))
             dev.sync()
             costs = time_f(*loc_args).results
         # pylint: disable=broad-except