squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
654 Commits 1 Branch 0 Tags 7.9 MiB
65f723bb9a
Commit Graph

3 Commits

Author SHA1 Message Date
janEbert
3c4053b75c
Make FA3 externally importable (#1053) 
Library name to import is `flash_attn_interface`, which matches the
test.
 2024-07-22 21:34:56 -07:00
Tri Dao
74b0761ff7 [FA3] BF16 forward 2024-07-14 23:39:46 -07:00
Tri Dao
7f67966cc7 FA3 initial code release 2024-07-11 09:53:36 -07:00