Wei Zhao 9d88ceb026 [KDA] Update flashKDA to support bf16 checkpoint state (#56485) ...

Signed-off-by: wzhao18 <wzhao18.sz@gmail.com>

2026-09-11 23:53:24 +00: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 reland (#54038)

2026-09-10 15:37:54 +08: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

[Kimi-K3][Perf] Make native CUDA AttnRes the SM100 default (#54261)

2026-08-31 23:18:12 +00:00

test_aux_attn_res_stream.py

[Core][MRV2] Support eagle3 spec decode with pipeline parallel (#50514)

2026-09-05 11:45:34 +08:00

test_eagle3.py

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

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

test_kda_metadata.py

[Kimi K3 Perf] Avoid KDA mixed-batch gather/scatter, 5.2%~7.7% E2E Throughput Improvement (#56159)

2026-09-10 17:57:01 +08:00

test_kda.py

[KDA] Update flashKDA to support bf16 checkpoint state (#56485)

2026-09-11 23:53:24 +00:00

test_latent_moe_tail.py

[Fast Start] Support fp4 (#55465)

2026-09-10 13:22:17 +00: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

test_weight_loading.py

[Bugfix] Fix Kimi K3 loading with interleaved weight streams (#53379)

2026-09-08 16:08:51 +02:00