sbjkx
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6416 [MSLITE][Develop] deconv fp16 bug 

Merge pull request !6416 from ling/deconv
This commit is contained in:
mindspore-ci-bot 2020-09-17 17:24:46 +08:00 committed by Gitee
parent efd0fde424 a31d14aec5
commit b759d402af
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/lite/nnacl/assembly/fp16/MatmulFp16.S
View File

@ -24,8 +24,8 @@ MatmulFp16Neon64:
st1 {v8.8h, v9.8h, v10.8h, v11.8h}, [sp], #64
st1 {v12.8h, v13.8h, v14.8h, v15.8h}, [sp], #64
mov w18, #16 // sizeof(float) * 8
mul w15, w5, w18 // block stride of lhs/rhs: sizeof(float) * 8 * depth
mov w18, #16 // sizeof(float16) * 8
mul w15, w5, w18 // block stride of lhs/rhs: sizeof(float16) * 8 * depth
mov x11, x3 // bias flag
mov x18, #2
ldr x17, [sp]
@ -57,7 +57,7 @@ L2:
dup v30.4s, wzr
dup v31.4s, wzr
cmp w13, #4
cmp w13, #8
blt CommLoopMul
OptLoopMul8: