FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness

Item #:

068431-1189

Details

DOI: https://doi.org/10.52202/068431-1189
Author(s): Tri Dao, Stefano Ermon, Dan Fu, Christopher Ré, Atri Rudra
Pages: 16344-16359 (16 pages)
Format: PDF Paper Download
Conference: Advances in Neural Information Processing Systems 35
Date/Location: Held 28 November - 9 December 2022, New Orleans, Louisiana, USA.
Series: Advances in Neural Information Processing Systems 35
Publisher: Neural Information Processing Systems Foundation, Inc. (NeurIPS)

DOI: https://doi.org/10.52202/068431-1189
Author(s): Tri Dao, Stefano Ermon, Dan Fu, Christopher Ré, Atri Rudra
Pages: 16344-16359 (16 pages)
Format: PDF Paper Download
Conference: Advances in Neural Information Processing Systems 35
Date/Location: Held 28 November - 9 December 2022, New Orleans, Louisiana, USA.
Series: Advances in Neural Information Processing Systems 35
Publisher: Neural Information Processing Systems Foundation, Inc. (NeurIPS)