tkonolige commented on a change in pull request #10123:
URL: https://github.com/apache/tvm/pull/10123#discussion_r796187587
##########
File path: python/tvm/auto_scheduler/measure.py
##########
@@ -1257,6 +1276,8 @@ def rpc_runner_run(
This is only has effect on CPU task.
verbose: int = 1
Verbosity level. 0 for silent, 1 to output information during program
measuring.
+ device: int = 0
+ Which device to run on if multiople are available.
Review comment:
```suggestion
Which device to run on if multiple are available.
```
##########
File path: tests/python/relay/test_auto_scheduler_tuning.py
##########
@@ -36,7 +36,7 @@ def tune_network(network, target):
log_file = fp.name
# Tuning
- measure_ctx = auto_scheduler.LocalRPCMeasureContext(timeout=60)
+ measure_ctx = auto_scheduler.LocalRPCMeasureContext(timeout=60,
device=0)
Review comment:
It would be nice if there was some way to test this with device != 0,
but I can't think of a good way.
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
