huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 11:37:11 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
74bf6994b172f364c6e8bea2ac6e1bfbc6ca0c25
candle/candle-flash-attn/src
History
OlivierDehaene 8d1a57c9a0 chore: update flash attention kernels (#1518) 
* chore: update flash attention kernels

* fmt

* remove unused kernels

* force f32

* correct stride
2024-01-05 18:28:55 +01:00
..
ffi.rs
chore: update flash attention kernels (#1518)
2024-01-05 18:28:55 +01:00
lib.rs
chore: update flash attention kernels (#1518)
2024-01-05 18:28:55 +01:00