huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 02:16:37 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
38fc86621ce7dfde7161b61edf94330a1797cb04
candle/candle-kernels
History
Laurent Mazare e3db30021f Support for "unbatched" rope. (#2926) 
* Support for (un)-batched rope.

* Use 3d rope in the rope/ropei/rope_thd functions.

* Get the CPU versions to work.

* Fix the cuda version.

* Adapt the metal side.

* Fix the metal tests.
2025-04-27 15:12:02 +02:00
..
src
Support for "unbatched" rope. (#2926)
2025-04-27 15:12:02 +02:00
build.rs
Update to cudarc 0.14 (breaking change). (#2858)
2025-04-03 09:12:19 +02:00
Cargo.toml
Bump the crate version to 0.9.0. (#2924)
2025-04-26 11:01:21 +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.