easonabc-public
/
paia-hw3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add cuda cfg

This commit is contained in:
Eason 2024-04-25 17:49:03 +08:00
parent 974c1764d4
commit bd58db47d1
2 changed files with 38 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
pyr/Cargo.lock generated
View File

@ -136,6 +136,17 @@ version = "1.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f1fdabc7756949593fe60f30ec81974b613357de856987752631dea1e3394c80"
[[package]]
name = "bindgen_cuda"
version = "0.1.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1f8489af5b7d17a81bffe37e0f4d6e1e4de87c87329d05447f22c35d95a1227d"
dependencies = [
"glob",
"num_cpus",
"rayon",
]
[[package]]
name = "bitflags"
version = "1.3.2"
@ -197,6 +208,8 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6f1b20174c1707e20f4cb364a355b449803c03e9b0c9193324623cf9787a4e00"
dependencies = [
"byteorder",
"candle-kernels",
"cudarc",
"gemm",
"half",
"memmap2",
@ -211,6 +224,15 @@ dependencies = [
"zip",
]
[[package]]
name = "candle-kernels"
version = "0.4.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5845911a44164ebb73b56a0e23793ba1b583bad102af7400fe4768babc5815b2"
dependencies = [
"bindgen_cuda",
]
[[package]]
name = "candle-nn"
version = "0.4.1"
@ -291,6 +313,15 @@ version = "0.2.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7a81dae078cea95a014a339291cec439d2f232ebe854a9d672b796c6afafa9b7"
[[package]]
name = "cudarc"
version = "0.10.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9395df0cab995685664e79cc35ad6302bf08fb9c5d82301875a183affe1278b1"
dependencies = [
"half",
]
[[package]]
name = "deranged"
version = "0.3.11"
@ -546,6 +577,12 @@ dependencies = [
"wasi",
]
[[package]]
name = "glob"
version = "0.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d2fabcfbdc87f4758337ca535fb41a6d701b65693ce38287d856d1674551ec9b"
[[package]]
name = "half"
version = "2.4.1"

2
pyr/Cargo.toml
View File

@ -25,4 +25,4 @@ serde = {version = "1.0.198", features = ["derive"]}
[features]
default = []
cuda = []
cuda = ["candle-nn/cuda"]