From 819a66e8fcddcd8424778694ceb7744541fda5c4 Mon Sep 17 00:00:00 2001 From: Jiajia Qin Date: Wed, 23 Apr 2025 21:23:48 +0800 Subject: [PATCH] [webgpu] Fix bug in 1D dispatch workgoups --- onnxruntime/contrib_ops/webgpu/bert/attention.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/contrib_ops/webgpu/bert/attention.cc b/onnxruntime/contrib_ops/webgpu/bert/attention.cc index 6e7919f281fb6..c6752b22ccc24 100644 --- a/onnxruntime/contrib_ops/webgpu/bert/attention.cc +++ b/onnxruntime/contrib_ops/webgpu/bert/attention.cc @@ -155,7 +155,7 @@ Status AttentionProbsProgram::GenerateShaderCode(ShaderHelper& shader) const { shader.MainFunctionBody() << "if (m + local_id.y < uniforms.M && n + local_id.x < total_sequence_length) {\n" << " let headOffset = batch_head_idx * uniforms.M * uniforms.N;\n" - << " let outputIdx = headOffset + m + local_id.y * uniforms.N + n + local_id.x;\n" + << " let outputIdx = headOffset + (m + local_id.y) * uniforms.N + n + local_id.x;\n" << " var sum: f32 = " << (components_ == 4 ? "value.x + value.y + value.z + value.w" : (components_ == 2 ? "value.x + value.y" : "value")) << ";\n"; shader.MainFunctionBody() << " output[outputIdx] = output_value_t(sum * uniforms.alpha)";