david/flash-attention

mirror of https://github.com/Dao-AILab/flash-attention

Author	SHA1 Message	Date
Tri Dao	db8ca79609 Use bf16 instead of fp16 in benchmark_gemm.py	3 weeks ago
Garrett Byrd	a69fc21595 Added hipBLAS/cuBLAS distinction in benchmark_gemm.py (#1393)	1 month ago
Tri Dao	c4b9015d74 Add benchmark_gemm.py	7 months ago