vllm/attention/backends/rocm_flash_attn.py · e97f802b2d74861af77997691a7d1c36498f6dca · OpenDAS / vllm_cscc

Gregory Shtrasberg authored Jan 23, 2025


Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Co-authored-by: Micah Williamson <micah.williamson@amd.com>

e97f802b

rocm_flash_attn.py 28 KB