candle/candle-kernels
Nicolas Patry 30313c3081
Moving to a proper build crate `bindgen_cuda`. (#1531)
* Moving to a proper build crate `bindgen_cuda`.

* Fmt.
2024-01-07 12:29:24 +01:00
..
src Rework the cuda casting bits. (#1112) 2023-10-17 09:44:51 +01:00
Cargo.toml Moving to a proper build crate `bindgen_cuda`. (#1531) 2024-01-07 12:29:24 +01:00
README.md Revert "Add the layer norm files. (#222)" (#223) 2023-07-22 16:51:11 +01:00
build.rs Moving to a proper build crate `bindgen_cuda`. (#1531) 2024-01-07 12:29:24 +01:00

README.md

candle-kernels

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