3 people 63b9931b73 [ROCm][DSv4.1] Paged MXFP4 sparse indexer on aiter's MQA-logits kernel (#58671) ...

Signed-off-by: Mehmet Cagri Kaymak <mehmet.kaymak@amd.com>
Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
Co-authored-by: Shanshan Shen <467638484@qq.com>

2026-09-29 14:37:57 +08:00

..

attention

[ROCm][DSv4.1] Paged MXFP4 sparse indexer on aiter's MQA-logits kernel (#58671)

2026-09-29 14:37:57 +08:00

core

[XPU] Preserve non-contiguous strides when pinning CPU tensors for UVA view (#54874)

2026-09-29 09:43:07 +08:00

fla

[CI] Split (B200) Miscellaneous Kernels into mHC, FLA Ops and Misc named jobs (#58609)

2026-09-26 04:10:14 -07:00

helion

[Refactor] Move auxiliary files out of the repository root (#58572)

2026-09-24 12:38:56 -07:00

ir

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

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

mamba

[Refactor] Remove dead tests utils (#58803)

2026-09-26 09:16:53 -04:00

mhc

[ROCm][DSv4.1][Perf] Run the delayed mHC seams through aiter's fused Triton kernel (#58655)

2026-09-29 10:19:10 +08:00

moe

[ROCm] Bump AITER to v0.1.23 (#58867)

2026-09-27 19:04:58 -05:00

quantization

[Quant] Use canonical N-first weight format for CT WNA16 MoE (#52798)

2026-09-28 11:22:44 -06: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

[Refactor] Remove dead tests utils (#58803)

2026-09-26 09:16:53 -04:00

test_apply_repetition_penalties.py

Add pydocstyle to the ruff rules (#52136)

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

test_awq_int4_to_int8.py

Add pydocstyle to the ruff rules (#52136)

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

test_bf16_skinny_gemm.py

[Bugfix][Kimi-K3] Fix KDA projection overlap on Hopper (#55426)

2026-09-11 19:07:30 +00:00

test_bf16x3_router_gemm_cutedsl.py

[Perf] Improve BF16x3 router GEMM accuracy and make it default on sm100 (#55899)

2026-09-09 07:33:12 -07:00

test_cache_kernels.py

Add pydocstyle to the ruff rules (#52136)

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

test_compressor_kv_cache.py

[ROCm][Perf] Enable layer-aware CSA2 multi-stream overlap for DeepSeek-V4.1-Flash (#57407)

2026-09-27 23:48:01 +08:00

test_concat_mla_q.py

[Bugfix][ROCm] Use the platform FP8 range in the concat MLA q test (#58153)

2026-09-22 18:05:33 -05:00

test_cp_gather_fp8.py

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

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

test_deepseek_v4_cpu_kernels.py

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

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

test_dsv41_mega_attn_layouts.py

[Model][DSv4.1] FlashMLA mega attention and the NVFP4 compressed KV cache (#56935)

2026-09-16 03:07:50 -07:00

test_engram.py

[Perf][DSv4.1] Shard the Engram wkv projection across TP ranks (#58678)

2026-09-26 01:15:57 +00: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

Add pydocstyle to the ruff rules (#52136)

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

test_fp32_router_gemm.py

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

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

test_fused_deepseek_v4_qnorm_rope_kv_insert.py

[CI][ROCm] Add ten AMD parity groups (#57876)

2026-09-21 11:08:06 +00:00

test_fused_deepseek_v32_norm_rope.py

[ROCm][Model][Bugfix] Enable GLM-5.2-MXFP4 on the deepseek_v32 path and fix sparse attention correctness (#51915)

2026-09-22 18:08:51 -05:00

test_fused_indexer_q_rope_quant.py

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

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

test_fused_inv_rope_fp8_quant.py

[Perf][DSv4.1] Fuse small-batch WO-A with inverse RoPE and MXFP8 quant on SM100/SM103 (#58634)

2026-09-26 21:16:12 -07: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

[Perf][Kernel][Quantization] Fuse ReLU2 with static FP8 activation quantization (#53793)

2026-09-14 10:26:15 -07:00

test_gate_linear_rocm_dispatch.py

[ROCm][Perf] Add low-M FP32 router GEMM for gfx950 (#54845)

2026-09-03 10:42:25 -07:00

test_gemm_rs_ar.py

[Kernel][DSV4.1] Fuse MXFP8 wo_b GEMM with sequence-parallel reduce-scatter (#57428)

2026-09-22 01:07:17 -07:00

test_kpool_decode_update_batched.py

[Bugfix][GLM-5.3-Flash] kpool corruption with speculative decoding (#58454)

2026-09-25 12:51:34 -04: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_minimax_m3_amd_ops.py

[Bugfix][Quantization] Fix MXFP8 startup crash on layers below mm_mxfp8 shape limits (#54223)

2026-09-25 02:54:11 +00: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] Speedup LM Head on Arm CPUs (#55352)

2026-09-11 16:21:14 +08:00

test_relu2_fp8_quant.py

[Perf][Kernel][Quantization] Fuse ReLU2 with static FP8 activation quantization (#53793)

2026-09-14 10:26:15 -07:00

test_rocm_fp32_router_gemm.py

[ROCm][Perf] Add low-M FP32 router GEMM for gfx950 (#54845)

2026-09-03 10:42:25 -07:00

test_shuffle_rows.py

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

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

test_top_k_per_row.py

[GLM-5.3-Flash] Route kpool indexer top-k through the shared SparseIndexerTopk dispatcher (#57546)

2026-09-20 11:17:59 +00:00

utils.py

[Refactor] Remove dead tests utils (#58803)

2026-09-26 09:16:53 -04:00