Hardcode dispatch.

microsoft · Sep 19, 2023 · eda748c · eda748c
1 parent a15c782
commit eda748c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/3rd-party/conv_backprop_mm_webgpu.ts b/js/web/lib/wasm/jsep/webgpu/ops/3rd-party/conv_backprop_mm_webgpu.ts
@@ -123,10 +123,9 @@ export const createConv2DTransposeMatMulProgramInfo =
                       (outWidth % 4 === 0 && !isChannelsLast)) &&
           outChannels % 4 === 0;
 
-      const dispatchX = !isChannelsLast ? outChannels : outWidth * outHeight;
-      const dispatchY = !isChannelsLast ? outWidth * outHeight : outChannels;
-      const workGroupSize: [number, number, number] =
-          isVec4 ? [8, 8, 1] : [dispatchX <= 4 ? 4 : 16, dispatchX > 4 && dispatchY <= 4 ? 4 : 16, 1];
+      const dispatchX = isChannelsLast ? outChannels : outWidth * outHeight;
+      const dispatchY = isChannelsLast ? outWidth * outHeight : outChannels;
+      const workGroupSize: [number, number, number] = isVec4 ? [8, 8, 1] : [4, 4, 1];
       const elementsPerThread =
           isVec4 ? [4, 4, 1] : [dispatchX <= 4 ? 1 : 2, dispatchX > 4 && dispatchY <= 4 ? 1 : 2, 1];
       const dispatch = [