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

!7286 [MS][LITE][CPU]fix winograd init asm bug 

Merge pull request !7286 from liuzhongkai/winograd_optimize
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
3879f1272e 9457409314
commit
f2c76ed745
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