Skip to content

Commit d8206a3

Browse files
authored
Fix linking bis (#549)
1 parent 943be86 commit d8206a3

File tree

3 files changed

+9
-13
lines changed

3 files changed

+9
-13
lines changed

Cargo.lock

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

Cargo.toml

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -46,29 +46,26 @@ serde_json = "1.0"
4646
thiserror = "1.0"
4747
rand = "0.9"
4848
serial_test = "2.0.0"
49-
cudarc = { version = "0.13" , features =["cuda-12020"]}
50-
intel-mkl-src = { version = "0.8" }
49+
cudarc = { version = "0.13" , features =["cuda-12020"], default-features = false}
50+
intel-mkl-src = { version = "0.8"}
5151
candle = { version = "0.8", package = "candle-core" }
5252
candle-nn = { version = "0.8", package = "candle-nn" }
5353
candle-transformers = { version = "0.8", package = "candle-transformers" }
5454
candle-flash-attn = { version = "0.8", package = "candle-flash-attn" }
5555
half = { version = "2.3.1", features = ["num-traits"] }
5656

5757
[patch.crates-io]
58-
cudarc = { git = "https://github.com/Narsil/cudarc" , rev = "18ae111a4e8779c11377636b9cc3379f686e99c6"}
58+
cudarc = { git = "https://github.com/Narsil/cudarc" , rev = "b2d6443329e559e9580204b55ecaf44cd6fb6d90"}
5959
candle = { git = "https://github.com/huggingface/candle", rev = "ec6d7ca7738f4052b6613edc8f4d2bb6866a7539", package = "candle-core" }
6060
candle-nn = { git = "https://github.com/huggingface/candle", rev = "ec6d7ca7738f4052b6613edc8f4d2bb6866a7539", package = "candle-nn" }
6161
candle-transformers = { git = "https://github.com/huggingface/candle", rev = "ec6d7ca7738f4052b6613edc8f4d2bb6866a7539", package = "candle-transformers" }
6262
candle-flash-attn = { git = "https://github.com/huggingface/candle", rev = "ec6d7ca7738f4052b6613edc8f4d2bb6866a7539", package = "candle-flash-attn" }
63-
# candle = { path = "../candle/candle-core", package = "candle-core" }
64-
# candle-nn = { path = "../candle/candle-nn" }
65-
# candle-flash-attn = { path = "../candle/candle-flash-attn" }
6663

6764
[profile.release]
6865
debug = 0
69-
# lto = "fat"
66+
lto = "fat"
7067
opt-level = 3
71-
# codegen-units = 1
68+
codegen-units = 1
7269
strip = "symbols"
7370
panic = "abort"
7471

router/Cargo.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -88,9 +88,9 @@ accelerate = ["text-embeddings-backend/accelerate"]
8888
python = ["text-embeddings-backend/python"]
8989
ort = ["text-embeddings-backend/ort"]
9090
candle = ["text-embeddings-backend/candle"]
91-
candle-cuda = ["candle", "text-embeddings-backend/flash-attn"]
92-
candle-cuda-turing = ["candle", "text-embeddings-backend/flash-attn-v1"]
93-
candle-cuda-volta = ["candle", "text-embeddings-backend/cuda"]
91+
candle-cuda = ["candle", "text-embeddings-backend/flash-attn", "dep:cudarc"]
92+
candle-cuda-turing = ["candle", "text-embeddings-backend/flash-attn-v1", "dep:cudarc"]
93+
candle-cuda-volta = ["candle", "text-embeddings-backend/cuda", "dep:cudarc"]
9494
static-linking = ["cudarc?/static-linking", "intel-mkl-src?/mkl-static-lp64-iomp"]
9595
dynamic-linking = ["cudarc?/dynamic-linking", "intel-mkl-src?/mkl-dynamic-lp64-iomp"]
9696
google = []

0 commit comments

Comments
 (0)