4 Commits

Author	SHA1	Message	Date
Tri Dao	5d079fdd7a	[Triton] Fix benchmark_causal, mention Triton version	2023-03-22 00:51:16 -07:00
Tri Dao	b0c0db81f6	Implement FlashAttention in Triton	2022-10-30 18:09:11 -07:00
Tri Dao	ed553e9238	Add Megatron attention implementation for benchmarking	2022-10-23 23:04:16 -07:00
Tri Dao	50ca23488d	Add Triton implementation for benchmarking	2022-10-23 17:25:56 -07:00