Skip to content

Commit 28d3f91

Browse files
authored
Speed up test_mxfp_matmul by reducing block sizes: 128->64 (#4778)
I ran the test twice, for 128 and 64 blocks, to see the impact of compilation and +- pure runtime, and got the following data: 128 blocks: 1. `192 passed, 96 xfailed in 1464.76s (0:24:24)` 2. `192 passed, 96 xfailed in 7.16s` 64 blocks: 1. `192 passed, 96 xfailed in 235.29s (0:03:55)` 3. `192 passed, 96 xfailed in 5.99s` It can be reverted in the future when compilation works better. Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 0f1ef29 commit 28d3f91

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/unit/intel/test_mxfp_matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def mxfp_matmul( #
9797

9898

9999
@pytest.mark.parametrize("M, N, K", [(1024, 512, 512)])
100-
@pytest.mark.parametrize("BLOCK_M, BLOCK_N, BLOCK_K", [(128, 128, 128)])
100+
@pytest.mark.parametrize("BLOCK_M, BLOCK_N, BLOCK_K", [(64, 64, 64)])
101101
@pytest.mark.parametrize("NUM_STAGES", [1, 3])
102102
@pytest.mark.parametrize("B_TRANS", [True, False])
103103
@pytest.mark.parametrize("PACK_B_ALONG_K", [True, False])

0 commit comments

Comments
 (0)