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
