3 people 46c8a161f4 [Bugfix] Handle padded routes in CUTLASS MoE permutations (#54747) ...

Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com>
Signed-off-by: Yongye Zhu <zyy1102000@gmail.com>
Co-authored-by: Codex <codex@openai.com>
Co-authored-by: Yongye Zhu <zyy1102000@gmail.com>

2026-09-01 19:08:17 -07:00

..

attention

[Bugfix] Gate sm_100-only kernel tests on the capability family, not >= (#54306)

2026-09-01 11:29:15 -04:00

core

[ROCm][CI] Add ROCm misc ops and env tests (#53279)

2026-09-01 19:03:37 +00:00

helion

[Kernel][Helion] Add numerics checks to benchmark script (#48968)

2026-07-31 20:16:02 -04:00

ir

[CI][Test] Deflake the rms_norm scaling-property assertions (#54271)

2026-08-29 08:08:20 +08:00

mamba

[Mamba] Add FlashInfer ReplaySSM backend (#52506)

2026-09-01 10:26:43 -04:00

moe

[Bugfix] Handle padded routes in CUTLASS MoE permutations (#54747)

2026-09-01 19:08:17 -07:00

quantization

[Perf][Kernel] Initialize NVFP4 padding in quant kernel (#53568)

2026-09-01 09:49:07 -07:00

scripts

[CI/Build][ROCm] Run the TileLang HIP symbol checks in their own interpreter (#53117)

2026-08-21 16:17:42 -05:00

turboquant

[Kernel][ROCm][Perf] FlyDSL decode-attention kernel for 4-bit TurboQuant KV cache (#47896)

2026-08-11 08:45:39 -07:00

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

allclose_default.py

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

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

conftest.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423)

2026-07-22 14:43:07 -06:00

quant_utils.py

[XPU][Tests] Make tests device-agnostic (#51968)

2026-08-20 02:25:41 -07:00

test_apply_repetition_penalties.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_awq_int4_to_int8.py

[CPU] Bump up to the latest CPU kernels (#41924)

2026-05-07 05:45:59 -07:00

test_bf16_skinny_gemm.py

[Kernel][Qwen] Add Hopper LL-GEMM tuning table for Qwen4Exp (#54560)

2026-09-01 11:21:41 +00:00

test_bf16x3_router_gemm_cutedsl.py

BF16x3 router GEMM (#47973)

2026-07-16 17:04:20 +08:00

test_cache_kernels.py

[Bugfix] Restore gather_and_maybe_dequant_cache OOB guard (#49427)

2026-07-22 13:47:37 -07:00

test_compressor_kv_cache.py

[ROCm] Fix DeepSeek V4 indexer numerics and coverage (#50803)

2026-08-20 09:34:00 +00:00

test_concat_mla_q.py

[Perf][GLM5.2] Fuse sparse MLA Q concatenation with head padding (#53878)

2026-08-27 11:54:32 -04:00

test_cp_gather_fp8.py

[Kernel] Optimize long-context MLA cache gathers (#51739)

2026-08-10 22:23:43 -07:00

test_fla_layernorm_guard.py

[Misc] Add and enable Triton kernel unit tests on XPU (#45694)

2026-08-07 02:53:03 -07:00

test_flashinfer_bf16_gemm.py

[kernel] Integrate FlashInfer BF16 CuTeDSL Low Latency GEMM (#50572)

2026-08-27 22:29:12 +00:00

test_flex_attention.py

[FlexAttention] Avoid encoder block-mask compile explosion (#50339)

2026-07-30 11:24:13 -04:00

test_fp32_router_gemm.py

[Kernel] Retire the DSv3 router GEMM CUDA kernel (#54040)

2026-08-31 07:15:31 +00:00

test_fused_deepseek_v4_qnorm_rope_kv_insert.py

Revert DSv4 eager workspace reuse (#52836)

2026-08-18 17:02:59 -07:00

test_fused_deepseek_v32_norm_rope.py

[Bugfix][DCP] Fix NVIDIA DeepSeek-V3.2 / GLM-5.2 fused attention (#50005)

2026-08-31 00:30:45 -07:00

test_fused_gdn_post_conv.py

[Model] Support Qwen3.8-Flash-Next (#53896)

2026-08-31 05:57:56 +00:00

test_fused_indexer_q_rope_quant.py

[ROCm] Fix DeepSeek V4 indexer numerics and coverage (#50803)

2026-08-20 09:34:00 +00:00

test_fused_inv_rope_fp8_quant.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423)

2026-07-22 14:43:07 -06:00

test_fused_minimax_m3_qknorm_rope_kv_insert.py

[Attention][MiniMax-M3] Add MSA speculative decode verification (#50032)

2026-08-02 13:46:55 -07:00

test_fused_qk_norm_rope_gate.py

[Kernel][Perf] Enable fused QK-norm + partial MRoPE + gate for Qwen3.6 (#52676)

2026-08-24 18:28:49 -07:00

test_fused_quant_activation.py

Add platform manual_seed_all API (#38468)

2026-04-10 13:43:50 +08:00

test_fused_recurrent_packed_decode.py

[Bugfix][Kernel] Keep packed GDN decode beta in FP32 (#53877)

2026-08-30 22:15:36 +08:00

test_fused_sigmoid_gating_delta_rule.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423)

2026-07-22 14:43:07 -06:00

test_gate_linear_rocm_dispatch.py

[Hardware][AMD] Enable fused bf16→fp32 router GEMM on ROCm (#50268)

2026-08-12 02:46:13 +00:00

test_kimi_k3_gemm_rs_ar.py

[Kimi-K3] Extend GEMM-RS to GEMM-AR (#53053)

2026-08-21 03:44:15 +00:00

test_ll_bf16_gemm.py

[Kimi-K3][Perf] Prefetch ll_bf16 router weights for M=1 (#53524)

2026-09-01 21:40:19 +00:00

test_mhc_kernels.py

[Bugfix] Fix DeepSeek V4 mHC broadcast buffer for weight sync (#52626)

2026-08-18 10:55:37 +08:00

test_mhc_tilelang_jit.py

[CI/Build][ROCm] Run the TileLang HIP symbol checks in their own interpreter (#53117)

2026-08-21 16:17:42 -05:00

test_minimax_m3_amd_ops.py

[ROCm][MoE] Fix expert_map vs AITER expert_mask for non-AITER experts under EP (#49758)

2026-08-11 15:33:02 -05:00

test_minimax_m3_sparse_attn_fp8_scale.py

[ROCm][MiniMax-M3] Add AITER sparse paged attention (#47287)

2026-07-12 19:27:29 -07:00

test_onednn.py

[CPU] Refactor CPU attention backend (#27954)

2025-11-12 09:43:06 +08:00

test_shuffle_rows.py

[Bugfix] Handle padded routes in CUTLASS MoE permutations (#54747)

2026-09-01 19:08:17 -07:00

test_top_k_per_row.py

[Perf][Kernel] Tune cooperative topk for medium batch-sizes (#53382)

2026-08-31 13:44:38 -07:00

utils.py

[ROCm][CI] Add More AITER quantization/MoE kernel tests (#49375)

2026-08-05 21:27:33 +00:00