Update js/web/lib/wasm/jsep/webgpu/ops/matmul.ts

microsoft · Oct 30, 2024 · e19f2a1 · e19f2a1
1 parent 7024cd6
commit e19f2a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/matmul.ts b/js/web/lib/wasm/jsep/webgpu/ops/matmul.ts
@@ -206,7 +206,7 @@ export const matMul = (context: ComputeContext): void => {
     const batchB = ShapeUtil.size(context.inputs[1].dims.slice(0, -2));
     if (batchA !== 1 && M === 1 && batchB === 1) {
       const reshapedA = context.inputs[0].reshape([1, batchA, K]);
-      const reshapedB = context.inputs[0].reshape([1, K, N]);
+      const reshapedB = context.inputs[1].reshape([1, K, N]);
       const matmulOutputShape = [1, batchA, N];
       const matmulInputs = [reshapedA, reshapedB];
       context.compute(createMatmulProgramInfo(matmulInputs, { activation: '' }, outputShape, matmulOutputShape), {