File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
onnxruntime/core/providers/webgpu/tensor Expand file tree Collapse file tree 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 {
6565
6666 shader.MainFunctionBody () << shader.GuardAgainstOutOfBoundsWorkgroupSizes (" uniforms.input_size" )
6767 << " var indices = " << input.OffsetToIndices (" global_idx" ) << " ;\n "
68- << " var index = indices[ " << axis_ << " ] ;\n "
68+ << " var index = " << input. IndicesGet ( " indices " , axis_) << " ;\n "
6969 << " let output_number = calculate_output_index(index);\n "
7070 << " if (output_number != 0u) {\n "
7171 << " index -= uniforms.sizes_in_split_axis[output_number - 1u];\n "
72- << " indices[ " << axis_ << " ] = index; \n "
72+ << " " << input. IndicesSet ( " indices " , axis_, " index " ) << " \n "
7373 << " }\n "
7474 << " write_buffer_data(output_number, global_idx, indices);\n " ;
7575
You can’t perform that action at this time.
0 commit comments