huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1df2bddccfbb4ab511a8cc3a87476d1fa72416bc
candle/candle-metal-kernels
History
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
tmp
Starting to fix some tests.
2023-11-30 11:30:31 +01:00
Cargo.toml
Bump the version number to 0.5.1. (#2155)
2024-05-03 11:17:05 +02:00
README.md
Adding the actual backend
2023-11-20 14:12:56 +01:00

README.md

candle-metal-kernels

This crate contains Metal kernels used from candle.