mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00

Files

Laurent Mazare 1df2bddccf Add the layernorm specialized op. (#2212 )

* Add the layernorm cuda kernels.

* Dedicated layer norm op.

* Add the slower variant.

* Plug the cuda implementation.

* Add the metal variant.

* Add a dedicated test.

* Bugfix.

2024-05-24 15:58:01 +02:00

src

Add the layernorm specialized op. (#2212 )

2024-05-24 15:58:01 +02:00

build.rs

Ensure that the kernels get rebuilt on cuh changes. (#1954 )

2024-03-28 06:56:48 +01:00

Cargo.toml

Bump the version number to 0.5.1. (#2155 )

2024-05-03 11:17:05 +02: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.