huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
18036c6ccbd216e0397fa6469bb50a8b8ea47892
candle/candle-kernels
History
Laurent Mazare af7f8b87d3 Custom op for RmsNorm (#1890) 
* Trying out a custom RmsNorm cuda kernel.

* CPU implementation for rms-norm.

* Cuda wrappers.

* Add some validation.

* Add some testing.

* More testing.
2024-03-21 06:36:28 +01:00
..
src
Custom op for RmsNorm (#1890)
2024-03-21 06:36:28 +01:00
build.rs
Moving to a proper build crate bindgen_cuda. (#1531)
2024-01-07 12:29:24 +01:00
Cargo.toml
Bump the crate versions to 0.4.2. (#1821)
2024-03-08 22:01:51 +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.