Skip to content

Commit 497a627

Browse files
authored
fix fp16 for skiplayernorm (microsoft#20381)
1 parent 42b700d commit 497a627

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

js/web/lib/wasm/jsep/webgpu/ops/skip-layer-norm.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ const createSkipLayerNormProgramInfo =
136136
var squareSum = ${fillVector('f32', components)};
137137
for (var i: u32 = 0; i < hidden_size_vectorized; i++) {
138138
let skip_value = skip[offset + i];
139-
let bias_value = ${hasBiasInput ? 'bias[i]' : '0.0'};
139+
let bias_value = ${hasBiasInput ? 'bias[i]' : dataType + '(0.0)'};
140140
let input_value = x[offset + i];
141141
let value = input_value + skip_value + bias_value;
142142
${hasInputSkipBiasSumOutput ? 'input_skip_bias_sum[offset + i] = value;' : ''}

0 commit comments

Comments
 (0)