huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 20:22:49 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
34f4b3187ec0e30e439a89ff7211a74050bc2358
candle/candle-kernels
History
Laurent Mazare 34f4b3187e Add a naive conv2d cuda kernel. (#438) 
* Add a naive conv2d cuda kernel.

* Proper conv2d support on the rust side.

* Conv1d testing on gpu.

* Also use the test on gpus.

* Fix the clean-ptx target.
2023-08-14 10:34:42 +01:00
..
src
Add a naive conv2d cuda kernel. (#438)
2023-08-14 10:34:42 +01:00
build.rs
Revert "Add the layer norm files. (#222)" (#223)
2023-07-22 16:51:11 +01:00
Cargo.toml
Add the license files. (#335)
2023-08-07 14:11:27 +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.