diff --git a/Cargo.toml b/Cargo.toml index c0d87680..e4cfc667 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -28,7 +28,7 @@ accelerate-src = { version = "0.3.2" } anyhow = { version = "1", features = ["backtrace"] } byteorder = "1.4.3" clap = { version = "4.2.4", features = ["derive"] } -cudarc = { version = "0.9.13", features = ["f16"] } +cudarc = { git = "https://github.com/Narsil/cudarc.git", branch = "sync_alloc", features = ["f16"] } # TODO: Switch back to the official gemm implementation once it has caught up. gemm = { version = "0.15.6", package = "candle-gemm" } ggblas = "0.1.2"