blob: 25262d3db99d0df98070114148944f5ec60b1136 [file] [log] [blame]
Viet-Hoa Do03b29712022-06-01 11:47:14 +01001/*
Michael Tyler7d9a6262023-02-01 16:37:07 +00002 * Copyright (c) 2022-2023 Arm Limited.
Viet-Hoa Do03b29712022-06-01 11:47:14 +01003 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
Michael Tyler7d9a6262023-02-01 16:37:07 +000013 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
Viet-Hoa Do03b29712022-06-01 11:47:14 +010015 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
Michael Tyler7d9a6262023-02-01 16:37:07 +000020 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
Viet-Hoa Do03b29712022-06-01 11:47:14 +010023 */
24
25#if defined(__ARM_FEATURE_SVE)
26
27template <>
28void interleave_block<2, 4, VLType::SME, true>(
29 int8_t * &out, const int8_t * const *in,
30 size_t width, size_t height, size_t row_offset, bool first
31)
32{
33 __asm__ __volatile__(
34 ".inst 0xd503477f // SMSTART ZA\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000035 "cntb x21\n"
36 "mov x23, %x[width]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010037 "mov z20.b, #0x1\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000038 "incb x23\n"
39 "mov x20, %x[width]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010040 "mov z19.s, #0x0\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010041 "mov z18.s, #0x0\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000042 "sub x17, x21, #0x1\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010043 "cntw x16\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000044 "sub x23, x23, #0x1\n"
45 "ands x17, x20, x17\n"
46 "udiv x23, x23, x21\n" // n_passes = ceildiv(width, VL<T>)
47 "csel x17, x17, x21, NE\n"
48 "lsl x22, %x[height], #0x1\n" // height * 2
49 "lsl x21, x16, #0x1\n"
50 "sub x20, x23, #0x1\n"
51 "add x17, x17, #0x3\n"
52 "whilelt p9.b, XZR, x22\n"
53 "whilelt p8.b, x21, x22\n"
54 "mov x15, #0x0\n"
55 "cntw x14, ALL, MUL #2\n"
56 "cntw x11, ALL, MUL #3\n"
57 "ptrue p4.b\n"
58 "lsr x20, x20, #0x1\n" // n_loops = (n_passes - 1) / 2
59 "and x10, x23, #0x1\n" // odd_tail = bool(n_passes & 0x1)
60 "lsr x17, x17, #0x2\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010061 "sub x9, x16, #0x2\n"
62 "ptrue p11.s\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010063 "zip1 p10.b, p9.b, p8.b\n"
64 "mov x28, %x[row_offset]\n"
65 "mov x27, %x[out]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000066 "whilelt p9.b, x15, %x[width]\n"
67 "whilelt p8.b, x15, %x[width]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010068 "cbnz %x[first], 1f\n"
69 "addvl x27, x27, #-2\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000070 "ld1w { z19.s }, p4/Z, [x27]\n"
71 "ld1w { z18.s }, p4/Z, [x27, #1, MUL VL]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010072 "1:" // K loop: Load row sums: End
Michael Tyler7d9a6262023-02-01 16:37:07 +000073 "mov x26, %x[in]\n"
74 "add x25, %x[in], x16, LSL #3\n"
75 "ldr x24, [x26, #0x0]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010076 "ldr x23, [x25, #0x0]\n"
Michael Tylerbe13cea2023-01-17 11:04:14 +000077 "mov x12, #0x0\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000078 "ldr x22, [x26, #0x8]\n"
79 "add x26, x26, #0x10\n"
80 "ldr x21, [x25, #0x8]\n"
81 "add x25, x25, #0x10\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010082 "cbz x9, 3f\n"
83 "2:" // K loop: Charge: Loop
Michael Tyler7d9a6262023-02-01 16:37:07 +000084 ".inst 0x25246140 // psel p0.b, p8.b/Z, p10.b[w12]\n"
85 ".inst 0xe01c0300 // ld1b { za0h.b[x12] }, p0/Z, [x24, x28]\n"
86 ".inst 0x252c6140 // psel p0.b, p8.b/Z, p10.b[w12, #1]\n"
87 "ldr x24, [x26, #0x0]\n"
88 ".inst 0xe01c02e1 // ld1b { za0h.b[x12, #1] }, p0/Z, [x23, x28]\n"
89 ".inst 0x25646141 // psel p1.b, p8.b/Z, p10.b[w12, #4]\n"
90 ".inst 0x256c6140 // psel p0.b, p8.b/Z, p10.b[w12, #5]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010091 "ldr x23, [x25, #0x0]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000092 ".inst 0xe01c06c4 // ld1b { za0h.b[x12, #4] }, p1/Z, [x22, x28]\n"
93 "ldr x22, [x26, #0x8]\n"
94 "add x26, x26, #0x10\n"
95 ".inst 0xe01c02a5 // ld1b { za0h.b[x12, #5] }, p0/Z, [x21, x28]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +010096 "add x12, x12, #0x8\n"
97 "cmp x12, x9, LSL #2\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +000098 "ldr x21, [x25, #0x8]\n"
99 "add x25, x25, #0x10\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100100 "blt 2b\n"
101 "3:" // K loop: Charge: End
Michael Tyler7d9a6262023-02-01 16:37:07 +0000102 ".inst 0x25246140 // psel p0.b, p8.b/Z, p10.b[w12]\n"
103 ".inst 0xe01c0300 // ld1b { za0h.b[x12] }, p0/Z, [x24, x28]\n"
104 ".inst 0x252c6140 // psel p0.b, p8.b/Z, p10.b[w12, #1]\n"
105 ".inst 0xe01c02e1 // ld1b { za0h.b[x12, #1] }, p0/Z, [x23, x28]\n"
106 ".inst 0x25646141 // psel p1.b, p8.b/Z, p10.b[w12, #4]\n"
107 ".inst 0x256c6140 // psel p0.b, p8.b/Z, p10.b[w12, #5]\n"
108 ".inst 0xe01c06c4 // ld1b { za0h.b[x12, #4] }, p1/Z, [x22, x28]\n"
109 "mov x26, %x[in]\n"
110 "add x25, %x[in], x16, LSL #3\n"
111 "ldr x24, [x26, #0x0]\n"
112 ".inst 0xe01c02a5 // ld1b { za0h.b[x12, #5] }, p0/Z, [x21, x28]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100113 "ldr x23, [x25, #0x0]\n"
Michael Tylerbe13cea2023-01-17 11:04:14 +0000114 "incb x28\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000115 "incb x15\n"
116 "ldr x22, [x26, #0x8]\n"
117 "add x26, x26, #0x10\n"
118 "ldr x21, [x25, #0x8]\n"
119 "add x25, x25, #0x10\n"
120 "cbz x20, 9f\n"
121 "mov x20, x20\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100122 "4:" // K loop: Main loop
Michael Tyler7d9a6262023-02-01 16:37:07 +0000123 "whilelt p8.b, x15, %x[width]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100124 "mov x13, #0x0\n"
125 "mov x12, #0x0\n"
126 "cbz x9, 6f\n"
127 "5:" // K loop: Main loop: First: Loop
Michael Tyler7d9a6262023-02-01 16:37:07 +0000128 ".inst 0x25356140 // psel p0.b, p8.b/Z, p10.b[w13, #2]\n"
129 ".inst 0xe01c2302 // ld1b { za0h.b[x13, #2] }, p0/Z, [x24, x28]\n"
130 ".inst 0x253d6140 // psel p0.b, p8.b/Z, p10.b[w13, #3]\n"
131 "ldr x24, [x26, #0x0]\n"
132 ".inst 0xe01c22e3 // ld1b { za0h.b[x13, #3] }, p0/Z, [x23, x28]\n"
133 ".inst 0x25756140 // psel p0.b, p8.b/Z, p10.b[w13, #6]\n"
134 ".inst 0x257d6142 // psel p2.b, p8.b/Z, p10.b[w13, #7]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100135 "ldr x23, [x25, #0x0]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000136 ".inst 0xe01c22c6 // ld1b { za0h.b[x13, #6] }, p0/Z, [x22, x28]\n"
137 ".inst 0x25306d21 // psel p1.s, p11.s/Z, p9.s[w12]\n"
138 "ldr x22, [x26, #0x8]\n"
139 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
140 ".inst 0xe01c2aa7 // ld1b { za0h.b[x13, #7] }, p2/Z, [x21, x28]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100141 "ldr x21, [x25, #0x8]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000142 ".inst 0xe0bf8760 // st1w { za0v.s[x12] }, p1/Z, [x27, XZR, LSL #2]\n"
143 ".inst 0xc0829010 // mova z16.s, p4/M, za0v.s[x12]\n"
144 ".inst 0xc0829091 // mova z17.s, p4/M, za1v.s[x12]\n"
145 "sdot z19.s, z16.b, z20.b\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100146 ".inst 0xe0b08364 // st1w { za1v.s[x12] }, p0/Z, [x27, x16, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000147 ".inst 0x25706d20 // psel p0.s, p11.s/Z, p9.s[w12, #1]\n"
148 "sdot z18.s, z17.b, z20.b\n"
149 ".inst 0xe0ae8361 // st1w { za0v.s[x12, #1] }, p0/Z, [x27, x14, LSL #2]\n"
150 ".inst 0x25706d20 // psel p0.s, p11.s/Z, p9.s[w12, #1]\n"
151 ".inst 0xc0829030 // mova z16.s, p4/M, za0v.s[x12, #1]\n"
152 ".inst 0xe0ab8365 // st1w { za1v.s[x12, #1] }, p0/Z, [x27, x11, LSL #2]\n"
153 ".inst 0xc08290b1 // mova z17.s, p4/M, za1v.s[x12, #1]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100154 "add x12, x12, #0x2\n"
155 "cmp x12, x9\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000156 "add x26, x26, #0x10\n"
157 "add x25, x25, #0x10\n"
158 "sdot z19.s, z16.b, z20.b\n"
159 "sdot z18.s, z17.b, z20.b\n"
160 "addvl x27, x27, #4\n"
161 "add x13, x13, #0x8\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100162 "blt 5b\n"
163 "6:" // K loop: Main loop: First: Tail
Michael Tyler7d9a6262023-02-01 16:37:07 +0000164 ".inst 0x25356140 // psel p0.b, p8.b/Z, p10.b[w13, #2]\n"
165 ".inst 0xe01c2302 // ld1b { za0h.b[x13, #2] }, p0/Z, [x24, x28]\n"
166 ".inst 0x253d6140 // psel p0.b, p8.b/Z, p10.b[w13, #3]\n"
167 ".inst 0xe01c22e3 // ld1b { za0h.b[x13, #3] }, p0/Z, [x23, x28]\n"
168 ".inst 0x25756141 // psel p1.b, p8.b/Z, p10.b[w13, #6]\n"
169 ".inst 0x257d6140 // psel p0.b, p8.b/Z, p10.b[w13, #7]\n"
170 ".inst 0xe01c26c6 // ld1b { za0h.b[x13, #6] }, p1/Z, [x22, x28]\n"
171 "mov x26, %x[in]\n"
172 "add x25, %x[in], x16, LSL #3\n"
173 "ldr x24, [x26, #0x0]\n"
174 ".inst 0xe01c22a7 // ld1b { za0h.b[x13, #7] }, p0/Z, [x21, x28]\n"
175 ".inst 0xc0829010 // mova z16.s, p4/M, za0v.s[x12]\n"
176 ".inst 0x25306d23 // psel p3.s, p11.s/Z, p9.s[w12]\n"
177 "sdot z19.s, z16.b, z20.b\n"
178 ".inst 0xc0829091 // mova z17.s, p4/M, za1v.s[x12]\n"
179 "sdot z18.s, z17.b, z20.b\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100180 "ldr x23, [x25, #0x0]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000181 ".inst 0x25306d22 // psel p2.s, p11.s/Z, p9.s[w12]\n"
182 "ldr x22, [x26, #0x8]\n"
183 ".inst 0x25706d21 // psel p1.s, p11.s/Z, p9.s[w12, #1]\n"
184 ".inst 0xc0829030 // mova z16.s, p4/M, za0v.s[x12, #1]\n"
185 ".inst 0x25706d20 // psel p0.s, p11.s/Z, p9.s[w12, #1]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100186 "ldr x21, [x25, #0x8]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000187 ".inst 0xe0bf8f60 // st1w { za0v.s[x12] }, p3/Z, [x27, XZR, LSL #2]\n"
188 ".inst 0xc08290b1 // mova z17.s, p4/M, za1v.s[x12, #1]\n"
189 "whilelt p9.b, x15, %x[width]\n"
190 ".inst 0xe0b08b64 // st1w { za1v.s[x12] }, p2/Z, [x27, x16, LSL #2]\n"
191 "incb x15\n"
192 "add x26, x26, #0x10\n"
193 "sdot z19.s, z16.b, z20.b\n"
194 ".inst 0xe0ae8761 // st1w { za0v.s[x12, #1] }, p1/Z, [x27, x14, LSL #2]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100195 "add x25, x25, #0x10\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000196 "sdot z18.s, z17.b, z20.b\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100197 "incb x28\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000198 ".inst 0xe0ab8365 // st1w { za1v.s[x12, #1] }, p0/Z, [x27, x11, LSL #2]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100199 "addvl x27, x27, #4\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000200 "whilelt p8.b, x15, %x[width]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100201 "mov x13, #0x0\n"
202 "mov x12, #0x0\n"
203 "cbz x9, 8f\n"
204 "7:" // K loop: Main loop: Second: Loop
Michael Tyler7d9a6262023-02-01 16:37:07 +0000205 ".inst 0x25256140 // psel p0.b, p8.b/Z, p10.b[w13]\n"
206 ".inst 0xe01c2300 // ld1b { za0h.b[x13] }, p0/Z, [x24, x28]\n"
207 ".inst 0x252d6140 // psel p0.b, p8.b/Z, p10.b[w13, #1]\n"
208 "ldr x24, [x26, #0x0]\n"
209 ".inst 0xe01c22e1 // ld1b { za0h.b[x13, #1] }, p0/Z, [x23, x28]\n"
210 ".inst 0x25656140 // psel p0.b, p8.b/Z, p10.b[w13, #4]\n"
211 ".inst 0x256d6142 // psel p2.b, p8.b/Z, p10.b[w13, #5]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100212 "ldr x23, [x25, #0x0]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000213 ".inst 0xe01c22c4 // ld1b { za0h.b[x13, #4] }, p0/Z, [x22, x28]\n"
214 ".inst 0x25306d21 // psel p1.s, p11.s/Z, p9.s[w12]\n"
215 "ldr x22, [x26, #0x8]\n"
216 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
217 ".inst 0xe01c2aa5 // ld1b { za0h.b[x13, #5] }, p2/Z, [x21, x28]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100218 "ldr x21, [x25, #0x8]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000219 ".inst 0xe0bf8768 // st1w { za2v.s[x12] }, p1/Z, [x27, XZR, LSL #2]\n"
220 ".inst 0xc0829110 // mova z16.s, p4/M, za2v.s[x12]\n"
221 ".inst 0xc0829191 // mova z17.s, p4/M, za3v.s[x12]\n"
222 "sdot z19.s, z16.b, z20.b\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100223 ".inst 0xe0b0836c // st1w { za3v.s[x12] }, p0/Z, [x27, x16, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000224 ".inst 0x25706d20 // psel p0.s, p11.s/Z, p9.s[w12, #1]\n"
225 "sdot z18.s, z17.b, z20.b\n"
226 ".inst 0xe0ae8369 // st1w { za2v.s[x12, #1] }, p0/Z, [x27, x14, LSL #2]\n"
227 ".inst 0x25706d20 // psel p0.s, p11.s/Z, p9.s[w12, #1]\n"
228 ".inst 0xc0829130 // mova z16.s, p4/M, za2v.s[x12, #1]\n"
229 ".inst 0xe0ab836d // st1w { za3v.s[x12, #1] }, p0/Z, [x27, x11, LSL #2]\n"
230 ".inst 0xc08291b1 // mova z17.s, p4/M, za3v.s[x12, #1]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100231 "add x12, x12, #0x2\n"
232 "cmp x12, x9\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000233 "add x26, x26, #0x10\n"
234 "add x25, x25, #0x10\n"
235 "sdot z19.s, z16.b, z20.b\n"
236 "sdot z18.s, z17.b, z20.b\n"
237 "addvl x27, x27, #4\n"
238 "add x13, x13, #0x8\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100239 "blt 7b\n"
240 "8:" // K loop: Main loop: Second: Tail
Michael Tyler7d9a6262023-02-01 16:37:07 +0000241 ".inst 0x25256140 // psel p0.b, p8.b/Z, p10.b[w13]\n"
242 ".inst 0xe01c2300 // ld1b { za0h.b[x13] }, p0/Z, [x24, x28]\n"
243 ".inst 0x252d6140 // psel p0.b, p8.b/Z, p10.b[w13, #1]\n"
244 ".inst 0xe01c22e1 // ld1b { za0h.b[x13, #1] }, p0/Z, [x23, x28]\n"
245 ".inst 0x25656141 // psel p1.b, p8.b/Z, p10.b[w13, #4]\n"
246 ".inst 0x256d6140 // psel p0.b, p8.b/Z, p10.b[w13, #5]\n"
247 ".inst 0xe01c26c4 // ld1b { za0h.b[x13, #4] }, p1/Z, [x22, x28]\n"
248 "mov x26, %x[in]\n"
249 "add x25, %x[in], x16, LSL #3\n"
250 "ldr x24, [x26, #0x0]\n"
251 ".inst 0xe01c22a5 // ld1b { za0h.b[x13, #5] }, p0/Z, [x21, x28]\n"
252 ".inst 0xc0829110 // mova z16.s, p4/M, za2v.s[x12]\n"
253 ".inst 0x25306d23 // psel p3.s, p11.s/Z, p9.s[w12]\n"
254 "sdot z19.s, z16.b, z20.b\n"
255 ".inst 0xc0829191 // mova z17.s, p4/M, za3v.s[x12]\n"
256 "sdot z18.s, z17.b, z20.b\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100257 "ldr x23, [x25, #0x0]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000258 ".inst 0x25306d22 // psel p2.s, p11.s/Z, p9.s[w12]\n"
259 "ldr x22, [x26, #0x8]\n"
260 ".inst 0x25706d21 // psel p1.s, p11.s/Z, p9.s[w12, #1]\n"
261 ".inst 0xc0829130 // mova z16.s, p4/M, za2v.s[x12, #1]\n"
262 ".inst 0x25706d20 // psel p0.s, p11.s/Z, p9.s[w12, #1]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100263 "ldr x21, [x25, #0x8]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000264 ".inst 0xe0bf8f68 // st1w { za2v.s[x12] }, p3/Z, [x27, XZR, LSL #2]\n"
265 ".inst 0xc08291b1 // mova z17.s, p4/M, za3v.s[x12, #1]\n"
266 "whilelt p9.b, x15, %x[width]\n"
267 ".inst 0xe0b08b6c // st1w { za3v.s[x12] }, p2/Z, [x27, x16, LSL #2]\n"
268 "subs x20, x20, #0x1\n"
269 "add x26, x26, #0x10\n"
270 "sdot z19.s, z16.b, z20.b\n"
271 ".inst 0xe0ae8769 // st1w { za2v.s[x12, #1] }, p1/Z, [x27, x14, LSL #2]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100272 "add x25, x25, #0x10\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000273 "sdot z18.s, z17.b, z20.b\n"
274 "incb x15\n"
275 ".inst 0xe0ab836d // st1w { za3v.s[x12, #1] }, p0/Z, [x27, x11, LSL #2]\n"
Michael Tylerbe13cea2023-01-17 11:04:14 +0000276 "addvl x27, x27, #4\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000277 "incb x28\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100278 "bgt 4b\n"
279 "9:" // K loop: Tails
Michael Tyler7d9a6262023-02-01 16:37:07 +0000280 "cbnz x10, 12f\n"
281 "mov x26, %x[in]\n"
282 "whilelt p8.b, x15, %x[width]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100283 "mov x13, #0x0\n"
284 "mov x12, #0x0\n"
285 "10:" // K loop: Tails: Even: First
Michael Tyler7d9a6262023-02-01 16:37:07 +0000286 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100287 ".inst 0xe0bf8360 // st1w { za0v.s[x12] }, p0/Z, [x27, XZR, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000288 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
289 ".inst 0xc0829010 // mova z16.s, p4/M, za0v.s[x12]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100290 ".inst 0xe0b08364 // st1w { za1v.s[x12] }, p0/Z, [x27, x16, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000291 "ldr x24, [x26, #0x0]\n"
292 ".inst 0x25356140 // psel p0.b, p8.b/Z, p10.b[w13, #2]\n"
293 ".inst 0xc0829091 // mova z17.s, p4/M, za1v.s[x12]\n"
294 "ldr x23, [x26, x16, LSL #0x3]\n"
295 ".inst 0xe01c2302 // ld1b { za0h.b[x13, #2] }, p0/Z, [x24, x28]\n"
Michael Tylerbe13cea2023-01-17 11:04:14 +0000296 "add x12, x12, #0x1\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000297 ".inst 0x253d6140 // psel p0.b, p8.b/Z, p10.b[w13, #3]\n"
Michael Tylerbe13cea2023-01-17 11:04:14 +0000298 "cmp x12, x16\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000299 "sdot z19.s, z16.b, z20.b\n"
300 "sdot z18.s, z17.b, z20.b\n"
301 ".inst 0xe01c22e3 // ld1b { za0h.b[x13, #3] }, p0/Z, [x23, x28]\n"
302 "add x26, x26, #0x8\n"
303 "addvl x27, x27, #2\n"
304 "add x13, x13, #0x4\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100305 "blt 10b\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000306 "whilelt p9.b, x15, %x[width]\n"
307 "whilelt p8.b, x15, %x[width]\n"
308 "mov x20, #0x0\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100309 "mov x12, #0x0\n"
310 "11:" // K loop: Tails: Even: Second
Michael Tyler7d9a6262023-02-01 16:37:07 +0000311 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100312 ".inst 0xe0bf8368 // st1w { za2v.s[x12] }, p0/Z, [x27, XZR, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000313 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
314 ".inst 0xc0829110 // mova z16.s, p4/M, za2v.s[x12]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100315 ".inst 0xe0b0836c // st1w { za3v.s[x12] }, p0/Z, [x27, x16, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000316 ".inst 0xc0829191 // mova z17.s, p4/M, za3v.s[x12]\n"
Michael Tylerbe13cea2023-01-17 11:04:14 +0000317 "add x12, x12, #0x1\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000318 "cmp x12, x17\n"
319 "sdot z19.s, z16.b, z20.b\n"
320 "sdot z18.s, z17.b, z20.b\n"
321 "addvl x27, x27, #2\n"
322 "add x20, x20, #0x4\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100323 "blt 11b\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000324 "whilelt p9.b, x15, %x[width]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100325 "b 14f\n"
326 "12:" // K loop: Tails: Odd
327 "mov x12, #0x0\n"
328 "13:" // K loop: Tails: Odd: Loop
Michael Tyler7d9a6262023-02-01 16:37:07 +0000329 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100330 ".inst 0xe0bf8360 // st1w { za0v.s[x12] }, p0/Z, [x27, XZR, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000331 ".inst 0x25306d20 // psel p0.s, p11.s/Z, p9.s[w12]\n"
332 ".inst 0xc0829010 // mova z16.s, p4/M, za0v.s[x12]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100333 ".inst 0xe0b08364 // st1w { za1v.s[x12] }, p0/Z, [x27, x16, LSL #2]\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000334 ".inst 0xc0829091 // mova z17.s, p4/M, za1v.s[x12]\n"
Michael Tylerbe13cea2023-01-17 11:04:14 +0000335 "add x12, x12, #0x1\n"
Michael Tyler7d9a6262023-02-01 16:37:07 +0000336 "cmp x12, x17\n"
337 "sdot z19.s, z16.b, z20.b\n"
338 "sdot z18.s, z17.b, z20.b\n"
339 "addvl x27, x27, #2\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100340 "blt 13b\n"
341 "14:" // K loop: End
Michael Tyler7d9a6262023-02-01 16:37:07 +0000342 "st1w { z19.s }, p4, [x27]\n"
343 "st1w { z18.s }, p4, [x27, #1, MUL VL]\n"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100344 "addvl x27, x27, #2\n"
345 "mov %x[out], x27\n"
346 ".inst 0xd503467f // SMSTOP\n"
347 : [out] "+&r" (out)
348 : [first] "r" (first), [height] "r" (height), [in] "r" (in), [row_offset] "r" (row_offset), [width] "r" (width)
Michael Tyler7d9a6262023-02-01 16:37:07 +0000349 : "cc", "memory", "p0", "p1", "p2", "p3", "p4", "p5", "p6", "p7", "p8", "p9", "p10", "p11", "p12", "p13", "p14", "p15", "x9", "x10", "x11", "x12", "x13", "x14", "x15", "x16", "x17", "x20", "x21", "x22", "x23", "x24", "x25", "x26", "x27", "x28", "z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7", "z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15", "z16", "z17", "z18", "z19", "z20", "z21", "z22", "z23", "z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31"
Viet-Hoa Do03b29712022-06-01 11:47:14 +0100350 );
351}
352
353#endif // defined(__ARM_FEATURE_SVE)