Browse Source

[MSLITE][Develop] fix bug of cpu arm32 MatVecMul

tags/v1.2.0-rc1
yangruoqi713 4 years ago
parent
commit
7e5e0671c8
2 changed files with 6 additions and 4 deletions
  1. +4
    -4
      mindspore/lite/nnacl/assembly/arm32/MatVecMulFp32.S
  2. +2
    -0
      mindspore/lite/test/models_arm32.cfg

+ 4
- 4
mindspore/lite/nnacl/assembly/arm32/MatVecMulFp32.S View File

@@ -20,8 +20,8 @@

asm_function MatVecMulFp32
// r4-r8 and q4-q7 must be saved according to https://static.docs.arm.com/ihi0042/i/aapcs32.pdf
push {r0-r8, r10, r11, lr}
add sp, sp, #48
push {r0-r8, r9, r10, r11, lr}
add sp, sp, #52

ldr r4, [sp]
ldr r5, [sp, #4]
@@ -180,7 +180,7 @@ Col1Loop:
b Col1Loop

End:
sub sp, sp, #48
pop {r0-r8, r10, r11, pc}
sub sp, sp, #52
pop {r0-r8, r9, r10, r11, pc}
#endif
#endif

+ 2
- 0
mindspore/lite/test/models_arm32.cfg View File

@@ -13,6 +13,8 @@ ml_face_landmark
ml_liveness_detect_landmark
deconv_test_model
deconvs_model
# onnx
ml_2012_ocr_cn.onnx
# aware_training
video_infer.tflite
mobilenet_v1_1.0_224_quant.tflite

Loading…
Cancel
Save