onnxruntime/js/web/lib/wasm/jsep/webgpu/ops
Jiajia Qin 92ee664f64
[js/webgpu] Fix shader errors in indicesGet/Set when rank > 4 (#18661)
### Description
Currently, for non-uniform variables, we still use `array<u32, N>` type
instead of array<vec4<u32>, N1>`. So we can't always treat all variables
with rank > 4 as uniforms to index.

This PR fixes below errors:
```
error(s) generated while compiling the shader:
:5:44 error: index 4 out of bounds [0..1]
             return uniforms.input_strides[4] * (outputIndices[4] % uniforms.input_shape[4])+uniforms.input_strides[3] * (outputIndices[3] % uniforms.input_shape[3])+uniforms.input_strides[2] * (outputIndices[2] % uniforms.input_shape[2])+uniforms.input_strides[1] * (outputIndices[1] % uniforms.input_shape[1])+uniforms.input_strides[0] * (outputIndices[0] % uniforms.input_shape[0]);
                                           ^
FAILED #OpTest# - expand.jsonc [webgpu]Expand - Expand 5D - float32 Expand 5 - float32
FAILED #OpTest# - expand.jsonc [webgpu]Expand - Expand 5D - float32 Expand 5 - shape < input.size()
2023-12-01 15:35:35 -08:00
..
3rd-party [js/web] allow ShaderHelper to use internal (non-I/O) variables (#18525) 2023-11-28 15:15:59 -08:00
argminmax.ts [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08:00
attention.ts [js/web] JSEP Attention & MultiHeadAttention (#17742) 2023-11-17 12:23:52 -08:00
batch-norm.ts [js/webgpu] Add BatchNormalization Op (#18468) 2023-11-22 15:58:06 -08:00
bias-add.ts
bias-split-gelu.ts [JS/Web] Resize & BiasSplitGelu fp16 support (#18536) 2023-11-22 12:12:07 -08:00
binary-op.ts [js/webgpu] Optimize broadcast binary. (#18185) 2023-11-20 16:52:17 -08:00
common.ts [js/webgpu] Fix shader errors in indicesGet/Set when rank > 4 (#18661) 2023-12-01 15:35:35 -08:00
concat.ts [js/webgpu] Add uniforms support to concat op (#18238) 2023-11-10 13:46:03 -08:00
conv-grouped.ts [js/webgpu] Fix conv2d with activation (#18388) 2023-11-10 12:54:35 -08:00
conv-transpose.ts [js/webgpu] Fix the transpose error when dims > 4D (#18027) 2023-10-23 11:02:19 -07:00
conv.ts [js/webgpu] Fix the transpose error when dims > 4D (#18027) 2023-10-23 11:02:19 -07:00
einsum.ts [JS/Web] Add uniforms to Einsum (#18531) 2023-11-29 15:30:33 -08:00
expand.ts [js/webgpu] add bool type for Expand/Gather (#18615) 2023-11-30 15:47:08 -08:00
fuse-utils.ts [js/webgpu] Fix conv2d with activation (#18388) 2023-11-10 12:54:35 -08:00
gather-elements.ts
gather.ts [js/webgpu] add bool type for Expand/Gather (#18615) 2023-11-30 15:47:08 -08:00
gemm.ts
instance-norm.ts [js/web] FP16 LayerNorm, InstanceNorm, SkipLayerNorm (#17630) 2023-10-18 10:47:41 -07:00
layer-norm.ts [js/web] FP16 LayerNorm, InstanceNorm, SkipLayerNorm (#17630) 2023-10-18 10:47:41 -07:00
matmul.ts
multi-head-attentiion.ts [js/web] JSEP Attention & MultiHeadAttention (#17742) 2023-11-17 12:23:52 -08:00
pad.ts [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08:00
pool.ts [JS/Web] Enabled 1d spacial input to GlobalAveragePool (#17973) 2023-10-23 16:02:50 -07:00
range.ts
reduce-shared.ts [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08:00
reduce.ts [js/web] optimize reduce related operators (#17957) 2023-11-02 12:51:48 -07:00
resize.ts [JS/Web] Resize & BiasSplitGelu fp16 support (#18536) 2023-11-22 12:12:07 -08:00
skip-layer-norm.ts [js/web] FP16 LayerNorm, InstanceNorm, SkipLayerNorm (#17630) 2023-10-18 10:47:41 -07:00
slice.ts [js/webgpu] Fix shader errors in indicesGet/Set when rank > 4 (#18661) 2023-12-01 15:35:35 -08:00
softmax.ts [js/webgpu] Support uniform for softmax (#18345) 2023-11-09 11:19:23 -08:00
split.ts
tile.ts
transpose.ts [js/webgpu] Fix the transpose error when dims > 4D (#18027) 2023-10-23 11:02:19 -07:00
unary-op.ts [js/web] unify resolve rules for "Clip" (#18527) 2023-11-20 23:18:06 -08:00
where.ts