candle/candle-metal-kernels
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
tmp Starting to fix some tests. 2023-11-30 11:30:31 +01:00
Cargo.toml Bumping the version number to 0.5.0. (#2009) 2024-04-04 17:48:45 +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.