File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
onnxruntime/contrib_ops/webgpu/bert Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -155,7 +155,7 @@ Status AttentionProbsProgram::GenerateShaderCode(ShaderHelper& shader) const {
155155
156156 shader.MainFunctionBody () << " if (m + local_id.y < uniforms.M && n + local_id.x < total_sequence_length) {\n "
157157 << " let headOffset = batch_head_idx * uniforms.M * uniforms.N;\n "
158- << " let outputIdx = headOffset + m + local_id.y * uniforms.N + n + local_id.x;\n "
158+ << " let outputIdx = headOffset + ( m + local_id.y) * uniforms.N + n + local_id.x;\n "
159159 << " var sum: f32 = " << (components_ == 4 ? " value.x + value.y + value.z + value.w" : (components_ == 2 ? " value.x + value.y" : " value" )) << " ;\n " ;
160160
161161 shader.MainFunctionBody () << " output[outputIdx] = output_value_t(sum * uniforms.alpha)" ;
You can’t perform that action at this time.
0 commit comments