Skip to content

Commit a44b6b7

Browse files
Update dependencies to include cudarc for enhanced CUDA support
- Added `cudarc` as an optional dependency in `Cargo.toml` with specific features for CUDA integration. - Updated `Cargo.lock` to reflect the inclusion of `cudarc` version 0.16.3, ensuring compatibility with the new features.
1 parent 0e4840e commit a44b6b7

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

Cargo.lock

Lines changed: 1 addition & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

python/Cargo.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ crate-type = ["cdylib"]
99

1010
[dependencies]
1111
embed_anything = {path = "../rust", features = ["ort"]}
12+
cudarc = { version = "0.16.3", features = ["std", "cublas", "cublaslt", "curand", "driver", "nvrtc", "f16", "cuda-version-from-build-system", "dynamic-loading"], default-features=false , optional=true}
13+
1214
processors-rs = {path = "../processors"}
1315
pyo3 = { version = "0.23.2"}
1416
tokio = { version = "1.39.0", features = ["rt-multi-thread"]}
@@ -19,7 +21,7 @@ strum_macros = {workspace = true}
1921
extension-module = ["pyo3/extension-module"]
2022
mkl = ["embed_anything/mkl"]
2123
accelerate = ["embed_anything/accelerate"]
22-
cuda = ["embed_anything/cuda"]
24+
cuda = ["embed_anything/cuda", "dep:cudarc"]
2325
cudnn = ["embed_anything/cudnn"]
2426
metal = ["embed_anything/metal"]
2527
ort = ["embed_anything/ort"]

0 commit comments

Comments
 (0)