Merge branch 'main' into triton

intel · Aug 26, 2024 · 7f3bc2e · 7f3bc2e
2 parents 860138d + 12d7ee7
commit 7f3bc2e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/xpu/xpu_test_utils.py b/test/xpu/xpu_test_utils.py
@@ -659,6 +659,11 @@ def align_supported_dtypes(self, db):
                 backward_dtypes.add(bfloat16)
                 opinfo.backward_dtypes = tuple(backward_dtypes)
 
+            if "has_fp64=0" in str(torch.xpu.get_device_properties(0)):
+                fp64_dtypes = [ torch.float64, torch.complex128, torch.double, ]
+                opinfo.dtypesIfXPU = set(filter(lambda x: (x not in fp64_dtypes), list(opinfo.dtypesIfXPU)))
+                opinfo.backward_dtypes = tuple(filter(lambda x: (x not in fp64_dtypes), list(opinfo.backward_dtypes)))
+
     def __enter__(self):
         # Monkey patch until we have a fancy way