Tri Dao 5badfb7848 Implement attention kernel that splits the batch into two		2022-10-13 20:49:02 -07:00
..
flash_attn	Implement attention kernel that splits the batch into two	2022-10-13 20:49:02 -07:00