Página inicial Explorar Ajuda
Registrar Entrar
david
/
flash-attention
mirror de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Arquivos Issues 0 Wiki
Tree: 9b6cba16c1
Branches Tags
flash-attention
/
csrc
/
flash_attn
Antoni Viros 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) 2 meses atrás
..
src 65f723bb9a Split bwd into more .cu files to speed up compilation 4 meses atrás
flash_api.cpp 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) 2 meses atrás