-
Notifications
You must be signed in to change notification settings - Fork 385
refactor accuracy eval script to be organized by hardware #3472
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
Conversation
|
Stack from ghstack (oldest at bottom): |
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3472
Note: Links to docs will display an error until the docs builds have been completed. This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Summary: We have a many-to-many mapping of recipe to device it can run on. Reorganizing the accuracy script to reflect this, since we don't have a machine that can run all the workflows. Test Plan: ``` // test on h100 benchmarks/quantization/eval_accuracy_for_readme.sh h100 ``` Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: 3c697ac ghstack-comment-id: 3633684836 Pull-Request: #3472
| QUANT_RECIPES_ALL=( | ||
| # no quantization (baseline) | ||
| "None" | ||
| "float8_rowwise" |
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.
Just checking, is this the only float8 recipe we want to showcase?
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 is covering all of the server GPU recipes from https://github.com/pytorch/ao/tree/main?tab=readme-ov-file#stable-workflows, and leaving out all other recipes
Summary:
We have a many-to-many mapping of recipe to device it can run on.
Reorganizing the accuracy script to reflect this, since we don't have a
machine that can run all the workflows.
Test Plan:
Reviewers:
Subscribers:
Tasks:
Tags: