huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 03:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4d04ac83c7fd65c6f6e5f08ab87f27ab6e5af21b
candle/candle-kernels/src
History
Gonzalo 8f7973958c fix: fix index_select cuda kernel for src target dim different than ids dim when selecting dim > 0 (#1037) 
* fix: fix index_select cuda kernel for src target dim different than ids dim when selecting dim > 0

* cargo fmt
2023-10-05 18:46:13 +01:00
..
affine.cu
Add support for i64 (#563)
2023-08-23 10:42:19 +01:00
binary_op_macros.cuh
Add some cmp tests. (#233)
2023-07-24 16:53:45 +01:00
binary.cu
Add support for i64 (#563)
2023-08-23 10:42:19 +01:00
cast.cu
cuda cast i64 (#925)
2023-09-21 19:52:39 +01:00
compatibility.cuh
Compat windows.
2023-08-10 17:46:47 +02:00
conv.cu
im2col version of the conv1d kernel. (#815)
2023-09-11 14:40:09 +01:00
cuda_utils.cuh
Add the rounding operators. (#1030)
2023-10-04 17:58:44 +01:00
fill.cu
fix: add missing gpu fill_* (#996)
2023-09-29 15:49:30 +01:00
indexing.cu
fix: fix index_select cuda kernel for src target dim different than ids dim when selecting dim > 0 (#1037)
2023-10-05 18:46:13 +01:00
lib.rs
Cuda kernels for IndexAdd/ScatterAdd. (#236)
2023-07-24 21:53:08 +01:00
reduce.cu
Add a dedicated cuda kernel for softmax. (#746)
2023-09-05 17:53:20 +02:00
ternary.cu
Add support for i64 (#563)
2023-08-23 10:42:19 +01:00
unary.cu
Add the rounding operators. (#1030)
2023-10-04 17:58:44 +01:00