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: 6b1d059eda
Branches Tags
flash-attention
/
csrc
/
rotary
sclarkson 1feb711f46 Fix compilation with clang on ARM64 (#1285) 1 month ago
..
rotary.cpp 1feb711f46 Fix compilation with clang on ARM64 (#1285) 1 month ago
rotary_cuda.cu e45a46a5b7 [Rotary] Implement GPT-J style (interleaved) rotary 1 year ago
setup.py 50896ec574 Make nvcc threads configurable via environment variable (#885) 10 months ago