Skip to content

Commit 0b772f3

Browse files
committed
Fix incorrectly dispatching to FMA
1 parent dab71a1 commit 0b772f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/unit/language/test_core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3548,7 +3548,7 @@ def test_dot3d(B, num_warps, M, N, K, BLOCK_M, BLOCK_N, in_dtype_str, out_dtype_
35483548
if out_dtype_str == "float16":
35493549
pytest.skip(f"{out_dtype_str} has low precision in WMMA dot")
35503550
else:
3551-
input_precision = "tf32" if is_cuda() and in_dtype_str == 'float32' else "ieee"
3551+
input_precision = "tf32" if (is_cuda() or is_xpu()) and in_dtype_str == 'float32' else "ieee"
35523552

35533553
if B == 8 and M == 64 and in_dtype_str == "float32" and out_dtype_str == "float32":
35543554
if not is_interpreter() and torch.cuda.is_available(

0 commit comments

Comments
 (0)