summaryrefslogtreecommitdiff
path: root/candle-flash-attn/kernels
Commit message (Expand)AuthorAgeFilesLines
* Update the flash attn kernels. (#2333)Laurent Mazare2024-07-1549-898/+2257
* Use flash-attn in gemma. (#2195)Laurent Mazare2024-05-181-0/+4
* chore: update flash attention kernels (#1518)OlivierDehaene2024-01-0526-451/+658
* Add back the bf16 flash-attn kernels. (#730)Laurent Mazare2023-09-041-13/+13
* Flash attention without padding (varlen). (#281)Laurent Mazare2023-07-311-3/+6
* Again set a few extra params in flash-attn. (#245)Laurent Mazare2023-07-2617-91/+379
* Proper flash-attn parameters. (#244)Laurent Mazare2023-07-261-1/+22
* Add flash attention (#241)Laurent Mazare2023-07-2610-0/+2361