Home Explore Help
Register Sign In
david
/
flash-attention
mirror of https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Files Issues 0 Wiki
Tree: fb9c9cbbe9
Branches Tags
flash-attention
/
csrc
/
flash_attn
Antoni Viros 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) 2 months ago
..
src 65f723bb9a Split bwd into more .cu files to speed up compilation 4 months ago
flash_api.cpp 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) 2 months ago