Chore/update cubecl (#2235)

tracel-ai · Sep 1, 2024 · 96a2340 · 96a2340
1 parent c1b6103
commit 96a2340
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 13 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -151,8 +151,8 @@ systemstat = "0.2.3"
 portable-atomic-util = { version = "0.2.2", features = ["alloc"] }
 
 ### For the main burn branch. ###
-cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e" }
-cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e" }
+cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "b0fa39bad9d5ecbd98a955e6f7f94c855b6e9a73" }
+cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "b0fa39bad9d5ecbd98a955e6f7f94c855b6e9a73" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl" }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common" }

diff --git a/backend-comparison/Cargo.toml b/backend-comparison/Cargo.toml
@@ -24,14 +24,13 @@ tch-cpu = ["burn/tch"]
 tch-gpu = ["burn/tch"]
 wgpu = ["burn/wgpu", "burn/autotune"]
 wgpu-fusion = ["wgpu", "burn/fusion"]
-# cuda-jit = ["burn-cuda"]
+cuda-jit = ["burn/cuda-jit"]
 
 [dependencies]
 arboard = { workspace = true }
 burn = { path = "../crates/burn", default-features = false }
 burn-common = { path = "../crates/burn-common", version = "0.15.0" }
 burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.15.0", optional = true }
-# burn-cuda = { path = "../crates/burn-cuda", version = "0.15.0", optional = true }
 clap = { workspace = true }
 colored = { workspace = true }
 derive-new = { workspace = true }

diff --git a/backend-comparison/src/lib.rs b/backend-comparison/src/lib.rs
@@ -129,7 +129,7 @@ macro_rules! bench_on_backend {
 
         #[cfg(feature = "cuda-jit")]
         {
-            use burn_cuda::{Cuda, CudaDevice};
+            use burn::backend::cuda_jit::{Cuda, CudaDevice};
 
             bench::<Cuda>(&CudaDevice::default(), feature_name, url, token);
         }