[PyTorch] Fix inductor CPU masked() body codegen when result dtype is bool and operator is where #138486
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
… bool and operator is where In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we don't make it a VecMask. Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/) [ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/138486
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 83b08e8 with merge base fb0da32 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
swolchok
added a commit
that referenced
this pull request
Oct 21, 2024
… bool and operator is where In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we don't make it a VecMask. Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/) ghstack-source-id: 249157267 Pull Request resolved: #138486
|
This pull request was exported from Phabricator. Differential Revision: D64702918 |
|
this should fix one of the inductor failures on #137426 . I need to figure out how to repro this issue without that PR and add a regression test. |
…lt dtype is bool and operator is where" In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we don't make it a VecMask. Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/) cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang aakhundov [ghstack-poisoned]
|
This pull request was exported from Phabricator. Differential Revision: D64702918 |
swolchok
added a commit
that referenced
this pull request
Oct 21, 2024
… bool and operator is where Pull Request resolved: #138486 In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we don't make it a VecMask. ghstack-source-id: 249215767 @exported-using-ghexport Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/)
I could use help adding a test here. The failing input to constant_pad_nd on #137426 is tensor size (0, 3, 3) and pad=(1, 1, 1, 1, 1, 1). I brute-forced all sizes in range(3, 10) for dim 1 and 2 on main without finding a repro. |
There was a problem hiding this comment.
Thanks for the fix. I think the changes in this PR looks good to me. But would like to study more about the failure.
It looks like body code generates a scalar var with bool dtype, and we mistake it as at::vec::Vectorized instead of at::vec::VecMask. It will be great to have a test case.
…lt dtype is bool and operator is where" In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we didn't make it a VecMask. Now we do. Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/) cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang aakhundov [ghstack-poisoned]
|
This pull request was exported from Phabricator. Differential Revision: D64702918 |
done! |
…lt dtype is bool and operator is where" In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we didn't make it a VecMask. Now we do. Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/) cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang aakhundov [ghstack-poisoned]
swolchok
added a commit
that referenced
this pull request
Oct 22, 2024
… bool and operator is where Pull Request resolved: #138486 In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we don't make it a VecMask. ghstack-source-id: 249288078 @exported-using-ghexport Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/)
|
This pull request was exported from Phabricator. Differential Revision: D64702918 |
leslie-fang-intel
approved these changes
Oct 22, 2024
This was referenced Oct 22, 2024
This was referenced Oct 22, 2024
…lt dtype is bool and operator is where" In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we didn't make it a VecMask. Now we do. Differential Revision: [D64702918](https://our.internmc.facebook.com/intern/diff/D64702918/) cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang aakhundov [ghstack-poisoned]
|
This pull request was exported from Phabricator. Differential Revision: D64702918 |
This was referenced Oct 22, 2024
malfet
approved these changes
Oct 24, 2024
pytorchmergebot
pushed a commit
that referenced
this pull request
Oct 24, 2024
This issue is one of two inductor bugs blocking land of #137426. Turned out to be simple Differential Revision: [D64734116](https://our.internmc.facebook.com/intern/diff/D64734116/) Pull Request resolved: #138542 Approved by: https://github.com/jgong5, https://github.com/malfet ghstack dependencies: #138486 Co-authored-by: leslie-fang-intel <leslie.fang@intel.com>
pytorchmergebot
pushed a commit
that referenced
this pull request
Oct 24, 2024
These are ternary ops, not binary ops. Differential Revision: [D64794253](https://our.internmc.facebook.com/intern/diff/D64794253/) Pull Request resolved: #138655 Approved by: https://github.com/jgong5, https://github.com/malfet ghstack dependencies: #138486, #138542
pytorchmergebot
pushed a commit
that referenced
this pull request
Oct 24, 2024
The size of the destination array was too small. Differential Revision: [D64843491](https://our.internmc.facebook.com/intern/diff/D64843491/) Pull Request resolved: #138716 Approved by: https://github.com/jgong5, https://github.com/malfet ghstack dependencies: #138486, #138542, #138655
pytorchmergebot
pushed a commit
that referenced
this pull request
Oct 24, 2024
There is no guarantee that `len` here is enough for a full vector. This was causing at least one test failure on #137426. Differential Revision: [D64857786](https://our.internmc.facebook.com/intern/diff/D64857786/) Pull Request resolved: #138744 Approved by: https://github.com/jgong5, https://github.com/malfet ghstack dependencies: #138486, #138542, #138655, #138716
pytorchmergebot
pushed a commit
that referenced
this pull request
Oct 26, 2024
The correct vector length for ARM64 is 128 bits (16 bytes). We were previously using double this, apparently just because that would be the same length as AVX2. Differential Revision: [D63984039](https://our.internmc.facebook.com/intern/diff/D63984039/) Pull Request resolved: #137426 Approved by: https://github.com/jgong5, https://github.com/malfet ghstack dependencies: #138486, #138542, #138655, #138716, #138744
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Stack from ghstack (oldest at bottom):
In this case, it looks like we expect the body to be a VecMask (unify_mask_base_type is called by where()), but we didn't make it a VecMask. Now we do.
Differential Revision: D64702918
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang @aakhundov