-
Notifications
You must be signed in to change notification settings - Fork 25
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
[tuner] Use JSON for benchmark output #256
[tuner] Use JSON for benchmark output #256
Conversation
tuner/tuner/libtuner_test.py
Outdated
path_config = libtuner.PathConfig() | ||
object.__setattr__(path_config, "output_json", "output.json") |
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.
Tests shouldn't write files in the current working directory. We can create a temporary directory for tests artifacts if needed.
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.
This hasn't been resolved
c965b6d
to
a49ef9c
Compare
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.
Nice, this looks good but we should also account for the time unit
a49ef9c
to
761bc2e
Compare
tuner/tuner/libtuner_test.py
Outdated
path_config = libtuner.PathConfig() | ||
object.__setattr__(path_config, "output_json", "output.json") |
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.
This hasn't been resolved
550f721
to
30ee7d7
Compare
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.
Overall looks good! Just a few minor comments. I will try it out locally before approving.
30ee7d7
to
6c250d9
Compare
6c250d9
to
20ae718
Compare
20ae718
to
49ec0ad
Compare
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.
I ran this on my machine and it works great. Thank you for this contribution!
### Notes - Adds `extract_benchmark_from_run_result` method to help in fetching the "benchmarks" data - Updates `IREEBenchmarkResult` model to reflect that the result is no longer stored as string but as a list of benchmarks - Updates the parsing of dispatches and models to read from Json ### Testing - Updates tests to verify that `get_mean_time` functions as expected - Updates tests to verify that Json data is properly parsed and processed
Notes
extract_benchmark_from_run_result
method to help in fetching the "benchmarks" dataIREEBenchmarkResult
model to reflect that the result is no longer stored as string but as a list of benchmarksTesting
get_mean_time
functions as expected