Skip to content

Commit 6620112

Browse files
Set GRF_mode==large for mxfp8 matmul tests (#5417)
Addresses #4062
1 parent f762b2b commit 6620112

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

python/test/unit/language/test_matmul.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1298,6 +1298,8 @@ def create_operand(dtype: str, size0: int, size1: int, k_dim: int, transpose: bo
12981298
kernel_kwargs["matrix_instr_nonkdim"] = nonKDim
12991299
if is_xpu() and (128, 256, 256) == (BLOCK_M, BLOCK_N, BLOCK_K) and not CONST_SCALE and not PACK_B_ALONG_K:
13001300
kernel_kwargs["num_warps"] = 8
1301+
if is_xpu():
1302+
kernel_kwargs["grf_mode"] = "large"
13011303
out = mxfp8_mxfp4_matmul[grid](a, b, output, a_scale, b_scale, M, N, K, stride_scale, a.stride(0), a.stride(1),
13021304
b.stride(0), b.stride(1), output.stride(0), output.stride(1), not CONST_SCALE,
13031305
dtype_converter[A_DATA_TYPE], dtype_converter[B_DATA_TYPE], BLOCK_M, BLOCK_N,

scripts/test-triton.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -731,9 +731,9 @@ run_triton_kernels_tests() {
731731
# FIXME: reconsider in the future
732732
max_procs=4
733733
fi
734-
734+
# skipping mxfp, they are part of mxfp_tests suite
735735
TRITON_TEST_SUITE=triton_kernels \
736-
run_pytest_command -vvv -n $max_procs --device xpu .
736+
run_pytest_command -vvv -n $max_procs --device xpu . -k 'not test_mxfp'
737737
}
738738

739739
test_triton() {

0 commit comments

Comments
 (0)