diff --git a/tests/kernels/test_block_fp8.py b/tests/kernels/test_block_fp8.py index 20eff1c2..6206cbd5 100644 --- a/tests/kernels/test_block_fp8.py +++ b/tests/kernels/test_block_fp8.py @@ -30,8 +30,8 @@ M_moe = [1, 7, 83, 512, 2048] N_moe = [4608] # [128, 4608, 13824] K_moe = [7168] # [256, 7168, 13824] BLOCK_SIZE = [[128, 128]] -E = [256] # [8, 24, 128, 256] -TOP_KS = [1] # [1, 2, 6] +E = [8, 24] # [8, 24, 128, 256] +TOP_KS = [2] # [1, 2, 6] OUT_DTYPES = [torch.bfloat16] # [torch.float32, torch.half, torch.bfloat16] SEEDS = [0]