Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

fix winograd init bug

tags/v1.1.0
liuzhongkai 5 years ago
parent
fb2442dad5
commit
9457409314
2 changed files with 2 additions and 2 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      mindspore/lite/nnacl/assembly/arm64/MatmulWinogradFp32.S
  2. +1
    -1
      mindspore/lite/nnacl/assembly/fp16/MatmulWinogradFp16.S

+ 1
- 1
mindspore/lite/nnacl/assembly/arm64/MatmulWinogradFp32.S View File

@@ -165,7 +165,7 @@ MatrixMultiplyWinograd:
b LoopM
EndLoopM:
sub sp, sp, #48
st1 {v8.4s}, [sp], #16
ld1 {v8.4s}, [sp], #16
ldp x19, x20, [sp], #16
ldp x21, x22, [sp], #16
ret


+ 1
- 1
mindspore/lite/nnacl/assembly/fp16/MatmulWinogradFp16.S View File

@@ -199,7 +199,7 @@ MatrixMultiplyWinogradFp16:

EndLoopM:
sub sp, sp, #48
st1 {v8.8h}, [sp], #16
ld1 {v8.8h}, [sp], #16
ldp x19, x20, [sp], #16
ldp x21, x22, [sp], #16
ret


Write Preview
Loading…
Cancel
Save