3 people 613abb50d5 [MoE] Nvfp4 Masked Gemm: Add flashinfer grouped_gemm_nt_masked (#25990) ...

Signed-off-by: Shu Wang. <shuw@nvidia.com>
Signed-off-by: mgoin <mgoin64@gmail.com>
Co-authored-by: Michael Goin <mgoin64@gmail.com>

2025-11-19 13:29:06 -08:00

..

modular_kernel_tools

[Feature] Prefill Context Parallel (PCP) basic support (#28718)

2025-11-19 15:52:44 -05:00

__init__.py

[Kernel] DeepEP dispatch-combine kernel integration (#18434)

2025-06-03 12:30:02 -07:00

parallel_utils.py

[Chore] Separate out optional dependency checks from vllm.utils (#27207)

2025-10-22 10:44:21 -04:00

test_batched_deepgemm.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_batched_moe.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_block_fp8.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_block_int8.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_count_expert_num_tokens.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

test_cutedsl_moe.py

[MoE] Nvfp4 Masked Gemm: Add flashinfer grouped_gemm_nt_masked (#25990)

2025-11-19 13:29:06 -08:00

test_cutlass_grouped_gemm.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

test_cutlass_moe.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_deepep_deepgemm_moe.py

[Performance][DeepGEMM] Estimate expected_m (#28694)

2025-11-15 13:52:14 +08:00

test_deepep_moe.py

[Performance][B200] silu_mul_quant: pack scales in int32 (#28358)

2025-11-13 10:16:55 -08:00

test_deepgemm.py

kernels/moe test pruning (#27053)

2025-10-30 12:10:34 +08:00

test_flashinfer_moe.py

kernels/moe test pruning (#27053)

2025-10-30 12:10:34 +08:00

test_flashinfer.py

Feature: Support Relu2 in FusedMoE fp8 cutlass path (#27261)

2025-11-16 13:39:44 -05:00

test_gpt_oss_triton_kernels.py

Update rope_scaling to rope_parameters in preparation for Transformers v5 (#28542)

2025-11-19 09:06:36 -08:00

test_grouped_topk.py

kernels/moe test pruning (#27053)

2025-10-30 12:10:34 +08:00

test_modular_kernel_combinations.py

kernels/moe test pruning (#27053)

2025-10-30 12:10:34 +08:00

test_moe_align_block_size.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

test_moe_permute_unpermute.py

[ROCm][FEAT] Fuse DeepSeek shared experts into AITER fused_moe ops (#24097)

2025-10-16 10:41:34 +08:00

test_moe.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_nvfp4_moe.py

kernels/moe test pruning (#27053)

2025-10-30 12:10:34 +08:00

test_ocp_mx_moe.py

[NVIDIA] [Perf] Update to leverage flashinfer trtllm FP4 MOE throughput kernel (#26714)

2025-10-16 16:20:25 -07:00

test_pplx_cutlass_moe.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_pplx_moe.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

test_rocm_aiter_topk.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_silu_mul_fp8_quant_deep_gemm.py

[Performance][B200] silu_mul_quant: pack scales in int32 (#28358)

2025-11-13 10:16:55 -08:00

test_triton_moe_ptpc_fp8.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733)

2025-11-15 01:59:31 -08:00

utils.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00