-
Notifications
You must be signed in to change notification settings - Fork 25.7k
[quant][pt2e] Add transform_for_annotation method in Quantizer #113115
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
Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/113115
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 9d1b047 with merge base 84d64d7 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: 8aca038 Pull Request resolved: #113115
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.
Left a comment for adding test to xnnpack quantizer but feel free to defer it. and left a update to the comment about transform_for_annotation
…izer" Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: 602e42c Pull Request resolved: #113115
| # For example quantizer can | ||
| # a) decompose a compound operator like scaled dot product attention, | ||
| # into bmm and softmax if quantizer knows how to quantize bmm/softmax but not sdpa | ||
| # or b) transform scalars to tensor to allow quantizing scalares. |
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.
| # or b) transform scalars to tensor to allow quantizing scalares. | |
| # or b) transform scalars to tensor to allow quantizing scalars. |
…izer" Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
|
@jerryzh168 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
…izer" Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D51141080](https://our.internmc.facebook.com/intern/diff/D51141080) [ghstack-poisoned]
Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: 5a7995d Pull Request resolved: #113115
|
@jerryzh168 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
Summary: X-link: pytorch/pytorch#113115 Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Differential Revision: D51141080
|
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
|
@pytorchbot revert -m "breaking executorch tests" -c ghfirst |
|
@pytorchbot successfully started a revert job. Check the current status here. |
|
Can't revert PR that was landed via phabricator as D51141080. Please revert by going to the internal diff and clicking Unland. |
…ch#113115) Summary: Adding the method so that people can do some transformations before annotation to make the graph easier to annotate Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D51141080](https://our.internmc.facebook.com/intern/diff/D51141080) Pull Request resolved: pytorch#113115 Approved by: https://github.com/kimishpatel
Stack from ghstack (oldest at bottom):
Summary:
Adding the method so that people can do some transformations before annotation to make the graph easier to annotate
Test Plan:
python test/test_quantization.py TestQuantizePT2E.test_transform_for_annotation
Reviewers:
Subscribers:
Tasks:
Tags:
Differential Revision: D51141080