Skip to content

Commit 844800e

Browse files
authored
Merge pull request #82 from jelmervdl/fix-wasm-intgemm-nullptr
Patch int8PrepareBias to handle nullptr
2 parents 53c4f7e + 6e40a8f commit 844800e

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

src/tensors/cpu/wasm_intgemm_fallback.cpp

+13-5
Original file line numberDiff line numberDiff line change
@@ -64,11 +64,19 @@ extern "C" void int8PrepareBiasFallback(const int8_t* input_B_prepared,
6464
const float* input_bias,
6565
float* output) {
6666
float unquant_factor = (-1) * ((127.0f / scale_A) * (127.0f / scale_B)) / (127.0f);
67-
intgemm::Int8Shift::PrepareBias(
68-
input_B_prepared,
69-
width,
70-
cols_B,
71-
intgemm::callbacks::UnquantizeAndAddBiasAndWrite(unquant_factor, input_bias, output));
67+
if (input_bias) {
68+
intgemm::Int8Shift::PrepareBias(
69+
input_B_prepared,
70+
width,
71+
cols_B,
72+
intgemm::callbacks::UnquantizeAndAddBiasAndWrite(unquant_factor, input_bias, output));
73+
} else {
74+
intgemm::Int8Shift::PrepareBias(
75+
input_B_prepared,
76+
width,
77+
cols_B,
78+
intgemm::callbacks::UnquantizeAndWrite(unquant_factor, output));
79+
}
7280
}
7381

7482
extern "C" void int8MultiplyAndAddBiasFallback(const int8_t* input_A_prepared,

0 commit comments

Comments
 (0)