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-kernels/src
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
..
affine.cu
Add the kernels.
2023-06-30 10:26:56 +01:00
binary_op_macros.cuh
Add some cmp tests. (#233)
2023-07-24 16:53:45 +01:00
binary.cu
Add some cmp tests. (#233)
2023-07-24 16:53:45 +01:00
cast.cu
Bugfix: remove the u8/bf16 conversion kernel as it is ambiguous.
2023-06-30 10:43:32 +01:00
compatibility.cuh
Add the bf16 cuda kernels.
2023-06-29 23:12:02 +01:00
conv.cu
Make it easier to use whisper samples from the repo. (#112)
2023-07-08 18:48:27 +01:00
cuda_utils.cuh
Cuda kernels for fast min/max reductions (#203)
2023-07-19 18:12:27 +01:00
fill.cu
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00
indexing.cu
Add a test for scatter add. (#238)
2023-07-25 09:12:14 +01:00
lib.rs
Cuda kernels for IndexAdd/ScatterAdd. (#236)
2023-07-24 21:53:08 +01:00
reduce.cu
Cleanup some todos. (#226)
2023-07-23 16:00:00 +01:00
ternary.cu
Add the kernels.
2023-06-30 10:26:56 +01:00
unary.cu
Add some flash attn test (#253)
2023-07-26 20:56:00 +01:00