huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,385 Commits 74 Branches 16 Tags
17313a4226a6c6bde444d28b4be4f0f96d155be7
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
a2e925462c Add the scatter in place ops. (#2923) 
* Add the scatter_set op.

* Metal op.

* Cuda version.

* Merge the checks.

* Add the actual ops.
 2025-04-26 07:36:49 +02:00
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
Laurent Mazare
665da30487 Backend refactoring. (#1966) 
* Backend refactoring.

* Metal tweaks.

* Move the cudnn module.
 2024-03-29 23:02:11 +01:00