-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[microTVM] Allow multiple runners in tuning micro models with meta-schedule #13811
[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
tests/micro/zephyr/test_ms_tuning.py
Outdated
platform=platform, | ||
options=project_options, | ||
session_timeout_sec=120, | ||
serial_numbers=["0", "1"], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @mehrdadh, made the change to use the serial number when available. PTAL.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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.