flash-attention/csrc/flash_attn
Grigory Sizov f816dee63c
Support unpadded LSE layout (#970)
* Support unpadded LSE layout.

Co-authored-by: Xinfeng Xie <xfxie.ceca@gmail.com>
Co-authored-by: Jianyu Huang <hjyahead@gmail.com>

* Cleanup

* Fix unpadded LSE on split-kv path

* Fix formatting and comments

* Fix inline vs forceinline

---------

Co-authored-by: Xinfeng Xie <xfxie.ceca@gmail.com>
Co-authored-by: Jianyu Huang <hjyahead@gmail.com>
2024-06-27 02:38:13 -07:00
..
src Support unpadded LSE layout (#970) 2024-06-27 02:38:13 -07:00
flash_api.cpp Support unpadded LSE layout (#970) 2024-06-27 02:38:13 -07:00