- Notifications
You must be signed in to change notification settings - Fork26.3k
[ROCm] Enable multiple NestedTensor tests on ROCm#170829
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 ourterms of service andprivacy statement. We’ll occasionally send you account related emails.
Already on GitHub?Sign in to your account
base:main
Are you sure you want to change the base?
Conversation
pytorch-botbot commentedDec 19, 2025 • edited
Loading Uh oh!
There was an error while loading.Please reload this page.
edited
Uh oh!
There was an error while loading.Please reload this page.
🔗 Helpful Links🧪 See artifacts and rendered test results athud.pytorch.org/pr/170829
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. |
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.
Pull request overview
This PR enables thetest_index_put_error test for ROCm by removing an outdated skip decorator. The skip was originally in place because ROCm disabled kernel assertions for performance reasons, but ROCm 7.1 now correctly raises errors on invalid index operations, allowing the test to pass.
Key Changes:
- Removed
@unittest.skipIf(TEST_WITH_ROCM, ...)decorator fromtest_index_put_errormethod
8ec16d2 toc6a1869CompareThis commit enables several NestedTensor tests (SDPA, compile, and error index)that now pass on ROCm 7.1.Fixespytorch#168785Fixespytorch#168786Fixespytorch#168787Fixespytorch#168788Fixespytorch#168789Fixespytorch#168790Fixespytorch#168792Fixespytorch#168793Fixespytorch#168794Fixespytorch#168795Fixespytorch#168796Fixespytorch#168797
c6a1869 to1f760b7Compare
Uh oh!
There was an error while loading.Please reload this page.
This PR enables several NestedTensor tests (SDPA, compile, and error index) that now pass on ROCm 7.1.
Fixes#168785
Fixes#168786
Fixes#168787
Fixes#168788
Fixes#168789
Fixes#168790
Fixes#168792
Fixes#168793
Fixes#168794
Fixes#168795
Fixes#168796
Fixes#168797
cc@jeffdaily@sunway513@jithunnair-amd@pruthvistony@ROCmSupport@jataylo@hongxiayang@naromero77amd@pragupta@jerrymannil@xinyazhang