Add cuda feature

This commit is contained in:
Louis Dureuil
2024-01-07 21:27:29 +01:00
parent 262b20fdba
commit 84f49d76cd
3 changed files with 25 additions and 1 deletions

21
Cargo.lock generated
View File

@ -706,6 +706,8 @@ version = "0.3.3"
source = "git+https://github.com/huggingface/candle.git#5270224f407502b82fe90bc2622894ce3871b002"
dependencies = [
"byteorder",
"candle-kernels",
"cudarc",
"gemm",
"half 2.3.1",
"memmap2 0.9.3",
@ -720,6 +722,16 @@ dependencies = [
"zip",
]
[[package]]
name = "candle-kernels"
version = "0.3.1"
source = "git+https://github.com/huggingface/candle.git#f4fcf6090045ac44122fd5f0a7e46db6e3e16528"
dependencies = [
"anyhow",
"glob",
"rayon",
]
[[package]]
name = "candle-nn"
version = "0.3.3"
@ -1163,6 +1175,15 @@ dependencies = [
"memchr",
]
[[package]]
name = "cudarc"
version = "0.9.15"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f1871a911a2b9a3f66a285896a719159985683bf9903aa2cf89e0c9f53e14552"
dependencies = [
"half 2.3.1",
]
[[package]]
name = "darling"
version = "0.14.4"