huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
3eb2bc6d07f192a5ce73ab6964745275f2c15213
candle/candle-flash-attn
History
Laurent Mazare 3eb2bc6d07 Softmax numerical stability. (#267) 
* Softmax numerical stability.

* Fix the flash-attn test.
2023-07-28 13:13:01 +01:00
..
cutlass @ c4f6b8c6bc
Add flash attention (#241)
2023-07-26 07:48:10 +01:00
kernels
Again set a few extra params in flash-attn. (#245)
2023-07-26 14:16:37 +01:00
src
Add some flash attn test (#253)
2023-07-26 20:56:00 +01:00
tests
Softmax numerical stability. (#267)
2023-07-28 13:13:01 +01:00
build.rs
Add some flash attn test (#253)
2023-07-26 20:56:00 +01:00
Cargo.toml
Softmax numerical stability. (#267)
2023-07-28 13:13:01 +01:00