flash-attention/csrc/flash_attn
Sophia Wisdom 37e32febba
Remove commented out code in bwd (#512)
* Remove lots of comments

* Remove unused traits
2023-09-01 16:43:58 -07:00
..
src Remove commented out code in bwd (#512) 2023-09-01 16:43:58 -07:00
flash_api.cpp Implement splitKV attention 2023-08-29 00:58:29 -07:00