aboutsummaryrefslogtreecommitdiff
path: root/src/core/NEON/kernels/arm_gemm/merges/sve_merge_fp32_3VLx8.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/merges/sve_merge_fp32_3VLx8.hpp')
-rw-r--r--src/core/NEON/kernels/arm_gemm/merges/sve_merge_fp32_3VLx8.hpp2597
1 files changed, 1456 insertions, 1141 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/merges/sve_merge_fp32_3VLx8.hpp b/src/core/NEON/kernels/arm_gemm/merges/sve_merge_fp32_3VLx8.hpp
index 4b51066ebe..f7feec4d83 100644
--- a/src/core/NEON/kernels/arm_gemm/merges/sve_merge_fp32_3VLx8.hpp
+++ b/src/core/NEON/kernels/arm_gemm/merges/sve_merge_fp32_3VLx8.hpp
@@ -26,11 +26,33 @@
#ifdef __ARM_FEATURE_SVE
template<>
-inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldout, const int y0, const int ymax, const int x0, const int xmax, const float alpha, const float beta)
+void MergeResults<3, 8, true>(float *out, const float *in, const int ldout, const int y0, const int ymax, const int x0, const int xmax, const float *bias, Activation act, bool append)
{
const float *inptr = in;
+ float nullbias[192] = { 0 };
+ float minval = - std::numeric_limits<float>::infinity();
+ float maxval = std::numeric_limits<float>::infinity();
- for (int y=y0; y<ymax; y+=8) {
+ switch(act.type)
+ {
+ default:
+ case Activation::Type::None:
+ break;
+ case Activation::Type::BoundedReLU:
+ maxval = static_cast<float>(act.param1);
+ /* fall through */
+ case Activation::Type::ReLU:
+ minval = 0.0f;
+ break;
+ }
+
+ if (!append && !bias)
+ {
+ memset(nullbias, 0, (3 * get_vector_length<float>() * sizeof(float)));
+ }
+
+ for (int y=y0; y<ymax; y+=8)
+ {
float *outptr0 = out + (y * ldout) + x0;
float *outptr1 = outptr0 + ldout;
float *outptr2 = outptr1 + ldout;
@@ -42,45 +64,52 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
const int height = ymax - y;
- for (int i=x0; i<xmax; i+=(3 * get_vector_length<float>())) {
- if (beta==0.0f)
+ for (int i=x0; i<xmax; i+=(3 * get_vector_length<float>()))
+ {
+ if (append)
{
- switch(height) {
+ switch(height)
+ {
case 1:
{
long w = xmax - i;
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z5.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr0], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "addvl %[outptr0], %[outptr0], #3\n"
- "1:\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
"addvl %[inptr], %[inptr], #24\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "addvl %[outptr0], %[outptr0], #3\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -91,47 +120,61 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z6.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z5.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z13.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z6.s, p1/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "ld1w z7.s, p2/z, [%[outptr1], #2, MUL VL]\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "st1w z13.s, p0, [%[outptr1]]\n"
+ "st1w z14.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "st1w z15.s, p2, [%[outptr1], #2, MUL VL]\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -142,59 +185,82 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z7.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z4.s, p0/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z5.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr2], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z13.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z6.s, p1/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "ld1w z7.s, p2/z, [%[outptr1], #2, MUL VL]\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z8.s, p0/z, [%[outptr2]]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "ld1w z16.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "ld1w z9.s, p1/z, [%[outptr2], #1, MUL VL]\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "ld1w z2.s, p2/z, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z10.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z16.s, z16.s, z8.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #-8, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "st1w z13.s, p0, [%[outptr1]]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z14.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "st1w z15.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fmin z10.s, p2/m, z10.s, z0.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z16.s, p0, [%[outptr2]]\n"
+ "fmax z10.s, p2/m, z10.s, z1.s\n"
+ "st1w z17.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "st1w z10.s, p2, [%[outptr2], #2, MUL VL]\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -205,70 +271,102 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z5.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z6.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z13.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z6.s, p1/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "ld1w z7.s, p2/z, [%[outptr1], #2, MUL VL]\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z8.s, p0/z, [%[outptr2]]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "ld1w z16.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "ld1w z9.s, p1/z, [%[outptr2], #1, MUL VL]\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "ld1w z2.s, p2/z, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z10.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "ld1w z3.s, p0/z, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "ld1w z4.s, p1/z, [%[outptr3], #1, MUL VL]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "ld1w z12.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z5.s, p2/z, [%[outptr3], #2, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z13.s, p0, [%[outptr1]]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "ld1w z13.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #-8, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #-5, MUL VL]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmin z10.s, p2/m, z10.s, z0.s\n"
+ "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z15.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z10.s, p2/m, z10.s, z1.s\n"
+ "st1w z16.s, p0, [%[outptr2]]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z11.s, p0/m, z11.s, z0.s\n"
+ "st1w z17.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fmin z12.s, p1/m, z12.s, z0.s\n"
+ "st1w z10.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fmax z11.s, p0/m, z11.s, z1.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "fmax z12.s, p1/m, z12.s, z1.s\n"
+ "st1w z11.s, p0, [%[outptr3]]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z12.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "st1w z13.s, p2, [%[outptr3], #2, MUL VL]\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -279,82 +377,123 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z5.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z6.s, p0/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z5.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr4], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z13.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z6.s, p1/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "ld1w z7.s, p2/z, [%[outptr1], #2, MUL VL]\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z8.s, p0/z, [%[outptr2]]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "ld1w z16.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "ld1w z9.s, p1/z, [%[outptr2], #1, MUL VL]\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "ld1w z2.s, p2/z, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z10.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "ld1w z3.s, p0/z, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "ld1w z4.s, p1/z, [%[outptr3], #1, MUL VL]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "ld1w z12.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z5.s, p2/z, [%[outptr3], #2, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z13.s, p0, [%[outptr1]]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "ld1w z13.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "ld1w z6.s, p0/z, [%[outptr4]]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmin z10.s, p2/m, z10.s, z0.s\n"
+ "ld1w z14.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z7.s, p1/z, [%[outptr4], #1, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z15.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z15.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fmax z10.s, p2/m, z10.s, z1.s\n"
+ "ld1w z8.s, p2/z, [%[outptr4], #2, MUL VL]\n"
+ "fmin z11.s, p0/m, z11.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr2]]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z16.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fmin z12.s, p1/m, z12.s, z0.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #-8, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8, #-5, MUL VL]\n"
+ "fmax z11.s, p0/m, z11.s, z1.s\n"
+ "st1w z17.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #-2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr4], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z12.s, p1/m, z12.s, z1.s\n"
+ "st1w z10.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z11.s, p0, [%[outptr3]]\n"
+ "fadd z15.s, z15.s, z7.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "st1w z12.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "st1w z13.s, p2, [%[outptr3], #2, MUL VL]\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
- "addvl %[outptr4], %[outptr4], #3\n"
- "1:\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
+ "st1w z14.s, p0, [%[outptr4]]\n"
+ "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
"addvl %[inptr], %[inptr], #24\n"
+ "st1w z15.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "st1w z16.s, p2, [%[outptr4], #2, MUL VL]\n"
+ "addvl %[outptr4], %[outptr4], #3\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -365,94 +504,144 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "ld1w z5.s, p0/z, [x8, #-1, MUL VL]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z6.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z5.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z4.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x280]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr4], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr5], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z13.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z6.s, p1/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "ld1w z7.s, p2/z, [%[outptr1], #2, MUL VL]\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z8.s, p0/z, [%[outptr2]]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "ld1w z16.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "ld1w z9.s, p1/z, [%[outptr2], #1, MUL VL]\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "ld1w z2.s, p2/z, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z10.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "ld1w z3.s, p0/z, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "ld1w z4.s, p1/z, [%[outptr3], #1, MUL VL]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "ld1w z12.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z5.s, p2/z, [%[outptr3], #2, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z13.s, p0, [%[outptr1]]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "ld1w z13.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "ld1w z6.s, p0/z, [%[outptr4]]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmin z10.s, p2/m, z10.s, z0.s\n"
+ "ld1w z14.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z7.s, p1/z, [%[outptr4], #1, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z15.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z15.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fmax z10.s, p2/m, z10.s, z1.s\n"
+ "ld1w z8.s, p2/z, [%[outptr4], #2, MUL VL]\n"
+ "fmin z11.s, p0/m, z11.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr2]]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z16.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fmin z12.s, p1/m, z12.s, z0.s\n"
+ "ld1w z9.s, p0/z, [%[outptr5]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "st1w z17.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmax z11.s, p0/m, z11.s, z1.s\n"
+ "ld1w z17.s, p0/z, [x8, #-1, MUL VL]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "ld1w z2.s, p1/z, [%[outptr5], #1, MUL VL]\n"
+ "fmax z12.s, p1/m, z12.s, z1.s\n"
+ "st1w z10.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "ld1w z10.s, p1/z, [x8]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z3.s, p2/z, [%[outptr5], #2, MUL VL]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z11.s, p0, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p2/z, [x8, #1, MUL VL]\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #-8, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #-5, MUL VL]\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "st1w z12.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #-2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8, #1, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "st1w z13.s, p2, [%[outptr3], #2, MUL VL]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "st1w z14.s, p0, [%[outptr4]]\n"
+ "fmin z17.s, p0/m, z17.s, z0.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z15.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z17.s, p0/m, z17.s, z1.s\n"
+ "st1w z16.s, p2, [%[outptr4], #2, MUL VL]\n"
+ "fmin z10.s, p1/m, z10.s, z0.s\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
+ "fmin z11.s, p2/m, z11.s, z0.s\n"
+ "st1w z17.s, p0, [%[outptr5]]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmax z10.s, p1/m, z10.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z11.s, p2/m, z11.s, z1.s\n"
"addvl %[outptr4], %[outptr4], #3\n"
- "prfm PSTL1KEEP, [%[outptr5], #0x60]\n"
- "addvl %[outptr5], %[outptr5], #3\n"
- "1:\n"
+ "st1w z10.s, p1, [%[outptr5], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x280]\n"
+ "prfm PLDL1KEEP, [%[outptr5], #0x60]\n"
"addvl %[inptr], %[inptr], #24\n"
+ "st1w z11.s, p2, [%[outptr5], #2, MUL VL]\n"
+ "addvl %[outptr5], %[outptr5], #3\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -463,106 +652,165 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "ld1w z5.s, p0/z, [x8, #-1, MUL VL]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5]]\n"
- "ld1w z6.s, p0/z, [x8, #2, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z7.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z4.s, p0/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z5.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x280]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr4], #1, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr5], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8, #3, MUL VL]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr6], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z13.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z6.s, p1/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "ld1w z7.s, p2/z, [%[outptr1], #2, MUL VL]\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z8.s, p0/z, [%[outptr2]]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "ld1w z16.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "ld1w z9.s, p1/z, [%[outptr2], #1, MUL VL]\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "ld1w z2.s, p2/z, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z10.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "ld1w z3.s, p0/z, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "ld1w z4.s, p1/z, [%[outptr3], #1, MUL VL]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "ld1w z12.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z5.s, p2/z, [%[outptr3], #2, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z13.s, p0, [%[outptr1]]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "ld1w z13.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "ld1w z6.s, p0/z, [%[outptr4]]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmin z10.s, p2/m, z10.s, z0.s\n"
+ "ld1w z14.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z7.s, p1/z, [%[outptr4], #1, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z15.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z15.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fmax z10.s, p2/m, z10.s, z1.s\n"
+ "ld1w z8.s, p2/z, [%[outptr4], #2, MUL VL]\n"
+ "fmin z11.s, p0/m, z11.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr2]]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z16.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fmin z12.s, p1/m, z12.s, z0.s\n"
+ "ld1w z9.s, p0/z, [%[outptr5]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "st1w z17.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmax z11.s, p0/m, z11.s, z1.s\n"
+ "ld1w z17.s, p0/z, [x8, #-1, MUL VL]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "ld1w z2.s, p1/z, [%[outptr5], #1, MUL VL]\n"
+ "fmax z12.s, p1/m, z12.s, z1.s\n"
+ "st1w z10.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "ld1w z10.s, p1/z, [x8]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z3.s, p2/z, [%[outptr5], #2, MUL VL]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z11.s, p0, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p2/z, [x8, #1, MUL VL]\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
+ "ld1w z4.s, p0/z, [%[outptr6]]\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "st1w z12.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "ld1w z12.s, p0/z, [x8, #2, MUL VL]\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "ld1w z5.s, p1/z, [%[outptr6], #1, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z13.s, p2, [%[outptr3], #2, MUL VL]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
+ "ld1w z13.s, p1/z, [x8, #3, MUL VL]\n"
+ "fmin z17.s, p0/m, z17.s, z0.s\n"
+ "ld1w z6.s, p2/z, [%[outptr6], #2, MUL VL]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "st1w z14.s, p0, [%[outptr4]]\n"
+ "fmin z10.s, p1/m, z10.s, z0.s\n"
+ "ld1w z14.s, p2/z, [x8, #4, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #-8, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8, #-5, MUL VL]\n"
+ "fmax z17.s, p0/m, z17.s, z1.s\n"
+ "st1w z15.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "fmax z10.s, p1/m, z10.s, z1.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #-2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr4], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #1, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr5], #2, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #4, MUL VL]\n"
+ "fmin z11.s, p2/m, z11.s, z0.s\n"
+ "st1w z16.s, p2, [%[outptr4], #2, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "st1w z17.s, p0, [%[outptr5]]\n"
+ "fmax z11.s, p2/m, z11.s, z1.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr6], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z12.s, p0/m, z12.s, z0.s\n"
+ "st1w z10.s, p1, [%[outptr5], #1, MUL VL]\n"
+ "fmin z13.s, p1/m, z13.s, z0.s\n"
+ "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "st1w z11.s, p2, [%[outptr5], #2, MUL VL]\n"
+ "fmax z12.s, p0/m, z12.s, z1.s\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z13.s, p1/m, z13.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmin z14.s, p2/m, z14.s, z0.s\n"
+ "st1w z12.s, p0, [%[outptr6]]\n"
+ "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
"addvl %[outptr4], %[outptr4], #3\n"
- "prfm PSTL1KEEP, [%[outptr5], #0x60]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x280]\n"
+ "fmax z14.s, p2/m, z14.s, z1.s\n"
+ "st1w z13.s, p1, [%[outptr6], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr5], #0x60]\n"
"addvl %[outptr5], %[outptr5], #3\n"
"prfm PLDL1KEEP, [%[inptr], #0x2c0]\n"
- "prfm PSTL1KEEP, [%[outptr6], #0x60]\n"
+ "st1w z14.s, p2, [%[outptr6], #2, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr6], #0x60]\n"
"addvl %[outptr6], %[outptr6], #3\n"
- "1:\n"
"addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -574,117 +822,185 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "ld1w z5.s, p0/z, [x8, #-1, MUL VL]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5]]\n"
- "ld1w z6.s, p0/z, [x8, #2, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6]]\n"
- "ld1w z7.s, p0/z, [x8, #5, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr7]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[outptr0]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z10.s, p0/z, [%[inptr]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z5.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z6.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x280]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4], #1, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5], #1, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #3, MUL VL]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6], #1, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #6, MUL VL]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr7], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "ld1w z3.s, p1/z, [%[outptr0], #1, MUL VL]\n"
+ "ld1w z11.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z13.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmin z10.s, p0/m, z10.s, z0.s\n"
+ "ld1w z4.s, p2/z, [%[outptr0], #2, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z12.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z6.s, p1/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fmax z10.s, p0/m, z10.s, z1.s\n"
+ "ld1w z7.s, p2/z, [%[outptr1], #2, MUL VL]\n"
+ "fmin z11.s, p1/m, z11.s, z0.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z8.s, p0/z, [%[outptr2]]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "st1w z10.s, p0, [%[outptr0]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "ld1w z16.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fmax z11.s, p1/m, z11.s, z1.s\n"
+ "ld1w z9.s, p1/z, [%[outptr2], #1, MUL VL]\n"
+ "fmin z12.s, p2/m, z12.s, z0.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "ld1w z2.s, p2/z, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "st1w z11.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z10.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fmax z12.s, p2/m, z12.s, z1.s\n"
+ "ld1w z3.s, p0/z, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "ld1w z4.s, p1/z, [%[outptr3], #1, MUL VL]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z12.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "ld1w z12.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z5.s, p2/z, [%[outptr3], #2, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z13.s, p0, [%[outptr1]]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "ld1w z13.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "ld1w z6.s, p0/z, [%[outptr4]]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmin z10.s, p2/m, z10.s, z0.s\n"
+ "ld1w z14.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z7.s, p1/z, [%[outptr4], #1, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z15.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z15.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fmax z10.s, p2/m, z10.s, z1.s\n"
+ "ld1w z8.s, p2/z, [%[outptr4], #2, MUL VL]\n"
+ "fmin z11.s, p0/m, z11.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr2]]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z16.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fmin z12.s, p1/m, z12.s, z0.s\n"
+ "ld1w z9.s, p0/z, [%[outptr5]]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "st1w z17.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmax z11.s, p0/m, z11.s, z1.s\n"
+ "ld1w z17.s, p0/z, [x8, #-1, MUL VL]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "ld1w z2.s, p1/z, [%[outptr5], #1, MUL VL]\n"
+ "fmax z12.s, p1/m, z12.s, z1.s\n"
+ "st1w z10.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "ld1w z10.s, p1/z, [x8]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "ld1w z3.s, p2/z, [%[outptr5], #2, MUL VL]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z11.s, p0, [%[outptr3]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
+ "ld1w z11.s, p2/z, [x8, #1, MUL VL]\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
+ "ld1w z4.s, p0/z, [%[outptr6]]\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "st1w z12.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "ld1w z12.s, p0/z, [x8, #2, MUL VL]\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "ld1w z5.s, p1/z, [%[outptr6], #1, MUL VL]\n"
+ "fadd z10.s, z10.s, z2.s\n"
+ "st1w z13.s, p2, [%[outptr3], #2, MUL VL]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
+ "ld1w z13.s, p1/z, [x8, #3, MUL VL]\n"
+ "fmin z17.s, p0/m, z17.s, z0.s\n"
+ "ld1w z6.s, p2/z, [%[outptr6], #2, MUL VL]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "st1w z14.s, p0, [%[outptr4]]\n"
+ "fmin z10.s, p1/m, z10.s, z0.s\n"
+ "ld1w z14.s, p2/z, [x8, #4, MUL VL]\n"
+ "fadd z11.s, z11.s, z3.s\n"
+ "ld1w z7.s, p0/z, [%[outptr7]]\n"
+ "fmax z17.s, p0/m, z17.s, z1.s\n"
+ "st1w z15.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "fadd z12.s, z12.s, z4.s\n"
+ "ld1w z15.s, p0/z, [x8, #5, MUL VL]\n"
+ "fmax z10.s, p1/m, z10.s, z1.s\n"
+ "ld1w z8.s, p1/z, [%[outptr7], #1, MUL VL]\n"
+ "fmin z11.s, p2/m, z11.s, z0.s\n"
+ "st1w z16.s, p2, [%[outptr4], #2, MUL VL]\n"
+ "fadd z13.s, z13.s, z5.s\n"
+ "ld1w z16.s, p1/z, [x8, #6, MUL VL]\n"
+ "fmin z12.s, p0/m, z12.s, z0.s\n"
+ "ld1w z9.s, p2/z, [%[outptr7], #2, MUL VL]\n"
+ "fadd z14.s, z14.s, z6.s\n"
+ "st1w z17.s, p0, [%[outptr5]]\n"
+ "fmax z11.s, p2/m, z11.s, z1.s\n"
+ "ld1w z17.s, p2/z, [x8, #7, MUL VL]\n"
+ "fmin z13.s, p1/m, z13.s, z0.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #-8, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #-5, MUL VL]\n"
+ "fmax z12.s, p0/m, z12.s, z1.s\n"
+ "st1w z10.s, p1, [%[outptr5], #1, MUL VL]\n"
+ "fmin z14.s, p2/m, z14.s, z0.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z4.s, p0/z, [x8, #-2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z8.s, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4], #2, MUL VL]\n"
- "ld1w z5.s, p0/z, [x8, #1, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
- "fmul z9.s, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5], #2, MUL VL]\n"
- "ld1w z6.s, p0/z, [x8, #4, MUL VL]\n"
+ "fmax z13.s, p1/m, z13.s, z1.s\n"
+ "st1w z11.s, p2, [%[outptr5], #2, MUL VL]\n"
+ "fadd z15.s, z15.s, z7.s\n"
+ "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z14.s, p2/m, z14.s, z1.s\n"
+ "st1w z12.s, p0, [%[outptr6]]\n"
+ "fadd z16.s, z16.s, z8.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "fmul z10.s, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6], #2, MUL VL]\n"
- "ld1w z7.s, p0/z, [x8, #7, MUL VL]\n"
- "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
- "fmul z11.s, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr7], #2, MUL VL]\n"
+ "fmin z15.s, p0/m, z15.s, z0.s\n"
+ "st1w z13.s, p1, [%[outptr6], #1, MUL VL]\n"
+ "fadd z17.s, z17.s, z9.s\n"
+ "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z16.s, p1/m, z16.s, z0.s\n"
+ "st1w z14.s, p2, [%[outptr6], #2, MUL VL]\n"
+ "fmax z15.s, p0/m, z15.s, z1.s\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
+ "fmin z17.s, p2/m, z17.s, z0.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmax z16.s, p1/m, z16.s, z1.s\n"
+ "st1w z15.s, p0, [%[outptr7]]\n"
+ "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z17.s, p2/m, z17.s, z1.s\n"
"addvl %[outptr4], %[outptr4], #3\n"
- "prfm PSTL1KEEP, [%[outptr5], #0x60]\n"
+ "st1w z16.s, p1, [%[outptr7], #1, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x280]\n"
+ "prfm PLDL1KEEP, [%[outptr5], #0x60]\n"
"addvl %[outptr5], %[outptr5], #3\n"
+ "st1w z17.s, p2, [%[outptr7], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x2c0]\n"
- "prfm PSTL1KEEP, [%[outptr6], #0x60]\n"
+ "prfm PLDL1KEEP, [%[outptr6], #0x60]\n"
"addvl %[outptr6], %[outptr6], #3\n"
- "prfm PSTL1KEEP, [%[outptr7], #0x60]\n"
+ "prfm PLDL1KEEP, [%[outptr7], #0x60]\n"
"addvl %[outptr7], %[outptr7], #3\n"
- "1:\n"
"addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -694,48 +1010,54 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
}
else
{
- switch(height) {
+ const float *biasptr = nullbias;
+ if (bias)
+ {
+ biasptr = bias + i;
+ }
+
+ switch(height)
+ {
case 1:
{
long w = xmax - i;
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z9.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr0], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z10.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "addvl %[outptr0], %[outptr0], #3\n"
- "1:\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
"addvl %[inptr], %[inptr], #24\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "addvl %[outptr0], %[outptr0], #3\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -746,59 +1068,58 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z9.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z10.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #4, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr1], #2, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "ld1w z16.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "ld1w z18.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "fadd z16.s, z16.s, z2.s\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fadd z17.s, z17.s, z3.s\n"
+ "fadd z18.s, z18.s, z4.s\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "fmin z18.s, p2/m, z18.s, z0.s\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z16.s, p0, [%[outptr1]]\n"
+ "fmax z18.s, p2/m, z18.s, z1.s\n"
+ "st1w z17.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "st1w z18.s, p2, [%[outptr1], #2, MUL VL]\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -809,77 +1130,76 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z9.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z10.s, p0/z, [%[outptr2]]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z11.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #4, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr2], #1, MUL VL]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr2], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z10.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr1], #2, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "ld1w z16.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "ld1w z18.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "ld1w z19.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fadd z16.s, z16.s, z2.s\n"
+ "ld1w z20.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z13.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fadd z17.s, z17.s, z3.s\n"
+ "fadd z18.s, z18.s, z4.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "fmin z18.s, p2/m, z18.s, z0.s\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z19.s, z19.s, z2.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #5, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr2], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-8, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #2, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z16.s, p0, [%[outptr1]]\n"
+ "fmax z18.s, p2/m, z18.s, z1.s\n"
+ "fmin z19.s, p0/m, z19.s, z0.s\n"
+ "fadd z20.s, z20.s, z3.s\n"
+ "st1w z17.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fadd z13.s, z13.s, z4.s\n"
+ "fmax z19.s, p0/m, z19.s, z1.s\n"
+ "st1w z18.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fmin z20.s, p1/m, z20.s, z0.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "st1w z19.s, p0, [%[outptr2]]\n"
+ "fmax z20.s, p1/m, z20.s, z1.s\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z20.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "st1w z13.s, p2, [%[outptr2], #2, MUL VL]\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -890,94 +1210,93 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z9.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z10.s, p0/z, [%[outptr2]]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z11.s, p0/z, [%[outptr3]]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #4, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr2], #1, MUL VL]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr3], #1, MUL VL]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr1], #2, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "ld1w z16.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "ld1w z18.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "ld1w z19.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fadd z16.s, z16.s, z2.s\n"
+ "ld1w z20.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z13.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fadd z17.s, z17.s, z3.s\n"
+ "fadd z18.s, z18.s, z4.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "ld1w z14.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "fmin z18.s, p2/m, z18.s, z0.s\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z19.s, z19.s, z2.s\n"
+ "ld1w z15.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fadd z20.s, z20.s, z3.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr2], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #-8, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr3], #2, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
+ "st1w z16.s, p0, [%[outptr1]]\n"
+ "fmax z18.s, p2/m, z18.s, z1.s\n"
+ "ld1w z16.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fmin z19.s, p0/m, z19.s, z0.s\n"
+ "fmin z20.s, p1/m, z20.s, z0.s\n"
+ "st1w z17.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fadd z13.s, z13.s, z4.s\n"
+ "fadd z14.s, z14.s, z2.s\n"
+ "fmax z19.s, p0/m, z19.s, z1.s\n"
+ "st1w z18.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fmax z20.s, p1/m, z20.s, z1.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-5, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "st1w z19.s, p0, [%[outptr2]]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "fadd z15.s, z15.s, z3.s\n"
+ "fadd z16.s, z16.s, z4.s\n"
+ "st1w z20.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "st1w z13.s, p2, [%[outptr2], #2, MUL VL]\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
+ "st1w z14.s, p0, [%[outptr3]]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "st1w z15.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "st1w z16.s, p2, [%[outptr3], #2, MUL VL]\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -988,112 +1307,111 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z9.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z10.s, p0/z, [%[outptr2]]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z11.s, p0/z, [%[outptr3]]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z8.s, p0/z, [%[outptr4]]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z9.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #4, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr2], #1, MUL VL]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr3], #1, MUL VL]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr4], #1, MUL VL]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr4], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z10.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr1], #2, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "ld1w z16.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "ld1w z18.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "ld1w z19.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fadd z16.s, z16.s, z2.s\n"
+ "ld1w z20.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z13.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fadd z17.s, z17.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "ld1w z14.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fadd z18.s, z18.s, z4.s\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z19.s, z19.s, z2.s\n"
+ "ld1w z15.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #5, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr2], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-8, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr3], #2, MUL VL]\n"
+ "fmin z18.s, p2/m, z18.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr1]]\n"
+ "fmin z19.s, p0/m, z19.s, z0.s\n"
+ "ld1w z16.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fadd z20.s, z20.s, z3.s\n"
+ "fadd z13.s, z13.s, z4.s\n"
+ "st1w z17.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmax z18.s, p2/m, z18.s, z1.s\n"
+ "ld1w z17.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fmax z19.s, p0/m, z19.s, z1.s\n"
+ "fmin z20.s, p1/m, z20.s, z0.s\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "st1w z18.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z14.s, z14.s, z2.s\n"
+ "ld1w z18.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fadd z15.s, z15.s, z3.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #-5, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr4], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #-2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr4], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z20.s, p1/m, z20.s, z1.s\n"
+ "st1w z19.s, p0, [%[outptr2]]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "ld1w z19.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "st1w z20.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fadd z16.s, z16.s, z4.s\n"
+ "fadd z17.s, z17.s, z2.s\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
+ "st1w z13.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "st1w z14.s, p0, [%[outptr3]]\n"
+ "fmin z17.s, p0/m, z17.s, z0.s\n"
+ "fadd z18.s, z18.s, z3.s\n"
+ "fadd z19.s, z19.s, z4.s\n"
+ "st1w z15.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "fmax z17.s, p0/m, z17.s, z1.s\n"
+ "fmin z18.s, p1/m, z18.s, z0.s\n"
+ "fmin z19.s, p2/m, z19.s, z0.s\n"
+ "st1w z16.s, p2, [%[outptr3], #2, MUL VL]\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z18.s, p1/m, z18.s, z1.s\n"
+ "st1w z17.s, p0, [%[outptr4]]\n"
+ "fmax z19.s, p2/m, z19.s, z1.s\n"
+ "st1w z18.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "st1w z19.s, p2, [%[outptr4], #2, MUL VL]\n"
"addvl %[outptr4], %[outptr4], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -1104,130 +1422,129 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z9.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z10.s, p0/z, [%[outptr2]]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z11.s, p0/z, [%[outptr3]]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z8.s, p0/z, [%[outptr4]]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "ld1w z9.s, p0/z, [%[outptr5]]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #-1, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z10.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #4, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr2], #1, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "ld1w z16.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "ld1w z18.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "ld1w z19.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fadd z16.s, z16.s, z2.s\n"
+ "ld1w z20.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z13.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fadd z17.s, z17.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "ld1w z14.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
"prfm PLDL1KEEP, [%[inptr], #0x280]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr3], #1, MUL VL]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr4], #1, MUL VL]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr4], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr5], #1, MUL VL]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr5], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr1], #2, MUL VL]\n"
+ "fadd z18.s, z18.s, z4.s\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z19.s, z19.s, z2.s\n"
+ "ld1w z15.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr2], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #-8, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr3], #2, MUL VL]\n"
+ "fmin z18.s, p2/m, z18.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr1]]\n"
+ "fmin z19.s, p0/m, z19.s, z0.s\n"
+ "ld1w z16.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fadd z20.s, z20.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr5], #0x60]\n"
+ "fmax z18.s, p2/m, z18.s, z1.s\n"
+ "st1w z17.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmax z19.s, p0/m, z19.s, z1.s\n"
+ "ld1w z17.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fmin z20.s, p1/m, z20.s, z0.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fadd z13.s, z13.s, z4.s\n"
+ "st1w z18.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z14.s, z14.s, z2.s\n"
+ "ld1w z18.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fmax z20.s, p1/m, z20.s, z1.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-5, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr4], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-2, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr5], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #1, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5], #2, MUL VL]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "st1w z19.s, p0, [%[outptr2]]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "ld1w z19.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fadd z15.s, z15.s, z3.s\n"
+ "fadd z16.s, z16.s, z4.s\n"
+ "st1w z20.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "ld1w z20.s, p0/z, [x8, #-1, MUL VL]\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "st1w z13.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z2.s\n"
+ "ld1w z13.s, p1/z, [x8]\n"
+ "fadd z18.s, z18.s, z3.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
+ "st1w z14.s, p0, [%[outptr3]]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "ld1w z14.s, p2/z, [x8, #1, MUL VL]\n"
+ "fmin z17.s, p0/m, z17.s, z0.s\n"
+ "fmin z18.s, p1/m, z18.s, z0.s\n"
+ "st1w z15.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fadd z19.s, z19.s, z4.s\n"
+ "fadd z20.s, z20.s, z2.s\n"
+ "fmax z17.s, p0/m, z17.s, z1.s\n"
+ "st1w z16.s, p2, [%[outptr3], #2, MUL VL]\n"
+ "fmax z18.s, p1/m, z18.s, z1.s\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
+ "fmin z19.s, p2/m, z19.s, z0.s\n"
+ "st1w z17.s, p0, [%[outptr4]]\n"
+ "fmin z20.s, p0/m, z20.s, z0.s\n"
+ "fadd z13.s, z13.s, z3.s\n"
+ "fadd z14.s, z14.s, z4.s\n"
+ "st1w z18.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "fmax z19.s, p2/m, z19.s, z1.s\n"
+ "fmax z20.s, p0/m, z20.s, z1.s\n"
+ "fmin z13.s, p1/m, z13.s, z0.s\n"
+ "fmin z14.s, p2/m, z14.s, z0.s\n"
+ "st1w z19.s, p2, [%[outptr4], #2, MUL VL]\n"
"addvl %[outptr4], %[outptr4], #3\n"
- "prfm PLDL1KEEP, [%[outptr5], #0x60]\n"
+ "fmax z13.s, p1/m, z13.s, z1.s\n"
+ "st1w z20.s, p0, [%[outptr5]]\n"
+ "fmax z14.s, p2/m, z14.s, z1.s\n"
+ "st1w z13.s, p1, [%[outptr5], #1, MUL VL]\n"
+ "st1w z14.s, p2, [%[outptr5], #2, MUL VL]\n"
"addvl %[outptr5], %[outptr5], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -1238,148 +1555,147 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z9.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z10.s, p0/z, [%[outptr2]]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z11.s, p0/z, [%[outptr3]]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z8.s, p0/z, [%[outptr4]]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "ld1w z9.s, p0/z, [%[outptr5]]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #-1, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5]]\n"
- "ld1w z10.s, p0/z, [%[outptr6]]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z11.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #4, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr2], #1, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "ld1w z16.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "ld1w z18.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "ld1w z19.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fadd z16.s, z16.s, z2.s\n"
+ "ld1w z20.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z13.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fadd z17.s, z17.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "ld1w z14.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
"prfm PLDL1KEEP, [%[inptr], #0x280]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr3], #1, MUL VL]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr4], #1, MUL VL]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr4], #1, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr5], #1, MUL VL]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr5], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr6], #1, MUL VL]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr6], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z10.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr1], #2, MUL VL]\n"
+ "fadd z18.s, z18.s, z4.s\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z19.s, z19.s, z2.s\n"
+ "ld1w z15.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [%[inptr], #5, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr2], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-8, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr3], #2, MUL VL]\n"
+ "fmin z18.s, p2/m, z18.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr1]]\n"
+ "fmin z19.s, p0/m, z19.s, z0.s\n"
+ "ld1w z16.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fadd z20.s, z20.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr5], #0x60]\n"
+ "fmax z18.s, p2/m, z18.s, z1.s\n"
+ "st1w z17.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmax z19.s, p0/m, z19.s, z1.s\n"
+ "ld1w z17.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fmin z20.s, p1/m, z20.s, z0.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x2c0]\n"
+ "fadd z13.s, z13.s, z4.s\n"
+ "st1w z18.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z14.s, z14.s, z2.s\n"
+ "ld1w z18.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fmax z20.s, p1/m, z20.s, z1.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #-5, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr4], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #-2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr4], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr5], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #1, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr5], #2, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr6], #2, MUL VL]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "st1w z19.s, p0, [%[outptr2]]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "ld1w z19.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fadd z15.s, z15.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr6], #0x60]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z20.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
+ "ld1w z20.s, p0/z, [x8, #-1, MUL VL]\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fadd z16.s, z16.s, z4.s\n"
+ "st1w z13.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z2.s\n"
+ "ld1w z13.s, p1/z, [x8]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #4, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr6], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "st1w z14.s, p0, [%[outptr3]]\n"
+ "fmin z17.s, p0/m, z17.s, z0.s\n"
+ "ld1w z14.s, p2/z, [x8, #1, MUL VL]\n"
+ "fadd z18.s, z18.s, z3.s\n"
+ "fadd z19.s, z19.s, z4.s\n"
+ "st1w z15.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "ld1w z15.s, p0/z, [x8, #2, MUL VL]\n"
+ "fmax z17.s, p0/m, z17.s, z1.s\n"
+ "fmin z18.s, p1/m, z18.s, z0.s\n"
+ "fmin z19.s, p2/m, z19.s, z0.s\n"
+ "st1w z16.s, p2, [%[outptr3], #2, MUL VL]\n"
+ "fadd z20.s, z20.s, z2.s\n"
+ "ld1w z16.s, p1/z, [x8, #3, MUL VL]\n"
+ "fadd z13.s, z13.s, z3.s\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z18.s, p1/m, z18.s, z1.s\n"
+ "st1w z17.s, p0, [%[outptr4]]\n"
+ "fmax z19.s, p2/m, z19.s, z1.s\n"
+ "ld1w z17.s, p2/z, [x8, #4, MUL VL]\n"
+ "fmin z20.s, p0/m, z20.s, z0.s\n"
+ "fmin z13.s, p1/m, z13.s, z0.s\n"
+ "st1w z18.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "fadd z14.s, z14.s, z4.s\n"
+ "fadd z15.s, z15.s, z2.s\n"
+ "fmax z20.s, p0/m, z20.s, z1.s\n"
+ "st1w z19.s, p2, [%[outptr4], #2, MUL VL]\n"
+ "fmax z13.s, p1/m, z13.s, z1.s\n"
"addvl %[outptr4], %[outptr4], #3\n"
- "prfm PLDL1KEEP, [%[outptr5], #0x60]\n"
+ "fmin z14.s, p2/m, z14.s, z0.s\n"
+ "st1w z20.s, p0, [%[outptr5]]\n"
+ "fmin z15.s, p0/m, z15.s, z0.s\n"
+ "fadd z16.s, z16.s, z3.s\n"
+ "fadd z17.s, z17.s, z4.s\n"
+ "st1w z13.s, p1, [%[outptr5], #1, MUL VL]\n"
+ "fmax z14.s, p2/m, z14.s, z1.s\n"
+ "fmax z15.s, p0/m, z15.s, z1.s\n"
+ "fmin z16.s, p1/m, z16.s, z0.s\n"
+ "fmin z17.s, p2/m, z17.s, z0.s\n"
+ "st1w z14.s, p2, [%[outptr5], #2, MUL VL]\n"
"addvl %[outptr5], %[outptr5], #3\n"
- "prfm PLDL1KEEP, [%[inptr], #0x2c0]\n"
- "prfm PLDL1KEEP, [%[outptr6], #0x60]\n"
+ "fmax z16.s, p1/m, z16.s, z1.s\n"
+ "st1w z15.s, p0, [%[outptr6]]\n"
+ "fmax z17.s, p2/m, z17.s, z1.s\n"
+ "st1w z16.s, p1, [%[outptr6], #1, MUL VL]\n"
+ "st1w z17.s, p2, [%[outptr6], #2, MUL VL]\n"
"addvl %[outptr6], %[outptr6], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;
@@ -1391,165 +1707,164 @@ inline void MergeResults<3, 8, true>(float *out, const float *in, const int ldou
long p = 0;
/* Optimized routine to copy an entire block */
__asm __volatile (
- "mov z2.s, %s[alpha]\n"
+ "mov z0.s, %s[maxval]\n"
"addvl x8, %[inptr], #16\n"
- "mov z3.s, %s[beta]\n"
+ "mov z1.s, %s[minval]\n"
"whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0]]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr]]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0]]\n"
- "ld1w z9.s, p0/z, [%[outptr1]]\n"
"prfm PLDL1KEEP, [%[inptr], #0x180]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #3, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1]]\n"
- "ld1w z10.s, p0/z, [%[outptr2]]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #6, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2]]\n"
- "ld1w z11.s, p0/z, [%[outptr3]]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-7, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3]]\n"
- "ld1w z8.s, p0/z, [%[outptr4]]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-4, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4]]\n"
- "ld1w z9.s, p0/z, [%[outptr5]]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #-1, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5]]\n"
- "ld1w z10.s, p0/z, [%[outptr6]]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #2, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6]]\n"
- "ld1w z11.s, p0/z, [%[outptr7]]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #5, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr7]]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr0], #0x60]\n"
+ "ld1w z2.s, p0/z, [%[biasptr]]\n"
+ "whilelt p1.s, %[p], %[w]\n"
+ "ld1w z3.s, p0/z, [%[biasptr], #1, MUL VL]\n"
"incw %[p], all, mul #1\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #1, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr1], #1, MUL VL]\n"
+ "ld1w z4.s, p0/z, [%[biasptr], #2, MUL VL]\n"
"prfm PLDL1KEEP, [%[inptr], #0x1c0]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #4, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #1, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr2], #1, MUL VL]\n"
+ "ld1w z13.s, p0/z, [%[inptr]]\n"
+ "whilelt p2.s, %[p], %[w]\n"
+ "ld1w z14.s, p1/z, [%[inptr], #1, MUL VL]\n"
+ "prfm PSTL1KEEP, [%[outptr1], #0x60]\n"
+ "fadd z13.s, z13.s, z2.s\n"
+ "ld1w z15.s, p2/z, [%[inptr], #2, MUL VL]\n"
+ "ld1w z16.s, p0/z, [%[inptr], #3, MUL VL]\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
+ "fadd z14.s, z14.s, z3.s\n"
+ "ld1w z17.s, p1/z, [%[inptr], #4, MUL VL]\n"
+ "fmin z13.s, p0/m, z13.s, z0.s\n"
+ "ld1w z18.s, p2/z, [%[inptr], #5, MUL VL]\n"
+ "fadd z15.s, z15.s, z4.s\n"
+ "ld1w z19.s, p0/z, [%[inptr], #6, MUL VL]\n"
+ "fadd z16.s, z16.s, z2.s\n"
+ "ld1w z20.s, p1/z, [%[inptr], #7, MUL VL]\n"
+ "fmin z14.s, p1/m, z14.s, z0.s\n"
+ "prfm PSTL1KEEP, [%[outptr2], #0x60]\n"
+ "fmax z13.s, p0/m, z13.s, z1.s\n"
+ "prfm PSTL1KEEP, [%[outptr3], #0x60]\n"
+ "fmax z14.s, p1/m, z14.s, z1.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x240]\n"
+ "fmin z15.s, p2/m, z15.s, z0.s\n"
+ "st1w z13.s, p0, [%[outptr0]]\n"
+ "fmin z16.s, p0/m, z16.s, z0.s\n"
+ "ld1w z13.s, p2/z, [x8, #-8, MUL VL]\n"
+ "fadd z17.s, z17.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z15.s, p2/m, z15.s, z1.s\n"
+ "st1w z14.s, p1, [%[outptr0], #1, MUL VL]\n"
+ "fmax z16.s, p0/m, z16.s, z1.s\n"
+ "ld1w z14.s, p0/z, [x8, #-7, MUL VL]\n"
+ "fmin z17.s, p1/m, z17.s, z0.s\n"
"prfm PLDL1KEEP, [%[inptr], #0x280]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [%[inptr], #7, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr3], #1, MUL VL]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-6, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #1, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr4], #1, MUL VL]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-3, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4], #1, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr5], #1, MUL VL]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5], #1, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr6], #1, MUL VL]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #3, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6], #1, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr7], #1, MUL VL]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #6, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr7], #1, MUL VL]\n"
- "whilelt p0.s, %[p], %[w]\n"
- "b.none 1f\n"
- "ld1w z8.s, p0/z, [%[outptr0], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr0], #0x60]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [%[inptr], #2, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr0], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr1], #2, MUL VL]\n"
+ "fadd z18.s, z18.s, z4.s\n"
+ "st1w z15.s, p2, [%[outptr0], #2, MUL VL]\n"
+ "fadd z19.s, z19.s, z2.s\n"
+ "ld1w z15.s, p1/z, [x8, #-6, MUL VL]\n"
+ "fmax z17.s, p1/m, z17.s, z1.s\n"
"addvl %[outptr0], %[outptr0], #3\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [%[inptr], #5, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr1], #2, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr2], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr1], #0x60]\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #-8, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr2], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr3], #2, MUL VL]\n"
+ "fmin z18.s, p2/m, z18.s, z0.s\n"
+ "st1w z16.s, p0, [%[outptr1]]\n"
+ "fmin z19.s, p0/m, z19.s, z0.s\n"
+ "ld1w z16.s, p2/z, [x8, #-5, MUL VL]\n"
+ "fadd z20.s, z20.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr5], #0x60]\n"
+ "fmax z18.s, p2/m, z18.s, z1.s\n"
+ "st1w z17.s, p1, [%[outptr1], #1, MUL VL]\n"
+ "fmax z19.s, p0/m, z19.s, z1.s\n"
+ "ld1w z17.s, p0/z, [x8, #-4, MUL VL]\n"
+ "fmin z20.s, p1/m, z20.s, z0.s\n"
+ "prfm PLDL1KEEP, [%[inptr], #0x2c0]\n"
+ "fadd z13.s, z13.s, z4.s\n"
+ "st1w z18.s, p2, [%[outptr1], #2, MUL VL]\n"
+ "fadd z14.s, z14.s, z2.s\n"
+ "ld1w z18.s, p1/z, [x8, #-3, MUL VL]\n"
+ "fmax z20.s, p1/m, z20.s, z1.s\n"
"addvl %[outptr1], %[outptr1], #3\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #-5, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr3], #2, MUL VL]\n"
- "ld1w z8.s, p0/z, [%[outptr4], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[inptr], #0x200]\n"
- "fmul z8.s, z8.s, z3.s\n"
- "ld1w z4.s, p0/z, [x8, #-2, MUL VL]\n"
- "fmla z8.s, p0/m, z4.s, z2.s\n"
- "st1w z8.s, p0, [%[outptr4], #2, MUL VL]\n"
- "ld1w z9.s, p0/z, [%[outptr5], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr2], #0x60]\n"
- "fmul z9.s, z9.s, z3.s\n"
- "ld1w z5.s, p0/z, [x8, #1, MUL VL]\n"
- "fmla z9.s, p0/m, z5.s, z2.s\n"
- "st1w z9.s, p0, [%[outptr5], #2, MUL VL]\n"
- "ld1w z10.s, p0/z, [%[outptr6], #2, MUL VL]\n"
+ "fmin z13.s, p2/m, z13.s, z0.s\n"
+ "st1w z19.s, p0, [%[outptr2]]\n"
+ "fmin z14.s, p0/m, z14.s, z0.s\n"
+ "ld1w z19.s, p2/z, [x8, #-2, MUL VL]\n"
+ "fadd z15.s, z15.s, z3.s\n"
+ "prfm PSTL1KEEP, [%[outptr6], #0x60]\n"
+ "fmax z13.s, p2/m, z13.s, z1.s\n"
+ "st1w z20.s, p1, [%[outptr2], #1, MUL VL]\n"
+ "fmax z14.s, p0/m, z14.s, z1.s\n"
+ "ld1w z20.s, p0/z, [x8, #-1, MUL VL]\n"
+ "fmin z15.s, p1/m, z15.s, z0.s\n"
+ "prfm PSTL1KEEP, [%[outptr7], #0x60]\n"
+ "fadd z16.s, z16.s, z4.s\n"
+ "st1w z13.s, p2, [%[outptr2], #2, MUL VL]\n"
+ "fadd z17.s, z17.s, z2.s\n"
+ "ld1w z13.s, p1/z, [x8]\n"
+ "fmax z15.s, p1/m, z15.s, z1.s\n"
"addvl %[outptr2], %[outptr2], #3\n"
- "fmul z10.s, z10.s, z3.s\n"
- "ld1w z6.s, p0/z, [x8, #4, MUL VL]\n"
- "fmla z10.s, p0/m, z6.s, z2.s\n"
- "st1w z10.s, p0, [%[outptr6], #2, MUL VL]\n"
- "ld1w z11.s, p0/z, [%[outptr7], #2, MUL VL]\n"
- "prfm PLDL1KEEP, [%[outptr3], #0x60]\n"
- "fmul z11.s, z11.s, z3.s\n"
- "ld1w z7.s, p0/z, [x8, #7, MUL VL]\n"
- "fmla z11.s, p0/m, z7.s, z2.s\n"
- "st1w z11.s, p0, [%[outptr7], #2, MUL VL]\n"
+ "fmin z16.s, p2/m, z16.s, z0.s\n"
+ "st1w z14.s, p0, [%[outptr3]]\n"
+ "fmin z17.s, p0/m, z17.s, z0.s\n"
+ "ld1w z14.s, p2/z, [x8, #1, MUL VL]\n"
+ "fadd z18.s, z18.s, z3.s\n"
+ "addvl %[inptr], %[inptr], #24\n"
+ "fmax z16.s, p2/m, z16.s, z1.s\n"
+ "st1w z15.s, p1, [%[outptr3], #1, MUL VL]\n"
+ "fmax z17.s, p0/m, z17.s, z1.s\n"
+ "ld1w z15.s, p0/z, [x8, #2, MUL VL]\n"
+ "fmin z18.s, p1/m, z18.s, z0.s\n"
+ "fadd z19.s, z19.s, z4.s\n"
+ "st1w z16.s, p2, [%[outptr3], #2, MUL VL]\n"
+ "fadd z20.s, z20.s, z2.s\n"
+ "ld1w z16.s, p1/z, [x8, #3, MUL VL]\n"
+ "fadd z13.s, z13.s, z3.s\n"
"addvl %[outptr3], %[outptr3], #3\n"
- "prfm PLDL1KEEP, [%[outptr4], #0x60]\n"
+ "fmax z18.s, p1/m, z18.s, z1.s\n"
+ "st1w z17.s, p0, [%[outptr4]]\n"
+ "fmin z19.s, p2/m, z19.s, z0.s\n"
+ "ld1w z17.s, p2/z, [x8, #4, MUL VL]\n"
+ "fmin z20.s, p0/m, z20.s, z0.s\n"
+ "fmin z13.s, p1/m, z13.s, z0.s\n"
+ "st1w z18.s, p1, [%[outptr4], #1, MUL VL]\n"
+ "fadd z14.s, z14.s, z4.s\n"
+ "ld1w z18.s, p0/z, [x8, #5, MUL VL]\n"
+ "fmax z19.s, p2/m, z19.s, z1.s\n"
+ "fmax z20.s, p0/m, z20.s, z1.s\n"
+ "fmax z13.s, p1/m, z13.s, z1.s\n"
+ "fmin z14.s, p2/m, z14.s, z0.s\n"
+ "st1w z19.s, p2, [%[outptr4], #2, MUL VL]\n"
+ "fadd z15.s, z15.s, z2.s\n"
+ "ld1w z19.s, p1/z, [x8, #6, MUL VL]\n"
+ "fadd z16.s, z16.s, z3.s\n"
"addvl %[outptr4], %[outptr4], #3\n"
- "prfm PLDL1KEEP, [%[outptr5], #0x60]\n"
+ "fmax z14.s, p2/m, z14.s, z1.s\n"
+ "st1w z20.s, p0, [%[outptr5]]\n"
+ "fmin z15.s, p0/m, z15.s, z0.s\n"
+ "ld1w z20.s, p2/z, [x8, #7, MUL VL]\n"
+ "fmin z16.s, p1/m, z16.s, z0.s\n"
+ "fadd z17.s, z17.s, z4.s\n"
+ "st1w z13.s, p1, [%[outptr5], #1, MUL VL]\n"
+ "fadd z18.s, z18.s, z2.s\n"
+ "fmax z15.s, p0/m, z15.s, z1.s\n"
+ "fmax z16.s, p1/m, z16.s, z1.s\n"
+ "st1w z14.s, p2, [%[outptr5], #2, MUL VL]\n"
+ "fmin z17.s, p2/m, z17.s, z0.s\n"
"addvl %[outptr5], %[outptr5], #3\n"
- "prfm PLDL1KEEP, [%[inptr], #0x2c0]\n"
- "prfm PLDL1KEEP, [%[outptr6], #0x60]\n"
+ "fmin z18.s, p0/m, z18.s, z0.s\n"
+ "st1w z15.s, p0, [%[outptr6]]\n"
+ "fadd z19.s, z19.s, z3.s\n"
+ "fmax z17.s, p2/m, z17.s, z1.s\n"
+ "fadd z20.s, z20.s, z4.s\n"
+ "st1w z16.s, p1, [%[outptr6], #1, MUL VL]\n"
+ "fmax z18.s, p0/m, z18.s, z1.s\n"
+ "fmin z19.s, p1/m, z19.s, z0.s\n"
+ "fmin z20.s, p2/m, z20.s, z0.s\n"
+ "st1w z17.s, p2, [%[outptr6], #2, MUL VL]\n"
"addvl %[outptr6], %[outptr6], #3\n"
- "prfm PLDL1KEEP, [%[outptr7], #0x60]\n"
+ "fmax z19.s, p1/m, z19.s, z1.s\n"
+ "fmax z20.s, p2/m, z20.s, z1.s\n"
+ "st1w z18.s, p0, [%[outptr7]]\n"
+ "st1w z19.s, p1, [%[outptr7], #1, MUL VL]\n"
+ "st1w z20.s, p2, [%[outptr7], #2, MUL VL]\n"
"addvl %[outptr7], %[outptr7], #3\n"
- "1:\n"
- "addvl %[inptr], %[inptr], #24\n"
: [outptr0] "+r" (outptr0), [outptr1] "+r" (outptr1), [outptr2] "+r" (outptr2), [outptr3] "+r" (outptr3), [outptr4] "+r" (outptr4), [outptr5] "+r" (outptr5), [outptr6] "+r" (outptr6), [outptr7] "+r" (outptr7),
[inptr] "+r" (inptr), [p] "+r" (p)
- : [alpha] "w" (alpha), [beta] "w" (beta), [w] "r" (w)
- : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "memory", "cc"
+ : [w] "r" (w), [biasptr] "r" (biasptr), [minval] "w" (minval), [maxval] "w" (maxval)
+ : "x8", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "memory", "cc"
);
}
break;