Tri Dao a8fec99a9a Skip flash_attn_split test		2022-11-13 12:27:48 -08:00
..
losses	Add fused cross entropy loss	2022-11-12 21:58:41 -08:00
test_flash_attn.py	Skip flash_attn_split test	2022-11-13 12:27:48 -08:00
test_rotary.py	Implement rotary embedding in CUDA	2022-11-04 22:42:01 -07:00