Files
candle/candle-kernels
Laurent Mazare b219903d0f Cuda backend optimization (#1886)
* Attempt at making the kernel faster.

* Also adapt the cast kernels.

* Also apply to binary ops.
2024-03-20 18:32:55 +01:00
..
2024-03-20 18:32:55 +01:00

candle-kernels

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