Merge fix

tracel-ai · Nov 22, 2024 · 1e29f78 · 1e29f78
1 parent b26b183
commit 1e29f78
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crates/cubecl-wgpu/src/compiler/wgsl/compiler.rs b/crates/cubecl-wgpu/src/compiler/wgsl/compiler.rs
@@ -65,12 +65,13 @@ impl cubecl_core::Compiler for WgslCompiler {
     fn compile(
         shader: cube::KernelDefinition,
         compilation_options: &Self::CompilationOptions,
+        mode: ExecutionMode,
     ) -> Self::Representation {
         let mut compiler = Self {
             compilation_options: compilation_options.clone(),
             ..Self::default()
         };
-        compiler.compile_shader(shader)
+        compiler.compile_shader(shader, mode)
     }
 
     fn elem_size(elem: cube::Elem) -> usize {