candle/candle-kernels
Laurent Mazare a4c56a958e
Add the const-set op. (#2910)
* Add the const-set op.

* Cuda implementation.

* Bugfix.

* Metal cleanup.

* Add the metal kernels.

* Add some testing.

* Finish the metal implementation.

* Bump the version.
2025-04-19 10:07:02 +02:00
..
src Add the const-set op. (#2910) 2025-04-19 10:07:02 +02:00
Cargo.toml Add the const-set op. (#2910) 2025-04-19 10:07:02 +02:00
README.md Revert "Add the layer norm files. (#222)" (#223) 2023-07-22 16:51:11 +01:00
build.rs Update to cudarc 0.14 (breaking change). (#2858) 2025-04-03 09:12:19 +02:00

README.md

candle-kernels

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