Skip to content

Commit 66a7171

Browse files
FarenlaytJaccovG
authored andcommitted
Fix relu codesize gain
1 parent 4445884 commit 66a7171

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

lib/src/kernels/transform/mli_krn_relu_fx.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ mli_status mli_krn_relu_fx8(const mli_tensor* in, const mli_relu_cfg* cfg, mli_t
5050
}
5151
} else {
5252
// Case with more tricky limits, which we're defining by separate function
53+
__builtin_assume(in->el_type == MLI_EL_FX_8);
5354
mli_minmax_t limits = mli_prv_get_relu_min_max(cfg, in);
5455
v2q15_t min_vec = fx_replic_v2q15(limits.min);
5556
v2q15_t max_vec = fx_replic_v2q15(limits.max);
@@ -101,6 +102,7 @@ mli_status mli_krn_relu_fx16(const mli_tensor* in, const mli_relu_cfg* cfg, mli_
101102
}
102103
} else {
103104
// Case with more tricky limits, which we're defining by separate function
105+
__builtin_assume(in->el_type == MLI_EL_FX_16);
104106
mli_minmax_t limits = mli_prv_get_relu_min_max(cfg, in);
105107
v2q15_t min_vec = fx_replic_v2q15(limits.min);
106108
v2q15_t max_vec = fx_replic_v2q15(limits.max);

0 commit comments

Comments
 (0)