Skip to content

Commit 661b04a

Browse files
wenscarlmgoin
authored andcommitted
[MoE] Nvfp4 Masked Gemm: Add flashinfer grouped_gemm_nt_masked (vllm-project#25990)
Signed-off-by: Shu Wang. <shuw@nvidia.com> Signed-off-by: mgoin <mgoin64@gmail.com> Co-authored-by: Michael Goin <mgoin64@gmail.com>
1 parent 78ce180 commit 661b04a

File tree

10 files changed

+1062
-33
lines changed

10 files changed

+1062
-33
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -921,6 +921,7 @@ steps:
921921
- pytest -v -s tests/kernels/moe/test_nvfp4_moe.py
922922
- pytest -v -s tests/kernels/moe/test_ocp_mx_moe.py
923923
- pytest -v -s tests/kernels/moe/test_flashinfer.py
924+
- pytest -v -s tests/kernels/moe/test_cutedsl_moe.py
924925

925926
- label: Blackwell Fusion and Compile Tests # 30 min
926927
timeout_in_minutes: 40

0 commit comments

Comments
 (0)