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

[js/webgpu] Fix shader compilation errors in cumsum (#18779)

Browse source 
### Description
This PR fixes below shader compilation errors:
```
Tint WGSL reader failure: :39:31 error: no matching overload for operator + (f32, i32)

5 candidate operators:
  operator + (T, T) -> T  where: T is abstract-float, abstract-int, f32, i32, u32 or f16
  operator + (vecN<T>, T) -> vecN<T>  where: T is abstract-float, abstract-int, f32, i32, u32 or f16
  operator + (T, vecN<T>) -> vecN<T>  where: T is abstract-float, abstract-int, f32, i32, u32 or f16
  operator + (vecN<T>, vecN<T>) -> vecN<T>  where: T is abstract-float, abstract-int, f32, i32, u32 or f16
  operator + (matNxM<T>, matNxM<T>) -> matNxM<T>  where: T is abstract-float, f32 or f16

                    sum = sum + get_inputByIndices(inputIndices);
                              ^


 - While validating [ShaderModuleDescriptor "CumSum"]
 - While calling [Device].CreateShaderModule([ShaderModuleDescriptor "CumSum"]).
This commit is contained in:
Jiajia Qin 2023-12-12 10:11:38 +08:00 committed by GitHub
parent a85ef652ed
commit b4be9e1bbb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 37 additions and 1 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

@ -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++) {

36
js/web/test/data/ops/cumsum.jsonc
View file

@ -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"
}
]
}
]
}
]