vllm/_aiter_ops.py · 130d6c9514856cb5a152329f0382d60ff6e8d97e · OpenDAS / vllm_cscc

[ROCm][Perf] Enable shuffle kv cache layout and assembly paged attention... · 130d6c95

Pleaplusone authored Jan 15, 2026


[ROCm][Perf] Enable shuffle kv cache layout and assembly paged attention kernel for `AiterFlashAttentionBackend` (#29887)
Signed-off-by: ganyi <ygan@amd.com>

130d6c95

_aiter_ops.py 46.7 KB

Replace _aiter_ops.py