Update to burn at head now that cube is updated

ArthurBrussee · Feb 3, 2025 · f23d631 · f23d631
1 parent 4749ece
commit f23d631
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 18 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -132,10 +132,6 @@ ecolor = { git = "https://github.com/emilk/egui", rev = "5b740f97acc3e0c24023609
 # cubecl = { path = "../cubecl/crates/cubecl" }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common" }
 
-# Temporary until Brun updates cube.
-cubecl = { git = "https://github.com/ArthurBrussee/cubecl", branch = "patched" }
-cubecl-common = { git = "https://github.com/ArthurBrussee/cubecl", branch = "patched" }
-
 [profile.dev]
 opt-level = 1
 debug = true

diff --git a/crates/brush-kernel/src/lib.rs b/crates/brush-kernel/src/lib.rs
@@ -4,13 +4,13 @@
 mod shaders;
 
 use burn::tensor::{DType, Shape};
+pub use burn_jit::cubecl::prelude::ExecutionMode;
 pub use burn_jit::cubecl::{
     client::ComputeClient, compute::CompiledKernel, compute::CubeTask, server::ComputeServer,
     CubeCount, CubeDim, KernelId,
 };
 pub use burn_jit::{cubecl::Compiler, tensor::JitTensor, JitRuntime};
 
-pub use burn_jit::cubecl::ExecutionMode;
 use bytemuck::Pod;
 use wgpu::naga;