huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
70f38c20693ad26c6fffa359e734bde92124c0dd
candle/candle-kernels
History
Laurent Mazare 98d1242b8f im2col based conv2d (#802) 
* im2col implementation for conv2d.

* Fix for the im2col implementation to match the current conv2d.

* Small optimization.

* Add a cuda kernel.

* Handle arbitrary layouts.

* Im2Col cuda code.
2023-09-10 21:02:42 +01:00
..
src
im2col based conv2d (#802)
2023-09-10 21:02:42 +01:00
build.rs
Revert "Add the layer norm files. (#222)" (#223)
2023-07-22 16:51:11 +01:00
Cargo.toml
Add some documentation. (#673)
2023-08-30 11:54:00 +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.