pytorch

0ee31ce8 - Revert "Use cub::BlockRadixSort to improve medium length sort performance"

Revert "Use cub::BlockRadixSort to improve medium length sort performance" This reverts commit 67a5d0bf40b10d8ebfb6b10b86f73583b9a8c461. Reverted https://github.com/pytorch/pytorch/pull/79628 on behalf of https://github.com/janeyx99 due to Sorry, reverting as it breaks ROCm build on trunk https://hud.pytorch.org/pytorch/pytorch/commit/67a5d0bf40b10d8ebfb6b10b86f73583b9a8c461