Startsida Utforska Hjälp
Registrera dig Logga in
david
/
flash-attention
spegling av https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Filer Ärenden 0 Wiki
Träd: 9b6cba16c1
Grenar Taggar
flash-attention
/
csrc
/
flash_attn
Antoni Viros 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) 2 månader sedan
..
src 65f723bb9a Split bwd into more .cu files to speed up compilation 4 månader sedan
flash_api.cpp 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) 2 månader sedan