[Testing] Unskip expm1 log1p for MPS (#151790)

malfet · pytorchmergebot · commit cea43f721a76 · 2025-04-21T17:18:47.000Z
But don't test them for unsupported dtypes (which is float64 for MPS) - Skip int64 for log1p for now (next PR will fix that) Pull Request resolved: #151790 Approved by: https://github.com/Skylion007
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -6541,12 +6541,14 @@ def fn(x):
             (torch.randn([64]),),
         )
 
-    @xfail_if_mps  # float64 is not MPS type
     def test_expm1(self):
         def fn(x):
             return torch.expm1(x), torch.expm1(x) * 2
 
         for dtype in (torch.float16, torch.float, torch.double, torch.int, torch.int64):
+            if not self.is_dtype_supported(dtype):
+                continue
+
             self.common(
                 fn,
                 (torch.randn([64]).to(dtype=dtype),),
@@ -6612,12 +6614,17 @@ def test_avg_pool_errors_with_uint(self):
                 ):
                     c_op(x, kernel_size=2, stride=2)
 
-    @xfail_if_mps  # float64 is not MPS type
     def test_log1p(self):
         def fn(x):
             return torch.log1p(x), torch.log1p(x) * 2
 
         for dtype in (torch.float16, torch.float, torch.double, torch.int, torch.int64):
+            if not self.is_dtype_supported(dtype):
+                continue
+
+            if dtype == torch.int64 and self.device == "mps":
+                continue
+
             self.common(
                 fn,
                 (torch.randn([64]).to(dtype=dtype),),
