Skip to content

Commit

Permalink
Revert layer-norm
Browse files Browse the repository at this point in the history
  • Loading branch information
axinging committed Jan 18, 2024
1 parent d3be951 commit d7f3603
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion js/web/lib/wasm/jsep/webgpu/ops/layer-norm.ts
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ const createLayerNormProgramInfo =
}
let mean = ${sumVector('meanVector', components)} / uniforms.norm_size;
let invStdDev =
inverseSqrt(${sumVector('meanSquareVector', components)} / uniforms.norm_size - mean * mean + uniforms.epsilon);
inverseSqrt(${sumVector('meanSquareVector', components)} / uniforms.norm_size - mean * mean + uniforms.epsilon);
for (var j: u32 = 0; j < uniforms.norm_size_vectorized; j++) {
let f32input = ${castToF32(dataType, components, 'x[j + offset]')};
Expand Down

0 comments on commit d7f3603

Please sign in to comment.