[ROCm][MLA][Bugfix] Reserve FP8 prefill workspace before lock for Kimi-K2.5 #42978
Fixes a ROCm AITER dense MLA FP8 prefill crash where the per-call scr…
3f08ba1d
xaguilar-amd
marked this pull request as ready for review 168 days ago
Adressed a Gemini comment about the understimation of the amount rese…
9c63edf5
xaguilar-amd
changed the title [ROCm][MLA][BUG] Reserve FP8 prefill workspace before lock for Kimi-K2.5 [ROCm][MLA][Bugfix] Reserve FP8 prefill workspace before lock for Kimi-K2.5 168 days ago
Address Rohan's nit: reduce_partial_map_size is always an int
7c39d0af
micah-wil
approved these changes
on 2026-06-02
[ROCm][MLA][Bugfix] Address review: drop hot-path refactor, fix final…
f0334982
Merge branch 'main' into fix_mla_ps_prefill_bug
daf7688d
Merge branch 'main' into fix_mla_ps_prefill_bug
366c146a
vllm-bot
merged
baacbfce
into main 146 days ago
AndreasKaratzas
added this to the v0.23.0 cherry picks milestone 145 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub