Forward fix to skip ROCm tests for #122836

[jbschlosser]

Stack from ghstack (oldest at bottom):

• (WIP) to_padded_tensor() triton kernel for NJT #121947
• Use return_and_correct_aliasing() for NJT + compatible storage setting #126552
• NJT <-> padded dense conversions #125947
• -> Forward fix to skip ROCm tests for #122836 #128891
• Short-term fix to preserve NJT metadata cache in torch.compile #122836
• Skip debug asserts for mixed dense, subclass views in autograd_not_implemented_fallback #128057
• Naive CPU kernels for jagged <-> padded dense conversions #127007

Fixes broken ROCm tests from #122836.

cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/128891

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

Note: Links to docs will display an error until the docs builds have been completed.

❌ 2 New Failures, 5 Unrelated Failures

As of commit c94ec99 with merge base edb45dc ():

NEW FAILURES - The following jobs have failed:

• rocm / linux-focal-rocm6.1-py3.8 / test (default, 5, 6, linux.rocm.gpu.2) (gh)
  Action 'https://api.github.com/repos/seemethere/upload-artifact-s3/tarball/baba72d0712b404f646cebe0730933554ebce96a' download has timed out. Error: The request was canceled due to the configured HttpClient.Timeout of 100 seconds elapsing.
• trunk / linux-focal-rocm6.1-py3.8 / test (distributed, 1, 1, linux.rocm.gpu) (gh)
  Resource temporarily unavailable (api.github.com:443)

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

• pull / linux-focal-py3.11-clang10 / test (dynamo, 3, 3, linux.2xlarge) (gh) (disabled by #123624 but the issue was closed recently and a rebase is needed to make it pass)
  profiler/test_torch_tidy.py::TestTorchTidyProfiler::test_optimizer_parameters_sgd
• pull / linux-focal-py3.12-clang10 / test (dynamo, 2, 3, linux.2xlarge) (gh) (disabled by #123624 but the issue was closed recently and a rebase is needed to make it pass)
  profiler/test_torch_tidy.py::TestTorchTidyProfiler::test_optimizer_parameters_sgd
• pull / linux-focal-py3.8-clang10 / test (dynamo, 1, 3, linux.2xlarge) (gh) (disabled by #123624 but the issue was closed recently and a rebase is needed to make it pass)
  profiler/test_torch_tidy.py::TestTorchTidyProfiler::test_optimizer_parameters_sgd
• pull / linux-jammy-py3.8-gcc11 / test (distributed, 1, 2, linux.2xlarge) (gh) (similar failure)
  Process completed with exit code 1.
• rocm / linux-focal-rocm6.1-py3.8 / test (default, 2, 6, linux.rocm.gpu.2) (gh) (detected as infra flaky with no log or failing log classifier)

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[jbschlosser]

@huydhn ping for stamp :)

[jbschlosser]

@pytorchbot merge

[pytorchmergebot]

Merge started

Your 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

Advanced Debugging

Check the merge workflow status
here

[jbschlosser]

@pytorchbot revert -m"reverting to revert parent PR" -c nosignal

[pytorchmergebot]

@pytorchbot successfully started a revert job. Check the current status here.
Questions? Feedback? Please reach out to the PyTorch DevX Team

[pytorchmergebot]

@jbschlosser your PR has been successfully reverted.

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.