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

Files

Gonzalo 8f7973958c fix: fix index_select cuda kernel for src target dim different than ids dim when selecting dim > 0 (#1037 )

* fix: fix index_select cuda kernel for src target dim different than ids dim when selecting dim > 0

* cargo fmt

2023-10-05 18:46:13 +01:00

2023-10-05 18:46:13 +01:00

build.rs

2023-09-13 11:39:22 +01:00

Cargo.toml

2023-10-01 13:51:57 +01:00

README.md

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

candle-kernels

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