From 64f184956ac8a9a52e8b89e8594f3d91583bb9f0 Mon Sep 17 00:00:00 2001
From: Xu Xing <xing.xu@intel.com>
Date: Tue, 30 Jul 2024 11:01:24 +0800
Subject: [PATCH] [js/webgpu] Fix wrong shape in maxpool

Bug: https://github.com/microsoft/onnxruntime/issues/21386
---
 js/web/lib/wasm/jsep/webgpu/ops/pool.ts | 10 ++++++++--
 js/web/test/suite-test-list.jsonc       |  1 +
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/js/web/lib/wasm/jsep/webgpu/ops/pool.ts b/js/web/lib/wasm/jsep/webgpu/ops/pool.ts
index 5521650e8ded4..e532437c99351 100644
--- a/js/web/lib/wasm/jsep/webgpu/ops/pool.ts
+++ b/js/web/lib/wasm/jsep/webgpu/ops/pool.ts
@@ -31,8 +31,14 @@ const getAdjustedPoolAttributesAndOutputShape = <AttributeType extends AveragePo
     inputShapeAsChannelFirst.splice(1, 0, inputShapeAsChannelFirst.pop()!);  // Move channel to the second position.
   }
   const hasDilations = Object.hasOwnProperty.call(attributes, 'dilations');
-  const kernelShape = attributes.kernelShape.slice();
-  const strides = attributes.strides.slice();
+  let kernelShape = attributes.kernelShape.slice();
+  if (kernelShape[kernelShape.length - 1] === 0) {
+    kernelShape.pop();
+  }
+  let strides = attributes.strides.slice();
+  if (strides[strides.length - 1] === 0) {
+    strides.pop();
+  }
   const dilations: number[] = hasDilations ? (attributes as MaxPoolAttributes).dilations.slice() : [];
   const pads = attributes.pads.slice();
   PoolConvUtil.adjustPoolAttributes(isGlobalOperator, inputShapeAsChannelFirst, kernelShape, strides, dilations, pads);
diff --git a/js/web/test/suite-test-list.jsonc b/js/web/test/suite-test-list.jsonc
index 4aaf9d16b2b0e..eb9a5c8c9d12e 100644
--- a/js/web/test/suite-test-list.jsonc
+++ b/js/web/test/suite-test-list.jsonc
@@ -1370,6 +1370,7 @@
       "matmul.jsonc",
       "matmulnbits.jsonc",
       "matmul-broadcast.jsonc",
+      "max-pool.jsonc",
       "mul.jsonc",
       "mul_int32.jsonc",
       "multihead-attention.jsonc",