[MPS] Remove remaining casts from 13.3 (#95870)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/95870
Approved by: https://github.com/kulinseth
diff --git a/test/test_mps.py b/test/test_mps.py
index 0c77fa5..9877612 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -3852,6 +3852,15 @@
helper(2, 8, 4, 4, "min", torch.float16)
helper(2, 8, 4, 4, "min", torch.int64)
+ @unittest.skipIf(product_version < 13.3, "Long data type supported from macOS 13.3 and above")
+ def test_reduction_sum_max_long_val(self):
+ x_mps = torch.tensor([sys.maxsize, sys.maxsize - 10, sys.maxsize - 5, sys.maxsize - 18], device="mps")
+ x_cpu = x_mps.detach().clone().cpu()
+
+ res_mps = torch.sum(x_mps)
+ res_cpu = torch.sum(x_cpu)
+ self.assertEqual(res_mps, res_cpu)
+
# Test forward max
# Note - don't test grad now
def test_max_el(self):
