This website requires JavaScript.
Explore
Help
Register
Sign In
squall
/
flash-attention
Watch
1
Star
0
Fork
0
You've already forked flash-attention
Code
Issues
Pull Requests
Actions
Packages
Projects
Releases
Wiki
Activity
226a1b721d
flash-attention
/
csrc
/
rotary
History
Tri Dao
ca81f32e04
Implement rotary embedding in CUDA
2022-11-04 22:42:01 -07:00
..
rotary_cuda.cu
Implement rotary embedding in CUDA
2022-11-04 22:42:01 -07:00
rotary.cpp
Implement rotary embedding in CUDA
2022-11-04 22:42:01 -07:00
setup.py
Implement rotary embedding in CUDA
2022-11-04 22:42:01 -07:00