Skip to content

Commit c3128cb

Browse files
mgoinAkshat-Tripathi
authored andcommitted
Fix test_block_fp8.py test for MoE (vllm-project#13915)
Signed-off-by: mgoin <[email protected]>
1 parent f0a2f15 commit c3128cb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/kernels/test_block_fp8.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030
N_moe = [4608] # [128, 4608, 13824]
3131
K_moe = [7168] # [256, 7168, 13824]
3232
BLOCK_SIZE = [[128, 128]]
33-
E = [256] # [8, 24, 128, 256]
34-
TOP_KS = [1] # [1, 2, 6]
33+
E = [8, 24] # [8, 24, 128, 256]
34+
TOP_KS = [2] # [1, 2, 6]
3535
OUT_DTYPES = [torch.bfloat16] # [torch.float32, torch.half, torch.bfloat16]
3636
SEEDS = [0]
3737

0 commit comments

Comments
 (0)