Página inicial Explorar Ajuda
Registrar Entrar
david
/
flash-attention
mirror de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Arquivos Issues 0 Wiki
Branch: changes_for_fp8
Branches Tags
flash-attention
/
csrc
/
rotary
Chirag Jain 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 meses atrás
..
rotary.cpp e45a46a5b7 [Rotary] Implement GPT-J style (interleaved) rotary 1 ano atrás
rotary_cuda.cu e45a46a5b7 [Rotary] Implement GPT-J style (interleaved) rotary 1 ano atrás
setup.py 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 meses atrás