-
Notifications
You must be signed in to change notification settings - Fork 25.7k
[Inductor] [Quant] Enable QConv2d Unary int8-mixed-bf16 Lowering #112550
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
[Inductor] [Quant] Enable QConv2d Unary int8-mixed-bf16 Lowering #112550
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/112550
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit c6718ec with merge base 8bdce9b ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
ghstack-source-id: f5bbb4a Pull Request resolved: pytorch#112550
…wering" **Summary** - PR 5 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor #111640. - Enable the QConv2d Unary int8-mixed-bf16 weight prepack and post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler [ghstack-poisoned]
ghstack-source-id: 1838155 Pull Request resolved: pytorch#112550
…wering" **Summary** - PR 5 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor #111640. - Enable the QConv2d Unary int8-mixed-bf16 weight prepack and post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler [ghstack-poisoned]
ghstack-source-id: 8e8e9d1 Pull Request resolved: pytorch#112550
…wering" **Summary** - PR 5 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor #111640. - Enable the QConv2d Unary int8-mixed-bf16 weight prepack and post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler [ghstack-poisoned]
…wering" **Summary** - PR 5 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor #111640. - Enable the QConv2d Unary int8-mixed-bf16 weight prepack and post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler [ghstack-poisoned]
|
Hi @eellison @jerryzh168, Could you kindly help to review these 3 PRs in this ghstack for the Inductor lowering of int8-mixed-bf16 quantization case? |
|
@jerryzh168 want to review this one ? |
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.
LG overall, please add more some docs
…wering" **Summary** - PR 5 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor #111640. - Enable the QConv2d Unary int8-mixed-bf16 weight prepack and post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler [ghstack-poisoned]
|
@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 |
…12551) **Summary** - PR 6 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor #111640. - Enable the QConv2d Binary int8-mixed-bf16 post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` Pull Request resolved: #112551 Approved by: https://github.com/jgong5, https://github.com/eellison, https://github.com/jerryzh168 ghstack dependencies: #112550
…orch#112550) **Summary** - PR 5 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor pytorch#111640. - Enable the QConv2d Unary int8-mixed-bf16 weight prepack and post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` Pull Request resolved: pytorch#112550 Approved by: https://github.com/jgong5, https://github.com/jerryzh168
…torch#112551) **Summary** - PR 6 for enabling Int8-Mixed-BF16 PT2E PTQ Quantization with Inductor pytorch#111640. - Enable the QConv2d Binary int8-mixed-bf16 post grad lowering inside inductor. **TestPlan** ``` python -m pytest test_mkldnn_pattern_matcher.py -k test_qconv2d ``` Pull Request resolved: pytorch#112551 Approved by: https://github.com/jgong5, https://github.com/eellison, https://github.com/jerryzh168 ghstack dependencies: pytorch#112550
* Most part of testcases work properly on Navi48(gfx1201) with TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1, in this commit enable it for this arch. No support of AOTriton currently for Navci44(gfx1200), so these testcases just skipped. * test_qconv2d_int8_mixed_bf16 skipped because it was originally skipped in pytorch#112550 but later lost. * test_sac_ilp_case1 skipped as per SWDEV-509011 * test_distributed_checkpoint_state_dict_type[0-1]_cuda fixed bug with arguments.
… SDPA or Navi4x (#2213) [release/2.6][SWDEV-523736] Skip some testcases for archs without SDPA or Navi4x [SWDEV-523736] Fix some unittests for Navi4x * Most part of testcases work properly on Navi48(gfx1201) with TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1, in this commit enable it for this arch. No support of AOTriton currently for Navci44(gfx1200), so these testcases just skipped. * test_qconv2d_int8_mixed_bf16 skipped because it was originally skipped in pytorch#112550 but later lost. * test_sac_ilp_case1 skipped as per SWDEV-509011 * test_distributed_checkpoint_state_dict_type[0-1]_cuda fixed bug with arguments. Fixes #SWDEV-523736
Stack from ghstack (oldest at bottom):
Summary
TestPlan
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @aakhundov @ColinPeppler