-
Notifications
You must be signed in to change notification settings - Fork 25.7k
[quant][pt2e] Add quant API re-entrant test #110125
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: Add the test to make sure we can call the quantize API multiple times Test Plan: python test/test_quantization.py TestQuantizePT2E.test_reentrant Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/110125
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit c20b62d with merge base ec5bbef ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Summary: Add the test to make sure we can call the quantize API multiple times Test Plan: python test/test_quantization.py TestQuantizePT2E.test_reentrant Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Add the test to make sure we can call the quantize API multiple times Test Plan: python test/test_quantization.py TestQuantizePT2E.test_reentrant Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
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 these. Looks great. I presume it would break when we move conv2d quantization from source partition to node.target based.
Also maybe worth adding linear in the model mix because that likely will break it today.
yeah I'll change the test when I refactor aten.conv2d quantization, and will need to add the None support in XNNPACKQuantizer at that time I think, will add this soon |
|
@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 |
Stack from ghstack (oldest at bottom):
prepare_pt2e,prepare_qat_pt2eandconvert_pt2e#110097Summary:
Add the test to make sure we can call the quantize API multiple times
Test Plan:
python test/test_quantization.py TestQuantizePT2E.test_reentrant
Reviewers:
Subscribers:
Tasks:
Tags: