vllm/attention/backends/rocm_flash_attn.py · f98548b9da39bce8aaff64974e1e68208aeada5e · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[torch.compile][ROCm] Fuse quantization onto attention using a torch.compile pass (#16756) · f98548b9
Luka Govedič authored Jun 12, 2025
```
Signed-off-by: Luka Govedič <lgovedic@redhat.com>
Co-authored-by: Sage Moore <sage@neuralmagic.com>
```
f98548b9

rocm_flash_attn.py 43.5 KB