dnrops
/
candle
mirror of https://github.com/huggingface/candle.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
candle/candle-kernels
History
Laurent Mazare 2ac302a5d1
Add the rope THD kernel. (#2014) 
* Add the rope THD kernel.

* Cuda kernel for rope-thd.

* Add the metal kernels.

* Add a dedicated test.
 		2024-04-05 08:32:58 +02:00
..
src Add the rope THD kernel. (#2014) 2024-04-05 08:32:58 +02:00
Cargo.toml Bumping the version number to 0.5.0. (#2009) 2024-04-04 17:48:45 +02:00
README.md Revert "Add the layer norm files. (#222)" (#223) 2023-07-22 16:51:11 +01:00
build.rs Ensure that the kernels get rebuilt on cuh changes. (#1954) 2024-03-28 06:56:48 +01:00

README.md

candle-kernels

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