[rocm7.1_internal_testing] fix large tensor sort on ROCm#2543

jithunnair-amd merged 1 commit intorocm7.1_internal_testingROCm/pytorch:rocm7.1_internal_testingfrom

dnikolaev/fix_large_tensor_sort_on_rocmROCm/pytorch:dnikolaev/fix_large_tensor_sort_on_rocmCopy head branch name to clipboard