floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update dgemm_kernel_4x4_cortexA53.c

This commit is contained in:
Martin Kroeker 2021-11-18 11:54:42 +01:00 committed by GitHub
parent 9605ba5b5a
commit a3d41cbbed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
kernel/arm64/dgemm_kernel_4x4_cortexA53.c
View File

@ -81,7 +81,7 @@ static inline void dgemm_kernel_arm_cortex_a53_4x4_m4n12(
/** register v0-v3 for loading A, v4-v7 for loading B, x0 for transporting data */
"ldp q0,q1,[%[sa]]; ldp q4,q5,[%[b1_]]\n\t"
"ldr d6,[%[b2_]]; ldr x0,[%[b2_],#8]\n\t"
"blt 3f; beq 2f;\n\t"
"blt 3f; beq 2f; .p2align 2;\n\t"
"1:\n\t"
/** main loop with unroll_k = 2, specially designed for cortex-A53 NEON pipeline */
"ldr d7,[%[b2_],#16]; fmov v6.d[1],x0\n\t"
@ -212,7 +212,8 @@ static inline void dgemm_kernel_arm_cortex_a53_4x4_m4n12(
"fmla v29.2d,v3.2d,v7.d[0]\n\t"
"fmla v30.2d,v2.2d,v7.d[1]\n\t"
"fmla v31.2d,v3.2d,v7.d[1]\n\t"
"b 4f; 3:\n\t"
"b 4f;\n\t"
"3:\n\t"
/** tail part with k = 1 */
"ldr d7,[%[b2_],#16]; fmov v6.d[1],x0\n\t"
"fmla v8.2d,v0.2d,v4.d[0]; ldr x0,[%[b2_],#24]\n\t"