🔗 Helpful Links

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

• 📄 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.

✅ No Failures

As of commit 712d30f with merge base a53d14d ():
💚 Looks good so far! There are no failures yet. 💚

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

approving assuming that this never ran before and therefore this is not a regression

Indeed, it never ran before. The only machine which has that much memory is GB300.

Hmm, those are strange failures

@pytorchbot rebase

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

Successfully rebased test_sort_large_float16 onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout test_sort_large_float16 && git pull --rebase)

@pytorchbot merge

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

approving assuming that this never ran before and therefore this is not a regression

Indeed, it never ran before. The only machine which has that much memory is GB300.

@Aidyn-A Did this updated test run in any of the CI jobs? It failed in ROCm CI (because the MI325 has >200GB memory) with the error Cannot sort dimension of length 8192 (link) (error coming from here), but I can't tell if it passed for CUDA at all.

approving assuming that this never ran before and therefore this is not a regression

Indeed, it never ran before. The only machine which has that much memory is GB300.

@Aidyn-A Did this updated test run in any of the CI jobs? It failed in ROCm CI (because the MI325 has >200GB memory) with the error Cannot sort dimension of length 8192 (link) (error coming from here), but I can't tell if it passed for CUDA at all.

Yes, it is passing on GB300. Was it passing prior my changes? I believe it should have failed, because var_mean does not support integer dtypes.