diff --git a/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts b/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts index e7208ce34d..85682f0b47 100644 --- a/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts +++ b/js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts @@ -37,7 +37,7 @@ const createCumsumProgramInfo = ${shaderHelper.mainStart()} ${shaderHelper.guardAgainstOutOfBoundsWorkgroupSizes('uniforms.outputSize')} var inputIndices = ${output.offsetToIndices('global_idx')}; - var sum = 0.0; + var sum = ${output.type.value}(0); let first : i32 = ${lowerLimit}; let last : i32 = ${upperLimit}; for (var i : i32 = first; i < last; i++) { diff --git a/js/web/test/data/ops/cumsum.jsonc b/js/web/test/data/ops/cumsum.jsonc index cac9be734b..b3173afb69 100644 --- a/js/web/test/data/ops/cumsum.jsonc +++ b/js/web/test/data/ops/cumsum.jsonc @@ -1322,5 +1322,41 @@ ] } ] + }, + { + "name": "CumSum", + "operator": "CumSum", + "attributes": [ + { "name": "exclusive", "data": 0, "type": "int" }, + { "name": "reverse", "data": 0, "type": "int" } + ], + "opset": { + "domain": "", + "version": 11 + }, + "cases": [ + { + "name": "CumSum int32; axis = 0; exclusive = 0, reverse = 0", + "inputs": [ + { + "data": [1, 2, 3, 4, 5], + "dims": [1, 1, 1, 1, 5], + "type": "int32" + }, + { + "data": [4], + "dims": [], + "type": "int32" + } + ], + "outputs": [ + { + "data": [1, 3, 6, 10, 15], + "dims": [1, 1, 1, 1, 5], + "type": "int32" + } + ] + } + ] } ]