Files
candle/candle-kernels
Laurent Mazare 9a5c7db91a Add support for i64 (#563)
* Add the i64 dtype.

* Adapt the cuda kernels.
2023-08-23 10:42:19 +01:00
..
2023-08-23 10:42:19 +01:00

candle-kernels

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