From bdb6d01385bc2a01a7bcd45a041417e9c5d792fd Mon Sep 17 00:00:00 2001 From: yangruoqi713 Date: Fri, 25 Dec 2020 16:03:41 +0800 Subject: [PATCH] [MSLITE][DEVELOP] fix bug of arm cpu fp32 op: deconv depthwise --- mindspore/lite/nnacl/assembly/arm64/DeconvDwFp32Border.S | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mindspore/lite/nnacl/assembly/arm64/DeconvDwFp32Border.S b/mindspore/lite/nnacl/assembly/arm64/DeconvDwFp32Border.S index dd65bcaed86..88e2e84ae3b 100644 --- a/mindspore/lite/nnacl/assembly/arm64/DeconvDwFp32Border.S +++ b/mindspore/lite/nnacl/assembly/arm64/DeconvDwFp32Border.S @@ -16,6 +16,10 @@ DeconvDwFp32Border: // https://github.com/ARM-software/abi-aa/blob/master/aapcs64/aapcs64.rst#simd-and-floating-point-registers // x19 ~ x29 should be also preserved // whereas our coding style do not permit such amount of parameters + cmp x3, #0 + beq End + cmp x4, #0 + beq End ld1 {v1.4s}, [x1] mov x13, x0 @@ -35,5 +39,6 @@ DeconvDwFp32Border: add x13, x13, x5 add x14, x14, x7 bne LoopH + End: ret #endif