huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
75629981bc2b101400a301803c027da2362a4ff9
candle/candle-kernels
History
OlivierDehaene 75629981bc feat: parse Cuda compute cap from env (#1066) 
* feat: add support for multiple compute caps

* Revert to one compute cap

* fmt

* fix
2023-10-16 15:37:38 +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
build.rs
feat: parse Cuda compute cap from env (#1066)
2023-10-16 15:37:38 +01:00
Cargo.toml
feat: parse Cuda compute cap from env (#1066)
2023-10-16 15:37:38 +01:00
README.md
Revert "Add the layer norm files. (#222)" (#223)
2023-07-22 16:51:11 +01:00

README.md

candle-kernels

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