vllm/attention/ops/rocm_aiter_mla.py · b2eb2b5ad7090ad3b3e002b200104a82eeb2fa7f · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[Kernel] Apply torch.Tag.needs_fixed_stride_order only for torch==2.6.0 (#19346) · b2eb2b5a
Richard Zou authored Jul 18, 2025
```
Signed-off-by: rzou <zou3519@gmail.com>
```
b2eb2b5a

rocm_aiter_mla.py 3.61 KB