From b73bb676ebdd9bf688ba3609e7e320fe36dda25b Mon Sep 17 00:00:00 2001 From: Corey Lowman Date: Thu, 6 Jul 2023 08:48:28 -0400 Subject: [PATCH] Bump half/cudarc versions --- Cargo.toml | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/Cargo.toml b/Cargo.toml index fb1c294b7..127efeeb9 100644 --- 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]