Nit

axinging · Sep 18, 2024 · 6300ce1 · 6300ce1
1 parent 2ddd64b
commit 6300ce1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/onnxruntime/core/providers/webgpu/tensor/transpose.cc b/onnxruntime/core/providers/webgpu/tensor/transpose.cc
@@ -149,7 +149,9 @@ Status Transpose::ComputeInternal(ComputeContext& context) const {
   uint32_t output_size = gsl::narrow_cast<int32_t>(input_tensor->Shape().Size());
   TransposeProgram program{*p_perm, use_shared};
   const auto tileSize = 16; //todo
-  program.SetWorkgroupSize(tileSize,tileSize,1);
+  if (use_shared) {
+    program.SetWorkgroupSize(tileSize,tileSize,1);
+  }
 
   /*
   useShared