Skip to content

Commit b901f28

Browse files
committed
Set GRF_mode==large for mxfp8 matmul tests
1 parent 9e23713 commit b901f28

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
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,

0 commit comments

Comments
 (0)