3 people 4d58fa254a [MoE] Unify DeepGEMM MegaMoE experts behind one feature-matrix layer; add NVFP4 (#61538) ...

Signed-off-by: Yongye Zhu <zyy1102000@gmail.com>
Co-authored-by: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
Co-authored-by: Kevin H. Luu <khluu000@gmail.com>

2026-10-15 19:02:01 +00:00

..

__init__.py

[Docs] Fix warnings in docs build (#22588)

2025-08-10 05:49:51 -07:00

attention_warmup.py

[Bugfix][MRV2] Fix FlashInfer attention and ReplaySSM warmups on MRV2 (#61013)

2026-10-11 16:04:41 -07:00

b12x_warmup.py

[Kernel] Add b12x FP4 MoE backend (#52018)

2026-08-21 08:04:36 -07:00

cutedsl_warmup.py

[2/N][Feat][Perf] Add new warmup infrastructure for JITs. Add predicate filtering for JIT warmup, and migrate Inkling FA4 (#49315)

2026-08-11 07:46:07 -04:00

deep_gemm_warmup.py

[MoE] Unify DeepGEMM MegaMoE experts behind one feature-matrix layer; add NVFP4 (#61538)

2026-10-15 19:02:01 +00:00

flashinfer_autotune_cache.py

[Fast Start] Preload the FlashInfer autotune table on the weight cache daemon (#60085)

2026-10-10 15:48:06 +00:00

flashinfer_sparse_mla_warmup.py

[ROCm][DSv4.1] Disambiguate ROCm attention class names (#58689)

2026-10-15 21:38:39 +08:00

jit_warmup_cutedsl_helper.py

[5/N][warmup][DSv4] Migrate NVIDIA CuTeDSL attention kernels (#53566)

2026-09-12 00:05:59 -04:00

jit_warmup_tilelang_helper.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

jit_warmup_triton_helper.py

[Refactor] Remove dead code multiple places (#60120)

2026-10-10 15:41:10 +00:00

jit_warmup.py

[Perf] Parallelize mHC pre-norm JIT warmup (#56683)

2026-09-13 18:58:20 -07:00

kernel_warmup.py

[Bugfix][MRV2] Fix FlashInfer attention and ReplaySSM warmups on MRV2 (#61013)

2026-10-11 16:04:41 -07:00

kimi_k3_triton_warmup.py

[Bugfix] Gate Kimi-K3 KDA warmup on sys.modules to skip Kimi import for non-Kimi models (#59257)

2026-10-01 09:25:45 +00:00

mamba_triton_warmup.py

[Perf] Extend Qwen Triton warmup to avoid first-request latency spikes (#54797)

2026-09-07 11:46:00 +08:00

minimax_m3_msa_warmup.py

[Bugfix][MRV2] Fix FlashInfer attention and ReplaySSM warmups on MRV2 (#61013)

2026-10-11 16:04:41 -07:00

qwen4_exp_qsa_warmup.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

qwen_triton_warmup.py

[Model] Add Qwen4Exp to the Qwen GDN Triton warmup (#56742)

2026-10-02 21:38:15 +08:00

qwen_vl_triton_warmup.py

[Perf] Skip the M-RoPE warmup when every holder of the RoPE fuses it (#60246)

2026-10-10 06:47:39 +00:00

replayssm_warmup.py

[Mamba] Add FlashInfer ReplaySSM support for MTP (#52928)

2026-09-30 21:41:17 +00:00

rocm_segmented_attn_autotune_warmup.py

[ROCM][Perf] Add ROCM_SEGMENTED_ATTN attention-backend works across RDNA3/3.5/4 (#59132)

2026-10-06 07:53:47 +00:00

watermark_sample_warmup.py

[watermarking] Dual-key gumbel-max watermarking for speculative decoding support (#56122)

2026-09-12 18:42:01 +00:00