huggingface/candle

mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00

Files

History

laurent 5ca309ecb0 Optimize the unary cuda kernels for the contiguous case.

2023-06-23 18:40:15 +01:00

..

Optimize the unary cuda kernels for the contiguous case.

2023-06-23 18:40:15 +01:00

build.rs

Integrate the kernels bits.

2023-06-22 09:59:00 +01:00

Cargo.toml

Integrate the kernels bits.

2023-06-22 09:59:00 +01:00

README.md

Add a couple cuda kernels from dfdx.

2023-06-22 12:56:29 +01:00

README.md

candle-kernels

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