Skip zero initialization of workgroup memory if possible (#125)

tracel-ai · Sep 17, 2024 · a090a47 · a090a47
1 parent c11e1d6
commit a090a47
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/crates/cubecl-wgpu/src/compute/server.rs b/crates/cubecl-wgpu/src/compute/server.rs
@@ -111,7 +111,10 @@ where
                     layout: None,
                     module: &module,
                     entry_point: "main",
-                    compilation_options: Default::default(),
+                    compilation_options: wgpu::PipelineCompilationOptions {
+                        zero_initialize_workgroup_memory: false,
+                        ..Default::default()
+                    },
                     cache: None,
                 }),
         )