saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-08 17:17:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix csum and enable ut (#20355)

Browse source
This commit is contained in:
Guenther Schmuelling 2024-04-17 15:01:06 -07:00 committed by GitHub
parent 4c3fc26255
commit a8a77ddfdc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
js/web/lib/wasm/jsep/webgpu/ops/cumsum.ts
View file

@ -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)
]

14
js/web/test/suite-test-list.jsonc
View file

@ -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",