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
c5626b827147e5029c6bd3e37352ec8ac501cfc3
candle/candle-kernels
History
Thomas Santerre c5626b8271 Add support for "sign" on tensors (#2012) 
* add the sign unary operator

* remove uneeded import

* remove uneeded import

* undo formatting

* undo formatting

* remove unnecessary redefintion

* allow gradient to flow through for sign and round

* fix cpu ops to ensure that negzero and positive zero are handled properly

* clippy fixes

* Properly avoid gradient tracking.

* Use a branchless version.

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>
2024-04-04 22:32:47 +02:00
..
src
Add support for "sign" on tensors (#2012)
2024-04-04 22:32:47 +02:00
build.rs
Ensure that the kernels get rebuilt on cuh changes. (#1954)
2024-03-28 06:56:48 +01: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

README.md

candle-kernels

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