Bump half/cudarc versions (#805)

coreylowman · Jul 6, 2023 · 91cf54a · 91cf54a
1 parent 610f124
commit 91cf54a
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -29,15 +29,13 @@ no-std-compat = { version = "0.4.1", default-features = false, features = [ "all
 spin = { version = "0.9.8", default-features = false, features = ["spin_mutex", "rwlock", "portable_atomic"], optional = true }
 rand = { version = "0.8.5", default-features = false, features = ["std_rng"] }
 rand_distr = { version = "0.4.3", default-features = false }
-zip = { version = "0.6.2", default-features = false, optional = true }
-cblas-sys = { version = "0.1.4", default-features = false, optional = true }
-libc = { version = "0.2", default-features = false, optional = true }
-cudarc = { git = "https://github.com/coreylowman/cudarc", branch = "dfdx-half", default-features = false, optional = true, features = ["driver", "cublas", "nvrtc"] }
+zip = { version = "0.6.6", default-features = false, optional = true }
+cudarc = { version = "0.9.11", default-features = false, optional = true, features = ["driver", "cublas", "nvrtc"] }
 num-traits = { version = "0.2.15", default-features = false }
 safetensors = { version = "0.3", default-features = false, optional = true }
 memmap2 = { version = "0.5", default-features = false, optional = true }
-half = { git = "https://github.com/starkat99/half-rs.git", branch = "main", optional = true, features = ["num-traits", "rand_distr"] }
-gemm = { version = "0.15.3", default-features = false, optional = true }
+half = { version = "2.3.1", optional = true, features = ["num-traits", "rand_distr"] }
+gemm = { version = "0.15.4", default-features = false, optional = true }
 rayon = { version = "1.7.0", optional = true }
 
 [dev-dependencies]