candle/candle-kernels
Charles Lew 1c09164021
Add `CANDLE_NVCC_CCBIN` support for `candle-kernels`, and eliminate warning. (#836)
2023-09-13 11:39:22 +01:00
..
src im2col version of the conv1d kernel. (#815) 2023-09-11 14:40:09 +01:00
Cargo.toml Bump the crate version + update the changelog. (#822) 2023-09-12 06:39:24 +01:00
README.md Revert "Add the layer norm files. (#222)" (#223) 2023-07-22 16:51:11 +01:00
build.rs Add `CANDLE_NVCC_CCBIN` support for `candle-kernels`, and eliminate warning. (#836) 2023-09-13 11:39:22 +01:00

README.md

candle-kernels

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