vllm/attention/backends/rocm_flash_attn.py · 558db8083cfd9b7ee76eafdae32b237d951c8b10 · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[V1][Kernel] Refactor the prefix_prefill kernel so that the caller no longer... · 558db808
Sage Moore authored Feb 22, 2025
```
[V1][Kernel] Refactor the prefix_prefill kernel so that the caller no longer has to pass in the context lengths (#13095)
```
558db808

rocm_flash_attn.py 37.4 KB