add float32,bf16 for xpu UTs

hoshibara · hoshibara · commit 7e69c402e551 · 2025-04-10T07:21:24.000Z
diff --git a/test/inductor/test_flex_attention.py b/test/inductor/test_flex_attention.py
@@ -117,7 +117,7 @@ def create_block_mask_test(score_mod, query, key):
         test_dtypes_fast = [torch.float16]
     elif HAS_XPU:
         test_device = "xpu"
-        test_dtypes = [torch.float16]
+        test_dtypes = [torch.float32, torch.bfloat16, torch.float16]
         test_dtypes_fast = [torch.float16]
 else:
     test_device = "cpu"
diff --git a/test/inductor/test_flex_decoding.py b/test/inductor/test_flex_decoding.py
@@ -64,7 +64,7 @@ def create_block_mask_test(score_mod, query, key):
         test_dtypes_fast = [torch.float16]
     elif HAS_XPU:
         test_device = "xpu"
-        test_dtypes = [torch.float16]
+        test_dtypes = [torch.float32, torch.bfloat16, torch.float16]
         test_dtypes_fast = [torch.float16]
 
 test_page_sizes = [64, 128, 256]
