Skip to content

Commit 6578b58

Browse files
authored
[AMD][TEST] Enable FP4 pack along M/N tests for CDNA4 (#7736)
This enables the testing for M/N packed FP4.
1 parent d429e2b commit 6578b58

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

python/test/unit/language/test_matmul.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -973,8 +973,6 @@ def test_block_scale_fp4(M, N, K, BLOCK_M, BLOCK_N, BLOCK_K, VEC_SIZE, with_a_sc
973973
if not (with_a_scale and with_b_scale):
974974
pytest.skip("None aScale/bScale is only tested on AMD backend for now")
975975
elif is_hip():
976-
if not pack_along_k:
977-
pytest.skip("Packing along M/N not implemented yet on AMD.")
978976
if not is_hip_cdna4():
979977
pytest.skip("Scaled fp4 matmul is only natively supported on CDNA4")
980978
if scale_type != 'float8_e8m0fnu':
@@ -1129,8 +1127,6 @@ def test_mxfp8_mxfp4_matmul(M, N, K, BLOCK_M, BLOCK_N, BLOCK_K, NUM_STAGES, B_TR
11291127
if not (A_DATA_TYPE == "float8e5" and B_DATA_TYPE == "float4"):
11301128
pytest.skip(f"(A: {A_DATA_TYPE}, B: {B_DATA_TYPE}) has not been tested on NV backend")
11311129
elif is_hip():
1132-
if not PACK_B_ALONG_K:
1133-
pytest.skip("Pack along M/N is not enabled on AMD backend")
11341130
if not is_hip_cdna4():
11351131
pytest.skip("Scaled mxfp4 & mxfp8 matmul is only natively supported on CDNA4")
11361132
if (nonKDim == 16 and BLOCK_K < 128) or (nonKDim == 32 and BLOCK_K < 64):

0 commit comments

Comments
 (0)