vllm/attention/ops/flashmla.py · cd9b9de1fb009cf607403ba08961f2a3f869931d · OpenDAS / vllm_cscc

[BugFix] Fix IMA FlashMLA full cuda-graph and DP + Update FlashMLA (#21691) · cd9b9de1

Lucas Wilkinson authored Aug 08, 2025


Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
Co-authored-by: yewentao256 <zhyanwentao@126.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>

cd9b9de1

flashmla.py 3.85 KB

Replace flashmla.py