vllm_flash_attn/flash_attn_interface.py · fdf6d72b48aea41f4ae6a89139a453dae554abc8 · gaoqiong / flash-attention · GitLab

Find file Blame History Permalink

Clean up API & Bypass torch.autograd.Function (#30) · fdf6d72b
Woosuk Kwon authored Nov 28, 2024
```
Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
```
fdf6d72b

flash_attn_interface.py 17.1 KB