huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 11:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity
659 Commits 74 Branches 16 Tags
d2dea11ef65566518b6d722e9dbe33822aa98b74
Commit Graph

1 Commits

Author SHA1 Message Date
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