Inicio Explorar Axuda
Rexistro Iniciar sesión
david
/
flash-attention
réplica de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Ficheiros Incidencias 0 Wiki
Árbore: 9b6cba16c1
Ramas Etiquetas
flash-attention
/
csrc
/
flash_attn
Antoni Viros 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) hai 2 meses
..
src 65f723bb9a Split bwd into more .cu files to speed up compilation hai 4 meses
flash_api.cpp 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) hai 2 meses