Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!30112 fix dynamic matmul CalcWeightSums bug 

Merge pull request !30112 from yeyunpeng2020/dynamic_quant
This commit is contained in:
i-robot 2022-02-16 09:00:48 +00:00 committed by Gitee
parent a97209dd00 1f619f0fd3
commit 5b48372717
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/dynamic_matmul_int8.c
View File

@ -321,9 +321,9 @@ void PackInput2Col4x4(const int8_t *src_input, int8_t *packed_input, int row, in
}
void CalcWeightSums(const int8_t *weight, int row, int col, int *dst, DataOrder order) {
int sum = 0;
if (order == RowMajor) {
for (int c = 0; c < col; ++c) {
int sum = 0;
for (int r = 0; r < row; ++r) {
sum += weight[r * col + c];
}
@ -331,6 +331,7 @@ void CalcWeightSums(const int8_t *weight, int row, int col, int *dst, DataOrder
}
} else {
for (int c = 0; c < col; ++c) {
int sum = 0;
for (int r = 0; r < row; ++r) {
sum += weight[c * row + r];
}
@ -341,9 +342,9 @@ void CalcWeightSums(const int8_t *weight, int row, int col, int *dst, DataOrder
}
void CalcPartWeightSums(const int8_t *weight, int row, int stride, int cur_col, int *dst, DataOrder order) {
int sum = 0;
if (order == RowMajor) {
for (int c = 0; c < cur_col; ++c) {
int sum = 0;
for (int r = 0; r < row; ++r) {
sum += weight[r * stride + c];
}
@ -351,6 +352,7 @@ void CalcPartWeightSums(const int8_t *weight, int row, int stride, int cur_col,
}
} else {
for (int c = 0; c < cur_col; ++c) {
int sum = 0;
for (int r = 0; r < row; ++r) {
sum += weight[c * row + r];
}