[microTVM] Allow multiple runners in tuning micro models with meta-schedule#13811
Conversation
|
Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment. Generated by tvm-bot |
2d437ce to
9676759
Compare
| platform=platform, | ||
| options=project_options, | ||
| session_timeout_sec=120, | ||
| serial_numbers=["0", "1"], |
There was a problem hiding this comment.
To be able to run this test in CI with physical devices, we need to pass the serial numbers. I suggest to reuse the serial_number pytest argument. We could use this argument in the format of SERIAL_1,SERIAL_2,...,SERIAL_N when we want to tune with N number of devices and parse the string in serial_number pytest argument function and return a list of serial numbers. Then we could pass the serial numbers to this function.
Then in this function you could check if serial_number arg is an empty list and set it to ["0", "1"] if it's empty.
There was a problem hiding this comment.
Thanks @mehrdadh, made the change to use the serial number when available. PTAL.
mehrdadh
left a comment
There was a problem hiding this comment.
Thanks for adding this support!
Looks good, we only need one change for the CI
9676759 to
cb61fc4
Compare
abddd94 to
a186551
Compare
|
@tvm-bot rerun |
…hedule (apache#13811) This PR adds support for running candidate projects on multiple physical boards when using meta-schedule to tune micro models, and helps reducing the tuning time.
This PR adds support for running candidate projects on multiple physical boards when using meta-schedule to tune micro models, and helps reducing the tuning time.