3 people 463aa5e30f [ROCm][Perf] Kimi-K3 Fused kernels for KDA prefill (#52606) ...

Signed-off-by: kliuae <kuanfu.liu@embeddedllm.com>
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Co-authored-by: Claude <noreply@anthropic.com>
Co-authored-by: TJian <tunjian.tan@embeddedllm.com>

2026-08-21 09:07:13 -07:00

..

__init__.py

[ROCm][Perf] Kimi-K3 Shard Latent MoE up-projection for ROCm path (#51253)

2026-08-07 06:11:18 +00:00

test_amd_attn_res.py

[Kimi-K3][AMD] Fuse AttnRes state updates and norms (#50593)

2026-08-04 12:32:13 -04:00

test_amd_kda_chunk.py

[ROCm][Perf] Kimi-K3 Fused kernels for KDA prefill (#52606)

2026-08-21 09:07:13 -07:00

test_amd_kda_decode.py

[ROCm][Perf] Enable fused KDA decode on gfx942 (MI325X) (#52293)

2026-08-18 18:01:00 +00:00

test_amd_latent_moe_runner.py

[ROCm][Perf] Kimi-K3 Shard Latent MoE up-projection for ROCm path (#51253)

2026-08-07 06:11:18 +00:00

test_attn_res.py

[Model] Add Kimi K3 support: model files and kernels [1/N] (#50089)

2026-07-29 14:10:58 +08:00

test_aux_attn_res_stream.py

[Model][Spec Decode] Tap the pre-norm AttnRes mixture as the Kimi K3 DFlash aux state (#50487)

2026-08-14 09:03:42 -07:00

test_eagle3.py

[Model][Spec Decode] Tap the pre-norm AttnRes mixture as the Kimi K3 DFlash aux state (#50487)

2026-08-14 09:03:42 -07:00

test_kda_metadata.py

[K3] support recoverssm for K3 (#51855)

2026-08-17 19:34:14 +08:00

test_kda.py

[K3] support recoverssm for K3 (#51855)

2026-08-17 19:34:14 +08:00

test_latent_moe_tail.py

[Model] Add Kimi K3 support: model files and kernels [1/N] (#50089)

2026-07-29 14:10:58 +08:00

test_mla_prefill_context.py

[Attention][MLA] Fuse Kimi-K3 chunked-context K/V packing (#51772)

2026-08-12 22:42:57 -07:00

test_sequence_parallel.py

[Kimi-K3] Add GEMM-RS for sequence parallelism (#52079)

2026-08-13 09:02:28 -07:00