[WebGPU] bug fix related to bounds checking in conv kernel (microsoft#26268)

prathikr · web-flow · commit e94153e82197 · 2025-10-09T16:52:07.000-07:00
Fixes chatterbox model for transformers.js
diff --git a/onnxruntime/core/providers/webgpu/nn/conv2d_mm_webgpu.cc b/onnxruntime/core/providers/webgpu/nn/conv2d_mm_webgpu.cc
@@ -103,7 +103,7 @@ std::string Conv2dMMProgram::Conv2dCommonSnippet(const ShaderVariableHelper& x,
     }
   } else {
     sample_w << "let col = colIn * " << inner_element_size_w << ";\n"
-             << "if (row < i32(uniforms.dim_inner) && col < i32(uniforms.dim_b_outer)) {\n"
+             << "if (row < i32(uniforms.dim_inner) && col < i32(uniforms.dim_a_outer)) {\n"
              << "  " << get_w_snippet(inner_element_size_w) << "\n"
              << "}\n"
              << "return " << TypeSnippet(inner_element_size_w, data_type) << "(0.0);\n";

Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ std::string Conv2dMMProgram::Conv2dCommonSnippet(const ShaderVariableHelper& x,`
`103`	`103`	`}`
`104`	`104`	`} else {`
`105`	`105`	`sample_w << "let col = colIn * " << inner_element_size_w << ";\n"`
`106`		`- << "if (row < i32(uniforms.dim_inner) && col < i32(uniforms.dim_b_outer)) {\n"`
	`106`	`+ << "if (row < i32(uniforms.dim_inner) && col < i32(uniforms.dim_a_outer)) {\n"`
`107`	`107`	`<< " " << get_w_snippet(inner_element_size_w) << "\n"`
`108`	`108`	`<< "}\n"`
`109`	`109`	`<< "return " << TypeSnippet(inner_element_size_w, data_type) << "(0.0);\n";`