Andreas KaratzasandClaude 67516b6c42 [Bugfix][ROCm][Kimi-K3] Stage fused KDA decode weights after loading (#61207) ...

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Co-authored-by: Claude <noreply@anthropic.com>

2026-10-13 14:55:57 -05: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

[ROCm][Kimi-K3] Use vLLM's FP8 max in the fused AttnRes quantization (#60717)

2026-10-09 04:45:13 +00:00

test_amd_kda_checkpoint.py

[Bugfix][Prefix caching] Fix inconsistent hybrid Mamba prefix cache boundaries (#60533)

2026-10-08 18:33:48 +00:00

test_amd_kda_chunk.py

[ROCm][Perf] Kimi-K3 enable prefill checkpoints on ROCm (#58344)

2026-10-02 06:58:04 +00:00

test_amd_kda_decode.py

[Bugfix][ROCm][Kimi-K3] Stage fused KDA decode weights after loading (#61207)

2026-10-13 14:55:57 -05:00

test_amd_kda_direct_return.py

[Kimi-K3][AMD] Return KDA and MLA projection outputs directly (#50592)

2026-09-21 15:44:41 -05:00

test_amd_latent_moe_runner.py

[ROCm][Perf] Kimi-K3 Store only the current rank's shards in latent MoE up-proj (#59591)

2026-10-08 14:55:20 +00:00

test_amd_mla_direct_return.py

[ROCm][Kimi-K3] Fuse AttnRes output with per-token FP8 input quantization (#59069)

2026-10-08 12:01:47 +00:00

test_attn_res.py

[CI][ROCm] Add an MI355 Kimi-K3 unit test group (#58012)

2026-09-23 11:25:51 -05:00

test_aux_attn_res_stream.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_eagle3.py

[CI/Build] Fix Kimi K3 Eagle3 test fixture (#54482)

2026-08-31 03:29:45 +00:00

test_gfx942_int4.py

[Quantization] Enable shared expert fusion compatibility with online shared_expert quantization (showcase: along Quark MXFP4 routed experts) (#55686)

2026-10-08 12:04:46 -04:00

test_kda_metadata.py

[MRV2] Switch CI tests still running on MRV1 unnecessarily (#61014)

2026-10-11 09:57:07 -07:00

test_kda.py

[Bugfix][RecoverSSM] Correct aligned state indices at exact block boundaries (#59962)

2026-10-12 02:08:14 +00:00

test_latent_moe_tail.py

[MoE] Defer the TRTLLM-Gen top-k finalize on the modular path (#58635)

2026-09-25 15:38:29 +08:00

test_mla_prefill_context.py

[Kimi-K3] Use the fused K/V pack kernel on the decode-context-parallel prefill path (#56881)

2026-10-06 00:13:55 +00:00

test_nvidia_kda_direct_return.py

[Kimi-K3][AMD] Return KDA and MLA projection outputs directly (#50592)

2026-09-21 15:44:41 -05:00

test_prefix_cache.py

[Bugfix][Mooncake] Save the EAGLE attention proof with hybrid Mamba partial tails (#60695)

2026-10-10 00:08:08 +00:00

test_sequence_parallel.py

[Perf] Enable Kimi K3 shared expert sharding with DeepEP v2 (#56875)

2026-10-08 16:42:15 +00:00