cao lei
|
95d65e4ec0
|
sync to flash attention kernel 2.5.9 and add document of how to write custom op (#757)
* sync to flash attention kernel 2.5.9
* support users to overload GetMayInplace and ReleaseMayInplace
* Undo the change for pybind11 dependency
|
2024-07-10 07:09:40 -07:00 |