huggingface/candle

mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00

Files

History

Laurent Mazare 2fe24ac5b1 Rework the cuda casting bits. (#1112 )

2023-10-17 09:44:51 +01:00

..

Rework the cuda casting bits. (#1112 )

2023-10-17 09:44:51 +01:00

build.rs

feat: parse Cuda compute cap from env (#1066 )

2023-10-16 15:37:38 +01:00

Cargo.toml

feat: parse Cuda compute cap from env (#1066 )

2023-10-16 15:37:38 +01:00

README.md

Revert "Add the layer norm files. (#222 )" (#223 )

2023-07-22 16:51:11 +01:00

README.md

candle-kernels

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