[inductor] Adjust test_log_fp64 to only run when float64 is supported. (pytorch#145686)

Pull Request resolved: pytorch#145686
Approved by: https://github.com/malfet, https://github.com/jansel

Co-authored-by: Nikita Shulga <[email protected]>

Author: 2 people

test/inductor/test_mps_basic.py

@@ -58,6 +58,7 @@ class MPSBasicTests(TestCase):
     test_inf = CommonTemplate.test_inf
     test_isinf = CommonTemplate.test_isinf
     test_isinf2 = CommonTemplate.test_isinf2
+    test_log_fp64 = CommonTemplate.test_log_fp64
     test_low_memory_max_pool = CommonTemplate.test_low_memory_max_pool
     test_max_min = CommonTemplate.test_max_min
     test_max_pool2d2 = CommonTemplate.test_max_pool2d2

test/inductor/test_torchinductor.py

@@ -6471,10 +6471,17 @@ def test_log_fp64(self):
         def fn(x):
             return torch.log(x), torch.log2(x)
 
-        self.common(
-            fn,
-            (torch.randn([1024], dtype=torch.float64) + 10,),
+        _dtype = torch.float64
+        ctx = (
+            contextlib.nullcontext()
+            if self.is_dtype_supported(_dtype)
+            else self.assertRaises(TypeError)
         )
+        with ctx:
+            self.common(
+                fn,
+                (torch.randn([1024], dtype=_dtype) + 10,),
+            )
 
     def test_bitwise(self):
         def fn(x, y):