File tree 1 file changed +2
-2
lines changed
onnxruntime/core/providers/webgpu/tensor
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -65,11 +65,11 @@ Status SplitProgram::GenerateShaderCode(ShaderHelper& shader) const {
65
65
66
66
shader.MainFunctionBody () << shader.GuardAgainstOutOfBoundsWorkgroupSizes (" uniforms.input_size" )
67
67
<< " var indices = " << input.OffsetToIndices (" global_idx" ) << " ;\n "
68
- << " var index = indices[ " << axis_ << " ] ;\n "
68
+ << " var index = " << input. IndicesGet ( " indices " , axis_) << " ;\n "
69
69
<< " let output_number = calculate_output_index(index);\n "
70
70
<< " if (output_number != 0u) {\n "
71
71
<< " index -= uniforms.sizes_in_split_axis[output_number - 1u];\n "
72
- << " indices[ " << axis_ << " ] = index; \n "
72
+ << " " << input. IndicesSet ( " indices " , axis_, " index " ) << " \n "
73
73
<< " }\n "
74
74
<< " write_buffer_data(output_number, global_idx, indices);\n " ;
75
75
You can’t perform that action at this time.
0 commit comments