HIP

triton-lang · Mar 28, 2024 · fe2b6a2 · fe2b6a2
1 parent a47fc35
commit fe2b6a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py
@@ -2948,6 +2948,8 @@ def test_dot(M, N, K, num_warps, col_a, col_b, epilogue, input_precision, in_dty
         pytest.skip(
             "numpy.dot with int8 inputs will overflow while tl.dot doesn't because MMA instruction's accumulator is 32-bit"
         )
+    if is_hip() and (input_precision != "ieee"):
+        pytest.skip(f"{input_precision} not supported on HIP")
 
     torch.backends.cuda.matmul.allow_tf32 = input_precision == "tf32"