dnrops
/
candle
mirror of https://github.com/huggingface/candle.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
candle/candle-kernels
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
..
src 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
Cargo.toml Bump the version to 0.3.0. (#1014) 2023-10-01 13:51:57 +01:00
README.md Revert "Add the layer norm files. (#222)" (#223) 2023-07-22 16:51:11 +01:00
build.rs Add `CANDLE_NVCC_CCBIN` support for `candle-kernels`, and eliminate warning. (#836) 2023-09-13 11:39:22 +01:00

README.md

candle-kernels

This crate contains CUDA kernels used from candle. Some of these implementations come from the dfdx crate.