From cd2a9ffbe58549cd97b90ae7c1fcd3e49db8d639 Mon Sep 17 00:00:00 2001 From: Arthur Date: Mon, 16 Sep 2024 21:57:01 -0400 Subject: [PATCH] Skip zero initialization of workgroup memory if possible --- crates/cubecl-wgpu/src/compute/server.rs | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/crates/cubecl-wgpu/src/compute/server.rs b/crates/cubecl-wgpu/src/compute/server.rs index b8ea2e022..29776b223 100644 --- 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, }), )