Use float8 mistralrs_cudarc_fork feature

EricLBuehler · Jan 8, 2025 · fd52b60 · fd52b60
1 parent af655eb
commit fd52b60
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -48,7 +48,7 @@ fancy-regex = "0.13.0"
 gemm = { version = "0.17.0", features = ["wasm-simd128-enable"] }
 hf-hub = { version = "0.3.3", package = "candle-hf-hub" }
 half = { version = "2.3.1", features = ["num-traits", "use-intrinsics", "rand_distr"] }
-float8 = { version = "0.1.0", features = ["num-traits", "rand_distr"] }
+float8 = { version = "0.1.2", features = ["num-traits", "rand_distr"] }
 hound = "3.5.1"
 image = { version = "0.25.2", default-features = false, features = ["jpeg", "png"] }
 imageproc = { version = "0.24.0", default-features = false }

diff --git a/candle-core/Cargo.toml b/candle-core/Cargo.toml
@@ -43,7 +43,7 @@ criterion = { workspace = true }
 
 [features]
 default = []
-cuda = ["cudarc", "dep:candle-kernels", "float8/cuda", "dep:ug-cuda"]
+cuda = ["cudarc", "dep:candle-kernels", "float8/cuda", "float8/mistralrs_cudarc_fork", "dep:ug-cuda"]
 cudnn = ["cuda", "cudarc/cudnn"]
 mkl = ["dep:libc", "dep:intel-mkl-src"]
 accelerate = ["dep:libc", "dep:accelerate-src"]