squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
d8aacc510c
flash-attention/csrc/rotary
History
Tri Dao dc08ea1c33 Support H100 for other CUDA extensions
2023-03-15 16:59:27 -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 Support H100 for other CUDA extensions 2023-03-15 16:59:27 -07:00