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

!13819 [MS][LITE][CPU]fix bug of floormod 

From: @fuzhiye
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-03-24 09:07:20 +08:00 committed by Gitee
parent a59f1814c1 b2a6e85a7a
commit e0b1afcfe5
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/nnacl/fp32/arithmetic_fp32.c
View File

@ -28,7 +28,9 @@ int ElementFloorMod(const float *in0, const float *in1, float *out, int size) {
int ElementFloorModInt(const int *in0, const int *in1, int *out, int size) {
for (int i = 0; i < size; i++) {
out[i] = in0[i] - (in0[i] / in1[i]) * in1[i];
NNACL_ASSERT(in1[i] != 0);
int remainder = in0[i] - (in0[i] / in1[i]) * in1[i];
out[i] = (remainder != 0) && ((in0[i] > 0) != (in1[i] > 0)) ? remainder + in1[i] : remainder;
}
return NNACL_OK;
}