Browse Source

!6958 [MSLITE] Fix the bug of fp16 fullconnection output incorrect

Merge pull request !6958 from zhanyuan/dev
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
2039f5f47c
1 changed files with 10 additions and 10 deletions
  1. +10
    -10
      mindspore/lite/nnacl/assembly/fp16/MatmulFp16.S

+ 10
- 10
mindspore/lite/nnacl/assembly/fp16/MatmulFp16.S View File

@@ -511,7 +511,7 @@ Write4:
st1 {v16.4h}, [x18], x17
cmp w10, #1
beq WriteEnd
st1 {v16.4h}, [x18], x17
st1 {v17.4h}, [x18], x17
cmp w10, #2
beq WriteEnd
st1 {v18.4h}, [x18], x17
@@ -561,8 +561,8 @@ Write5:
st1 {v16.h}[4], [x13], x17
cmp w10, #1
beq WriteEnd
st1 {v16.4h}, [x18], x17
st1 {v16.h}[4], [x13], x17
st1 {v17.4h}, [x18], x17
st1 {v17.h}[4], [x13], x17
cmp w10, #2
beq WriteEnd
st1 {v18.4h}, [x18], x17
@@ -628,9 +628,9 @@ Write6:
st1 {v16.h}[5], [x14], x17
cmp w10, #1
beq WriteEnd
st1 {v16.4h}, [x18], x17
st1 {v16.h}[4], [x13], x17
st1 {v16.h}[5], [x14], x17
st1 {v17.4h}, [x18], x17
st1 {v17.h}[4], [x13], x17
st1 {v17.h}[5], [x14], x17
cmp w10, #2
beq WriteEnd
st1 {v18.4h}, [x18], x17
@@ -712,10 +712,10 @@ Write7:
st1 {v16.h}[6], [x16], x17
cmp w10, #1
beq WriteEnd
st1 {v16.4h}, [x18], x17
st1 {v16.h}[4], [x13], x17
st1 {v16.h}[5], [x14], x17
st1 {v16.h}[6], [x16], x17
st1 {v17.4h}, [x18], x17
st1 {v17.h}[4], [x13], x17
st1 {v17.h}[5], [x14], x17
st1 {v17.h}[6], [x16], x17
cmp w10, #2
beq WriteEnd
st1 {v18.4h}, [x18], x17


Loading…
Cancel
Save