diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/kernels/sve_native_s8s32_dot_4VLx4/generic.cpp')
| -rw-r--r-- | src/core/NEON/kernels/arm_gemm/kernels/sve_native_s8s32_dot_4VLx4/generic.cpp | 1368 |
1 files changed, 596 insertions, 772 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/kernels/sve_native_s8s32_dot_4VLx4/generic.cpp b/src/core/NEON/kernels/arm_gemm/kernels/sve_native_s8s32_dot_4VLx4/generic.cpp index abee1bbe1f..95cf88ab54 100644 --- a/src/core/NEON/kernels/arm_gemm/kernels/sve_native_s8s32_dot_4VLx4/generic.cpp +++ b/src/core/NEON/kernels/arm_gemm/kernels/sve_native_s8s32_dot_4VLx4/generic.cpp @@ -25,14 +25,14 @@ #include <algorithm> +#include "arm_gemm.hpp" #include <cstdint> #include "../../asmlib.hpp" #include "../../utils.hpp" namespace arm_gemm { -void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int ldb, int32_t *C, int ldc, int32_t beta, int M, int N, int K) { - const long beta0 = (beta == 0); +void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int ldb, int32_t *C, int ldc, int M, int N, int K, const int32_t *bias, Activation act, bool append) { const long loops_count = ((K + 16) / 32) - 1; K -= loops_count * 32; const long regs_count = (K / 16) - 1; @@ -46,11 +46,9 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l const unsigned long ldab = lda * sizeof(int8_t); int32_t *c_ptr0 = C + (y * ldc); - const unsigned long ldcb = ldc * sizeof(int32_t); for (int x0=0; x0<N; x0+=(4 * get_vector_length<int32_t>())) { const long width = std::min((unsigned long)N-x0, (4 * get_vector_length<int32_t>())); - const int32_t *betaptr = β long loops = loops_count; long regs = regs_count; long temp = 0; @@ -62,95 +60,68 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l const int8_t *b_ptr2 = b_ptr1 + ldb; const int8_t *b_ptr3 = b_ptr2 + ldb; long ldbb = ldb * sizeof(int8_t) * 4; + const unsigned long ldcb = ldc * sizeof(int32_t); switch(M-y) { case 1: __asm __volatile ( + "mov z16.s, #0\n" "whilelt p6.b, %[temp], %[leftovers]\n" + "mov z17.s, #0\n" "whilelt p0.s, %[temp], %[width]\n" + "mov z18.s, #0\n" "whilelt p4.b, %[temp], %[width]\n" + "mov z19.s, #0\n" "incw %[temp], all, mul #1\n" "ptrue p7.b\n" - "whilelt p1.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p2.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p3.s, %[temp], %[width]\n" - "cbz %[beta0], 1f\n" - "mov z16.s, #0\n" - "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" - "mov z17.s, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "mov z18.s, #0\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "mov z19.s, #0\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "add %[a_ptr0], %[a_ptr0], #0x10\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" + "ld1b z9.b, p4/z, [%[b_ptr2]]\n" + "whilelt p1.s, %[temp], %[width]\n" + "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" + "incw %[temp], all, mul #1\n" "zip2 z11.b, z8.b, z9.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "zip1 z9.b, z8.b, z9.b\n" "ld1b z8.b, p4/z, [%[b_ptr3]]\n" - "ld1b z13.b, p4/z, [%[b_ptr2]]\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "zip2 z12.b, z10.b, z8.b\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "cbz %[loops], 2f\n" - "b 3f\n" - "1:\n" - "ld1rw z15.s, p7/z, [%[betaptr]]\n" - "ld1w z16.s, p0/z, [%[c_ptr0]]\n" - "ld1w z17.s, p1/z, [%[c_ptr0], #1, MUL VL]\n" - "ld1w z18.s, p2/z, [%[c_ptr0], #2, MUL VL]\n" - "ld1w z19.s, p3/z, [%[c_ptr0], #3, MUL VL]\n" - "mul z16.s, p7/m, z16.s, z15.s\n" - "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" - "mul z17.s, p7/m, z17.s, z15.s\n" - "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "mul z18.s, p7/m, z18.s, z15.s\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "mul z19.s, p7/m, z19.s, z15.s\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "add %[a_ptr0], %[a_ptr0], #0x10\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "zip2 z11.b, z8.b, z9.b\n" + "whilelt p2.s, %[temp], %[width]\n" + "incw %[temp], all, mul #1\n" + "zip2 z12.b, z10.b, z8.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "zip1 z9.b, z8.b, z9.b\n" - "ld1b z8.b, p4/z, [%[b_ptr3]]\n" + "zip1 z10.b, z10.b, z8.b\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" + "whilelt p3.s, %[temp], %[width]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "zip2 z12.b, z10.b, z8.b\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "cbz %[loops], 2f\n" - "3:\n" - "zip1 z10.b, z10.b, z8.b\n" - "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" - "subs %[loops], %[loops], #0x1\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip1 z8.b, z9.b, z10.b\n" - "add %[a_ptr0], %[a_ptr0], #0x20\n" + "ld1b z14.b, p4/z, [%[b_ptr1]]\n" + "cbz %[loops], 1f\n" + "2:\n" "zip2 z9.b, z9.b, z10.b\n" + "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" "zip1 z10.b, z11.b, z12.b\n" + "subs %[loops], %[loops], #0x1\n" "zip2 z11.b, z11.b, z12.b\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "sdot z16.s, z8.b, z0.b[0]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "sdot z17.s, z9.b, z0.b[0]\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "zip2 z15.b, z12.b, z13.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "zip2 z15.b, z12.b, z13.b\n" + "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "zip1 z13.b, z12.b, z13.b\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" "sdot z18.s, z10.b, z0.b[0]\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" + "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "sdot z19.s, z11.b, z0.b[0]\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" + "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "zip2 z8.b, z14.b, z12.b\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" + "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr1], %[b_ptr1], %[ldb]\n" + "add %[a_ptr0], %[a_ptr0], #0x20\n" "zip1 z12.b, z13.b, z14.b\n" "zip2 z13.b, z13.b, z14.b\n" "zip1 z14.b, z15.b, z8.b\n" @@ -289,32 +260,32 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z4.b[3]\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z12.b, z10.b, z8.b\n" - "b.ne 3b\n" - "2:\n" "zip1 z10.b, z10.b, z8.b\n" "zip1 z8.b, z9.b, z10.b\n" + "b.ne 2b\n" + "1:\n" "zip2 z9.b, z9.b, z10.b\n" "zip1 z10.b, z11.b, z12.b\n" "zip2 z11.b, z11.b, z12.b\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" - "cbz %[regs], 4f\n" - "sdot z16.s, z8.b, z0.b[0]\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z17.s, z9.b, z0.b[0]\n" + "cbz %[regs], 3f\n" + "sdot z16.s, z8.b, z0.b[0]\n" "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" - "sdot z18.s, z10.b, z0.b[0]\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip2 z8.b, z14.b, z12.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "sdot z19.s, z11.b, z0.b[0]\n" + "sdot z18.s, z10.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "zip1 z12.b, z13.b, z14.b\n" + "sdot z19.s, z11.b, z0.b[0]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" + "zip1 z12.b, z13.b, z14.b\n" + "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z13.b, z13.b, z14.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "zip1 z14.b, z15.b, z8.b\n" @@ -377,6 +348,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "zip2 z12.b, z10.b, z8.b\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip1 z10.b, z10.b, z8.b\n" + "addvl %[a_ptr0], %[a_ptr0], #2\n" "zip1 z8.b, z9.b, z10.b\n" "zip2 z9.b, z9.b, z10.b\n" "zip1 z10.b, z11.b, z12.b\n" @@ -437,7 +409,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z4.b[3]\n" "sdot z18.s, z14.b, z4.b[3]\n" "sdot z19.s, z15.b, z4.b[3]\n" - "cbz %[blocks], 5f\n" + "cbz %[blocks], 4f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -459,7 +431,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z0.b[0]\n" "sdot z18.s, z10.b, z0.b[0]\n" "sdot z19.s, z11.b, z0.b[0]\n" - "b.eq 6f\n" + "b.eq 5f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -481,7 +453,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z0.b[1]\n" "sdot z18.s, z14.b, z0.b[1]\n" "sdot z19.s, z15.b, z0.b[1]\n" - "b.eq 7f\n" + "b.eq 6f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -502,31 +474,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z0.b[2]\n" "sdot z18.s, z10.b, z0.b[2]\n" "sdot z19.s, z11.b, z0.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 9f\n" + "b.eq 8f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 10f\n" + "b.eq 9f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "10:\n" + "b 10f\n" + "9:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "9:\n" + "b 10f\n" + "8:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "11:\n" + "10:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -540,33 +512,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z0.b[3]\n" "sdot z18.s, z14.b, z0.b[3]\n" "sdot z19.s, z15.b, z0.b[3]\n" - "b 8f\n" - "7:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "6:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 12f\n" + "b.eq 11f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 13f\n" + "b.eq 12f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "13:\n" + "b 13f\n" + "12:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "12:\n" + "b 13f\n" + "11:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "14:\n" + "13:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -580,33 +552,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z0.b[2]\n" "sdot z18.s, z10.b, z0.b[2]\n" "sdot z19.s, z11.b, z0.b[2]\n" - "b 8f\n" - "6:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "5:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 15f\n" + "b.eq 14f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 16f\n" + "b.eq 15f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "16:\n" + "b 16f\n" + "15:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "15:\n" + "b 16f\n" + "14:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "17:\n" + "16:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -620,33 +592,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z0.b[1]\n" "sdot z18.s, z14.b, z0.b[1]\n" "sdot z19.s, z15.b, z0.b[1]\n" - "b 8f\n" - "5:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "4:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 18f\n" + "b.eq 17f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 19f\n" + "b.eq 18f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "19:\n" + "b 19f\n" + "18:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "18:\n" + "b 19f\n" + "17:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "20:\n" + "19:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -660,36 +632,36 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z0.b[0]\n" "sdot z18.s, z10.b, z0.b[0]\n" "sdot z19.s, z11.b, z0.b[0]\n" - "b 8f\n" - "4:\n" + "b 7f\n" + "3:\n" "sdot z16.s, z8.b, z0.b[0]\n" - "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z17.s, z9.b, z0.b[0]\n" "ld1rqb z4.b, p6/z, [%[a_ptr0]]\n" - "sdot z18.s, z10.b, z0.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip2 z8.b, z14.b, z12.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "sdot z17.s, z9.b, z0.b[0]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "sdot z19.s, z11.b, z0.b[0]\n" + "sdot z18.s, z10.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip1 z12.b, z13.b, z14.b\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "zip2 z13.b, z13.b, z14.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip1 z14.b, z15.b, z8.b\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "sdot z16.s, z12.b, z0.b[1]\n" + "sdot z19.s, z11.b, z0.b[0]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "sdot z17.s, z13.b, z0.b[1]\n" - "ld1b z13.b, p4/z, [%[b_ptr2]]\n" + "sdot z16.s, z12.b, z0.b[1]\n" + "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip2 z11.b, z8.b, z9.b\n" + "addvl %[a_ptr0], %[a_ptr0], #1\n" "zip1 z9.b, z8.b, z9.b\n" "ld1b z8.b, p4/z, [%[b_ptr3]]\n" + "sdot z17.s, z13.b, z0.b[1]\n" + "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "sdot z18.s, z14.b, z0.b[1]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" "sdot z19.s, z15.b, z0.b[1]\n" @@ -718,7 +690,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z0.b[3]\n" "sdot z18.s, z14.b, z0.b[3]\n" "sdot z19.s, z15.b, z0.b[3]\n" - "cbz %[blocks], 21f\n" + "cbz %[blocks], 20f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -740,7 +712,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z4.b[0]\n" "sdot z18.s, z10.b, z4.b[0]\n" "sdot z19.s, z11.b, z4.b[0]\n" - "b.eq 22f\n" + "b.eq 21f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -762,7 +734,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z4.b[1]\n" "sdot z18.s, z14.b, z4.b[1]\n" "sdot z19.s, z15.b, z4.b[1]\n" - "b.eq 23f\n" + "b.eq 22f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -783,31 +755,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z4.b[2]\n" "sdot z18.s, z10.b, z4.b[2]\n" "sdot z19.s, z11.b, z4.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 24f\n" + "b.eq 23f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 25f\n" + "b.eq 24f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "25:\n" + "b 25f\n" + "24:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "24:\n" + "b 25f\n" + "23:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "26:\n" + "25:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -821,33 +793,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z4.b[3]\n" "sdot z18.s, z14.b, z4.b[3]\n" "sdot z19.s, z15.b, z4.b[3]\n" - "b 8f\n" - "23:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "22:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 27f\n" + "b.eq 26f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 28f\n" + "b.eq 27f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "28:\n" + "b 28f\n" + "27:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "27:\n" + "b 28f\n" + "26:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "29:\n" + "28:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -861,33 +833,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z4.b[2]\n" "sdot z18.s, z10.b, z4.b[2]\n" "sdot z19.s, z11.b, z4.b[2]\n" - "b 8f\n" - "22:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "21:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 30f\n" + "b.eq 29f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 31f\n" + "b.eq 30f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "31:\n" + "b 31f\n" + "30:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "30:\n" + "b 31f\n" + "29:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "32:\n" + "31:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -901,33 +873,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z13.b, z4.b[1]\n" "sdot z18.s, z14.b, z4.b[1]\n" "sdot z19.s, z15.b, z4.b[1]\n" - "b 8f\n" - "21:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "20:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 33f\n" + "b.eq 32f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 34f\n" + "b.eq 33f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "34:\n" + "b 34f\n" + "33:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "33:\n" + "b 34f\n" + "32:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "35:\n" + "34:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -941,14 +913,14 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z17.s, z9.b, z4.b[0]\n" "sdot z18.s, z10.b, z4.b[0]\n" "sdot z19.s, z11.b, z4.b[0]\n" - "8:\n" + "7:\n" "st1w z16.s, p0, [%[c_ptr0]]\n" "st1w z17.s, p1, [%[c_ptr0], #1, MUL VL]\n" "st1w z18.s, p2, [%[c_ptr0], #2, MUL VL]\n" "st1w z19.s, p3, [%[c_ptr0], #3, MUL VL]\n" "addvl %[c_ptr0], %[c_ptr0], #4\n" : [a_ptr0] "+r" (a_ptr0), [b_ptr0] "+r" (b_ptr0), [b_ptr1] "+r" (b_ptr1), [b_ptr2] "+r" (b_ptr2), [b_ptr3] "+r" (b_ptr3), [c_ptr0] "+r" (c_ptr0), [loops] "+r" (loops), [regs] "+r" (regs), [temp] "+r" (temp), [blocks] "+r" (blocks), [odds] "+r" (odds) - : [betaptr] "r" (betaptr), [width] "r" (width), [beta0] "r" (beta0), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) + : [width] "r" (width), [append] "r" (static_cast<uint64_t>(append)), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) : "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "z21", "z22", "z23", "z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31", "cc", "memory" ); break; @@ -956,125 +928,81 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l __asm __volatile ( "a_ptr1 .req X0\n" "c_ptr1 .req X1\n" + "mov z16.s, #0\n" "add a_ptr1, %[a_ptr0], %[lda]\n" + "mov z17.s, #0\n" "add c_ptr1, %[c_ptr0], %[ldc]\n" + "mov z18.s, #0\n" "whilelt p6.b, %[temp], %[leftovers]\n" + "mov z19.s, #0\n" "whilelt p0.s, %[temp], %[width]\n" + "mov z20.s, #0\n" "whilelt p4.b, %[temp], %[width]\n" + "mov z21.s, #0\n" "incw %[temp], all, mul #1\n" - "ptrue p7.b\n" - "whilelt p1.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p2.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p3.s, %[temp], %[width]\n" - "cbz %[beta0], 1f\n" - "mov z16.s, #0\n" - "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" - "mov z17.s, #0\n" - "ld1rqb z1.b, p7/z, [a_ptr1]\n" - "mov z18.s, #0\n" + "mov z22.s, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "mov z19.s, #0\n" + "mov z23.s, #0\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "mov z20.s, #0\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "mov z21.s, #0\n" - "add %[a_ptr0], %[a_ptr0], #0x10\n" + "whilelt p1.s, %[temp], %[width]\n" + "incw %[temp], all, mul #1\n" "zip2 z11.b, z8.b, z9.b\n" - "add a_ptr1, a_ptr1, #0x10\n" + "ptrue p7.b\n" "zip1 z9.b, z8.b, z9.b\n" "ld1b z8.b, p4/z, [%[b_ptr3]]\n" - "mov z22.s, #0\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "mov z23.s, #0\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "zip2 z12.b, z10.b, z8.b\n" - "ld1b z13.b, p4/z, [%[b_ptr2]]\n" - "zip1 z10.b, z10.b, z8.b\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "zip1 z8.b, z9.b, z10.b\n" - "zip2 z9.b, z9.b, z10.b\n" - "zip1 z10.b, z11.b, z12.b\n" - "zip2 z11.b, z11.b, z12.b\n" - "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "cbz %[loops], 2f\n" - "b 3f\n" - "1:\n" - "ld1rw z15.s, p7/z, [%[betaptr]]\n" - "ld1w z16.s, p0/z, [%[c_ptr0]]\n" - "ld1w z17.s, p1/z, [%[c_ptr0], #1, MUL VL]\n" - "ld1w z18.s, p2/z, [%[c_ptr0], #2, MUL VL]\n" - "ld1w z19.s, p3/z, [%[c_ptr0], #3, MUL VL]\n" - "mul z16.s, p7/m, z16.s, z15.s\n" - "ld1w z20.s, p0/z, [c_ptr1]\n" - "mul z17.s, p7/m, z17.s, z15.s\n" - "ld1w z21.s, p1/z, [c_ptr1, #1, MUL VL]\n" - "mul z18.s, p7/m, z18.s, z15.s\n" - "ld1w z22.s, p2/z, [c_ptr1, #2, MUL VL]\n" - "mul z19.s, p7/m, z19.s, z15.s\n" - "ld1w z23.s, p3/z, [c_ptr1, #3, MUL VL]\n" - "mul z20.s, p7/m, z20.s, z15.s\n" + "whilelt p2.s, %[temp], %[width]\n" + "incw %[temp], all, mul #1\n" "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" - "mul z21.s, p7/m, z21.s, z15.s\n" - "ld1rqb z1.b, p7/z, [a_ptr1]\n" - "mul z22.s, p7/m, z22.s, z15.s\n" - "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "mul z23.s, p7/m, z23.s, z15.s\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "add %[a_ptr0], %[a_ptr0], #0x10\n" - "add a_ptr1, a_ptr1, #0x10\n" - "zip2 z11.b, z8.b, z9.b\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "zip1 z9.b, z8.b, z9.b\n" - "ld1b z8.b, p4/z, [%[b_ptr3]]\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "ld1b z13.b, p4/z, [%[b_ptr2]]\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z12.b, z10.b, z8.b\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" + "ld1rqb z1.b, p7/z, [a_ptr1]\n" "zip1 z10.b, z10.b, z8.b\n" + "whilelt p3.s, %[temp], %[width]\n" + "add a_ptr1, a_ptr1, #0x10\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z8.b, z9.b, z10.b\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "zip2 z9.b, z9.b, z10.b\n" + "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "zip1 z10.b, z11.b, z12.b\n" + "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip2 z11.b, z11.b, z12.b\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "cbz %[loops], 2f\n" - "3:\n" - "sdot z16.s, z8.b, z0.b[0]\n" - "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" + "ld1b z14.b, p4/z, [%[b_ptr1]]\n" + "add %[b_ptr3], %[b_ptr3], %[ldb]\n" + "cbz %[loops], 1f\n" + "2:\n" "zip2 z15.b, z12.b, z13.b\n" - "ld1rqb z5.b, p7/z, [a_ptr1]\n" + "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" "zip1 z13.b, z12.b, z13.b\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" + "sdot z16.s, z8.b, z0.b[0]\n" + "ld1rqb z5.b, p7/z, [a_ptr1]\n" "sdot z20.s, z8.b, z1.b[0]\n" "subs %[loops], %[loops], #0x1\n" - "sdot z17.s, z9.b, z0.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip2 z8.b, z14.b, z12.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "sdot z21.s, z9.b, z1.b[0]\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "sdot z18.s, z10.b, z0.b[0]\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip1 z12.b, z13.b, z14.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z13.b, z13.b, z14.b\n" "add %[a_ptr0], %[a_ptr0], #0x20\n" "zip1 z14.b, z15.b, z8.b\n" "add a_ptr1, a_ptr1, #0x20\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" + "sdot z21.s, z9.b, z1.b[0]\n" + "ld1b z9.b, p4/z, [%[b_ptr2]]\n" + "sdot z18.s, z10.b, z0.b[0]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "sdot z22.s, z10.b, z1.b[0]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "sdot z19.s, z11.b, z0.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "sdot z23.s, z11.b, z1.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip2 z11.b, z8.b, z9.b\n" @@ -1244,38 +1172,38 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "ld1b z14.b, p4/z, [%[b_ptr1]]\n" "sdot z19.s, z15.b, z4.b[3]\n" "sdot z23.s, z15.b, z5.b[3]\n" - "b.ne 3b\n" - "2:\n" + "b.ne 2b\n" + "1:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" - "cbz %[regs], 4f\n" - "sdot z16.s, z8.b, z0.b[0]\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z20.s, z8.b, z1.b[0]\n" + "cbz %[regs], 3f\n" + "sdot z16.s, z8.b, z0.b[0]\n" "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" - "sdot z17.s, z9.b, z0.b[0]\n" + "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z5.b, p7/z, [a_ptr1]\n" - "sdot z21.s, z9.b, z1.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip2 z8.b, z14.b, z12.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "sdot z18.s, z10.b, z0.b[0]\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "sdot z22.s, z10.b, z1.b[0]\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "zip1 z12.b, z13.b, z14.b\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z13.b, z13.b, z14.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "zip1 z14.b, z15.b, z8.b\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "sdot z19.s, z11.b, z0.b[0]\n" + "sdot z21.s, z9.b, z1.b[0]\n" + "ld1b z9.b, p4/z, [%[b_ptr2]]\n" + "sdot z18.s, z10.b, z0.b[0]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" + "sdot z22.s, z10.b, z1.b[0]\n" + "ld1b z10.b, p4/z, [%[b_ptr1]]\n" + "sdot z19.s, z11.b, z0.b[0]\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "sdot z23.s, z11.b, z1.b[0]\n" + "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "ld1b z8.b, p4/z, [%[b_ptr3]]\n" @@ -1357,23 +1285,25 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z20.s, z8.b, z5.b[0]\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z8.b, z14.b, z12.b\n" + "addvl %[a_ptr0], %[a_ptr0], #2\n" "zip1 z14.b, z14.b, z12.b\n" + "addvl a_ptr1, a_ptr1, #2\n" "sdot z17.s, z9.b, z4.b[0]\n" "sdot z21.s, z9.b, z5.b[0]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "sdot z18.s, z10.b, z4.b[0]\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "zip1 z12.b, z13.b, z14.b\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "zip2 z13.b, z13.b, z14.b\n" "zip1 z14.b, z15.b, z8.b\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" + "sdot z18.s, z10.b, z4.b[0]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "sdot z22.s, z10.b, z5.b[0]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "sdot z19.s, z11.b, z4.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "sdot z23.s, z11.b, z5.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" + "sdot z23.s, z11.b, z5.b[0]\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "ld1b z8.b, p4/z, [%[b_ptr3]]\n" @@ -1420,7 +1350,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z5.b[3]\n" "sdot z19.s, z15.b, z4.b[3]\n" "sdot z23.s, z15.b, z5.b[3]\n" - "cbz %[blocks], 5f\n" + "cbz %[blocks], 4f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -1446,7 +1376,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z1.b[0]\n" "sdot z19.s, z11.b, z0.b[0]\n" "sdot z23.s, z11.b, z1.b[0]\n" - "b.eq 6f\n" + "b.eq 5f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -1472,7 +1402,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z1.b[1]\n" "sdot z19.s, z15.b, z0.b[1]\n" "sdot z23.s, z15.b, z1.b[1]\n" - "b.eq 7f\n" + "b.eq 6f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -1497,31 +1427,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z1.b[2]\n" "sdot z19.s, z11.b, z0.b[2]\n" "sdot z23.s, z11.b, z1.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 9f\n" + "b.eq 8f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 10f\n" + "b.eq 9f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "10:\n" + "b 10f\n" + "9:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "9:\n" + "b 10f\n" + "8:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "11:\n" + "10:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -1539,33 +1469,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z1.b[3]\n" "sdot z19.s, z15.b, z0.b[3]\n" "sdot z23.s, z15.b, z1.b[3]\n" - "b 8f\n" - "7:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "6:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 12f\n" + "b.eq 11f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 13f\n" + "b.eq 12f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "13:\n" + "b 13f\n" + "12:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "12:\n" + "b 13f\n" + "11:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "14:\n" + "13:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -1583,33 +1513,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z1.b[2]\n" "sdot z19.s, z11.b, z0.b[2]\n" "sdot z23.s, z11.b, z1.b[2]\n" - "b 8f\n" - "6:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "5:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 15f\n" + "b.eq 14f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 16f\n" + "b.eq 15f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "16:\n" + "b 16f\n" + "15:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "15:\n" + "b 16f\n" + "14:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "17:\n" + "16:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -1627,33 +1557,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z1.b[1]\n" "sdot z19.s, z15.b, z0.b[1]\n" "sdot z23.s, z15.b, z1.b[1]\n" - "b 8f\n" - "5:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "4:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 18f\n" + "b.eq 17f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 19f\n" + "b.eq 18f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "19:\n" + "b 19f\n" + "18:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "18:\n" + "b 19f\n" + "17:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "20:\n" + "19:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -1671,35 +1601,36 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z1.b[0]\n" "sdot z19.s, z11.b, z0.b[0]\n" "sdot z23.s, z11.b, z1.b[0]\n" - "b 8f\n" - "4:\n" + "b 7f\n" + "3:\n" "sdot z16.s, z8.b, z0.b[0]\n" - "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z4.b, p6/z, [%[a_ptr0]]\n" - "sdot z17.s, z9.b, z0.b[0]\n" + "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z5.b, p6/z, [a_ptr1]\n" - "sdot z21.s, z9.b, z1.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip2 z8.b, z14.b, z12.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "sdot z18.s, z10.b, z0.b[0]\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "sdot z22.s, z10.b, z1.b[0]\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "zip1 z12.b, z13.b, z14.b\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z13.b, z13.b, z14.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "addvl %[a_ptr0], %[a_ptr0], #1\n" "zip1 z14.b, z15.b, z8.b\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" + "addvl a_ptr1, a_ptr1, #1\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "sdot z19.s, z11.b, z0.b[0]\n" + "sdot z21.s, z9.b, z1.b[0]\n" + "ld1b z9.b, p4/z, [%[b_ptr2]]\n" + "sdot z18.s, z10.b, z0.b[0]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" + "sdot z22.s, z10.b, z1.b[0]\n" + "ld1b z10.b, p4/z, [%[b_ptr1]]\n" + "sdot z19.s, z11.b, z0.b[0]\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "sdot z23.s, z11.b, z1.b[0]\n" + "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "ld1b z8.b, p4/z, [%[b_ptr3]]\n" @@ -1746,7 +1677,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z1.b[3]\n" "sdot z19.s, z15.b, z0.b[3]\n" "sdot z23.s, z15.b, z1.b[3]\n" - "cbz %[blocks], 21f\n" + "cbz %[blocks], 20f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -1772,7 +1703,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z5.b[0]\n" "sdot z19.s, z11.b, z4.b[0]\n" "sdot z23.s, z11.b, z5.b[0]\n" - "b.eq 22f\n" + "b.eq 21f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -1798,7 +1729,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z5.b[1]\n" "sdot z19.s, z15.b, z4.b[1]\n" "sdot z23.s, z15.b, z5.b[1]\n" - "b.eq 23f\n" + "b.eq 22f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -1823,31 +1754,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z5.b[2]\n" "sdot z19.s, z11.b, z4.b[2]\n" "sdot z23.s, z11.b, z5.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 24f\n" + "b.eq 23f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 25f\n" + "b.eq 24f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "25:\n" + "b 25f\n" + "24:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "24:\n" + "b 25f\n" + "23:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "26:\n" + "25:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -1865,33 +1796,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z5.b[3]\n" "sdot z19.s, z15.b, z4.b[3]\n" "sdot z23.s, z15.b, z5.b[3]\n" - "b 8f\n" - "23:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "22:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 27f\n" + "b.eq 26f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 28f\n" + "b.eq 27f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "28:\n" + "b 28f\n" + "27:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "27:\n" + "b 28f\n" + "26:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "29:\n" + "28:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -1909,33 +1840,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z5.b[2]\n" "sdot z19.s, z11.b, z4.b[2]\n" "sdot z23.s, z11.b, z5.b[2]\n" - "b 8f\n" - "22:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "21:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 30f\n" + "b.eq 29f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 31f\n" + "b.eq 30f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "31:\n" + "b 31f\n" + "30:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "30:\n" + "b 31f\n" + "29:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "32:\n" + "31:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -1953,33 +1884,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z14.b, z5.b[1]\n" "sdot z19.s, z15.b, z4.b[1]\n" "sdot z23.s, z15.b, z5.b[1]\n" - "b 8f\n" - "21:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "20:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 33f\n" + "b.eq 32f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 34f\n" + "b.eq 33f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "34:\n" + "b 34f\n" + "33:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "33:\n" + "b 34f\n" + "32:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "35:\n" + "34:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -1997,7 +1928,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z22.s, z10.b, z5.b[0]\n" "sdot z19.s, z11.b, z4.b[0]\n" "sdot z23.s, z11.b, z5.b[0]\n" - "8:\n" + "7:\n" "st1w z16.s, p0, [%[c_ptr0]]\n" "st1w z17.s, p1, [%[c_ptr0], #1, MUL VL]\n" "st1w z18.s, p2, [%[c_ptr0], #2, MUL VL]\n" @@ -2010,7 +1941,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l ".unreq a_ptr1\n" ".unreq c_ptr1\n" : [a_ptr0] "+r" (a_ptr0), [b_ptr0] "+r" (b_ptr0), [b_ptr1] "+r" (b_ptr1), [b_ptr2] "+r" (b_ptr2), [b_ptr3] "+r" (b_ptr3), [c_ptr0] "+r" (c_ptr0), [loops] "+r" (loops), [regs] "+r" (regs), [temp] "+r" (temp), [blocks] "+r" (blocks), [odds] "+r" (odds) - : [betaptr] "r" (betaptr), [width] "r" (width), [beta0] "r" (beta0), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) + : [width] "r" (width), [append] "r" (static_cast<uint64_t>(append)), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) : "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "z21", "z22", "z23", "z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31", "x0", "x1", "cc", "memory" ); break; @@ -2020,119 +1951,65 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "a_ptr2 .req X1\n" "c_ptr1 .req X2\n" "c_ptr2 .req X3\n" + "mov z16.s, #0\n" "add a_ptr1, %[a_ptr0], %[lda]\n" - "add c_ptr1, %[c_ptr0], %[ldc]\n" + "mov z17.s, #0\n" "add a_ptr2, a_ptr1, %[lda]\n" + "mov z18.s, #0\n" + "add c_ptr1, %[c_ptr0], %[ldc]\n" + "mov z19.s, #0\n" "add c_ptr2, c_ptr1, %[ldc]\n" + "mov z20.s, #0\n" "whilelt p6.b, %[temp], %[leftovers]\n" + "mov z21.s, #0\n" "whilelt p0.s, %[temp], %[width]\n" + "mov z22.s, #0\n" "whilelt p4.b, %[temp], %[width]\n" + "mov z23.s, #0\n" "incw %[temp], all, mul #1\n" - "ptrue p7.b\n" - "whilelt p1.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p2.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p3.s, %[temp], %[width]\n" - "cbz %[beta0], 1f\n" - "mov z16.s, #0\n" - "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" - "mov z17.s, #0\n" - "ld1rqb z1.b, p7/z, [a_ptr1]\n" - "mov z18.s, #0\n" - "ld1rqb z2.b, p7/z, [a_ptr2]\n" - "mov z19.s, #0\n" + "mov z24.s, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "mov z20.s, #0\n" + "mov z25.s, #0\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" - "mov z21.s, #0\n" + "mov z26.s, #0\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "mov z22.s, #0\n" - "add %[a_ptr0], %[a_ptr0], #0x10\n" + "mov z27.s, #0\n" + "whilelt p1.s, %[temp], %[width]\n" "zip2 z11.b, z8.b, z9.b\n" - "add a_ptr1, a_ptr1, #0x10\n" + "incw %[temp], all, mul #1\n" "zip1 z9.b, z8.b, z9.b\n" "ld1b z8.b, p4/z, [%[b_ptr3]]\n" - "mov z23.s, #0\n" - "add a_ptr2, a_ptr2, #0x10\n" - "mov z24.s, #0\n" + "ptrue p7.b\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" + "whilelt p2.s, %[temp], %[width]\n" "zip2 z12.b, z10.b, z8.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "zip1 z10.b, z10.b, z8.b\n" - "ld1b z13.b, p4/z, [%[b_ptr2]]\n" - "mov z25.s, #0\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "mov z26.s, #0\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "zip1 z8.b, z9.b, z10.b\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "zip2 z9.b, z9.b, z10.b\n" - "zip1 z10.b, z11.b, z12.b\n" - "zip2 z11.b, z11.b, z12.b\n" - "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "mov z27.s, #0\n" - "cbz %[loops], 2f\n" - "b 3f\n" - "1:\n" - "ld1rw z15.s, p7/z, [%[betaptr]]\n" - "ld1w z16.s, p0/z, [%[c_ptr0]]\n" - "ld1w z17.s, p1/z, [%[c_ptr0], #1, MUL VL]\n" - "ld1w z18.s, p2/z, [%[c_ptr0], #2, MUL VL]\n" - "ld1w z19.s, p3/z, [%[c_ptr0], #3, MUL VL]\n" - "mul z16.s, p7/m, z16.s, z15.s\n" - "ld1w z20.s, p0/z, [c_ptr1]\n" - "mul z17.s, p7/m, z17.s, z15.s\n" - "ld1w z21.s, p1/z, [c_ptr1, #1, MUL VL]\n" - "mul z18.s, p7/m, z18.s, z15.s\n" - "ld1w z22.s, p2/z, [c_ptr1, #2, MUL VL]\n" - "mul z19.s, p7/m, z19.s, z15.s\n" - "ld1w z23.s, p3/z, [c_ptr1, #3, MUL VL]\n" - "mul z20.s, p7/m, z20.s, z15.s\n" - "ld1w z24.s, p0/z, [c_ptr2]\n" - "mul z21.s, p7/m, z21.s, z15.s\n" - "ld1w z25.s, p1/z, [c_ptr2, #1, MUL VL]\n" - "mul z22.s, p7/m, z22.s, z15.s\n" - "ld1w z26.s, p2/z, [c_ptr2, #2, MUL VL]\n" - "mul z23.s, p7/m, z23.s, z15.s\n" - "ld1w z27.s, p3/z, [c_ptr2, #3, MUL VL]\n" - "mul z24.s, p7/m, z24.s, z15.s\n" "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" - "mul z25.s, p7/m, z25.s, z15.s\n" + "zip1 z10.b, z10.b, z8.b\n" "ld1rqb z1.b, p7/z, [a_ptr1]\n" - "mul z26.s, p7/m, z26.s, z15.s\n" "ld1rqb z2.b, p7/z, [a_ptr2]\n" - "mul z27.s, p7/m, z27.s, z15.s\n" - "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" + "incw %[temp], all, mul #1\n" "add %[a_ptr0], %[a_ptr0], #0x10\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" + "zip1 z8.b, z9.b, z10.b\n" "add a_ptr1, a_ptr1, #0x10\n" - "zip2 z11.b, z8.b, z9.b\n" + "zip2 z9.b, z9.b, z10.b\n" + "whilelt p3.s, %[temp], %[width]\n" + "zip1 z10.b, z11.b, z12.b\n" "add a_ptr2, a_ptr2, #0x10\n" - "zip1 z9.b, z8.b, z9.b\n" - "ld1b z8.b, p4/z, [%[b_ptr3]]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" + "zip2 z11.b, z11.b, z12.b\n" + "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "zip2 z12.b, z10.b, z8.b\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" - "zip1 z10.b, z10.b, z8.b\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "zip1 z8.b, z9.b, z10.b\n" - "zip2 z9.b, z9.b, z10.b\n" - "zip1 z10.b, z11.b, z12.b\n" - "zip2 z11.b, z11.b, z12.b\n" - "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "cbz %[loops], 2f\n" - "3:\n" - "sdot z16.s, z8.b, z0.b[0]\n" - "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" + "ld1b z14.b, p4/z, [%[b_ptr1]]\n" + "cbz %[loops], 1f\n" + "2:\n" "zip2 z15.b, z12.b, z13.b\n" - "ld1rqb z5.b, p7/z, [a_ptr1]\n" + "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" "zip1 z13.b, z12.b, z13.b\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" + "sdot z16.s, z8.b, z0.b[0]\n" + "ld1rqb z5.b, p7/z, [a_ptr1]\n" "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z6.b, p7/z, [a_ptr2]\n" "sdot z24.s, z8.b, z2.b[0]\n" @@ -2361,24 +2238,23 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z4.b[3]\n" "sdot z23.s, z15.b, z5.b[3]\n" "sdot z27.s, z15.b, z6.b[3]\n" - "b.ne 3b\n" - "2:\n" + "b.ne 2b\n" + "1:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" - "cbz %[regs], 4f\n" - "sdot z16.s, z8.b, z0.b[0]\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z20.s, z8.b, z1.b[0]\n" + "cbz %[regs], 3f\n" + "sdot z16.s, z8.b, z0.b[0]\n" "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" - "sdot z24.s, z8.b, z2.b[0]\n" + "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z5.b, p7/z, [a_ptr1]\n" - "sdot z17.s, z9.b, z0.b[0]\n" + "sdot z24.s, z8.b, z2.b[0]\n" "ld1rqb z6.b, p7/z, [a_ptr2]\n" "zip2 z8.b, z14.b, z12.b\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "sdot z21.s, z9.b, z1.b[0]\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip1 z12.b, z13.b, z14.b\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" @@ -2386,12 +2262,13 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "zip1 z14.b, z15.b, z8.b\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" + "sdot z21.s, z9.b, z1.b[0]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "sdot z25.s, z9.b, z2.b[0]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "sdot z18.s, z10.b, z0.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "sdot z22.s, z10.b, z1.b[0]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "sdot z22.s, z10.b, z1.b[0]\n" "sdot z26.s, z10.b, z2.b[0]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "sdot z19.s, z11.b, z0.b[0]\n" @@ -2491,8 +2368,11 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z16.s, z8.b, z4.b[0]\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "sdot z20.s, z8.b, z5.b[0]\n" + "addvl %[a_ptr0], %[a_ptr0], #2\n" "sdot z24.s, z8.b, z6.b[0]\n" + "addvl a_ptr1, a_ptr1, #2\n" "zip2 z8.b, z14.b, z12.b\n" + "addvl a_ptr2, a_ptr2, #2\n" "zip1 z14.b, z14.b, z12.b\n" "sdot z17.s, z9.b, z4.b[0]\n" "sdot z21.s, z9.b, z5.b[0]\n" @@ -2571,7 +2451,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z4.b[3]\n" "sdot z23.s, z15.b, z5.b[3]\n" "sdot z27.s, z15.b, z6.b[3]\n" - "cbz %[blocks], 5f\n" + "cbz %[blocks], 4f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -2601,7 +2481,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z0.b[0]\n" "sdot z23.s, z11.b, z1.b[0]\n" "sdot z27.s, z11.b, z2.b[0]\n" - "b.eq 6f\n" + "b.eq 5f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -2631,7 +2511,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z0.b[1]\n" "sdot z23.s, z15.b, z1.b[1]\n" "sdot z27.s, z15.b, z2.b[1]\n" - "b.eq 7f\n" + "b.eq 6f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -2660,31 +2540,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z0.b[2]\n" "sdot z23.s, z11.b, z1.b[2]\n" "sdot z27.s, z11.b, z2.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 9f\n" + "b.eq 8f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 10f\n" + "b.eq 9f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "10:\n" + "b 10f\n" + "9:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "9:\n" + "b 10f\n" + "8:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "11:\n" + "10:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -2706,33 +2586,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z0.b[3]\n" "sdot z23.s, z15.b, z1.b[3]\n" "sdot z27.s, z15.b, z2.b[3]\n" - "b 8f\n" - "7:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "6:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 12f\n" + "b.eq 11f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 13f\n" + "b.eq 12f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "13:\n" + "b 13f\n" + "12:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "12:\n" + "b 13f\n" + "11:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "14:\n" + "13:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -2754,33 +2634,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z0.b[2]\n" "sdot z23.s, z11.b, z1.b[2]\n" "sdot z27.s, z11.b, z2.b[2]\n" - "b 8f\n" - "6:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "5:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 15f\n" + "b.eq 14f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 16f\n" + "b.eq 15f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "16:\n" + "b 16f\n" + "15:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "15:\n" + "b 16f\n" + "14:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "17:\n" + "16:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -2802,33 +2682,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z0.b[1]\n" "sdot z23.s, z15.b, z1.b[1]\n" "sdot z27.s, z15.b, z2.b[1]\n" - "b 8f\n" - "5:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "4:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 18f\n" + "b.eq 17f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 19f\n" + "b.eq 18f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "19:\n" + "b 19f\n" + "18:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "18:\n" + "b 19f\n" + "17:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "20:\n" + "19:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -2850,34 +2730,36 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z0.b[0]\n" "sdot z23.s, z11.b, z1.b[0]\n" "sdot z27.s, z11.b, z2.b[0]\n" - "b 8f\n" - "4:\n" + "b 7f\n" + "3:\n" "sdot z16.s, z8.b, z0.b[0]\n" - "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z4.b, p6/z, [%[a_ptr0]]\n" - "sdot z24.s, z8.b, z2.b[0]\n" + "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z5.b, p6/z, [a_ptr1]\n" - "sdot z17.s, z9.b, z0.b[0]\n" + "sdot z24.s, z8.b, z2.b[0]\n" "ld1rqb z6.b, p6/z, [a_ptr2]\n" "zip2 z8.b, z14.b, z12.b\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "sdot z21.s, z9.b, z1.b[0]\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip1 z12.b, z13.b, z14.b\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z13.b, z13.b, z14.b\n" + "addvl %[a_ptr0], %[a_ptr0], #1\n" "zip1 z14.b, z15.b, z8.b\n" + "addvl a_ptr1, a_ptr1, #1\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" + "sdot z21.s, z9.b, z1.b[0]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "sdot z25.s, z9.b, z2.b[0]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "sdot z18.s, z10.b, z0.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "sdot z22.s, z10.b, z1.b[0]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "sdot z22.s, z10.b, z1.b[0]\n" + "addvl a_ptr2, a_ptr2, #1\n" "sdot z26.s, z10.b, z2.b[0]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "sdot z19.s, z11.b, z0.b[0]\n" @@ -2942,7 +2824,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z0.b[3]\n" "sdot z23.s, z15.b, z1.b[3]\n" "sdot z27.s, z15.b, z2.b[3]\n" - "cbz %[blocks], 21f\n" + "cbz %[blocks], 20f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -2972,7 +2854,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z4.b[0]\n" "sdot z23.s, z11.b, z5.b[0]\n" "sdot z27.s, z11.b, z6.b[0]\n" - "b.eq 22f\n" + "b.eq 21f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -3002,7 +2884,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z4.b[1]\n" "sdot z23.s, z15.b, z5.b[1]\n" "sdot z27.s, z15.b, z6.b[1]\n" - "b.eq 23f\n" + "b.eq 22f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -3031,31 +2913,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z4.b[2]\n" "sdot z23.s, z11.b, z5.b[2]\n" "sdot z27.s, z11.b, z6.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 24f\n" + "b.eq 23f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 25f\n" + "b.eq 24f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "25:\n" + "b 25f\n" + "24:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "24:\n" + "b 25f\n" + "23:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "26:\n" + "25:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -3077,33 +2959,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z4.b[3]\n" "sdot z23.s, z15.b, z5.b[3]\n" "sdot z27.s, z15.b, z6.b[3]\n" - "b 8f\n" - "23:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "22:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 27f\n" + "b.eq 26f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 28f\n" + "b.eq 27f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "28:\n" + "b 28f\n" + "27:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "27:\n" + "b 28f\n" + "26:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "29:\n" + "28:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -3125,33 +3007,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z4.b[2]\n" "sdot z23.s, z11.b, z5.b[2]\n" "sdot z27.s, z11.b, z6.b[2]\n" - "b 8f\n" - "22:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "21:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 30f\n" + "b.eq 29f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 31f\n" + "b.eq 30f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "31:\n" + "b 31f\n" + "30:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "30:\n" + "b 31f\n" + "29:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "32:\n" + "31:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -3173,33 +3055,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z15.b, z4.b[1]\n" "sdot z23.s, z15.b, z5.b[1]\n" "sdot z27.s, z15.b, z6.b[1]\n" - "b 8f\n" - "21:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "20:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 33f\n" + "b.eq 32f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 34f\n" + "b.eq 33f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "34:\n" + "b 34f\n" + "33:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "33:\n" + "b 34f\n" + "32:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "35:\n" + "34:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -3221,7 +3103,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z19.s, z11.b, z4.b[0]\n" "sdot z23.s, z11.b, z5.b[0]\n" "sdot z27.s, z11.b, z6.b[0]\n" - "8:\n" + "7:\n" "st1w z16.s, p0, [%[c_ptr0]]\n" "st1w z17.s, p1, [%[c_ptr0], #1, MUL VL]\n" "st1w z18.s, p2, [%[c_ptr0], #2, MUL VL]\n" @@ -3240,7 +3122,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l ".unreq c_ptr1\n" ".unreq c_ptr2\n" : [a_ptr0] "+r" (a_ptr0), [b_ptr0] "+r" (b_ptr0), [b_ptr1] "+r" (b_ptr1), [b_ptr2] "+r" (b_ptr2), [b_ptr3] "+r" (b_ptr3), [c_ptr0] "+r" (c_ptr0), [loops] "+r" (loops), [regs] "+r" (regs), [temp] "+r" (temp), [blocks] "+r" (blocks), [odds] "+r" (odds) - : [betaptr] "r" (betaptr), [width] "r" (width), [beta0] "r" (beta0), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) + : [width] "r" (width), [append] "r" (static_cast<uint64_t>(append)), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) : "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "z21", "z22", "z23", "z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31", "x0", "x1", "x2", "x3", "cc", "memory" ); break; @@ -3253,175 +3135,109 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "c_ptr1 .req X3\n" "c_ptr2 .req X4\n" "c_ptr3 .req X5\n" - "add a_ptr1, %[a_ptr0], %[lda]\n" - "add c_ptr1, %[c_ptr0], %[ldc]\n" - "add a_ptr2, a_ptr1, %[lda]\n" - "add c_ptr2, c_ptr1, %[ldc]\n" - "add a_ptr3, a_ptr2, %[lda]\n" - "add c_ptr3, c_ptr2, %[ldc]\n" - "whilelt p6.b, %[temp], %[leftovers]\n" - "whilelt p0.s, %[temp], %[width]\n" - "whilelt p4.b, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "ptrue p7.b\n" - "whilelt p1.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p2.s, %[temp], %[width]\n" - "incw %[temp], all, mul #1\n" - "whilelt p3.s, %[temp], %[width]\n" - "cbz %[beta0], 1f\n" "mov z16.s, #0\n" - "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" + "add a_ptr1, %[a_ptr0], %[lda]\n" "mov z17.s, #0\n" - "ld1rqb z1.b, p7/z, [a_ptr1]\n" + "add a_ptr2, a_ptr1, %[lda]\n" "mov z18.s, #0\n" - "ld1rqb z2.b, p7/z, [a_ptr2]\n" + "add a_ptr3, a_ptr2, %[lda]\n" "mov z19.s, #0\n" - "ld1rqb z3.b, p7/z, [a_ptr3]\n" + "add c_ptr1, %[c_ptr0], %[ldc]\n" "mov z20.s, #0\n" - "ld1b z8.b, p4/z, [%[b_ptr0]]\n" + "add c_ptr2, c_ptr1, %[ldc]\n" "mov z21.s, #0\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" + "add c_ptr3, c_ptr2, %[ldc]\n" "mov z22.s, #0\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" + "whilelt p6.b, %[temp], %[leftovers]\n" "mov z23.s, #0\n" - "add %[a_ptr0], %[a_ptr0], #0x10\n" - "zip2 z11.b, z8.b, z9.b\n" - "add a_ptr1, a_ptr1, #0x10\n" - "zip1 z9.b, z8.b, z9.b\n" - "ld1b z8.b, p4/z, [%[b_ptr3]]\n" + "whilelt p0.s, %[temp], %[width]\n" "mov z24.s, #0\n" - "add a_ptr2, a_ptr2, #0x10\n" + "whilelt p4.b, %[temp], %[width]\n" "mov z25.s, #0\n" - "add a_ptr3, a_ptr3, #0x10\n" - "zip2 z12.b, z10.b, z8.b\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "zip1 z10.b, z10.b, z8.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "incw %[temp], all, mul #1\n" "mov z26.s, #0\n" - "ld1b z13.b, p4/z, [%[b_ptr2]]\n" + "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "mov z27.s, #0\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "zip1 z8.b, z9.b, z10.b\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "zip2 z9.b, z9.b, z10.b\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "zip1 z10.b, z11.b, z12.b\n" - "zip2 z11.b, z11.b, z12.b\n" - "ld1b z12.b, p4/z, [%[b_ptr0]]\n" + "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "mov z28.s, #0\n" + "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "mov z29.s, #0\n" + "whilelt p1.s, %[temp], %[width]\n" + "zip2 z11.b, z8.b, z9.b\n" + "incw %[temp], all, mul #1\n" + "zip1 z9.b, z8.b, z9.b\n" + "ld1b z8.b, p4/z, [%[b_ptr3]]\n" "mov z30.s, #0\n" - "zip2 z15.b, z12.b, z13.b\n" - "zip1 z13.b, z12.b, z13.b\n" + "ptrue p7.b\n" "mov z31.s, #0\n" - "cbz %[loops], 2f\n" - "b 3f\n" - "1:\n" - "ld1rw z15.s, p7/z, [%[betaptr]]\n" - "ld1w z16.s, p0/z, [%[c_ptr0]]\n" - "ld1w z17.s, p1/z, [%[c_ptr0], #1, MUL VL]\n" - "ld1w z18.s, p2/z, [%[c_ptr0], #2, MUL VL]\n" - "ld1w z19.s, p3/z, [%[c_ptr0], #3, MUL VL]\n" - "mul z16.s, p7/m, z16.s, z15.s\n" - "ld1w z20.s, p0/z, [c_ptr1]\n" - "mul z17.s, p7/m, z17.s, z15.s\n" - "ld1w z21.s, p1/z, [c_ptr1, #1, MUL VL]\n" - "mul z18.s, p7/m, z18.s, z15.s\n" - "ld1w z22.s, p2/z, [c_ptr1, #2, MUL VL]\n" - "mul z19.s, p7/m, z19.s, z15.s\n" - "ld1w z23.s, p3/z, [c_ptr1, #3, MUL VL]\n" - "mul z20.s, p7/m, z20.s, z15.s\n" - "ld1w z24.s, p0/z, [c_ptr2]\n" - "mul z21.s, p7/m, z21.s, z15.s\n" - "ld1w z25.s, p1/z, [c_ptr2, #1, MUL VL]\n" - "mul z22.s, p7/m, z22.s, z15.s\n" - "ld1w z26.s, p2/z, [c_ptr2, #2, MUL VL]\n" - "mul z23.s, p7/m, z23.s, z15.s\n" - "ld1w z27.s, p3/z, [c_ptr2, #3, MUL VL]\n" - "mul z24.s, p7/m, z24.s, z15.s\n" - "ld1w z28.s, p0/z, [c_ptr3]\n" - "mul z25.s, p7/m, z25.s, z15.s\n" - "ld1w z29.s, p1/z, [c_ptr3, #1, MUL VL]\n" - "mul z26.s, p7/m, z26.s, z15.s\n" - "ld1w z30.s, p2/z, [c_ptr3, #2, MUL VL]\n" - "mul z27.s, p7/m, z27.s, z15.s\n" - "ld1w z31.s, p3/z, [c_ptr3, #3, MUL VL]\n" - "mul z28.s, p7/m, z28.s, z15.s\n" + "whilelt p2.s, %[temp], %[width]\n" + "zip2 z12.b, z10.b, z8.b\n" "ld1rqb z0.b, p7/z, [%[a_ptr0]]\n" - "mul z29.s, p7/m, z29.s, z15.s\n" + "zip1 z10.b, z10.b, z8.b\n" "ld1rqb z1.b, p7/z, [a_ptr1]\n" - "mul z30.s, p7/m, z30.s, z15.s\n" "ld1rqb z2.b, p7/z, [a_ptr2]\n" - "mul z31.s, p7/m, z31.s, z15.s\n" + "incw %[temp], all, mul #1\n" "ld1rqb z3.b, p7/z, [a_ptr3]\n" - "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[a_ptr0], %[a_ptr0], #0x10\n" - "ld1b z9.b, p4/z, [%[b_ptr2]]\n" + "zip1 z8.b, z9.b, z10.b\n" + "whilelt p3.s, %[temp], %[width]\n" + "zip2 z9.b, z9.b, z10.b\n" "add a_ptr1, a_ptr1, #0x10\n" - "ld1b z10.b, p4/z, [%[b_ptr1]]\n" + "zip1 z10.b, z11.b, z12.b\n" "add a_ptr2, a_ptr2, #0x10\n" - "zip2 z11.b, z8.b, z9.b\n" + "zip2 z11.b, z11.b, z12.b\n" "add a_ptr3, a_ptr3, #0x10\n" - "zip1 z9.b, z8.b, z9.b\n" - "ld1b z8.b, p4/z, [%[b_ptr3]]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "zip2 z12.b, z10.b, z8.b\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" - "zip1 z10.b, z10.b, z8.b\n" - "ld1b z14.b, p4/z, [%[b_ptr1]]\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" - "zip1 z8.b, z9.b, z10.b\n" - "zip2 z9.b, z9.b, z10.b\n" - "zip1 z10.b, z11.b, z12.b\n" - "zip2 z11.b, z11.b, z12.b\n" - "ld1b z12.b, p4/z, [%[b_ptr0]]\n" + "ld1b z14.b, p4/z, [%[b_ptr1]]\n" + "cbz %[loops], 1f\n" + "2:\n" "zip2 z15.b, z12.b, z13.b\n" + "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" "zip1 z13.b, z12.b, z13.b\n" - "cbz %[loops], 2f\n" - "3:\n" - "sdot z16.s, z8.b, z0.b[0]\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" + "sdot z16.s, z8.b, z0.b[0]\n" + "ld1rqb z5.b, p7/z, [a_ptr1]\n" "sdot z20.s, z8.b, z1.b[0]\n" - "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" + "ld1rqb z6.b, p7/z, [a_ptr2]\n" "sdot z24.s, z8.b, z2.b[0]\n" - "ld1rqb z5.b, p7/z, [a_ptr1]\n" + "ld1rqb z7.b, p7/z, [a_ptr3]\n" "sdot z28.s, z8.b, z3.b[0]\n" - "ld1rqb z6.b, p7/z, [a_ptr2]\n" + "subs %[loops], %[loops], #0x1\n" "zip2 z8.b, z14.b, z12.b\n" - "ld1rqb z7.b, p7/z, [a_ptr3]\n" + "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "zip1 z14.b, z14.b, z12.b\n" - "subs %[loops], %[loops], #0x1\n" + "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "sdot z17.s, z9.b, z0.b[0]\n" - "add %[b_ptr0], %[b_ptr0], %[ldb]\n" + "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "zip1 z12.b, z13.b, z14.b\n" - "add %[b_ptr2], %[b_ptr2], %[ldb]\n" + "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "zip2 z13.b, z13.b, z14.b\n" - "add %[b_ptr1], %[b_ptr1], %[ldb]\n" + "add %[a_ptr0], %[a_ptr0], #0x20\n" "zip1 z14.b, z15.b, z8.b\n" - "add %[b_ptr3], %[b_ptr3], %[ldb]\n" + "add a_ptr1, a_ptr1, #0x20\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "sdot z21.s, z9.b, z1.b[0]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "sdot z25.s, z9.b, z2.b[0]\n" - "add %[a_ptr0], %[a_ptr0], #0x20\n" + "add a_ptr2, a_ptr2, #0x20\n" "sdot z29.s, z9.b, z3.b[0]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "sdot z18.s, z10.b, z0.b[0]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "sdot z22.s, z10.b, z1.b[0]\n" - "add a_ptr1, a_ptr1, #0x20\n" + "add a_ptr3, a_ptr3, #0x20\n" "sdot z26.s, z10.b, z2.b[0]\n" - "add a_ptr2, a_ptr2, #0x20\n" "sdot z30.s, z10.b, z3.b[0]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" "sdot z19.s, z11.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "sdot z23.s, z11.b, z1.b[0]\n" - "add a_ptr3, a_ptr3, #0x20\n" "sdot z27.s, z11.b, z2.b[0]\n" "sdot z31.s, z11.b, z3.b[0]\n" "zip2 z11.b, z8.b, z9.b\n" @@ -3649,29 +3465,29 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z5.b[3]\n" "sdot z27.s, z15.b, z6.b[3]\n" "sdot z31.s, z15.b, z7.b[3]\n" + "b.ne 2b\n" + "1:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" - "b.ne 3b\n" - "2:\n" - "cbz %[regs], 4f\n" - "sdot z16.s, z8.b, z0.b[0]\n" "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z20.s, z8.b, z1.b[0]\n" + "cbz %[regs], 3f\n" + "sdot z16.s, z8.b, z0.b[0]\n" "ld1rqb z4.b, p7/z, [%[a_ptr0]]\n" - "sdot z24.s, z8.b, z2.b[0]\n" + "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z5.b, p7/z, [a_ptr1]\n" - "sdot z28.s, z8.b, z3.b[0]\n" + "sdot z24.s, z8.b, z2.b[0]\n" "ld1rqb z6.b, p7/z, [a_ptr2]\n" - "zip2 z8.b, z14.b, z12.b\n" + "sdot z28.s, z8.b, z3.b[0]\n" "ld1rqb z7.b, p7/z, [a_ptr3]\n" - "zip1 z14.b, z14.b, z12.b\n" + "zip2 z8.b, z14.b, z12.b\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "sdot z17.s, z9.b, z0.b[0]\n" + "zip1 z14.b, z14.b, z12.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "zip1 z12.b, z13.b, z14.b\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "zip2 z13.b, z13.b, z14.b\n" + "zip1 z12.b, z13.b, z14.b\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" + "zip2 z13.b, z13.b, z14.b\n" "zip1 z14.b, z15.b, z8.b\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -3797,9 +3613,13 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z16.s, z8.b, z4.b[0]\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" "sdot z20.s, z8.b, z5.b[0]\n" + "addvl %[a_ptr0], %[a_ptr0], #2\n" "sdot z24.s, z8.b, z6.b[0]\n" + "addvl a_ptr1, a_ptr1, #2\n" "sdot z28.s, z8.b, z7.b[0]\n" + "addvl a_ptr2, a_ptr2, #2\n" "zip2 z8.b, z14.b, z12.b\n" + "addvl a_ptr3, a_ptr3, #2\n" "zip1 z14.b, z14.b, z12.b\n" "sdot z17.s, z9.b, z4.b[0]\n" "sdot z21.s, z9.b, z5.b[0]\n" @@ -3893,7 +3713,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z5.b[3]\n" "sdot z27.s, z15.b, z6.b[3]\n" "sdot z31.s, z15.b, z7.b[3]\n" - "cbz %[blocks], 5f\n" + "cbz %[blocks], 4f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -3927,7 +3747,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z1.b[0]\n" "sdot z27.s, z11.b, z2.b[0]\n" "sdot z31.s, z11.b, z3.b[0]\n" - "b.eq 6f\n" + "b.eq 5f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -3961,7 +3781,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z1.b[1]\n" "sdot z27.s, z15.b, z2.b[1]\n" "sdot z31.s, z15.b, z3.b[1]\n" - "b.eq 7f\n" + "b.eq 6f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -3994,31 +3814,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z1.b[2]\n" "sdot z27.s, z11.b, z2.b[2]\n" "sdot z31.s, z11.b, z3.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 9f\n" + "b.eq 8f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 10f\n" + "b.eq 9f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "10:\n" + "b 10f\n" + "9:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 11f\n" - "9:\n" + "b 10f\n" + "8:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "11:\n" + "10:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -4044,33 +3864,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z1.b[3]\n" "sdot z27.s, z15.b, z2.b[3]\n" "sdot z31.s, z15.b, z3.b[3]\n" - "b 8f\n" - "7:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "6:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 12f\n" + "b.eq 11f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 13f\n" + "b.eq 12f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "13:\n" + "b 13f\n" + "12:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 14f\n" - "12:\n" + "b 13f\n" + "11:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "14:\n" + "13:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -4096,33 +3916,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z1.b[2]\n" "sdot z27.s, z11.b, z2.b[2]\n" "sdot z31.s, z11.b, z3.b[2]\n" - "b 8f\n" - "6:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "5:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 15f\n" + "b.eq 14f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 16f\n" + "b.eq 15f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "16:\n" + "b 16f\n" + "15:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 17f\n" - "15:\n" + "b 16f\n" + "14:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "17:\n" + "16:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -4148,33 +3968,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z1.b[1]\n" "sdot z27.s, z15.b, z2.b[1]\n" "sdot z31.s, z15.b, z3.b[1]\n" - "b 8f\n" - "5:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "4:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 18f\n" + "b.eq 17f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 19f\n" + "b.eq 18f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "19:\n" + "b 19f\n" + "18:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 20f\n" - "18:\n" + "b 19f\n" + "17:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "20:\n" + "19:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -4200,37 +4020,40 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z1.b[0]\n" "sdot z27.s, z11.b, z2.b[0]\n" "sdot z31.s, z11.b, z3.b[0]\n" - "b 8f\n" - "4:\n" + "b 7f\n" + "3:\n" "sdot z16.s, z8.b, z0.b[0]\n" - "ld1b z12.b, p4/z, [%[b_ptr3]]\n" - "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z4.b, p6/z, [%[a_ptr0]]\n" - "sdot z24.s, z8.b, z2.b[0]\n" + "sdot z20.s, z8.b, z1.b[0]\n" "ld1rqb z5.b, p6/z, [a_ptr1]\n" - "sdot z28.s, z8.b, z3.b[0]\n" + "sdot z24.s, z8.b, z2.b[0]\n" "ld1rqb z6.b, p6/z, [a_ptr2]\n" - "zip2 z8.b, z14.b, z12.b\n" + "sdot z28.s, z8.b, z3.b[0]\n" "ld1rqb z7.b, p6/z, [a_ptr3]\n" - "zip1 z14.b, z14.b, z12.b\n" + "zip2 z8.b, z14.b, z12.b\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" - "sdot z17.s, z9.b, z0.b[0]\n" + "zip1 z14.b, z14.b, z12.b\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" - "zip1 z12.b, z13.b, z14.b\n" + "sdot z17.s, z9.b, z0.b[0]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" - "zip2 z13.b, z13.b, z14.b\n" + "zip1 z12.b, z13.b, z14.b\n" "add %[b_ptr3], %[b_ptr3], %[ldb]\n" + "zip2 z13.b, z13.b, z14.b\n" + "addvl %[a_ptr0], %[a_ptr0], #1\n" "zip1 z14.b, z15.b, z8.b\n" + "addvl a_ptr1, a_ptr1, #1\n" "zip2 z15.b, z15.b, z8.b\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "sdot z21.s, z9.b, z1.b[0]\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "sdot z25.s, z9.b, z2.b[0]\n" + "addvl a_ptr2, a_ptr2, #1\n" "sdot z29.s, z9.b, z3.b[0]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "sdot z18.s, z10.b, z0.b[0]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "sdot z22.s, z10.b, z1.b[0]\n" + "addvl a_ptr3, a_ptr3, #1\n" "sdot z26.s, z10.b, z2.b[0]\n" "sdot z30.s, z10.b, z3.b[0]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" @@ -4309,7 +4132,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z1.b[3]\n" "sdot z27.s, z15.b, z2.b[3]\n" "sdot z31.s, z15.b, z3.b[3]\n" - "cbz %[blocks], 21f\n" + "cbz %[blocks], 20f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -4343,7 +4166,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z5.b[0]\n" "sdot z27.s, z11.b, z6.b[0]\n" "sdot z31.s, z11.b, z7.b[0]\n" - "b.eq 22f\n" + "b.eq 21f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" @@ -4377,7 +4200,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z5.b[1]\n" "sdot z27.s, z15.b, z6.b[1]\n" "sdot z31.s, z15.b, z7.b[1]\n" - "b.eq 23f\n" + "b.eq 22f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" @@ -4410,31 +4233,31 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z5.b[2]\n" "sdot z27.s, z11.b, z6.b[2]\n" "sdot z31.s, z11.b, z7.b[2]\n" - "cbz %[odds], 8f\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 24f\n" + "b.eq 23f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 25f\n" + "b.eq 24f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "25:\n" + "b 25f\n" + "24:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 26f\n" - "24:\n" + "b 25f\n" + "23:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "26:\n" + "25:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -4460,33 +4283,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z5.b[3]\n" "sdot z27.s, z15.b, z6.b[3]\n" "sdot z31.s, z15.b, z7.b[3]\n" - "b 8f\n" - "23:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "22:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 27f\n" + "b.eq 26f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 28f\n" + "b.eq 27f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "28:\n" + "b 28f\n" + "27:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 29f\n" - "27:\n" + "b 28f\n" + "26:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "29:\n" + "28:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -4512,33 +4335,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z5.b[2]\n" "sdot z27.s, z11.b, z6.b[2]\n" "sdot z31.s, z11.b, z7.b[2]\n" - "b 8f\n" - "22:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "21:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 30f\n" + "b.eq 29f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 31f\n" + "b.eq 30f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z13.b, p4/z, [%[b_ptr2]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "31:\n" + "b 31f\n" + "30:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" "ld1b z14.b, p4/z, [%[b_ptr1]]\n" - "b 32f\n" - "30:\n" + "b 31f\n" + "29:\n" "mov z13.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z14.b, #0\n" "ld1b z12.b, p4/z, [%[b_ptr0]]\n" - "32:\n" + "31:\n" "zip2 z15.b, z12.b, z13.b\n" "zip1 z13.b, z12.b, z13.b\n" "mov z12.b, #0\n" @@ -4564,33 +4387,33 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z15.b, z5.b[1]\n" "sdot z27.s, z15.b, z6.b[1]\n" "sdot z31.s, z15.b, z7.b[1]\n" - "b 8f\n" - "21:\n" - "cbz %[odds], 8f\n" + "b 7f\n" + "20:\n" + "cbz %[odds], 7f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 33f\n" + "b.eq 32f\n" "subs %[odds], %[odds], #0x1\n" - "b.eq 34f\n" + "b.eq 33f\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr2], %[b_ptr2], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z9.b, p4/z, [%[b_ptr2]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "34:\n" + "b 34f\n" + "33:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "add %[b_ptr1], %[b_ptr1], %[ldb]\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" "ld1b z10.b, p4/z, [%[b_ptr1]]\n" - "b 35f\n" - "33:\n" + "b 34f\n" + "32:\n" "mov z9.b, #0\n" "add %[b_ptr0], %[b_ptr0], %[ldb]\n" "mov z10.b, #0\n" "ld1b z8.b, p4/z, [%[b_ptr0]]\n" - "35:\n" + "34:\n" "zip2 z11.b, z8.b, z9.b\n" "zip1 z9.b, z8.b, z9.b\n" "mov z8.b, #0\n" @@ -4616,7 +4439,7 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l "sdot z23.s, z11.b, z5.b[0]\n" "sdot z27.s, z11.b, z6.b[0]\n" "sdot z31.s, z11.b, z7.b[0]\n" - "8:\n" + "7:\n" "st1w z16.s, p0, [%[c_ptr0]]\n" "st1w z17.s, p1, [%[c_ptr0], #1, MUL VL]\n" "st1w z18.s, p2, [%[c_ptr0], #2, MUL VL]\n" @@ -4641,11 +4464,12 @@ void sve_native_s8s32_dot_4VLx4(const int8_t *A, int lda, const int8_t *B, int l ".unreq c_ptr2\n" ".unreq c_ptr3\n" : [a_ptr0] "+r" (a_ptr0), [b_ptr0] "+r" (b_ptr0), [b_ptr1] "+r" (b_ptr1), [b_ptr2] "+r" (b_ptr2), [b_ptr3] "+r" (b_ptr3), [c_ptr0] "+r" (c_ptr0), [loops] "+r" (loops), [regs] "+r" (regs), [temp] "+r" (temp), [blocks] "+r" (blocks), [odds] "+r" (odds) - : [betaptr] "r" (betaptr), [width] "r" (width), [beta0] "r" (beta0), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) + : [width] "r" (width), [append] "r" (static_cast<uint64_t>(append)), [lda] "r" (ldab), [ldc] "r" (ldcb), [leftovers] "r" (leftovers), [ldb] "r" (ldbb) : "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "z21", "z22", "z23", "z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31", "x0", "x1", "x2", "x3", "x4", "x5", "cc", "memory" ); break; } + } } } |