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
d2dea11ef65566518b6d722e9dbe33822aa98b74
candle/candle-flash-attn/tests
History
Laurent Mazare 4f92420132 Add some flash attn test (#253) 
* Add some flash-attn test.

* Add the cpu test.

* Fail when the head is not a multiple of 8.

* Polish the flash attention test.
2023-07-26 20:56:00 +01:00
..
flash_attn_tests.rs
Add some flash attn test (#253)
2023-07-26 20:56:00 +01:00