Revert "Replaced isBroadcast flag with !vectorize to determine to set…

… useShapeUniforms." This reverts commit 6717dd0.
microsoft · Nov 4, 2023 · f06c768 · f06c768
1 parent a44e6ac
commit f06c768
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/binary-op.ts b/js/web/lib/wasm/jsep/webgpu/ops/binary-op.ts
@@ -152,8 +152,7 @@ const createBinaryOpProgramInfo =
         vectorize = true;
       }
       cacheKeyAux.push(vectorize);
-      // Don't need to use shape/stride uniforms if input/output tensors are vectorized.
-      const useShapesUniforms = !vectorize && enableShapesUniforms(a.dims.length) &&
+      const useShapesUniforms = enableShapesUniforms(a.dims.length) &&
           enableShapesUniforms(b.dims.length) && enableShapesUniforms(outputShape.length);
       return {
         name,