huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 19:58:35 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
5c3864f9f765f3283fc82bbe1ea1aafd45adbad6
candle/candle-kernels
History
Laurent Mazare e676f85f00 Sketch a fast cuda kernel for reduce-sum. (#109) 
* Sketch a fast cuda kernel for reduce-sum.

* Sketch the rust support code for the fast sum kernel.

* More work on the fast kernel.

* Add some testing ground.

* A couple fixes for the fast sum kernel.
2023-07-08 12:43:56 +01:00
..
src
Sketch a fast cuda kernel for reduce-sum. (#109)
2023-07-08 12:43:56 +01:00
build.rs
Fixing the cached build.
2023-07-05 18:12:17 +02:00
Cargo.toml
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00
README.md
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00

README.md

candle-kernels

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