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
e45a46a5b7
flash-attention
/
csrc
/
rotary
History
Tri Dao
e45a46a5b7
[Rotary] Implement GPT-J style (interleaved) rotary
2023-03-14 14:35:53 -07:00
..
rotary_cuda.cu
[Rotary] Implement GPT-J style (interleaved) rotary
2023-03-14 14:35:53 -07:00
rotary.cpp
[Rotary] Implement GPT-J style (interleaved) rotary
2023-03-14 14:35:53 -07:00
setup.py
Implement rotary embedding in CUDA
2022-11-04 22:42:01 -07:00