We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 002eb0d commit 0c96542Copy full SHA for 0c96542
onnxruntime/core/providers/webgpu/tensor/gather_nd.cc
@@ -43,7 +43,7 @@ Status GatherNDProgram::GenerateShaderCode(ShaderHelper& shader) const {
43
data_dim += indices_innerest_dim_;
44
45
for (uint32_t i = 0; i < static_cast<uint32_t>(data.Rank() - data_dim); i++) {
46
- shader.MainFunctionBody() << " " << data.IndicesSet("data_indices", data_dim, output.IndicesGet("output_indices", indices.Rank() - 1 + i)) << "\n";
+ shader.MainFunctionBody() << " " << data.IndicesSet("data_indices", data_dim + i, output.IndicesGet("output_indices", indices.Rank() - 1 + i)) << "\n";
47
}
48
49
shader.MainFunctionBody() << " " << output.SetByOffset("global_idx", data.GetByIndices("data_indices"));
0 commit comments