Files
candle/candle-kernels
Laurent Mazare 536c5e702e Cuda kernels for fast min/max reductions (#203)
* Add the min/max cuda kernels.

* Better integration of the cuda kernels.
2023-07-19 18:12:27 +01:00
..
2023-07-05 18:12:17 +02:00
2023-06-27 11:57:27 +02:00
2023-06-27 11:57:27 +02:00

candle-kernels

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