[Test][Easy] Use float16 dtype in test_sort_large (#159939)

Aidyn-A · pytorchmergebot · commit 556e2a73f4f0 · 2025-08-08T09:56:44.000Z
The test fails with: >RuntimeError: var_mean only support floating point and complex dtypes Pull Request resolved: #159939 Approved by: https://github.com/eqy
diff --git a/test/test_sort_and_select.py b/test/test_sort_and_select.py
@@ -215,7 +215,7 @@ def test_stable_sort(self, device, dtype):
             )
 
     @onlyCUDA
-    @dtypes(torch.uint8)
+    @dtypes(torch.float16)
     @largeTensorTest("200GB")  # Unfortunately 80GB A100 is not large enough
     def test_sort_large(self, device, dtype):
         t0 = torch.randperm(8192, device=device).to(dtype)

Original file line number	Diff line number	Diff line change
`@@ -215,7 +215,7 @@ def test_stable_sort(self, device, dtype):`
`215`	`215`	`)`
`216`	`216`
`217`	`217`	`@onlyCUDA`
`218`		`- @dtypes(torch.uint8)`
	`218`	`+ @dtypes(torch.float16)`
`219`	`219`	`@largeTensorTest("200GB") # Unfortunately 80GB A100 is not large enough`
`220`	`220`	`def test_sort_large(self, device, dtype):`
`221`	`221`	`t0 = torch.randperm(8192, device=device).to(dtype)`