Tri Dao 9e5e8bc91e Change causal mask to be aligned to bottom-right instead of top-left		2023-08-24 23:41:07 -07:00
..
benchmark_causal.py	Change causal mask to be aligned to bottom-right instead of top-left	2023-08-24 23:41:07 -07:00
benchmark_flash_attention.py	[Benchmark] Add script to benchmark FlashAttention	2023-07-28 00:26:52 -10:00