[GitHub] [tvm] jcf94 commented on a change in pull request #7632: [AutoScheduler] Fix incorrectly array context device and hide info at the beginning

[GitBox <gi...@apache.org>]

jcf94 commented on a change in pull request #7632:
URL: https://github.com/apache/tvm/pull/7632#discussion_r592033976



##########
File path: python/tvm/auto_scheduler/measure.py
##########
@@ -1079,7 +1079,7 @@ def _timed_rpc_run(
                 if arg in tensor_input_map:
                     tensor_name = tensor_input_map[arg]
                     if tensor_name in task_input_names:
-                        args.append(get_task_input_buffer(inp.task.workload_key, tensor_name))
+                        args.append(ndarray.array(get_task_input_buffer(inp.task.workload_key, tensor_name), ctx))

Review comment:
       Thanks!
   Let's add an extra UT to `tests/python/unittest/test_auto_scheduler_measure.py`, copy the `test_measure_special_inputs_map_by_name()` and modifiy the runner to a `LocalRPCMeasureContext().runner`.




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org