tests/kernels/test_flash_attn.py · 1356df53bd5d6877358aff3d2bbd95f28f8009a4 · OpenDAS / vllm_cscc

Stephen Krider authored May 13, 2024


Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Co-authored-by: LiuXiaoxuanPKU <lilyliupku@gmail.com>

1356df53

test_flash_attn.py 7.12 KB