squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
5ca83a9c71
flash-attention/csrc/rotary
History
Chirag Jain 50896ec574
Make nvcc threads configurable via environment variable (#885)
2024-03-13 20:46:57 -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 Make nvcc threads configurable via environment variable (#885) 2024-03-13 20:46:57 -07:00