Tri Dao 7c9953815a Add fused cross entropy loss		2022-11-12 21:58:41 -08:00
..
flash_attn	Fix out-of-bound memory read	2022-11-09 09:34:14 -08:00
fused_softmax	Add Megatron attention implementation for benchmarking	2022-10-23 23:04:16 -07:00
rotary	Implement rotary embedding in CUDA	2022-11-04 22:42:01 -07:00
xentropy	Add fused cross entropy loss	2022-11-12 21:58:41 -08:00