From a8a77ddfdc15e3c170e13d7c89f4555f23b8ef67 Mon Sep 17 00:00:00 2001
From: Guenther Schmuelling <guschmue@microsoft.com>
Date: Wed, 17 Apr 2024 15:01:06 -0700
Subject: [PATCH] fix csum and enable ut (#20355)

---
 js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts |  2 +-
 js/web/test/suite-test-list.jsonc         | 14 +++++++-------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts b/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts
index 6080301d99..b8b50b3565 100644
--- a/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts
+++ b/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts
@@ -54,7 +54,7 @@ const createCumsumProgramInfo =
               outputs: [{dims: inputShape, dataType: inputType}],
               dispatchGroup: {x: Math.ceil(outputSize / 64 /* workgroup size */)},
               programUniforms: [
-                {type: DataType.uint32, data: outputSize}, {type: DataType.int32, data: axis},
+                {type: DataType.uint32, data: outputSize}, {type: DataType.uint32, data: axis},
                 ...createTensorShapeVariables(inputShape, inputShape)
               ]
 
diff --git a/js/web/test/suite-test-list.jsonc b/js/web/test/suite-test-list.jsonc
index 811e3659b5..142a4df06a 100644
--- a/js/web/test/suite-test-list.jsonc
+++ b/js/web/test/suite-test-list.jsonc
@@ -465,13 +465,13 @@
       "test_cos",
       "test_cosh_example",
       "test_cosh",
-      // "test_cumsum_1d_exclusive",
-      // "test_cumsum_1d_reverse_exclusive",
-      // "test_cumsum_1d_reverse",
-      // "test_cumsum_1d",
-      // "test_cumsum_2d_axis_0",
-      // "test_cumsum_2d_axis_1",
-      // "test_cumsum_2d_negative_axis",
+      "test_cumsum_1d_exclusive",
+      "test_cumsum_1d_reverse_exclusive",
+      "test_cumsum_1d_reverse",
+      "test_cumsum_1d",
+      "test_cumsum_2d_axis_0",
+      "test_cumsum_2d_axis_1",
+      "test_cumsum_2d_negative_axis",
       "test_depthtospace_crd_mode_example",
       "test_depthtospace_crd_mode",
       "test_depthtospace_dcr_mode",