From 41ce6dc30d967ce99e2328aa869517fb52156911 Mon Sep 17 00:00:00 2001 From: fuzhiye Date: Tue, 23 Mar 2021 13:09:58 +0800 Subject: [PATCH] fix bug of floorMod --- mindspore/lite/nnacl/fp32/arithmetic_fp32.c | 4 +++- mindspore/lite/test/models_npu.cfg | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/nnacl/fp32/arithmetic_fp32.c b/mindspore/lite/nnacl/fp32/arithmetic_fp32.c index cda6db6bce..bdae1dc197 100644 --- a/mindspore/lite/nnacl/fp32/arithmetic_fp32.c +++ b/mindspore/lite/nnacl/fp32/arithmetic_fp32.c @@ -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; } diff --git a/mindspore/lite/test/models_npu.cfg b/mindspore/lite/test/models_npu.cfg index 4bf2a39bfe..95724c0d48 100644 --- a/mindspore/lite/test/models_npu.cfg +++ b/mindspore/lite/test/models_npu.cfg @@ -71,7 +71,7 @@ ml_edu_kit_hand_key_position.onnx 2 #ml_video_edit_oneclick_adaptis.pb #too many subgraphs densenet.tflite 3 ml_video_edit_enhance.pb 2 -ml_video_edit_video_segment_gauss_adaptis_part2_pb2tflite.tflite 3 2 +ml_video_edit_video_segment_gauss_adaptis_part2_pb2tflite.tflite 10 2 ml_video_edit_img_segment_adaptise_pb2tflite.tflite 0.5 2 ml_video_edit_imitate_filter.onnx 103 hdc_mobilenet_1w_class.onnx 20