masahi commented on a change in pull request #7233:
URL: https://github.com/apache/tvm/pull/7233#discussion_r559871247
##########
File path: python/tvm/autotvm/measure/measure_methods.py
##########
@@ -560,10 +561,11 @@ def run_through_rpc(
raise AttributeError(
"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()
+ args = [nd.array(np.zeros(x[0], dtype=x[1]), ctx=ctx) for x in
build_result.arg_info]
+ if "scatter" not in measure_input.task.name:
+ # the index tensor of scatter op cannot be randomly initialized
+ for arg in args:
+ random_fill(arg)
Review comment:
oops bad merge, fixed. thanks
----------------------------------------------------------------
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:
[email protected]
