Skip to content

Commit 188048c

Browse files
authored
[AMD] Enable passing test_cast_matmul cases (#6277)
1 parent dad5367 commit 188048c

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

python/test/regression/test_cast_matmul.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
import triton
1313
import triton.language as tl
14-
from triton._internal_testing import is_hip_cdna3, is_cuda, is_hip
14+
from triton._internal_testing import is_hip_cdna3, is_cuda
1515

1616
input_dtypes = ["bfloat16", "float16", "float32", "float64"]
1717
if is_cuda():
@@ -89,8 +89,6 @@ def matmul_kernel(A, B, C, M, N, K, #
8989
def test_cast_matmul(M, K, N, BLOCK_K, BLOCK_M, BLOCK_N, w_dtype, x_dtype, out_dtype, device):
9090
if x_dtype == w_dtype:
9191
pytest.skip("skip the same input dtype")
92-
if is_hip() and BLOCK_M == 64 and w_dtype in ["float8_e5m2", "float8_e4m3fnuz"]:
93-
pytest.skip("skip due to bug on HIP path")
9492
x_dtype: torch.dtype = getattr(torch, x_dtype)
9593
w_dtype: torch.dtype = getattr(torch, w_dtype)
9694

0 commit comments

Comments
 (0)