-
Notifications
You must be signed in to change notification settings - Fork 631
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
Rewrite benchmarking scripts with Python. #5082
Conversation
This causes a lot of issues....
|
||
import configuration | ||
|
||
PATTERN = re.compile(r"BM_(\w+)(.+)/real_time(\s+) (?P<ms>[.0-9]+) ms") |
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 still think we should be using the more structured output instead of parsing the human-readable output... but this is ok for now
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.
Yes, I think we can let it report a csv or something. I keep this because this is the same behavior as before. I will consider this.
Co-authored-by: Geoffrey Martin-Noble <[email protected]>
* 9ae2e49 Rename compiler2 to compiler (#5113) * 4168916 [CodeGen] Add a pattern to canonicalize HAL interface load/store (#5103) * 5b6e7d1 Rewrite benchmarking scripts with Python. (#5082) * 36912bc Merge google -> main (#5117) * ec9f244 Merge branch 'main' into google-to-main * e479591 Disable native-py check from buildifier action * 02bb838 Enable concat, gather and torch_index_select on Linalg on tensors path. (#5053) * 31c71df Integrate MLIR-EmitC at iml130/mlir-emitc@3483f16 (#5114) * 2839b02 Fix missing <cstddef> header file. (#5112) * 5670121 Fixing flow.dispatch.workgroups tied operands when removing args. (#5108) COPYBARA_INTEGRATE_REVIEW=#5125 from not-jenni:main-to-google 4936e55 PiperOrigin-RevId: 363244672
* 9ae2e49 Rename compiler2 to compiler (#5113) * 4168916 [CodeGen] Add a pattern to canonicalize HAL interface load/store (#5103) * 5b6e7d1 Rewrite benchmarking scripts with Python. (#5082) * 36912bc Merge google -> main (#5117) * ec9f244 Merge branch 'main' into google-to-main * e479591 Disable native-py check from buildifier action * 02bb838 Enable concat, gather and torch_index_select on Linalg on tensors path. (#5053) * 31c71df Integrate MLIR-EmitC at iml130/mlir-emitc@3483f16 (#5114) * 2839b02 Fix missing <cstddef> header file. (#5112) * 5670121 Fixing flow.dispatch.workgroups tied operands when removing args. (#5108) PiperOrigin-RevId: 363244672
This moves the config to utils.py, which makes the compilation script and benchmarking script share the same config. A ModelBenchmarkInfo contains a phone list. This allows us to set different config for different models. This is useful for doing batch mode benchmarking.