tests/kernels/test_flash_attn.py · 29bc01bf3bc26642e4cee15ebd36a6ce5799326d · 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