Tri Dao ca81f32e04 Implement rotary embedding in CUDA		2022-11-04 22:42:01 -07:00
..
test_flash_attn.py	Fix race condition in Triton bwd when there's bias	2022-11-04 11:20:27 -07:00
test_rotary.py	Implement rotary embedding in CUDA	2022-11-04 22:42:01 -07:00