tests/kernels/test_flash_attn.py · 073a4bd1c04164af29843cb5478740e9839d2d8a · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[Kernel] Use `out` arg in flash_attn_varlen_func (#10811) · 073a4bd1
Woosuk Kwon authored Dec 01, 2024
```
Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
```
073a4bd1

test_flash_attn.py 8.67 KB