huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
afc10a3232b218dcdb8c3b0989f1066940ea992b
candle/candle-kernels
History
Laurent Mazare ca318a6ec7 Add to the cuda example a reproduction of the issue. (#579) 
* Add to the cuda example a reproduction of the issue.

* Tweak.

* Add a test using non-square matrixes.

* Fix the conv2d kernel.

* Display the error.

* And tweak the comment.
2023-08-24 12:07:31 +01:00
..
src
Add to the cuda example a reproduction of the issue. (#579)
2023-08-24 12:07:31 +01:00
build.rs
Revert "Add the layer norm files. (#222)" (#223)
2023-07-22 16:51:11 +01:00
Cargo.toml
Add some group parameter to convolutions. (#566)
2023-08-23 12:58:55 +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.