Skip to content

Commit b8105de

Browse files
Hakim7267JaccovG
authored andcommitted
fix copy sa8 parameters in mov
1 parent 67771c1 commit b8105de

File tree

1 file changed

+5
-13
lines changed

1 file changed

+5
-13
lines changed

lib/src/move/mli_mov_api.cc

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -115,22 +115,14 @@ mli_status mli_mov_prepare(mli_mov_handle_t* h, const mli_tensor* src, const mli
115115
if ((dst->el_params.sa.scale.mem.pi16 != src->el_params.sa.scale.mem.pi16) &&
116116
(dst->el_params.sa.scale.mem.pi16 != NULL)) {
117117
mli::mov::mli_mov_memcpy<int8_t>(h, src->el_params.sa.scale.mem.pi8,
118-
dst->el_params.sa.scale.mem.pi8, src->el_params.sa.scale.capacity, 1, 1, src_in_vccm, dst_in_vccm, true, true, false);
118+
dst->el_params.sa.scale.mem.pi8, src->el_params.sa.scale.capacity, 1, 1, false, false, true, true, false);
119+
mli::mov::mli_mov_memcpy<int8_t>(h, src->el_params.sa.zero_point.mem.pi8,
120+
dst->el_params.sa.zero_point.mem.pi8, src->el_params.sa.zero_point.capacity, 1, 1, false, false, true, true, false);
121+
mli::mov::mli_mov_memcpy<int8_t>(h, src->el_params.sa.scale_frac_bits.mem.pi8,
122+
dst->el_params.sa.scale_frac_bits.mem.pi8, src->el_params.sa.scale_frac_bits.capacity, 1, 1, false, false, true, true, false);
119123
} else {
120124
dst->el_params.sa.scale = src->el_params.sa.scale;
121-
}
122-
if ((dst->el_params.sa.zero_point.mem.pi16 != src->el_params.sa.scale.mem.pi16) &&
123-
(dst->el_params.sa.zero_point.mem.pi16 != NULL)) {
124-
mli::mov::mli_mov_memcpy<int8_t>(h, src->el_params.sa.scale.mem.pi8,
125-
dst->el_params.sa.scale.mem.pi8, src->el_params.sa.scale.capacity, 1, 1, src_in_vccm, dst_in_vccm, true, true, false);
126-
} else {
127125
dst->el_params.sa.zero_point = src->el_params.sa.zero_point;
128-
}
129-
if ((dst->el_params.sa.scale_frac_bits.mem.pi8 != src->el_params.sa.scale_frac_bits.mem.pi8)
130-
&& (dst->el_params.sa.scale_frac_bits.mem.pi8 != NULL)) {
131-
mli::mov::mli_mov_memcpy<int8_t>(h, src->el_params.sa.scale.mem.pi8,
132-
dst->el_params.sa.scale.mem.pi8, src->el_params.sa.scale.capacity, 1, 1, src_in_vccm, dst_in_vccm, true, true, false);
133-
} else {
134126
dst->el_params.sa.scale_frac_bits = src->el_params.sa.scale_frac_bits;
135127
}
136128
dst->el_params.sa.dim = src->el_params.sa.dim;

0 commit comments

Comments
 (0)