]> git.proxmox.com Git - mirror_qemu.git/blame - target/arm/translate-sve.c
target/arm: Implement SVE2 saturating multiply-add high (indexed)
[mirror_qemu.git] / target / arm / translate-sve.c
CommitLineData
38388f7e
RH
1/*
2 * AArch64 SVE translation
3 *
4 * Copyright (c) 2018 Linaro, Ltd
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
50f57e09 9 * version 2.1 of the License, or (at your option) any later version.
38388f7e
RH
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
18 */
19
20#include "qemu/osdep.h"
21#include "cpu.h"
22#include "exec/exec-all.h"
dcb32f1d
PMD
23#include "tcg/tcg-op.h"
24#include "tcg/tcg-op-gvec.h"
25#include "tcg/tcg-gvec-desc.h"
38388f7e
RH
26#include "qemu/log.h"
27#include "arm_ldst.h"
28#include "translate.h"
29#include "internals.h"
30#include "exec/helper-proto.h"
31#include "exec/helper-gen.h"
32#include "exec/log.h"
33#include "trace-tcg.h"
34#include "translate-a64.h"
cc48affe 35#include "fpu/softfloat.h"
38388f7e 36
757f9cff 37
9ee3a611
RH
38typedef void GVecGen2sFn(unsigned, uint32_t, uint32_t,
39 TCGv_i64, uint32_t, uint32_t);
40
38cadeba
RH
41typedef void gen_helper_gvec_flags_3(TCGv_i32, TCGv_ptr, TCGv_ptr,
42 TCGv_ptr, TCGv_i32);
757f9cff
RH
43typedef void gen_helper_gvec_flags_4(TCGv_i32, TCGv_ptr, TCGv_ptr,
44 TCGv_ptr, TCGv_ptr, TCGv_i32);
45
c4e7c493 46typedef void gen_helper_gvec_mem(TCGv_env, TCGv_ptr, TCGv_i64, TCGv_i32);
f6dbf62a
RH
47typedef void gen_helper_gvec_mem_scatter(TCGv_env, TCGv_ptr, TCGv_ptr,
48 TCGv_ptr, TCGv_i64, TCGv_i32);
c4e7c493 49
ccd841c3
RH
50/*
51 * Helpers for extracting complex instruction fields.
52 */
53
54/* See e.g. ASR (immediate, predicated).
55 * Returns -1 for unallocated encoding; diagnose later.
56 */
451e4ffd 57static int tszimm_esz(DisasContext *s, int x)
ccd841c3
RH
58{
59 x >>= 3; /* discard imm3 */
60 return 31 - clz32(x);
61}
62
451e4ffd 63static int tszimm_shr(DisasContext *s, int x)
ccd841c3 64{
451e4ffd 65 return (16 << tszimm_esz(s, x)) - x;
ccd841c3
RH
66}
67
68/* See e.g. LSL (immediate, predicated). */
451e4ffd 69static int tszimm_shl(DisasContext *s, int x)
ccd841c3 70{
451e4ffd 71 return x - (8 << tszimm_esz(s, x));
ccd841c3
RH
72}
73
451e4ffd 74static inline int plus1(DisasContext *s, int x)
24e82e68
RH
75{
76 return x + 1;
77}
78
f25a2361 79/* The SH bit is in bit 8. Extract the low 8 and shift. */
451e4ffd 80static inline int expand_imm_sh8s(DisasContext *s, int x)
f25a2361
RH
81{
82 return (int8_t)x << (x & 0x100 ? 8 : 0);
83}
84
451e4ffd 85static inline int expand_imm_sh8u(DisasContext *s, int x)
6e6a157d
RH
86{
87 return (uint8_t)x << (x & 0x100 ? 8 : 0);
88}
89
c4e7c493
RH
90/* Convert a 2-bit memory size (msz) to a 4-bit data type (dtype)
91 * with unsigned data. C.f. SVE Memory Contiguous Load Group.
92 */
451e4ffd 93static inline int msz_dtype(DisasContext *s, int msz)
c4e7c493
RH
94{
95 static const uint8_t dtype[4] = { 0, 5, 10, 15 };
96 return dtype[msz];
97}
98
38388f7e
RH
99/*
100 * Include the generated decoder.
101 */
102
139c1837 103#include "decode-sve.c.inc"
38388f7e
RH
104
105/*
106 * Implement all of the translator functions referenced by the decoder.
107 */
108
d1822297
RH
109/* Return the offset info CPUARMState of the predicate vector register Pn.
110 * Note for this purpose, FFR is P16.
111 */
112static inline int pred_full_reg_offset(DisasContext *s, int regno)
113{
114 return offsetof(CPUARMState, vfp.pregs[regno]);
115}
116
117/* Return the byte size of the whole predicate register, VL / 64. */
118static inline int pred_full_reg_size(DisasContext *s)
119{
120 return s->sve_len >> 3;
121}
122
516e246a
RH
123/* Round up the size of a register to a size allowed by
124 * the tcg vector infrastructure. Any operation which uses this
125 * size may assume that the bits above pred_full_reg_size are zero,
126 * and must leave them the same way.
127 *
128 * Note that this is not needed for the vector registers as they
129 * are always properly sized for tcg vectors.
130 */
131static int size_for_gvec(int size)
132{
133 if (size <= 8) {
134 return 8;
135 } else {
136 return QEMU_ALIGN_UP(size, 16);
137 }
138}
139
140static int pred_gvec_reg_size(DisasContext *s)
141{
142 return size_for_gvec(pred_full_reg_size(s));
143}
144
40e32e5a
RH
145/* Invoke an out-of-line helper on 2 Zregs. */
146static void gen_gvec_ool_zz(DisasContext *s, gen_helper_gvec_2 *fn,
147 int rd, int rn, int data)
148{
149 unsigned vsz = vec_full_reg_size(s);
150 tcg_gen_gvec_2_ool(vec_full_reg_offset(s, rd),
151 vec_full_reg_offset(s, rn),
152 vsz, vsz, data, fn);
153}
154
e645d1a1
RH
155/* Invoke an out-of-line helper on 3 Zregs. */
156static void gen_gvec_ool_zzz(DisasContext *s, gen_helper_gvec_3 *fn,
157 int rd, int rn, int rm, int data)
158{
159 unsigned vsz = vec_full_reg_size(s);
160 tcg_gen_gvec_3_ool(vec_full_reg_offset(s, rd),
161 vec_full_reg_offset(s, rn),
162 vec_full_reg_offset(s, rm),
163 vsz, vsz, data, fn);
164}
165
38650638
RH
166/* Invoke an out-of-line helper on 4 Zregs. */
167static void gen_gvec_ool_zzzz(DisasContext *s, gen_helper_gvec_4 *fn,
168 int rd, int rn, int rm, int ra, int data)
169{
170 unsigned vsz = vec_full_reg_size(s);
171 tcg_gen_gvec_4_ool(vec_full_reg_offset(s, rd),
172 vec_full_reg_offset(s, rn),
173 vec_full_reg_offset(s, rm),
174 vec_full_reg_offset(s, ra),
175 vsz, vsz, data, fn);
176}
177
96a461f7
RH
178/* Invoke an out-of-line helper on 2 Zregs and a predicate. */
179static void gen_gvec_ool_zzp(DisasContext *s, gen_helper_gvec_3 *fn,
180 int rd, int rn, int pg, int data)
181{
182 unsigned vsz = vec_full_reg_size(s);
183 tcg_gen_gvec_3_ool(vec_full_reg_offset(s, rd),
184 vec_full_reg_offset(s, rn),
185 pred_full_reg_offset(s, pg),
186 vsz, vsz, data, fn);
187}
188
36cbb7a8
RH
189/* Invoke an out-of-line helper on 3 Zregs and a predicate. */
190static void gen_gvec_ool_zzzp(DisasContext *s, gen_helper_gvec_4 *fn,
191 int rd, int rn, int rm, int pg, int data)
192{
193 unsigned vsz = vec_full_reg_size(s);
194 tcg_gen_gvec_4_ool(vec_full_reg_offset(s, rd),
195 vec_full_reg_offset(s, rn),
196 vec_full_reg_offset(s, rm),
197 pred_full_reg_offset(s, pg),
198 vsz, vsz, data, fn);
199}
f7d79c41 200
36cbb7a8 201/* Invoke a vector expander on two Zregs. */
f7d79c41
RH
202static void gen_gvec_fn_zz(DisasContext *s, GVecGen2Fn *gvec_fn,
203 int esz, int rd, int rn)
38388f7e 204{
f7d79c41
RH
205 unsigned vsz = vec_full_reg_size(s);
206 gvec_fn(esz, vec_full_reg_offset(s, rd),
207 vec_full_reg_offset(s, rn), vsz, vsz);
38388f7e
RH
208}
209
39eea561 210/* Invoke a vector expander on three Zregs. */
28c4da31
RH
211static void gen_gvec_fn_zzz(DisasContext *s, GVecGen3Fn *gvec_fn,
212 int esz, int rd, int rn, int rm)
38388f7e 213{
28c4da31
RH
214 unsigned vsz = vec_full_reg_size(s);
215 gvec_fn(esz, vec_full_reg_offset(s, rd),
216 vec_full_reg_offset(s, rn),
217 vec_full_reg_offset(s, rm), vsz, vsz);
38388f7e
RH
218}
219
911cdc6d
RH
220/* Invoke a vector expander on four Zregs. */
221static void gen_gvec_fn_zzzz(DisasContext *s, GVecGen4Fn *gvec_fn,
222 int esz, int rd, int rn, int rm, int ra)
223{
224 unsigned vsz = vec_full_reg_size(s);
225 gvec_fn(esz, vec_full_reg_offset(s, rd),
226 vec_full_reg_offset(s, rn),
227 vec_full_reg_offset(s, rm),
228 vec_full_reg_offset(s, ra), vsz, vsz);
229}
230
39eea561
RH
231/* Invoke a vector move on two Zregs. */
232static bool do_mov_z(DisasContext *s, int rd, int rn)
38388f7e 233{
f7d79c41
RH
234 if (sve_access_check(s)) {
235 gen_gvec_fn_zz(s, tcg_gen_gvec_mov, MO_8, rd, rn);
236 }
237 return true;
38388f7e
RH
238}
239
d9d78dcc
RH
240/* Initialize a Zreg with replications of a 64-bit immediate. */
241static void do_dupi_z(DisasContext *s, int rd, uint64_t word)
242{
243 unsigned vsz = vec_full_reg_size(s);
8711e71f 244 tcg_gen_gvec_dup_imm(MO_64, vec_full_reg_offset(s, rd), vsz, vsz, word);
d9d78dcc
RH
245}
246
516e246a 247/* Invoke a vector expander on three Pregs. */
dd81a8d7
RH
248static void gen_gvec_fn_ppp(DisasContext *s, GVecGen3Fn *gvec_fn,
249 int rd, int rn, int rm)
516e246a 250{
dd81a8d7
RH
251 unsigned psz = pred_gvec_reg_size(s);
252 gvec_fn(MO_64, pred_full_reg_offset(s, rd),
253 pred_full_reg_offset(s, rn),
254 pred_full_reg_offset(s, rm), psz, psz);
516e246a
RH
255}
256
257/* Invoke a vector move on two Pregs. */
258static bool do_mov_p(DisasContext *s, int rd, int rn)
259{
d0b2df5a
RH
260 if (sve_access_check(s)) {
261 unsigned psz = pred_gvec_reg_size(s);
262 tcg_gen_gvec_mov(MO_8, pred_full_reg_offset(s, rd),
263 pred_full_reg_offset(s, rn), psz, psz);
264 }
265 return true;
516e246a
RH
266}
267
9e18d7a6
RH
268/* Set the cpu flags as per a return from an SVE helper. */
269static void do_pred_flags(TCGv_i32 t)
270{
271 tcg_gen_mov_i32(cpu_NF, t);
272 tcg_gen_andi_i32(cpu_ZF, t, 2);
273 tcg_gen_andi_i32(cpu_CF, t, 1);
274 tcg_gen_movi_i32(cpu_VF, 0);
275}
276
277/* Subroutines computing the ARM PredTest psuedofunction. */
278static void do_predtest1(TCGv_i64 d, TCGv_i64 g)
279{
280 TCGv_i32 t = tcg_temp_new_i32();
281
282 gen_helper_sve_predtest1(t, d, g);
283 do_pred_flags(t);
284 tcg_temp_free_i32(t);
285}
286
287static void do_predtest(DisasContext *s, int dofs, int gofs, int words)
288{
289 TCGv_ptr dptr = tcg_temp_new_ptr();
290 TCGv_ptr gptr = tcg_temp_new_ptr();
291 TCGv_i32 t;
292
293 tcg_gen_addi_ptr(dptr, cpu_env, dofs);
294 tcg_gen_addi_ptr(gptr, cpu_env, gofs);
295 t = tcg_const_i32(words);
296
297 gen_helper_sve_predtest(t, dptr, gptr, t);
298 tcg_temp_free_ptr(dptr);
299 tcg_temp_free_ptr(gptr);
300
301 do_pred_flags(t);
302 tcg_temp_free_i32(t);
303}
304
028e2a7b
RH
305/* For each element size, the bits within a predicate word that are active. */
306const uint64_t pred_esz_masks[4] = {
307 0xffffffffffffffffull, 0x5555555555555555ull,
308 0x1111111111111111ull, 0x0101010101010101ull
309};
310
39eea561
RH
311/*
312 *** SVE Logical - Unpredicated Group
313 */
314
28c4da31
RH
315static bool do_zzz_fn(DisasContext *s, arg_rrr_esz *a, GVecGen3Fn *gvec_fn)
316{
317 if (sve_access_check(s)) {
318 gen_gvec_fn_zzz(s, gvec_fn, a->esz, a->rd, a->rn, a->rm);
319 }
320 return true;
321}
322
3a7be554 323static bool trans_AND_zzz(DisasContext *s, arg_rrr_esz *a)
39eea561 324{
28c4da31 325 return do_zzz_fn(s, a, tcg_gen_gvec_and);
39eea561
RH
326}
327
3a7be554 328static bool trans_ORR_zzz(DisasContext *s, arg_rrr_esz *a)
39eea561 329{
28c4da31 330 return do_zzz_fn(s, a, tcg_gen_gvec_or);
39eea561
RH
331}
332
3a7be554 333static bool trans_EOR_zzz(DisasContext *s, arg_rrr_esz *a)
39eea561 334{
28c4da31 335 return do_zzz_fn(s, a, tcg_gen_gvec_xor);
39eea561
RH
336}
337
3a7be554 338static bool trans_BIC_zzz(DisasContext *s, arg_rrr_esz *a)
38388f7e 339{
28c4da31 340 return do_zzz_fn(s, a, tcg_gen_gvec_andc);
38388f7e 341}
d1822297 342
e6eba6e5
RH
343static void gen_xar8_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, int64_t sh)
344{
345 TCGv_i64 t = tcg_temp_new_i64();
346 uint64_t mask = dup_const(MO_8, 0xff >> sh);
347
348 tcg_gen_xor_i64(t, n, m);
349 tcg_gen_shri_i64(d, t, sh);
350 tcg_gen_shli_i64(t, t, 8 - sh);
351 tcg_gen_andi_i64(d, d, mask);
352 tcg_gen_andi_i64(t, t, ~mask);
353 tcg_gen_or_i64(d, d, t);
354 tcg_temp_free_i64(t);
355}
356
357static void gen_xar16_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, int64_t sh)
358{
359 TCGv_i64 t = tcg_temp_new_i64();
360 uint64_t mask = dup_const(MO_16, 0xffff >> sh);
361
362 tcg_gen_xor_i64(t, n, m);
363 tcg_gen_shri_i64(d, t, sh);
364 tcg_gen_shli_i64(t, t, 16 - sh);
365 tcg_gen_andi_i64(d, d, mask);
366 tcg_gen_andi_i64(t, t, ~mask);
367 tcg_gen_or_i64(d, d, t);
368 tcg_temp_free_i64(t);
369}
370
371static void gen_xar_i32(TCGv_i32 d, TCGv_i32 n, TCGv_i32 m, int32_t sh)
372{
373 tcg_gen_xor_i32(d, n, m);
374 tcg_gen_rotri_i32(d, d, sh);
375}
376
377static void gen_xar_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, int64_t sh)
378{
379 tcg_gen_xor_i64(d, n, m);
380 tcg_gen_rotri_i64(d, d, sh);
381}
382
383static void gen_xar_vec(unsigned vece, TCGv_vec d, TCGv_vec n,
384 TCGv_vec m, int64_t sh)
385{
386 tcg_gen_xor_vec(vece, d, n, m);
387 tcg_gen_rotri_vec(vece, d, d, sh);
388}
389
390void gen_gvec_xar(unsigned vece, uint32_t rd_ofs, uint32_t rn_ofs,
391 uint32_t rm_ofs, int64_t shift,
392 uint32_t opr_sz, uint32_t max_sz)
393{
394 static const TCGOpcode vecop[] = { INDEX_op_rotli_vec, 0 };
395 static const GVecGen3i ops[4] = {
396 { .fni8 = gen_xar8_i64,
397 .fniv = gen_xar_vec,
398 .fno = gen_helper_sve2_xar_b,
399 .opt_opc = vecop,
400 .vece = MO_8 },
401 { .fni8 = gen_xar16_i64,
402 .fniv = gen_xar_vec,
403 .fno = gen_helper_sve2_xar_h,
404 .opt_opc = vecop,
405 .vece = MO_16 },
406 { .fni4 = gen_xar_i32,
407 .fniv = gen_xar_vec,
408 .fno = gen_helper_sve2_xar_s,
409 .opt_opc = vecop,
410 .vece = MO_32 },
411 { .fni8 = gen_xar_i64,
412 .fniv = gen_xar_vec,
413 .fno = gen_helper_gvec_xar_d,
414 .opt_opc = vecop,
415 .vece = MO_64 }
416 };
417 int esize = 8 << vece;
418
419 /* The SVE2 range is 1 .. esize; the AdvSIMD range is 0 .. esize-1. */
420 tcg_debug_assert(shift >= 0);
421 tcg_debug_assert(shift <= esize);
422 shift &= esize - 1;
423
424 if (shift == 0) {
425 /* xar with no rotate devolves to xor. */
426 tcg_gen_gvec_xor(vece, rd_ofs, rn_ofs, rm_ofs, opr_sz, max_sz);
427 } else {
428 tcg_gen_gvec_3i(rd_ofs, rn_ofs, rm_ofs, opr_sz, max_sz,
429 shift, &ops[vece]);
430 }
431}
432
433static bool trans_XAR(DisasContext *s, arg_rrri_esz *a)
434{
435 if (a->esz < 0 || !dc_isar_feature(aa64_sve2, s)) {
436 return false;
437 }
438 if (sve_access_check(s)) {
439 unsigned vsz = vec_full_reg_size(s);
440 gen_gvec_xar(a->esz, vec_full_reg_offset(s, a->rd),
441 vec_full_reg_offset(s, a->rn),
442 vec_full_reg_offset(s, a->rm), a->imm, vsz, vsz);
443 }
444 return true;
445}
446
911cdc6d
RH
447static bool do_sve2_zzzz_fn(DisasContext *s, arg_rrrr_esz *a, GVecGen4Fn *fn)
448{
449 if (!dc_isar_feature(aa64_sve2, s)) {
450 return false;
451 }
452 if (sve_access_check(s)) {
453 gen_gvec_fn_zzzz(s, fn, a->esz, a->rd, a->rn, a->rm, a->ra);
454 }
455 return true;
456}
457
458static void gen_eor3_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, TCGv_i64 k)
459{
460 tcg_gen_xor_i64(d, n, m);
461 tcg_gen_xor_i64(d, d, k);
462}
463
464static void gen_eor3_vec(unsigned vece, TCGv_vec d, TCGv_vec n,
465 TCGv_vec m, TCGv_vec k)
466{
467 tcg_gen_xor_vec(vece, d, n, m);
468 tcg_gen_xor_vec(vece, d, d, k);
469}
470
471static void gen_eor3(unsigned vece, uint32_t d, uint32_t n, uint32_t m,
472 uint32_t a, uint32_t oprsz, uint32_t maxsz)
473{
474 static const GVecGen4 op = {
475 .fni8 = gen_eor3_i64,
476 .fniv = gen_eor3_vec,
477 .fno = gen_helper_sve2_eor3,
478 .vece = MO_64,
479 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
480 };
481 tcg_gen_gvec_4(d, n, m, a, oprsz, maxsz, &op);
482}
483
484static bool trans_EOR3(DisasContext *s, arg_rrrr_esz *a)
485{
486 return do_sve2_zzzz_fn(s, a, gen_eor3);
487}
488
489static void gen_bcax_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, TCGv_i64 k)
490{
491 tcg_gen_andc_i64(d, m, k);
492 tcg_gen_xor_i64(d, d, n);
493}
494
495static void gen_bcax_vec(unsigned vece, TCGv_vec d, TCGv_vec n,
496 TCGv_vec m, TCGv_vec k)
497{
498 tcg_gen_andc_vec(vece, d, m, k);
499 tcg_gen_xor_vec(vece, d, d, n);
500}
501
502static void gen_bcax(unsigned vece, uint32_t d, uint32_t n, uint32_t m,
503 uint32_t a, uint32_t oprsz, uint32_t maxsz)
504{
505 static const GVecGen4 op = {
506 .fni8 = gen_bcax_i64,
507 .fniv = gen_bcax_vec,
508 .fno = gen_helper_sve2_bcax,
509 .vece = MO_64,
510 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
511 };
512 tcg_gen_gvec_4(d, n, m, a, oprsz, maxsz, &op);
513}
514
515static bool trans_BCAX(DisasContext *s, arg_rrrr_esz *a)
516{
517 return do_sve2_zzzz_fn(s, a, gen_bcax);
518}
519
520static void gen_bsl(unsigned vece, uint32_t d, uint32_t n, uint32_t m,
521 uint32_t a, uint32_t oprsz, uint32_t maxsz)
522{
523 /* BSL differs from the generic bitsel in argument ordering. */
524 tcg_gen_gvec_bitsel(vece, d, a, n, m, oprsz, maxsz);
525}
526
527static bool trans_BSL(DisasContext *s, arg_rrrr_esz *a)
528{
529 return do_sve2_zzzz_fn(s, a, gen_bsl);
530}
531
532static void gen_bsl1n_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, TCGv_i64 k)
533{
534 tcg_gen_andc_i64(n, k, n);
535 tcg_gen_andc_i64(m, m, k);
536 tcg_gen_or_i64(d, n, m);
537}
538
539static void gen_bsl1n_vec(unsigned vece, TCGv_vec d, TCGv_vec n,
540 TCGv_vec m, TCGv_vec k)
541{
542 if (TCG_TARGET_HAS_bitsel_vec) {
543 tcg_gen_not_vec(vece, n, n);
544 tcg_gen_bitsel_vec(vece, d, k, n, m);
545 } else {
546 tcg_gen_andc_vec(vece, n, k, n);
547 tcg_gen_andc_vec(vece, m, m, k);
548 tcg_gen_or_vec(vece, d, n, m);
549 }
550}
551
552static void gen_bsl1n(unsigned vece, uint32_t d, uint32_t n, uint32_t m,
553 uint32_t a, uint32_t oprsz, uint32_t maxsz)
554{
555 static const GVecGen4 op = {
556 .fni8 = gen_bsl1n_i64,
557 .fniv = gen_bsl1n_vec,
558 .fno = gen_helper_sve2_bsl1n,
559 .vece = MO_64,
560 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
561 };
562 tcg_gen_gvec_4(d, n, m, a, oprsz, maxsz, &op);
563}
564
565static bool trans_BSL1N(DisasContext *s, arg_rrrr_esz *a)
566{
567 return do_sve2_zzzz_fn(s, a, gen_bsl1n);
568}
569
570static void gen_bsl2n_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, TCGv_i64 k)
571{
572 /*
573 * Z[dn] = (n & k) | (~m & ~k)
574 * = | ~(m | k)
575 */
576 tcg_gen_and_i64(n, n, k);
577 if (TCG_TARGET_HAS_orc_i64) {
578 tcg_gen_or_i64(m, m, k);
579 tcg_gen_orc_i64(d, n, m);
580 } else {
581 tcg_gen_nor_i64(m, m, k);
582 tcg_gen_or_i64(d, n, m);
583 }
584}
585
586static void gen_bsl2n_vec(unsigned vece, TCGv_vec d, TCGv_vec n,
587 TCGv_vec m, TCGv_vec k)
588{
589 if (TCG_TARGET_HAS_bitsel_vec) {
590 tcg_gen_not_vec(vece, m, m);
591 tcg_gen_bitsel_vec(vece, d, k, n, m);
592 } else {
593 tcg_gen_and_vec(vece, n, n, k);
594 tcg_gen_or_vec(vece, m, m, k);
595 tcg_gen_orc_vec(vece, d, n, m);
596 }
597}
598
599static void gen_bsl2n(unsigned vece, uint32_t d, uint32_t n, uint32_t m,
600 uint32_t a, uint32_t oprsz, uint32_t maxsz)
601{
602 static const GVecGen4 op = {
603 .fni8 = gen_bsl2n_i64,
604 .fniv = gen_bsl2n_vec,
605 .fno = gen_helper_sve2_bsl2n,
606 .vece = MO_64,
607 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
608 };
609 tcg_gen_gvec_4(d, n, m, a, oprsz, maxsz, &op);
610}
611
612static bool trans_BSL2N(DisasContext *s, arg_rrrr_esz *a)
613{
614 return do_sve2_zzzz_fn(s, a, gen_bsl2n);
615}
616
617static void gen_nbsl_i64(TCGv_i64 d, TCGv_i64 n, TCGv_i64 m, TCGv_i64 k)
618{
619 tcg_gen_and_i64(n, n, k);
620 tcg_gen_andc_i64(m, m, k);
621 tcg_gen_nor_i64(d, n, m);
622}
623
624static void gen_nbsl_vec(unsigned vece, TCGv_vec d, TCGv_vec n,
625 TCGv_vec m, TCGv_vec k)
626{
627 tcg_gen_bitsel_vec(vece, d, k, n, m);
628 tcg_gen_not_vec(vece, d, d);
629}
630
631static void gen_nbsl(unsigned vece, uint32_t d, uint32_t n, uint32_t m,
632 uint32_t a, uint32_t oprsz, uint32_t maxsz)
633{
634 static const GVecGen4 op = {
635 .fni8 = gen_nbsl_i64,
636 .fniv = gen_nbsl_vec,
637 .fno = gen_helper_sve2_nbsl,
638 .vece = MO_64,
639 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
640 };
641 tcg_gen_gvec_4(d, n, m, a, oprsz, maxsz, &op);
642}
643
644static bool trans_NBSL(DisasContext *s, arg_rrrr_esz *a)
645{
646 return do_sve2_zzzz_fn(s, a, gen_nbsl);
647}
648
fea98f9c
RH
649/*
650 *** SVE Integer Arithmetic - Unpredicated Group
651 */
652
3a7be554 653static bool trans_ADD_zzz(DisasContext *s, arg_rrr_esz *a)
fea98f9c 654{
28c4da31 655 return do_zzz_fn(s, a, tcg_gen_gvec_add);
fea98f9c
RH
656}
657
3a7be554 658static bool trans_SUB_zzz(DisasContext *s, arg_rrr_esz *a)
fea98f9c 659{
28c4da31 660 return do_zzz_fn(s, a, tcg_gen_gvec_sub);
fea98f9c
RH
661}
662
3a7be554 663static bool trans_SQADD_zzz(DisasContext *s, arg_rrr_esz *a)
fea98f9c 664{
28c4da31 665 return do_zzz_fn(s, a, tcg_gen_gvec_ssadd);
fea98f9c
RH
666}
667
3a7be554 668static bool trans_SQSUB_zzz(DisasContext *s, arg_rrr_esz *a)
fea98f9c 669{
28c4da31 670 return do_zzz_fn(s, a, tcg_gen_gvec_sssub);
fea98f9c
RH
671}
672
3a7be554 673static bool trans_UQADD_zzz(DisasContext *s, arg_rrr_esz *a)
fea98f9c 674{
28c4da31 675 return do_zzz_fn(s, a, tcg_gen_gvec_usadd);
fea98f9c
RH
676}
677
3a7be554 678static bool trans_UQSUB_zzz(DisasContext *s, arg_rrr_esz *a)
fea98f9c 679{
28c4da31 680 return do_zzz_fn(s, a, tcg_gen_gvec_ussub);
fea98f9c
RH
681}
682
f97cfd59
RH
683/*
684 *** SVE Integer Arithmetic - Binary Predicated Group
685 */
686
687static bool do_zpzz_ool(DisasContext *s, arg_rprr_esz *a, gen_helper_gvec_4 *fn)
688{
f97cfd59
RH
689 if (fn == NULL) {
690 return false;
691 }
692 if (sve_access_check(s)) {
36cbb7a8 693 gen_gvec_ool_zzzp(s, fn, a->rd, a->rn, a->rm, a->pg, 0);
f97cfd59
RH
694 }
695 return true;
696}
697
a2103582
RH
698/* Select active elememnts from Zn and inactive elements from Zm,
699 * storing the result in Zd.
700 */
701static void do_sel_z(DisasContext *s, int rd, int rn, int rm, int pg, int esz)
702{
703 static gen_helper_gvec_4 * const fns[4] = {
704 gen_helper_sve_sel_zpzz_b, gen_helper_sve_sel_zpzz_h,
705 gen_helper_sve_sel_zpzz_s, gen_helper_sve_sel_zpzz_d
706 };
36cbb7a8 707 gen_gvec_ool_zzzp(s, fns[esz], rd, rn, rm, pg, 0);
a2103582
RH
708}
709
f97cfd59 710#define DO_ZPZZ(NAME, name) \
3a7be554 711static bool trans_##NAME##_zpzz(DisasContext *s, arg_rprr_esz *a) \
f97cfd59
RH
712{ \
713 static gen_helper_gvec_4 * const fns[4] = { \
714 gen_helper_sve_##name##_zpzz_b, gen_helper_sve_##name##_zpzz_h, \
715 gen_helper_sve_##name##_zpzz_s, gen_helper_sve_##name##_zpzz_d, \
716 }; \
717 return do_zpzz_ool(s, a, fns[a->esz]); \
718}
719
720DO_ZPZZ(AND, and)
721DO_ZPZZ(EOR, eor)
722DO_ZPZZ(ORR, orr)
723DO_ZPZZ(BIC, bic)
724
725DO_ZPZZ(ADD, add)
726DO_ZPZZ(SUB, sub)
727
728DO_ZPZZ(SMAX, smax)
729DO_ZPZZ(UMAX, umax)
730DO_ZPZZ(SMIN, smin)
731DO_ZPZZ(UMIN, umin)
732DO_ZPZZ(SABD, sabd)
733DO_ZPZZ(UABD, uabd)
734
735DO_ZPZZ(MUL, mul)
736DO_ZPZZ(SMULH, smulh)
737DO_ZPZZ(UMULH, umulh)
738
27721dbb
RH
739DO_ZPZZ(ASR, asr)
740DO_ZPZZ(LSR, lsr)
741DO_ZPZZ(LSL, lsl)
742
3a7be554 743static bool trans_SDIV_zpzz(DisasContext *s, arg_rprr_esz *a)
f97cfd59
RH
744{
745 static gen_helper_gvec_4 * const fns[4] = {
746 NULL, NULL, gen_helper_sve_sdiv_zpzz_s, gen_helper_sve_sdiv_zpzz_d
747 };
748 return do_zpzz_ool(s, a, fns[a->esz]);
749}
750
3a7be554 751static bool trans_UDIV_zpzz(DisasContext *s, arg_rprr_esz *a)
f97cfd59
RH
752{
753 static gen_helper_gvec_4 * const fns[4] = {
754 NULL, NULL, gen_helper_sve_udiv_zpzz_s, gen_helper_sve_udiv_zpzz_d
755 };
756 return do_zpzz_ool(s, a, fns[a->esz]);
757}
758
3a7be554 759static bool trans_SEL_zpzz(DisasContext *s, arg_rprr_esz *a)
a2103582
RH
760{
761 if (sve_access_check(s)) {
762 do_sel_z(s, a->rd, a->rn, a->rm, a->pg, a->esz);
763 }
764 return true;
765}
d3fe4a29 766
f97cfd59
RH
767#undef DO_ZPZZ
768
afac6d04
RH
769/*
770 *** SVE Integer Arithmetic - Unary Predicated Group
771 */
772
773static bool do_zpz_ool(DisasContext *s, arg_rpr_esz *a, gen_helper_gvec_3 *fn)
774{
775 if (fn == NULL) {
776 return false;
777 }
778 if (sve_access_check(s)) {
96a461f7 779 gen_gvec_ool_zzp(s, fn, a->rd, a->rn, a->pg, 0);
afac6d04
RH
780 }
781 return true;
782}
783
784#define DO_ZPZ(NAME, name) \
3a7be554 785static bool trans_##NAME(DisasContext *s, arg_rpr_esz *a) \
afac6d04
RH
786{ \
787 static gen_helper_gvec_3 * const fns[4] = { \
788 gen_helper_sve_##name##_b, gen_helper_sve_##name##_h, \
789 gen_helper_sve_##name##_s, gen_helper_sve_##name##_d, \
790 }; \
791 return do_zpz_ool(s, a, fns[a->esz]); \
792}
793
794DO_ZPZ(CLS, cls)
795DO_ZPZ(CLZ, clz)
796DO_ZPZ(CNT_zpz, cnt_zpz)
797DO_ZPZ(CNOT, cnot)
798DO_ZPZ(NOT_zpz, not_zpz)
799DO_ZPZ(ABS, abs)
800DO_ZPZ(NEG, neg)
801
3a7be554 802static bool trans_FABS(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
803{
804 static gen_helper_gvec_3 * const fns[4] = {
805 NULL,
806 gen_helper_sve_fabs_h,
807 gen_helper_sve_fabs_s,
808 gen_helper_sve_fabs_d
809 };
810 return do_zpz_ool(s, a, fns[a->esz]);
811}
812
3a7be554 813static bool trans_FNEG(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
814{
815 static gen_helper_gvec_3 * const fns[4] = {
816 NULL,
817 gen_helper_sve_fneg_h,
818 gen_helper_sve_fneg_s,
819 gen_helper_sve_fneg_d
820 };
821 return do_zpz_ool(s, a, fns[a->esz]);
822}
823
3a7be554 824static bool trans_SXTB(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
825{
826 static gen_helper_gvec_3 * const fns[4] = {
827 NULL,
828 gen_helper_sve_sxtb_h,
829 gen_helper_sve_sxtb_s,
830 gen_helper_sve_sxtb_d
831 };
832 return do_zpz_ool(s, a, fns[a->esz]);
833}
834
3a7be554 835static bool trans_UXTB(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
836{
837 static gen_helper_gvec_3 * const fns[4] = {
838 NULL,
839 gen_helper_sve_uxtb_h,
840 gen_helper_sve_uxtb_s,
841 gen_helper_sve_uxtb_d
842 };
843 return do_zpz_ool(s, a, fns[a->esz]);
844}
845
3a7be554 846static bool trans_SXTH(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
847{
848 static gen_helper_gvec_3 * const fns[4] = {
849 NULL, NULL,
850 gen_helper_sve_sxth_s,
851 gen_helper_sve_sxth_d
852 };
853 return do_zpz_ool(s, a, fns[a->esz]);
854}
855
3a7be554 856static bool trans_UXTH(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
857{
858 static gen_helper_gvec_3 * const fns[4] = {
859 NULL, NULL,
860 gen_helper_sve_uxth_s,
861 gen_helper_sve_uxth_d
862 };
863 return do_zpz_ool(s, a, fns[a->esz]);
864}
865
3a7be554 866static bool trans_SXTW(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
867{
868 return do_zpz_ool(s, a, a->esz == 3 ? gen_helper_sve_sxtw_d : NULL);
869}
870
3a7be554 871static bool trans_UXTW(DisasContext *s, arg_rpr_esz *a)
afac6d04
RH
872{
873 return do_zpz_ool(s, a, a->esz == 3 ? gen_helper_sve_uxtw_d : NULL);
874}
875
876#undef DO_ZPZ
877
047cec97
RH
878/*
879 *** SVE Integer Reduction Group
880 */
881
882typedef void gen_helper_gvec_reduc(TCGv_i64, TCGv_ptr, TCGv_ptr, TCGv_i32);
883static bool do_vpz_ool(DisasContext *s, arg_rpr_esz *a,
884 gen_helper_gvec_reduc *fn)
885{
886 unsigned vsz = vec_full_reg_size(s);
887 TCGv_ptr t_zn, t_pg;
888 TCGv_i32 desc;
889 TCGv_i64 temp;
890
891 if (fn == NULL) {
892 return false;
893 }
894 if (!sve_access_check(s)) {
895 return true;
896 }
897
898 desc = tcg_const_i32(simd_desc(vsz, vsz, 0));
899 temp = tcg_temp_new_i64();
900 t_zn = tcg_temp_new_ptr();
901 t_pg = tcg_temp_new_ptr();
902
903 tcg_gen_addi_ptr(t_zn, cpu_env, vec_full_reg_offset(s, a->rn));
904 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, a->pg));
905 fn(temp, t_zn, t_pg, desc);
906 tcg_temp_free_ptr(t_zn);
907 tcg_temp_free_ptr(t_pg);
908 tcg_temp_free_i32(desc);
909
910 write_fp_dreg(s, a->rd, temp);
911 tcg_temp_free_i64(temp);
912 return true;
913}
914
915#define DO_VPZ(NAME, name) \
3a7be554 916static bool trans_##NAME(DisasContext *s, arg_rpr_esz *a) \
047cec97
RH
917{ \
918 static gen_helper_gvec_reduc * const fns[4] = { \
919 gen_helper_sve_##name##_b, gen_helper_sve_##name##_h, \
920 gen_helper_sve_##name##_s, gen_helper_sve_##name##_d, \
921 }; \
922 return do_vpz_ool(s, a, fns[a->esz]); \
923}
924
925DO_VPZ(ORV, orv)
926DO_VPZ(ANDV, andv)
927DO_VPZ(EORV, eorv)
928
929DO_VPZ(UADDV, uaddv)
930DO_VPZ(SMAXV, smaxv)
931DO_VPZ(UMAXV, umaxv)
932DO_VPZ(SMINV, sminv)
933DO_VPZ(UMINV, uminv)
934
3a7be554 935static bool trans_SADDV(DisasContext *s, arg_rpr_esz *a)
047cec97
RH
936{
937 static gen_helper_gvec_reduc * const fns[4] = {
938 gen_helper_sve_saddv_b, gen_helper_sve_saddv_h,
939 gen_helper_sve_saddv_s, NULL
940 };
941 return do_vpz_ool(s, a, fns[a->esz]);
942}
943
944#undef DO_VPZ
945
ccd841c3
RH
946/*
947 *** SVE Shift by Immediate - Predicated Group
948 */
949
60245996
RH
950/*
951 * Copy Zn into Zd, storing zeros into inactive elements.
952 * If invert, store zeros into the active elements.
ccd841c3 953 */
60245996
RH
954static bool do_movz_zpz(DisasContext *s, int rd, int rn, int pg,
955 int esz, bool invert)
ccd841c3 956{
60245996
RH
957 static gen_helper_gvec_3 * const fns[4] = {
958 gen_helper_sve_movz_b, gen_helper_sve_movz_h,
959 gen_helper_sve_movz_s, gen_helper_sve_movz_d,
ccd841c3 960 };
60245996 961
ccd841c3 962 if (sve_access_check(s)) {
96a461f7 963 gen_gvec_ool_zzp(s, fns[esz], rd, rn, pg, invert);
ccd841c3
RH
964 }
965 return true;
966}
967
968static bool do_zpzi_ool(DisasContext *s, arg_rpri_esz *a,
969 gen_helper_gvec_3 *fn)
970{
971 if (sve_access_check(s)) {
96a461f7 972 gen_gvec_ool_zzp(s, fn, a->rd, a->rn, a->pg, a->imm);
ccd841c3
RH
973 }
974 return true;
975}
976
3a7be554 977static bool trans_ASR_zpzi(DisasContext *s, arg_rpri_esz *a)
ccd841c3
RH
978{
979 static gen_helper_gvec_3 * const fns[4] = {
980 gen_helper_sve_asr_zpzi_b, gen_helper_sve_asr_zpzi_h,
981 gen_helper_sve_asr_zpzi_s, gen_helper_sve_asr_zpzi_d,
982 };
983 if (a->esz < 0) {
984 /* Invalid tsz encoding -- see tszimm_esz. */
985 return false;
986 }
987 /* Shift by element size is architecturally valid. For
988 arithmetic right-shift, it's the same as by one less. */
989 a->imm = MIN(a->imm, (8 << a->esz) - 1);
990 return do_zpzi_ool(s, a, fns[a->esz]);
991}
992
3a7be554 993static bool trans_LSR_zpzi(DisasContext *s, arg_rpri_esz *a)
ccd841c3
RH
994{
995 static gen_helper_gvec_3 * const fns[4] = {
996 gen_helper_sve_lsr_zpzi_b, gen_helper_sve_lsr_zpzi_h,
997 gen_helper_sve_lsr_zpzi_s, gen_helper_sve_lsr_zpzi_d,
998 };
999 if (a->esz < 0) {
1000 return false;
1001 }
1002 /* Shift by element size is architecturally valid.
1003 For logical shifts, it is a zeroing operation. */
1004 if (a->imm >= (8 << a->esz)) {
60245996 1005 return do_movz_zpz(s, a->rd, a->rd, a->pg, a->esz, true);
ccd841c3
RH
1006 } else {
1007 return do_zpzi_ool(s, a, fns[a->esz]);
1008 }
1009}
1010
3a7be554 1011static bool trans_LSL_zpzi(DisasContext *s, arg_rpri_esz *a)
ccd841c3
RH
1012{
1013 static gen_helper_gvec_3 * const fns[4] = {
1014 gen_helper_sve_lsl_zpzi_b, gen_helper_sve_lsl_zpzi_h,
1015 gen_helper_sve_lsl_zpzi_s, gen_helper_sve_lsl_zpzi_d,
1016 };
1017 if (a->esz < 0) {
1018 return false;
1019 }
1020 /* Shift by element size is architecturally valid.
1021 For logical shifts, it is a zeroing operation. */
1022 if (a->imm >= (8 << a->esz)) {
60245996 1023 return do_movz_zpz(s, a->rd, a->rd, a->pg, a->esz, true);
ccd841c3
RH
1024 } else {
1025 return do_zpzi_ool(s, a, fns[a->esz]);
1026 }
1027}
1028
3a7be554 1029static bool trans_ASRD(DisasContext *s, arg_rpri_esz *a)
ccd841c3
RH
1030{
1031 static gen_helper_gvec_3 * const fns[4] = {
1032 gen_helper_sve_asrd_b, gen_helper_sve_asrd_h,
1033 gen_helper_sve_asrd_s, gen_helper_sve_asrd_d,
1034 };
1035 if (a->esz < 0) {
1036 return false;
1037 }
1038 /* Shift by element size is architecturally valid. For arithmetic
1039 right shift for division, it is a zeroing operation. */
1040 if (a->imm >= (8 << a->esz)) {
60245996 1041 return do_movz_zpz(s, a->rd, a->rd, a->pg, a->esz, true);
ccd841c3
RH
1042 } else {
1043 return do_zpzi_ool(s, a, fns[a->esz]);
1044 }
1045}
1046
fe7f8dfb
RH
1047/*
1048 *** SVE Bitwise Shift - Predicated Group
1049 */
1050
1051#define DO_ZPZW(NAME, name) \
3a7be554 1052static bool trans_##NAME##_zpzw(DisasContext *s, arg_rprr_esz *a) \
fe7f8dfb
RH
1053{ \
1054 static gen_helper_gvec_4 * const fns[3] = { \
1055 gen_helper_sve_##name##_zpzw_b, gen_helper_sve_##name##_zpzw_h, \
1056 gen_helper_sve_##name##_zpzw_s, \
1057 }; \
1058 if (a->esz < 0 || a->esz >= 3) { \
1059 return false; \
1060 } \
1061 return do_zpzz_ool(s, a, fns[a->esz]); \
1062}
1063
1064DO_ZPZW(ASR, asr)
1065DO_ZPZW(LSR, lsr)
1066DO_ZPZW(LSL, lsl)
1067
1068#undef DO_ZPZW
1069
d9d78dcc
RH
1070/*
1071 *** SVE Bitwise Shift - Unpredicated Group
1072 */
1073
1074static bool do_shift_imm(DisasContext *s, arg_rri_esz *a, bool asr,
1075 void (*gvec_fn)(unsigned, uint32_t, uint32_t,
1076 int64_t, uint32_t, uint32_t))
1077{
1078 if (a->esz < 0) {
1079 /* Invalid tsz encoding -- see tszimm_esz. */
1080 return false;
1081 }
1082 if (sve_access_check(s)) {
1083 unsigned vsz = vec_full_reg_size(s);
1084 /* Shift by element size is architecturally valid. For
1085 arithmetic right-shift, it's the same as by one less.
1086 Otherwise it is a zeroing operation. */
1087 if (a->imm >= 8 << a->esz) {
1088 if (asr) {
1089 a->imm = (8 << a->esz) - 1;
1090 } else {
1091 do_dupi_z(s, a->rd, 0);
1092 return true;
1093 }
1094 }
1095 gvec_fn(a->esz, vec_full_reg_offset(s, a->rd),
1096 vec_full_reg_offset(s, a->rn), a->imm, vsz, vsz);
1097 }
1098 return true;
1099}
1100
3a7be554 1101static bool trans_ASR_zzi(DisasContext *s, arg_rri_esz *a)
d9d78dcc
RH
1102{
1103 return do_shift_imm(s, a, true, tcg_gen_gvec_sari);
1104}
1105
3a7be554 1106static bool trans_LSR_zzi(DisasContext *s, arg_rri_esz *a)
d9d78dcc
RH
1107{
1108 return do_shift_imm(s, a, false, tcg_gen_gvec_shri);
1109}
1110
3a7be554 1111static bool trans_LSL_zzi(DisasContext *s, arg_rri_esz *a)
d9d78dcc
RH
1112{
1113 return do_shift_imm(s, a, false, tcg_gen_gvec_shli);
1114}
1115
1116static bool do_zzw_ool(DisasContext *s, arg_rrr_esz *a, gen_helper_gvec_3 *fn)
1117{
1118 if (fn == NULL) {
1119 return false;
1120 }
1121 if (sve_access_check(s)) {
e645d1a1 1122 gen_gvec_ool_zzz(s, fn, a->rd, a->rn, a->rm, 0);
d9d78dcc
RH
1123 }
1124 return true;
1125}
1126
1127#define DO_ZZW(NAME, name) \
3a7be554 1128static bool trans_##NAME##_zzw(DisasContext *s, arg_rrr_esz *a) \
d9d78dcc
RH
1129{ \
1130 static gen_helper_gvec_3 * const fns[4] = { \
1131 gen_helper_sve_##name##_zzw_b, gen_helper_sve_##name##_zzw_h, \
1132 gen_helper_sve_##name##_zzw_s, NULL \
1133 }; \
1134 return do_zzw_ool(s, a, fns[a->esz]); \
1135}
1136
1137DO_ZZW(ASR, asr)
1138DO_ZZW(LSR, lsr)
1139DO_ZZW(LSL, lsl)
1140
1141#undef DO_ZZW
1142
96a36e4a
RH
1143/*
1144 *** SVE Integer Multiply-Add Group
1145 */
1146
1147static bool do_zpzzz_ool(DisasContext *s, arg_rprrr_esz *a,
1148 gen_helper_gvec_5 *fn)
1149{
1150 if (sve_access_check(s)) {
1151 unsigned vsz = vec_full_reg_size(s);
1152 tcg_gen_gvec_5_ool(vec_full_reg_offset(s, a->rd),
1153 vec_full_reg_offset(s, a->ra),
1154 vec_full_reg_offset(s, a->rn),
1155 vec_full_reg_offset(s, a->rm),
1156 pred_full_reg_offset(s, a->pg),
1157 vsz, vsz, 0, fn);
1158 }
1159 return true;
1160}
1161
1162#define DO_ZPZZZ(NAME, name) \
3a7be554 1163static bool trans_##NAME(DisasContext *s, arg_rprrr_esz *a) \
96a36e4a
RH
1164{ \
1165 static gen_helper_gvec_5 * const fns[4] = { \
1166 gen_helper_sve_##name##_b, gen_helper_sve_##name##_h, \
1167 gen_helper_sve_##name##_s, gen_helper_sve_##name##_d, \
1168 }; \
1169 return do_zpzzz_ool(s, a, fns[a->esz]); \
1170}
1171
1172DO_ZPZZZ(MLA, mla)
1173DO_ZPZZZ(MLS, mls)
1174
1175#undef DO_ZPZZZ
1176
9a56c9c3
RH
1177/*
1178 *** SVE Index Generation Group
1179 */
1180
1181static void do_index(DisasContext *s, int esz, int rd,
1182 TCGv_i64 start, TCGv_i64 incr)
1183{
1184 unsigned vsz = vec_full_reg_size(s);
1185 TCGv_i32 desc = tcg_const_i32(simd_desc(vsz, vsz, 0));
1186 TCGv_ptr t_zd = tcg_temp_new_ptr();
1187
1188 tcg_gen_addi_ptr(t_zd, cpu_env, vec_full_reg_offset(s, rd));
1189 if (esz == 3) {
1190 gen_helper_sve_index_d(t_zd, start, incr, desc);
1191 } else {
1192 typedef void index_fn(TCGv_ptr, TCGv_i32, TCGv_i32, TCGv_i32);
1193 static index_fn * const fns[3] = {
1194 gen_helper_sve_index_b,
1195 gen_helper_sve_index_h,
1196 gen_helper_sve_index_s,
1197 };
1198 TCGv_i32 s32 = tcg_temp_new_i32();
1199 TCGv_i32 i32 = tcg_temp_new_i32();
1200
1201 tcg_gen_extrl_i64_i32(s32, start);
1202 tcg_gen_extrl_i64_i32(i32, incr);
1203 fns[esz](t_zd, s32, i32, desc);
1204
1205 tcg_temp_free_i32(s32);
1206 tcg_temp_free_i32(i32);
1207 }
1208 tcg_temp_free_ptr(t_zd);
1209 tcg_temp_free_i32(desc);
1210}
1211
3a7be554 1212static bool trans_INDEX_ii(DisasContext *s, arg_INDEX_ii *a)
9a56c9c3
RH
1213{
1214 if (sve_access_check(s)) {
1215 TCGv_i64 start = tcg_const_i64(a->imm1);
1216 TCGv_i64 incr = tcg_const_i64(a->imm2);
1217 do_index(s, a->esz, a->rd, start, incr);
1218 tcg_temp_free_i64(start);
1219 tcg_temp_free_i64(incr);
1220 }
1221 return true;
1222}
1223
3a7be554 1224static bool trans_INDEX_ir(DisasContext *s, arg_INDEX_ir *a)
9a56c9c3
RH
1225{
1226 if (sve_access_check(s)) {
1227 TCGv_i64 start = tcg_const_i64(a->imm);
1228 TCGv_i64 incr = cpu_reg(s, a->rm);
1229 do_index(s, a->esz, a->rd, start, incr);
1230 tcg_temp_free_i64(start);
1231 }
1232 return true;
1233}
1234
3a7be554 1235static bool trans_INDEX_ri(DisasContext *s, arg_INDEX_ri *a)
9a56c9c3
RH
1236{
1237 if (sve_access_check(s)) {
1238 TCGv_i64 start = cpu_reg(s, a->rn);
1239 TCGv_i64 incr = tcg_const_i64(a->imm);
1240 do_index(s, a->esz, a->rd, start, incr);
1241 tcg_temp_free_i64(incr);
1242 }
1243 return true;
1244}
1245
3a7be554 1246static bool trans_INDEX_rr(DisasContext *s, arg_INDEX_rr *a)
9a56c9c3
RH
1247{
1248 if (sve_access_check(s)) {
1249 TCGv_i64 start = cpu_reg(s, a->rn);
1250 TCGv_i64 incr = cpu_reg(s, a->rm);
1251 do_index(s, a->esz, a->rd, start, incr);
1252 }
1253 return true;
1254}
1255
96f922cc
RH
1256/*
1257 *** SVE Stack Allocation Group
1258 */
1259
3a7be554 1260static bool trans_ADDVL(DisasContext *s, arg_ADDVL *a)
96f922cc 1261{
5de56742
AC
1262 if (sve_access_check(s)) {
1263 TCGv_i64 rd = cpu_reg_sp(s, a->rd);
1264 TCGv_i64 rn = cpu_reg_sp(s, a->rn);
1265 tcg_gen_addi_i64(rd, rn, a->imm * vec_full_reg_size(s));
1266 }
96f922cc
RH
1267 return true;
1268}
1269
3a7be554 1270static bool trans_ADDPL(DisasContext *s, arg_ADDPL *a)
96f922cc 1271{
5de56742
AC
1272 if (sve_access_check(s)) {
1273 TCGv_i64 rd = cpu_reg_sp(s, a->rd);
1274 TCGv_i64 rn = cpu_reg_sp(s, a->rn);
1275 tcg_gen_addi_i64(rd, rn, a->imm * pred_full_reg_size(s));
1276 }
96f922cc
RH
1277 return true;
1278}
1279
3a7be554 1280static bool trans_RDVL(DisasContext *s, arg_RDVL *a)
96f922cc 1281{
5de56742
AC
1282 if (sve_access_check(s)) {
1283 TCGv_i64 reg = cpu_reg(s, a->rd);
1284 tcg_gen_movi_i64(reg, a->imm * vec_full_reg_size(s));
1285 }
96f922cc
RH
1286 return true;
1287}
1288
4b242d9c
RH
1289/*
1290 *** SVE Compute Vector Address Group
1291 */
1292
1293static bool do_adr(DisasContext *s, arg_rrri *a, gen_helper_gvec_3 *fn)
1294{
1295 if (sve_access_check(s)) {
e645d1a1 1296 gen_gvec_ool_zzz(s, fn, a->rd, a->rn, a->rm, a->imm);
4b242d9c
RH
1297 }
1298 return true;
1299}
1300
3a7be554 1301static bool trans_ADR_p32(DisasContext *s, arg_rrri *a)
4b242d9c
RH
1302{
1303 return do_adr(s, a, gen_helper_sve_adr_p32);
1304}
1305
3a7be554 1306static bool trans_ADR_p64(DisasContext *s, arg_rrri *a)
4b242d9c
RH
1307{
1308 return do_adr(s, a, gen_helper_sve_adr_p64);
1309}
1310
3a7be554 1311static bool trans_ADR_s32(DisasContext *s, arg_rrri *a)
4b242d9c
RH
1312{
1313 return do_adr(s, a, gen_helper_sve_adr_s32);
1314}
1315
3a7be554 1316static bool trans_ADR_u32(DisasContext *s, arg_rrri *a)
4b242d9c
RH
1317{
1318 return do_adr(s, a, gen_helper_sve_adr_u32);
1319}
1320
0762cd42
RH
1321/*
1322 *** SVE Integer Misc - Unpredicated Group
1323 */
1324
3a7be554 1325static bool trans_FEXPA(DisasContext *s, arg_rr_esz *a)
0762cd42
RH
1326{
1327 static gen_helper_gvec_2 * const fns[4] = {
1328 NULL,
1329 gen_helper_sve_fexpa_h,
1330 gen_helper_sve_fexpa_s,
1331 gen_helper_sve_fexpa_d,
1332 };
1333 if (a->esz == 0) {
1334 return false;
1335 }
1336 if (sve_access_check(s)) {
40e32e5a 1337 gen_gvec_ool_zz(s, fns[a->esz], a->rd, a->rn, 0);
0762cd42
RH
1338 }
1339 return true;
1340}
1341
3a7be554 1342static bool trans_FTSSEL(DisasContext *s, arg_rrr_esz *a)
a1f233f2
RH
1343{
1344 static gen_helper_gvec_3 * const fns[4] = {
1345 NULL,
1346 gen_helper_sve_ftssel_h,
1347 gen_helper_sve_ftssel_s,
1348 gen_helper_sve_ftssel_d,
1349 };
1350 if (a->esz == 0) {
1351 return false;
1352 }
1353 if (sve_access_check(s)) {
e645d1a1 1354 gen_gvec_ool_zzz(s, fns[a->esz], a->rd, a->rn, a->rm, 0);
a1f233f2
RH
1355 }
1356 return true;
1357}
1358
516e246a
RH
1359/*
1360 *** SVE Predicate Logical Operations Group
1361 */
1362
1363static bool do_pppp_flags(DisasContext *s, arg_rprr_s *a,
1364 const GVecGen4 *gvec_op)
1365{
1366 if (!sve_access_check(s)) {
1367 return true;
1368 }
1369
1370 unsigned psz = pred_gvec_reg_size(s);
1371 int dofs = pred_full_reg_offset(s, a->rd);
1372 int nofs = pred_full_reg_offset(s, a->rn);
1373 int mofs = pred_full_reg_offset(s, a->rm);
1374 int gofs = pred_full_reg_offset(s, a->pg);
1375
dd81a8d7
RH
1376 if (!a->s) {
1377 tcg_gen_gvec_4(dofs, nofs, mofs, gofs, psz, psz, gvec_op);
1378 return true;
1379 }
1380
516e246a
RH
1381 if (psz == 8) {
1382 /* Do the operation and the flags generation in temps. */
1383 TCGv_i64 pd = tcg_temp_new_i64();
1384 TCGv_i64 pn = tcg_temp_new_i64();
1385 TCGv_i64 pm = tcg_temp_new_i64();
1386 TCGv_i64 pg = tcg_temp_new_i64();
1387
1388 tcg_gen_ld_i64(pn, cpu_env, nofs);
1389 tcg_gen_ld_i64(pm, cpu_env, mofs);
1390 tcg_gen_ld_i64(pg, cpu_env, gofs);
1391
1392 gvec_op->fni8(pd, pn, pm, pg);
1393 tcg_gen_st_i64(pd, cpu_env, dofs);
1394
1395 do_predtest1(pd, pg);
1396
1397 tcg_temp_free_i64(pd);
1398 tcg_temp_free_i64(pn);
1399 tcg_temp_free_i64(pm);
1400 tcg_temp_free_i64(pg);
1401 } else {
1402 /* The operation and flags generation is large. The computation
1403 * of the flags depends on the original contents of the guarding
1404 * predicate. If the destination overwrites the guarding predicate,
1405 * then the easiest way to get this right is to save a copy.
1406 */
1407 int tofs = gofs;
1408 if (a->rd == a->pg) {
1409 tofs = offsetof(CPUARMState, vfp.preg_tmp);
1410 tcg_gen_gvec_mov(0, tofs, gofs, psz, psz);
1411 }
1412
1413 tcg_gen_gvec_4(dofs, nofs, mofs, gofs, psz, psz, gvec_op);
1414 do_predtest(s, dofs, tofs, psz / 8);
1415 }
1416 return true;
1417}
1418
1419static void gen_and_pg_i64(TCGv_i64 pd, TCGv_i64 pn, TCGv_i64 pm, TCGv_i64 pg)
1420{
1421 tcg_gen_and_i64(pd, pn, pm);
1422 tcg_gen_and_i64(pd, pd, pg);
1423}
1424
1425static void gen_and_pg_vec(unsigned vece, TCGv_vec pd, TCGv_vec pn,
1426 TCGv_vec pm, TCGv_vec pg)
1427{
1428 tcg_gen_and_vec(vece, pd, pn, pm);
1429 tcg_gen_and_vec(vece, pd, pd, pg);
1430}
1431
3a7be554 1432static bool trans_AND_pppp(DisasContext *s, arg_rprr_s *a)
516e246a
RH
1433{
1434 static const GVecGen4 op = {
1435 .fni8 = gen_and_pg_i64,
1436 .fniv = gen_and_pg_vec,
1437 .fno = gen_helper_sve_and_pppp,
1438 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
1439 };
dd81a8d7
RH
1440
1441 if (!a->s) {
1442 if (!sve_access_check(s)) {
1443 return true;
1444 }
1445 if (a->rn == a->rm) {
1446 if (a->pg == a->rn) {
1447 do_mov_p(s, a->rd, a->rn);
1448 } else {
1449 gen_gvec_fn_ppp(s, tcg_gen_gvec_and, a->rd, a->rn, a->pg);
1450 }
1451 return true;
1452 } else if (a->pg == a->rn || a->pg == a->rm) {
1453 gen_gvec_fn_ppp(s, tcg_gen_gvec_and, a->rd, a->rn, a->rm);
1454 return true;
516e246a 1455 }
516e246a 1456 }
dd81a8d7 1457 return do_pppp_flags(s, a, &op);
516e246a
RH
1458}
1459
1460static void gen_bic_pg_i64(TCGv_i64 pd, TCGv_i64 pn, TCGv_i64 pm, TCGv_i64 pg)
1461{
1462 tcg_gen_andc_i64(pd, pn, pm);
1463 tcg_gen_and_i64(pd, pd, pg);
1464}
1465
1466static void gen_bic_pg_vec(unsigned vece, TCGv_vec pd, TCGv_vec pn,
1467 TCGv_vec pm, TCGv_vec pg)
1468{
1469 tcg_gen_andc_vec(vece, pd, pn, pm);
1470 tcg_gen_and_vec(vece, pd, pd, pg);
1471}
1472
3a7be554 1473static bool trans_BIC_pppp(DisasContext *s, arg_rprr_s *a)
516e246a
RH
1474{
1475 static const GVecGen4 op = {
1476 .fni8 = gen_bic_pg_i64,
1477 .fniv = gen_bic_pg_vec,
1478 .fno = gen_helper_sve_bic_pppp,
1479 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
1480 };
dd81a8d7
RH
1481
1482 if (!a->s && a->pg == a->rn) {
1483 if (sve_access_check(s)) {
1484 gen_gvec_fn_ppp(s, tcg_gen_gvec_andc, a->rd, a->rn, a->rm);
1485 }
1486 return true;
516e246a 1487 }
dd81a8d7 1488 return do_pppp_flags(s, a, &op);
516e246a
RH
1489}
1490
1491static void gen_eor_pg_i64(TCGv_i64 pd, TCGv_i64 pn, TCGv_i64 pm, TCGv_i64 pg)
1492{
1493 tcg_gen_xor_i64(pd, pn, pm);
1494 tcg_gen_and_i64(pd, pd, pg);
1495}
1496
1497static void gen_eor_pg_vec(unsigned vece, TCGv_vec pd, TCGv_vec pn,
1498 TCGv_vec pm, TCGv_vec pg)
1499{
1500 tcg_gen_xor_vec(vece, pd, pn, pm);
1501 tcg_gen_and_vec(vece, pd, pd, pg);
1502}
1503
3a7be554 1504static bool trans_EOR_pppp(DisasContext *s, arg_rprr_s *a)
516e246a
RH
1505{
1506 static const GVecGen4 op = {
1507 .fni8 = gen_eor_pg_i64,
1508 .fniv = gen_eor_pg_vec,
1509 .fno = gen_helper_sve_eor_pppp,
1510 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
1511 };
dd81a8d7 1512 return do_pppp_flags(s, a, &op);
516e246a
RH
1513}
1514
3a7be554 1515static bool trans_SEL_pppp(DisasContext *s, arg_rprr_s *a)
516e246a 1516{
516e246a
RH
1517 if (a->s) {
1518 return false;
516e246a 1519 }
d4bc6232
RH
1520 if (sve_access_check(s)) {
1521 unsigned psz = pred_gvec_reg_size(s);
1522 tcg_gen_gvec_bitsel(MO_8, pred_full_reg_offset(s, a->rd),
1523 pred_full_reg_offset(s, a->pg),
1524 pred_full_reg_offset(s, a->rn),
1525 pred_full_reg_offset(s, a->rm), psz, psz);
1526 }
1527 return true;
516e246a
RH
1528}
1529
1530static void gen_orr_pg_i64(TCGv_i64 pd, TCGv_i64 pn, TCGv_i64 pm, TCGv_i64 pg)
1531{
1532 tcg_gen_or_i64(pd, pn, pm);
1533 tcg_gen_and_i64(pd, pd, pg);
1534}
1535
1536static void gen_orr_pg_vec(unsigned vece, TCGv_vec pd, TCGv_vec pn,
1537 TCGv_vec pm, TCGv_vec pg)
1538{
1539 tcg_gen_or_vec(vece, pd, pn, pm);
1540 tcg_gen_and_vec(vece, pd, pd, pg);
1541}
1542
3a7be554 1543static bool trans_ORR_pppp(DisasContext *s, arg_rprr_s *a)
516e246a
RH
1544{
1545 static const GVecGen4 op = {
1546 .fni8 = gen_orr_pg_i64,
1547 .fniv = gen_orr_pg_vec,
1548 .fno = gen_helper_sve_orr_pppp,
1549 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
1550 };
dd81a8d7
RH
1551
1552 if (!a->s && a->pg == a->rn && a->rn == a->rm) {
516e246a 1553 return do_mov_p(s, a->rd, a->rn);
516e246a 1554 }
dd81a8d7 1555 return do_pppp_flags(s, a, &op);
516e246a
RH
1556}
1557
1558static void gen_orn_pg_i64(TCGv_i64 pd, TCGv_i64 pn, TCGv_i64 pm, TCGv_i64 pg)
1559{
1560 tcg_gen_orc_i64(pd, pn, pm);
1561 tcg_gen_and_i64(pd, pd, pg);
1562}
1563
1564static void gen_orn_pg_vec(unsigned vece, TCGv_vec pd, TCGv_vec pn,
1565 TCGv_vec pm, TCGv_vec pg)
1566{
1567 tcg_gen_orc_vec(vece, pd, pn, pm);
1568 tcg_gen_and_vec(vece, pd, pd, pg);
1569}
1570
3a7be554 1571static bool trans_ORN_pppp(DisasContext *s, arg_rprr_s *a)
516e246a
RH
1572{
1573 static const GVecGen4 op = {
1574 .fni8 = gen_orn_pg_i64,
1575 .fniv = gen_orn_pg_vec,
1576 .fno = gen_helper_sve_orn_pppp,
1577 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
1578 };
dd81a8d7 1579 return do_pppp_flags(s, a, &op);
516e246a
RH
1580}
1581
1582static void gen_nor_pg_i64(TCGv_i64 pd, TCGv_i64 pn, TCGv_i64 pm, TCGv_i64 pg)
1583{
1584 tcg_gen_or_i64(pd, pn, pm);
1585 tcg_gen_andc_i64(pd, pg, pd);
1586}
1587
1588static void gen_nor_pg_vec(unsigned vece, TCGv_vec pd, TCGv_vec pn,
1589 TCGv_vec pm, TCGv_vec pg)
1590{
1591 tcg_gen_or_vec(vece, pd, pn, pm);
1592 tcg_gen_andc_vec(vece, pd, pg, pd);
1593}
1594
3a7be554 1595static bool trans_NOR_pppp(DisasContext *s, arg_rprr_s *a)
516e246a
RH
1596{
1597 static const GVecGen4 op = {
1598 .fni8 = gen_nor_pg_i64,
1599 .fniv = gen_nor_pg_vec,
1600 .fno = gen_helper_sve_nor_pppp,
1601 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
1602 };
dd81a8d7 1603 return do_pppp_flags(s, a, &op);
516e246a
RH
1604}
1605
1606static void gen_nand_pg_i64(TCGv_i64 pd, TCGv_i64 pn, TCGv_i64 pm, TCGv_i64 pg)
1607{
1608 tcg_gen_and_i64(pd, pn, pm);
1609 tcg_gen_andc_i64(pd, pg, pd);
1610}
1611
1612static void gen_nand_pg_vec(unsigned vece, TCGv_vec pd, TCGv_vec pn,
1613 TCGv_vec pm, TCGv_vec pg)
1614{
1615 tcg_gen_and_vec(vece, pd, pn, pm);
1616 tcg_gen_andc_vec(vece, pd, pg, pd);
1617}
1618
3a7be554 1619static bool trans_NAND_pppp(DisasContext *s, arg_rprr_s *a)
516e246a
RH
1620{
1621 static const GVecGen4 op = {
1622 .fni8 = gen_nand_pg_i64,
1623 .fniv = gen_nand_pg_vec,
1624 .fno = gen_helper_sve_nand_pppp,
1625 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
1626 };
dd81a8d7 1627 return do_pppp_flags(s, a, &op);
516e246a
RH
1628}
1629
9e18d7a6
RH
1630/*
1631 *** SVE Predicate Misc Group
1632 */
1633
3a7be554 1634static bool trans_PTEST(DisasContext *s, arg_PTEST *a)
9e18d7a6
RH
1635{
1636 if (sve_access_check(s)) {
1637 int nofs = pred_full_reg_offset(s, a->rn);
1638 int gofs = pred_full_reg_offset(s, a->pg);
1639 int words = DIV_ROUND_UP(pred_full_reg_size(s), 8);
1640
1641 if (words == 1) {
1642 TCGv_i64 pn = tcg_temp_new_i64();
1643 TCGv_i64 pg = tcg_temp_new_i64();
1644
1645 tcg_gen_ld_i64(pn, cpu_env, nofs);
1646 tcg_gen_ld_i64(pg, cpu_env, gofs);
1647 do_predtest1(pn, pg);
1648
1649 tcg_temp_free_i64(pn);
1650 tcg_temp_free_i64(pg);
1651 } else {
1652 do_predtest(s, nofs, gofs, words);
1653 }
1654 }
1655 return true;
1656}
1657
028e2a7b
RH
1658/* See the ARM pseudocode DecodePredCount. */
1659static unsigned decode_pred_count(unsigned fullsz, int pattern, int esz)
1660{
1661 unsigned elements = fullsz >> esz;
1662 unsigned bound;
1663
1664 switch (pattern) {
1665 case 0x0: /* POW2 */
1666 return pow2floor(elements);
1667 case 0x1: /* VL1 */
1668 case 0x2: /* VL2 */
1669 case 0x3: /* VL3 */
1670 case 0x4: /* VL4 */
1671 case 0x5: /* VL5 */
1672 case 0x6: /* VL6 */
1673 case 0x7: /* VL7 */
1674 case 0x8: /* VL8 */
1675 bound = pattern;
1676 break;
1677 case 0x9: /* VL16 */
1678 case 0xa: /* VL32 */
1679 case 0xb: /* VL64 */
1680 case 0xc: /* VL128 */
1681 case 0xd: /* VL256 */
1682 bound = 16 << (pattern - 9);
1683 break;
1684 case 0x1d: /* MUL4 */
1685 return elements - elements % 4;
1686 case 0x1e: /* MUL3 */
1687 return elements - elements % 3;
1688 case 0x1f: /* ALL */
1689 return elements;
1690 default: /* #uimm5 */
1691 return 0;
1692 }
1693 return elements >= bound ? bound : 0;
1694}
1695
1696/* This handles all of the predicate initialization instructions,
1697 * PTRUE, PFALSE, SETFFR. For PFALSE, we will have set PAT == 32
1698 * so that decode_pred_count returns 0. For SETFFR, we will have
1699 * set RD == 16 == FFR.
1700 */
1701static bool do_predset(DisasContext *s, int esz, int rd, int pat, bool setflag)
1702{
1703 if (!sve_access_check(s)) {
1704 return true;
1705 }
1706
1707 unsigned fullsz = vec_full_reg_size(s);
1708 unsigned ofs = pred_full_reg_offset(s, rd);
1709 unsigned numelem, setsz, i;
1710 uint64_t word, lastword;
1711 TCGv_i64 t;
1712
1713 numelem = decode_pred_count(fullsz, pat, esz);
1714
1715 /* Determine what we must store into each bit, and how many. */
1716 if (numelem == 0) {
1717 lastword = word = 0;
1718 setsz = fullsz;
1719 } else {
1720 setsz = numelem << esz;
1721 lastword = word = pred_esz_masks[esz];
1722 if (setsz % 64) {
973558a3 1723 lastword &= MAKE_64BIT_MASK(0, setsz % 64);
028e2a7b
RH
1724 }
1725 }
1726
1727 t = tcg_temp_new_i64();
1728 if (fullsz <= 64) {
1729 tcg_gen_movi_i64(t, lastword);
1730 tcg_gen_st_i64(t, cpu_env, ofs);
1731 goto done;
1732 }
1733
1734 if (word == lastword) {
1735 unsigned maxsz = size_for_gvec(fullsz / 8);
1736 unsigned oprsz = size_for_gvec(setsz / 8);
1737
1738 if (oprsz * 8 == setsz) {
8711e71f 1739 tcg_gen_gvec_dup_imm(MO_64, ofs, oprsz, maxsz, word);
028e2a7b
RH
1740 goto done;
1741 }
028e2a7b
RH
1742 }
1743
1744 setsz /= 8;
1745 fullsz /= 8;
1746
1747 tcg_gen_movi_i64(t, word);
973558a3 1748 for (i = 0; i < QEMU_ALIGN_DOWN(setsz, 8); i += 8) {
028e2a7b
RH
1749 tcg_gen_st_i64(t, cpu_env, ofs + i);
1750 }
1751 if (lastword != word) {
1752 tcg_gen_movi_i64(t, lastword);
1753 tcg_gen_st_i64(t, cpu_env, ofs + i);
1754 i += 8;
1755 }
1756 if (i < fullsz) {
1757 tcg_gen_movi_i64(t, 0);
1758 for (; i < fullsz; i += 8) {
1759 tcg_gen_st_i64(t, cpu_env, ofs + i);
1760 }
1761 }
1762
1763 done:
1764 tcg_temp_free_i64(t);
1765
1766 /* PTRUES */
1767 if (setflag) {
1768 tcg_gen_movi_i32(cpu_NF, -(word != 0));
1769 tcg_gen_movi_i32(cpu_CF, word == 0);
1770 tcg_gen_movi_i32(cpu_VF, 0);
1771 tcg_gen_mov_i32(cpu_ZF, cpu_NF);
1772 }
1773 return true;
1774}
1775
3a7be554 1776static bool trans_PTRUE(DisasContext *s, arg_PTRUE *a)
028e2a7b
RH
1777{
1778 return do_predset(s, a->esz, a->rd, a->pat, a->s);
1779}
1780
3a7be554 1781static bool trans_SETFFR(DisasContext *s, arg_SETFFR *a)
028e2a7b
RH
1782{
1783 /* Note pat == 31 is #all, to set all elements. */
1784 return do_predset(s, 0, FFR_PRED_NUM, 31, false);
1785}
1786
3a7be554 1787static bool trans_PFALSE(DisasContext *s, arg_PFALSE *a)
028e2a7b
RH
1788{
1789 /* Note pat == 32 is #unimp, to set no elements. */
1790 return do_predset(s, 0, a->rd, 32, false);
1791}
1792
3a7be554 1793static bool trans_RDFFR_p(DisasContext *s, arg_RDFFR_p *a)
028e2a7b
RH
1794{
1795 /* The path through do_pppp_flags is complicated enough to want to avoid
1796 * duplication. Frob the arguments into the form of a predicated AND.
1797 */
1798 arg_rprr_s alt_a = {
1799 .rd = a->rd, .pg = a->pg, .s = a->s,
1800 .rn = FFR_PRED_NUM, .rm = FFR_PRED_NUM,
1801 };
3a7be554 1802 return trans_AND_pppp(s, &alt_a);
028e2a7b
RH
1803}
1804
3a7be554 1805static bool trans_RDFFR(DisasContext *s, arg_RDFFR *a)
028e2a7b
RH
1806{
1807 return do_mov_p(s, a->rd, FFR_PRED_NUM);
1808}
1809
3a7be554 1810static bool trans_WRFFR(DisasContext *s, arg_WRFFR *a)
028e2a7b
RH
1811{
1812 return do_mov_p(s, FFR_PRED_NUM, a->rn);
1813}
1814
1815static bool do_pfirst_pnext(DisasContext *s, arg_rr_esz *a,
1816 void (*gen_fn)(TCGv_i32, TCGv_ptr,
1817 TCGv_ptr, TCGv_i32))
1818{
1819 if (!sve_access_check(s)) {
1820 return true;
1821 }
1822
1823 TCGv_ptr t_pd = tcg_temp_new_ptr();
1824 TCGv_ptr t_pg = tcg_temp_new_ptr();
1825 TCGv_i32 t;
86300b5d 1826 unsigned desc = 0;
028e2a7b 1827
86300b5d
RH
1828 desc = FIELD_DP32(desc, PREDDESC, OPRSZ, pred_full_reg_size(s));
1829 desc = FIELD_DP32(desc, PREDDESC, ESZ, a->esz);
028e2a7b
RH
1830
1831 tcg_gen_addi_ptr(t_pd, cpu_env, pred_full_reg_offset(s, a->rd));
1832 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, a->rn));
1833 t = tcg_const_i32(desc);
1834
1835 gen_fn(t, t_pd, t_pg, t);
1836 tcg_temp_free_ptr(t_pd);
1837 tcg_temp_free_ptr(t_pg);
1838
1839 do_pred_flags(t);
1840 tcg_temp_free_i32(t);
1841 return true;
1842}
1843
3a7be554 1844static bool trans_PFIRST(DisasContext *s, arg_rr_esz *a)
028e2a7b
RH
1845{
1846 return do_pfirst_pnext(s, a, gen_helper_sve_pfirst);
1847}
1848
3a7be554 1849static bool trans_PNEXT(DisasContext *s, arg_rr_esz *a)
028e2a7b
RH
1850{
1851 return do_pfirst_pnext(s, a, gen_helper_sve_pnext);
1852}
1853
24e82e68
RH
1854/*
1855 *** SVE Element Count Group
1856 */
1857
1858/* Perform an inline saturating addition of a 32-bit value within
1859 * a 64-bit register. The second operand is known to be positive,
1860 * which halves the comparisions we must perform to bound the result.
1861 */
1862static void do_sat_addsub_32(TCGv_i64 reg, TCGv_i64 val, bool u, bool d)
1863{
1864 int64_t ibound;
1865 TCGv_i64 bound;
1866 TCGCond cond;
1867
1868 /* Use normal 64-bit arithmetic to detect 32-bit overflow. */
1869 if (u) {
1870 tcg_gen_ext32u_i64(reg, reg);
1871 } else {
1872 tcg_gen_ext32s_i64(reg, reg);
1873 }
1874 if (d) {
1875 tcg_gen_sub_i64(reg, reg, val);
1876 ibound = (u ? 0 : INT32_MIN);
1877 cond = TCG_COND_LT;
1878 } else {
1879 tcg_gen_add_i64(reg, reg, val);
1880 ibound = (u ? UINT32_MAX : INT32_MAX);
1881 cond = TCG_COND_GT;
1882 }
1883 bound = tcg_const_i64(ibound);
1884 tcg_gen_movcond_i64(cond, reg, reg, bound, bound, reg);
1885 tcg_temp_free_i64(bound);
1886}
1887
1888/* Similarly with 64-bit values. */
1889static void do_sat_addsub_64(TCGv_i64 reg, TCGv_i64 val, bool u, bool d)
1890{
1891 TCGv_i64 t0 = tcg_temp_new_i64();
1892 TCGv_i64 t1 = tcg_temp_new_i64();
1893 TCGv_i64 t2;
1894
1895 if (u) {
1896 if (d) {
1897 tcg_gen_sub_i64(t0, reg, val);
1898 tcg_gen_movi_i64(t1, 0);
1899 tcg_gen_movcond_i64(TCG_COND_LTU, reg, reg, val, t1, t0);
1900 } else {
1901 tcg_gen_add_i64(t0, reg, val);
1902 tcg_gen_movi_i64(t1, -1);
1903 tcg_gen_movcond_i64(TCG_COND_LTU, reg, t0, reg, t1, t0);
1904 }
1905 } else {
1906 if (d) {
1907 /* Detect signed overflow for subtraction. */
1908 tcg_gen_xor_i64(t0, reg, val);
1909 tcg_gen_sub_i64(t1, reg, val);
7a31e0c6 1910 tcg_gen_xor_i64(reg, reg, t1);
24e82e68
RH
1911 tcg_gen_and_i64(t0, t0, reg);
1912
1913 /* Bound the result. */
1914 tcg_gen_movi_i64(reg, INT64_MIN);
1915 t2 = tcg_const_i64(0);
1916 tcg_gen_movcond_i64(TCG_COND_LT, reg, t0, t2, reg, t1);
1917 } else {
1918 /* Detect signed overflow for addition. */
1919 tcg_gen_xor_i64(t0, reg, val);
1920 tcg_gen_add_i64(reg, reg, val);
1921 tcg_gen_xor_i64(t1, reg, val);
1922 tcg_gen_andc_i64(t0, t1, t0);
1923
1924 /* Bound the result. */
1925 tcg_gen_movi_i64(t1, INT64_MAX);
1926 t2 = tcg_const_i64(0);
1927 tcg_gen_movcond_i64(TCG_COND_LT, reg, t0, t2, t1, reg);
1928 }
1929 tcg_temp_free_i64(t2);
1930 }
1931 tcg_temp_free_i64(t0);
1932 tcg_temp_free_i64(t1);
1933}
1934
1935/* Similarly with a vector and a scalar operand. */
1936static void do_sat_addsub_vec(DisasContext *s, int esz, int rd, int rn,
1937 TCGv_i64 val, bool u, bool d)
1938{
1939 unsigned vsz = vec_full_reg_size(s);
1940 TCGv_ptr dptr, nptr;
1941 TCGv_i32 t32, desc;
1942 TCGv_i64 t64;
1943
1944 dptr = tcg_temp_new_ptr();
1945 nptr = tcg_temp_new_ptr();
1946 tcg_gen_addi_ptr(dptr, cpu_env, vec_full_reg_offset(s, rd));
1947 tcg_gen_addi_ptr(nptr, cpu_env, vec_full_reg_offset(s, rn));
1948 desc = tcg_const_i32(simd_desc(vsz, vsz, 0));
1949
1950 switch (esz) {
1951 case MO_8:
1952 t32 = tcg_temp_new_i32();
1953 tcg_gen_extrl_i64_i32(t32, val);
1954 if (d) {
1955 tcg_gen_neg_i32(t32, t32);
1956 }
1957 if (u) {
1958 gen_helper_sve_uqaddi_b(dptr, nptr, t32, desc);
1959 } else {
1960 gen_helper_sve_sqaddi_b(dptr, nptr, t32, desc);
1961 }
1962 tcg_temp_free_i32(t32);
1963 break;
1964
1965 case MO_16:
1966 t32 = tcg_temp_new_i32();
1967 tcg_gen_extrl_i64_i32(t32, val);
1968 if (d) {
1969 tcg_gen_neg_i32(t32, t32);
1970 }
1971 if (u) {
1972 gen_helper_sve_uqaddi_h(dptr, nptr, t32, desc);
1973 } else {
1974 gen_helper_sve_sqaddi_h(dptr, nptr, t32, desc);
1975 }
1976 tcg_temp_free_i32(t32);
1977 break;
1978
1979 case MO_32:
1980 t64 = tcg_temp_new_i64();
1981 if (d) {
1982 tcg_gen_neg_i64(t64, val);
1983 } else {
1984 tcg_gen_mov_i64(t64, val);
1985 }
1986 if (u) {
1987 gen_helper_sve_uqaddi_s(dptr, nptr, t64, desc);
1988 } else {
1989 gen_helper_sve_sqaddi_s(dptr, nptr, t64, desc);
1990 }
1991 tcg_temp_free_i64(t64);
1992 break;
1993
1994 case MO_64:
1995 if (u) {
1996 if (d) {
1997 gen_helper_sve_uqsubi_d(dptr, nptr, val, desc);
1998 } else {
1999 gen_helper_sve_uqaddi_d(dptr, nptr, val, desc);
2000 }
2001 } else if (d) {
2002 t64 = tcg_temp_new_i64();
2003 tcg_gen_neg_i64(t64, val);
2004 gen_helper_sve_sqaddi_d(dptr, nptr, t64, desc);
2005 tcg_temp_free_i64(t64);
2006 } else {
2007 gen_helper_sve_sqaddi_d(dptr, nptr, val, desc);
2008 }
2009 break;
2010
2011 default:
2012 g_assert_not_reached();
2013 }
2014
2015 tcg_temp_free_ptr(dptr);
2016 tcg_temp_free_ptr(nptr);
2017 tcg_temp_free_i32(desc);
2018}
2019
3a7be554 2020static bool trans_CNT_r(DisasContext *s, arg_CNT_r *a)
24e82e68
RH
2021{
2022 if (sve_access_check(s)) {
2023 unsigned fullsz = vec_full_reg_size(s);
2024 unsigned numelem = decode_pred_count(fullsz, a->pat, a->esz);
2025 tcg_gen_movi_i64(cpu_reg(s, a->rd), numelem * a->imm);
2026 }
2027 return true;
2028}
2029
3a7be554 2030static bool trans_INCDEC_r(DisasContext *s, arg_incdec_cnt *a)
24e82e68
RH
2031{
2032 if (sve_access_check(s)) {
2033 unsigned fullsz = vec_full_reg_size(s);
2034 unsigned numelem = decode_pred_count(fullsz, a->pat, a->esz);
2035 int inc = numelem * a->imm * (a->d ? -1 : 1);
2036 TCGv_i64 reg = cpu_reg(s, a->rd);
2037
2038 tcg_gen_addi_i64(reg, reg, inc);
2039 }
2040 return true;
2041}
2042
3a7be554 2043static bool trans_SINCDEC_r_32(DisasContext *s, arg_incdec_cnt *a)
24e82e68
RH
2044{
2045 if (!sve_access_check(s)) {
2046 return true;
2047 }
2048
2049 unsigned fullsz = vec_full_reg_size(s);
2050 unsigned numelem = decode_pred_count(fullsz, a->pat, a->esz);
2051 int inc = numelem * a->imm;
2052 TCGv_i64 reg = cpu_reg(s, a->rd);
2053
2054 /* Use normal 64-bit arithmetic to detect 32-bit overflow. */
2055 if (inc == 0) {
2056 if (a->u) {
2057 tcg_gen_ext32u_i64(reg, reg);
2058 } else {
2059 tcg_gen_ext32s_i64(reg, reg);
2060 }
2061 } else {
2062 TCGv_i64 t = tcg_const_i64(inc);
2063 do_sat_addsub_32(reg, t, a->u, a->d);
2064 tcg_temp_free_i64(t);
2065 }
2066 return true;
2067}
2068
3a7be554 2069static bool trans_SINCDEC_r_64(DisasContext *s, arg_incdec_cnt *a)
24e82e68
RH
2070{
2071 if (!sve_access_check(s)) {
2072 return true;
2073 }
2074
2075 unsigned fullsz = vec_full_reg_size(s);
2076 unsigned numelem = decode_pred_count(fullsz, a->pat, a->esz);
2077 int inc = numelem * a->imm;
2078 TCGv_i64 reg = cpu_reg(s, a->rd);
2079
2080 if (inc != 0) {
2081 TCGv_i64 t = tcg_const_i64(inc);
2082 do_sat_addsub_64(reg, t, a->u, a->d);
2083 tcg_temp_free_i64(t);
2084 }
2085 return true;
2086}
2087
3a7be554 2088static bool trans_INCDEC_v(DisasContext *s, arg_incdec2_cnt *a)
24e82e68
RH
2089{
2090 if (a->esz == 0) {
2091 return false;
2092 }
2093
2094 unsigned fullsz = vec_full_reg_size(s);
2095 unsigned numelem = decode_pred_count(fullsz, a->pat, a->esz);
2096 int inc = numelem * a->imm;
2097
2098 if (inc != 0) {
2099 if (sve_access_check(s)) {
2100 TCGv_i64 t = tcg_const_i64(a->d ? -inc : inc);
2101 tcg_gen_gvec_adds(a->esz, vec_full_reg_offset(s, a->rd),
2102 vec_full_reg_offset(s, a->rn),
2103 t, fullsz, fullsz);
2104 tcg_temp_free_i64(t);
2105 }
2106 } else {
2107 do_mov_z(s, a->rd, a->rn);
2108 }
2109 return true;
2110}
2111
3a7be554 2112static bool trans_SINCDEC_v(DisasContext *s, arg_incdec2_cnt *a)
24e82e68
RH
2113{
2114 if (a->esz == 0) {
2115 return false;
2116 }
2117
2118 unsigned fullsz = vec_full_reg_size(s);
2119 unsigned numelem = decode_pred_count(fullsz, a->pat, a->esz);
2120 int inc = numelem * a->imm;
2121
2122 if (inc != 0) {
2123 if (sve_access_check(s)) {
2124 TCGv_i64 t = tcg_const_i64(inc);
2125 do_sat_addsub_vec(s, a->esz, a->rd, a->rn, t, a->u, a->d);
2126 tcg_temp_free_i64(t);
2127 }
2128 } else {
2129 do_mov_z(s, a->rd, a->rn);
2130 }
2131 return true;
2132}
2133
e1fa1164
RH
2134/*
2135 *** SVE Bitwise Immediate Group
2136 */
2137
2138static bool do_zz_dbm(DisasContext *s, arg_rr_dbm *a, GVecGen2iFn *gvec_fn)
2139{
2140 uint64_t imm;
2141 if (!logic_imm_decode_wmask(&imm, extract32(a->dbm, 12, 1),
2142 extract32(a->dbm, 0, 6),
2143 extract32(a->dbm, 6, 6))) {
2144 return false;
2145 }
2146 if (sve_access_check(s)) {
2147 unsigned vsz = vec_full_reg_size(s);
2148 gvec_fn(MO_64, vec_full_reg_offset(s, a->rd),
2149 vec_full_reg_offset(s, a->rn), imm, vsz, vsz);
2150 }
2151 return true;
2152}
2153
3a7be554 2154static bool trans_AND_zzi(DisasContext *s, arg_rr_dbm *a)
e1fa1164
RH
2155{
2156 return do_zz_dbm(s, a, tcg_gen_gvec_andi);
2157}
2158
3a7be554 2159static bool trans_ORR_zzi(DisasContext *s, arg_rr_dbm *a)
e1fa1164
RH
2160{
2161 return do_zz_dbm(s, a, tcg_gen_gvec_ori);
2162}
2163
3a7be554 2164static bool trans_EOR_zzi(DisasContext *s, arg_rr_dbm *a)
e1fa1164
RH
2165{
2166 return do_zz_dbm(s, a, tcg_gen_gvec_xori);
2167}
2168
3a7be554 2169static bool trans_DUPM(DisasContext *s, arg_DUPM *a)
e1fa1164
RH
2170{
2171 uint64_t imm;
2172 if (!logic_imm_decode_wmask(&imm, extract32(a->dbm, 12, 1),
2173 extract32(a->dbm, 0, 6),
2174 extract32(a->dbm, 6, 6))) {
2175 return false;
2176 }
2177 if (sve_access_check(s)) {
2178 do_dupi_z(s, a->rd, imm);
2179 }
2180 return true;
2181}
2182
f25a2361
RH
2183/*
2184 *** SVE Integer Wide Immediate - Predicated Group
2185 */
2186
2187/* Implement all merging copies. This is used for CPY (immediate),
2188 * FCPY, CPY (scalar), CPY (SIMD&FP scalar).
2189 */
2190static void do_cpy_m(DisasContext *s, int esz, int rd, int rn, int pg,
2191 TCGv_i64 val)
2192{
2193 typedef void gen_cpy(TCGv_ptr, TCGv_ptr, TCGv_ptr, TCGv_i64, TCGv_i32);
2194 static gen_cpy * const fns[4] = {
2195 gen_helper_sve_cpy_m_b, gen_helper_sve_cpy_m_h,
2196 gen_helper_sve_cpy_m_s, gen_helper_sve_cpy_m_d,
2197 };
2198 unsigned vsz = vec_full_reg_size(s);
2199 TCGv_i32 desc = tcg_const_i32(simd_desc(vsz, vsz, 0));
2200 TCGv_ptr t_zd = tcg_temp_new_ptr();
2201 TCGv_ptr t_zn = tcg_temp_new_ptr();
2202 TCGv_ptr t_pg = tcg_temp_new_ptr();
2203
2204 tcg_gen_addi_ptr(t_zd, cpu_env, vec_full_reg_offset(s, rd));
2205 tcg_gen_addi_ptr(t_zn, cpu_env, vec_full_reg_offset(s, rn));
2206 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, pg));
2207
2208 fns[esz](t_zd, t_zn, t_pg, val, desc);
2209
2210 tcg_temp_free_ptr(t_zd);
2211 tcg_temp_free_ptr(t_zn);
2212 tcg_temp_free_ptr(t_pg);
2213 tcg_temp_free_i32(desc);
2214}
2215
3a7be554 2216static bool trans_FCPY(DisasContext *s, arg_FCPY *a)
f25a2361
RH
2217{
2218 if (a->esz == 0) {
2219 return false;
2220 }
2221 if (sve_access_check(s)) {
2222 /* Decode the VFP immediate. */
2223 uint64_t imm = vfp_expand_imm(a->esz, a->imm);
2224 TCGv_i64 t_imm = tcg_const_i64(imm);
2225 do_cpy_m(s, a->esz, a->rd, a->rn, a->pg, t_imm);
2226 tcg_temp_free_i64(t_imm);
2227 }
2228 return true;
2229}
2230
3a7be554 2231static bool trans_CPY_m_i(DisasContext *s, arg_rpri_esz *a)
f25a2361 2232{
3a7be554 2233 if (a->esz == 0 && extract32(s->insn, 13, 1)) {
f25a2361
RH
2234 return false;
2235 }
2236 if (sve_access_check(s)) {
2237 TCGv_i64 t_imm = tcg_const_i64(a->imm);
2238 do_cpy_m(s, a->esz, a->rd, a->rn, a->pg, t_imm);
2239 tcg_temp_free_i64(t_imm);
2240 }
2241 return true;
2242}
2243
3a7be554 2244static bool trans_CPY_z_i(DisasContext *s, arg_CPY_z_i *a)
f25a2361
RH
2245{
2246 static gen_helper_gvec_2i * const fns[4] = {
2247 gen_helper_sve_cpy_z_b, gen_helper_sve_cpy_z_h,
2248 gen_helper_sve_cpy_z_s, gen_helper_sve_cpy_z_d,
2249 };
2250
3a7be554 2251 if (a->esz == 0 && extract32(s->insn, 13, 1)) {
f25a2361
RH
2252 return false;
2253 }
2254 if (sve_access_check(s)) {
2255 unsigned vsz = vec_full_reg_size(s);
2256 TCGv_i64 t_imm = tcg_const_i64(a->imm);
2257 tcg_gen_gvec_2i_ool(vec_full_reg_offset(s, a->rd),
2258 pred_full_reg_offset(s, a->pg),
2259 t_imm, vsz, vsz, 0, fns[a->esz]);
2260 tcg_temp_free_i64(t_imm);
2261 }
2262 return true;
2263}
2264
b94f8f60
RH
2265/*
2266 *** SVE Permute Extract Group
2267 */
2268
75114792 2269static bool do_EXT(DisasContext *s, int rd, int rn, int rm, int imm)
b94f8f60
RH
2270{
2271 if (!sve_access_check(s)) {
2272 return true;
2273 }
2274
2275 unsigned vsz = vec_full_reg_size(s);
75114792 2276 unsigned n_ofs = imm >= vsz ? 0 : imm;
b94f8f60 2277 unsigned n_siz = vsz - n_ofs;
75114792
SL
2278 unsigned d = vec_full_reg_offset(s, rd);
2279 unsigned n = vec_full_reg_offset(s, rn);
2280 unsigned m = vec_full_reg_offset(s, rm);
b94f8f60
RH
2281
2282 /* Use host vector move insns if we have appropriate sizes
2283 * and no unfortunate overlap.
2284 */
2285 if (m != d
2286 && n_ofs == size_for_gvec(n_ofs)
2287 && n_siz == size_for_gvec(n_siz)
2288 && (d != n || n_siz <= n_ofs)) {
2289 tcg_gen_gvec_mov(0, d, n + n_ofs, n_siz, n_siz);
2290 if (n_ofs != 0) {
2291 tcg_gen_gvec_mov(0, d + n_siz, m, n_ofs, n_ofs);
2292 }
2293 } else {
2294 tcg_gen_gvec_3_ool(d, n, m, vsz, vsz, n_ofs, gen_helper_sve_ext);
2295 }
2296 return true;
2297}
2298
75114792
SL
2299static bool trans_EXT(DisasContext *s, arg_EXT *a)
2300{
2301 return do_EXT(s, a->rd, a->rn, a->rm, a->imm);
2302}
2303
2304static bool trans_EXT_sve2(DisasContext *s, arg_rri *a)
2305{
2306 if (!dc_isar_feature(aa64_sve2, s)) {
2307 return false;
2308 }
2309 return do_EXT(s, a->rd, a->rn, (a->rn + 1) % 32, a->imm);
2310}
2311
30562ab7
RH
2312/*
2313 *** SVE Permute - Unpredicated Group
2314 */
2315
3a7be554 2316static bool trans_DUP_s(DisasContext *s, arg_DUP_s *a)
30562ab7
RH
2317{
2318 if (sve_access_check(s)) {
2319 unsigned vsz = vec_full_reg_size(s);
2320 tcg_gen_gvec_dup_i64(a->esz, vec_full_reg_offset(s, a->rd),
2321 vsz, vsz, cpu_reg_sp(s, a->rn));
2322 }
2323 return true;
2324}
2325
3a7be554 2326static bool trans_DUP_x(DisasContext *s, arg_DUP_x *a)
30562ab7
RH
2327{
2328 if ((a->imm & 0x1f) == 0) {
2329 return false;
2330 }
2331 if (sve_access_check(s)) {
2332 unsigned vsz = vec_full_reg_size(s);
2333 unsigned dofs = vec_full_reg_offset(s, a->rd);
2334 unsigned esz, index;
2335
2336 esz = ctz32(a->imm);
2337 index = a->imm >> (esz + 1);
2338
2339 if ((index << esz) < vsz) {
2340 unsigned nofs = vec_reg_offset(s, a->rn, index, esz);
2341 tcg_gen_gvec_dup_mem(esz, dofs, nofs, vsz, vsz);
2342 } else {
7e17d50e
RH
2343 /*
2344 * While dup_mem handles 128-bit elements, dup_imm does not.
2345 * Thankfully element size doesn't matter for splatting zero.
2346 */
2347 tcg_gen_gvec_dup_imm(MO_64, dofs, vsz, vsz, 0);
30562ab7
RH
2348 }
2349 }
2350 return true;
2351}
2352
2353static void do_insr_i64(DisasContext *s, arg_rrr_esz *a, TCGv_i64 val)
2354{
2355 typedef void gen_insr(TCGv_ptr, TCGv_ptr, TCGv_i64, TCGv_i32);
2356 static gen_insr * const fns[4] = {
2357 gen_helper_sve_insr_b, gen_helper_sve_insr_h,
2358 gen_helper_sve_insr_s, gen_helper_sve_insr_d,
2359 };
2360 unsigned vsz = vec_full_reg_size(s);
2361 TCGv_i32 desc = tcg_const_i32(simd_desc(vsz, vsz, 0));
2362 TCGv_ptr t_zd = tcg_temp_new_ptr();
2363 TCGv_ptr t_zn = tcg_temp_new_ptr();
2364
2365 tcg_gen_addi_ptr(t_zd, cpu_env, vec_full_reg_offset(s, a->rd));
2366 tcg_gen_addi_ptr(t_zn, cpu_env, vec_full_reg_offset(s, a->rn));
2367
2368 fns[a->esz](t_zd, t_zn, val, desc);
2369
2370 tcg_temp_free_ptr(t_zd);
2371 tcg_temp_free_ptr(t_zn);
2372 tcg_temp_free_i32(desc);
2373}
2374
3a7be554 2375static bool trans_INSR_f(DisasContext *s, arg_rrr_esz *a)
30562ab7
RH
2376{
2377 if (sve_access_check(s)) {
2378 TCGv_i64 t = tcg_temp_new_i64();
2379 tcg_gen_ld_i64(t, cpu_env, vec_reg_offset(s, a->rm, 0, MO_64));
2380 do_insr_i64(s, a, t);
2381 tcg_temp_free_i64(t);
2382 }
2383 return true;
2384}
2385
3a7be554 2386static bool trans_INSR_r(DisasContext *s, arg_rrr_esz *a)
30562ab7
RH
2387{
2388 if (sve_access_check(s)) {
2389 do_insr_i64(s, a, cpu_reg(s, a->rm));
2390 }
2391 return true;
2392}
2393
3a7be554 2394static bool trans_REV_v(DisasContext *s, arg_rr_esz *a)
30562ab7
RH
2395{
2396 static gen_helper_gvec_2 * const fns[4] = {
2397 gen_helper_sve_rev_b, gen_helper_sve_rev_h,
2398 gen_helper_sve_rev_s, gen_helper_sve_rev_d
2399 };
2400
2401 if (sve_access_check(s)) {
40e32e5a 2402 gen_gvec_ool_zz(s, fns[a->esz], a->rd, a->rn, 0);
30562ab7
RH
2403 }
2404 return true;
2405}
2406
3a7be554 2407static bool trans_TBL(DisasContext *s, arg_rrr_esz *a)
30562ab7
RH
2408{
2409 static gen_helper_gvec_3 * const fns[4] = {
2410 gen_helper_sve_tbl_b, gen_helper_sve_tbl_h,
2411 gen_helper_sve_tbl_s, gen_helper_sve_tbl_d
2412 };
2413
2414 if (sve_access_check(s)) {
e645d1a1 2415 gen_gvec_ool_zzz(s, fns[a->esz], a->rd, a->rn, a->rm, 0);
30562ab7
RH
2416 }
2417 return true;
2418}
2419
3a7be554 2420static bool trans_UNPK(DisasContext *s, arg_UNPK *a)
30562ab7
RH
2421{
2422 static gen_helper_gvec_2 * const fns[4][2] = {
2423 { NULL, NULL },
2424 { gen_helper_sve_sunpk_h, gen_helper_sve_uunpk_h },
2425 { gen_helper_sve_sunpk_s, gen_helper_sve_uunpk_s },
2426 { gen_helper_sve_sunpk_d, gen_helper_sve_uunpk_d },
2427 };
2428
2429 if (a->esz == 0) {
2430 return false;
2431 }
2432 if (sve_access_check(s)) {
2433 unsigned vsz = vec_full_reg_size(s);
2434 tcg_gen_gvec_2_ool(vec_full_reg_offset(s, a->rd),
2435 vec_full_reg_offset(s, a->rn)
2436 + (a->h ? vsz / 2 : 0),
2437 vsz, vsz, 0, fns[a->esz][a->u]);
2438 }
2439 return true;
2440}
2441
d731d8cb
RH
2442/*
2443 *** SVE Permute - Predicates Group
2444 */
2445
2446static bool do_perm_pred3(DisasContext *s, arg_rrr_esz *a, bool high_odd,
2447 gen_helper_gvec_3 *fn)
2448{
2449 if (!sve_access_check(s)) {
2450 return true;
2451 }
2452
2453 unsigned vsz = pred_full_reg_size(s);
2454
d731d8cb
RH
2455 TCGv_ptr t_d = tcg_temp_new_ptr();
2456 TCGv_ptr t_n = tcg_temp_new_ptr();
2457 TCGv_ptr t_m = tcg_temp_new_ptr();
2458 TCGv_i32 t_desc;
f9b0fcce 2459 uint32_t desc = 0;
d731d8cb 2460
f9b0fcce
RH
2461 desc = FIELD_DP32(desc, PREDDESC, OPRSZ, vsz);
2462 desc = FIELD_DP32(desc, PREDDESC, ESZ, a->esz);
2463 desc = FIELD_DP32(desc, PREDDESC, DATA, high_odd);
d731d8cb
RH
2464
2465 tcg_gen_addi_ptr(t_d, cpu_env, pred_full_reg_offset(s, a->rd));
2466 tcg_gen_addi_ptr(t_n, cpu_env, pred_full_reg_offset(s, a->rn));
2467 tcg_gen_addi_ptr(t_m, cpu_env, pred_full_reg_offset(s, a->rm));
2468 t_desc = tcg_const_i32(desc);
2469
2470 fn(t_d, t_n, t_m, t_desc);
2471
2472 tcg_temp_free_ptr(t_d);
2473 tcg_temp_free_ptr(t_n);
2474 tcg_temp_free_ptr(t_m);
2475 tcg_temp_free_i32(t_desc);
2476 return true;
2477}
2478
2479static bool do_perm_pred2(DisasContext *s, arg_rr_esz *a, bool high_odd,
2480 gen_helper_gvec_2 *fn)
2481{
2482 if (!sve_access_check(s)) {
2483 return true;
2484 }
2485
2486 unsigned vsz = pred_full_reg_size(s);
2487 TCGv_ptr t_d = tcg_temp_new_ptr();
2488 TCGv_ptr t_n = tcg_temp_new_ptr();
2489 TCGv_i32 t_desc;
70acaafe 2490 uint32_t desc = 0;
d731d8cb
RH
2491
2492 tcg_gen_addi_ptr(t_d, cpu_env, pred_full_reg_offset(s, a->rd));
2493 tcg_gen_addi_ptr(t_n, cpu_env, pred_full_reg_offset(s, a->rn));
2494
70acaafe
RH
2495 desc = FIELD_DP32(desc, PREDDESC, OPRSZ, vsz);
2496 desc = FIELD_DP32(desc, PREDDESC, ESZ, a->esz);
2497 desc = FIELD_DP32(desc, PREDDESC, DATA, high_odd);
d731d8cb
RH
2498 t_desc = tcg_const_i32(desc);
2499
2500 fn(t_d, t_n, t_desc);
2501
2502 tcg_temp_free_i32(t_desc);
2503 tcg_temp_free_ptr(t_d);
2504 tcg_temp_free_ptr(t_n);
2505 return true;
2506}
2507
3a7be554 2508static bool trans_ZIP1_p(DisasContext *s, arg_rrr_esz *a)
d731d8cb
RH
2509{
2510 return do_perm_pred3(s, a, 0, gen_helper_sve_zip_p);
2511}
2512
3a7be554 2513static bool trans_ZIP2_p(DisasContext *s, arg_rrr_esz *a)
d731d8cb
RH
2514{
2515 return do_perm_pred3(s, a, 1, gen_helper_sve_zip_p);
2516}
2517
3a7be554 2518static bool trans_UZP1_p(DisasContext *s, arg_rrr_esz *a)
d731d8cb
RH
2519{
2520 return do_perm_pred3(s, a, 0, gen_helper_sve_uzp_p);
2521}
2522
3a7be554 2523static bool trans_UZP2_p(DisasContext *s, arg_rrr_esz *a)
d731d8cb
RH
2524{
2525 return do_perm_pred3(s, a, 1, gen_helper_sve_uzp_p);
2526}
2527
3a7be554 2528static bool trans_TRN1_p(DisasContext *s, arg_rrr_esz *a)
d731d8cb
RH
2529{
2530 return do_perm_pred3(s, a, 0, gen_helper_sve_trn_p);
2531}
2532
3a7be554 2533static bool trans_TRN2_p(DisasContext *s, arg_rrr_esz *a)
d731d8cb
RH
2534{
2535 return do_perm_pred3(s, a, 1, gen_helper_sve_trn_p);
2536}
2537
3a7be554 2538static bool trans_REV_p(DisasContext *s, arg_rr_esz *a)
d731d8cb
RH
2539{
2540 return do_perm_pred2(s, a, 0, gen_helper_sve_rev_p);
2541}
2542
3a7be554 2543static bool trans_PUNPKLO(DisasContext *s, arg_PUNPKLO *a)
d731d8cb
RH
2544{
2545 return do_perm_pred2(s, a, 0, gen_helper_sve_punpk_p);
2546}
2547
3a7be554 2548static bool trans_PUNPKHI(DisasContext *s, arg_PUNPKHI *a)
d731d8cb
RH
2549{
2550 return do_perm_pred2(s, a, 1, gen_helper_sve_punpk_p);
2551}
2552
234b48e9
RH
2553/*
2554 *** SVE Permute - Interleaving Group
2555 */
2556
2557static bool do_zip(DisasContext *s, arg_rrr_esz *a, bool high)
2558{
2559 static gen_helper_gvec_3 * const fns[4] = {
2560 gen_helper_sve_zip_b, gen_helper_sve_zip_h,
2561 gen_helper_sve_zip_s, gen_helper_sve_zip_d,
2562 };
2563
2564 if (sve_access_check(s)) {
2565 unsigned vsz = vec_full_reg_size(s);
2566 unsigned high_ofs = high ? vsz / 2 : 0;
2567 tcg_gen_gvec_3_ool(vec_full_reg_offset(s, a->rd),
2568 vec_full_reg_offset(s, a->rn) + high_ofs,
2569 vec_full_reg_offset(s, a->rm) + high_ofs,
2570 vsz, vsz, 0, fns[a->esz]);
2571 }
2572 return true;
2573}
2574
2575static bool do_zzz_data_ool(DisasContext *s, arg_rrr_esz *a, int data,
2576 gen_helper_gvec_3 *fn)
2577{
2578 if (sve_access_check(s)) {
e645d1a1 2579 gen_gvec_ool_zzz(s, fn, a->rd, a->rn, a->rm, data);
234b48e9
RH
2580 }
2581 return true;
2582}
2583
3a7be554 2584static bool trans_ZIP1_z(DisasContext *s, arg_rrr_esz *a)
234b48e9
RH
2585{
2586 return do_zip(s, a, false);
2587}
2588
3a7be554 2589static bool trans_ZIP2_z(DisasContext *s, arg_rrr_esz *a)
234b48e9
RH
2590{
2591 return do_zip(s, a, true);
2592}
2593
2594static gen_helper_gvec_3 * const uzp_fns[4] = {
2595 gen_helper_sve_uzp_b, gen_helper_sve_uzp_h,
2596 gen_helper_sve_uzp_s, gen_helper_sve_uzp_d,
2597};
2598
3a7be554 2599static bool trans_UZP1_z(DisasContext *s, arg_rrr_esz *a)
234b48e9
RH
2600{
2601 return do_zzz_data_ool(s, a, 0, uzp_fns[a->esz]);
2602}
2603
3a7be554 2604static bool trans_UZP2_z(DisasContext *s, arg_rrr_esz *a)
234b48e9
RH
2605{
2606 return do_zzz_data_ool(s, a, 1 << a->esz, uzp_fns[a->esz]);
2607}
2608
2609static gen_helper_gvec_3 * const trn_fns[4] = {
2610 gen_helper_sve_trn_b, gen_helper_sve_trn_h,
2611 gen_helper_sve_trn_s, gen_helper_sve_trn_d,
2612};
2613
3a7be554 2614static bool trans_TRN1_z(DisasContext *s, arg_rrr_esz *a)
234b48e9
RH
2615{
2616 return do_zzz_data_ool(s, a, 0, trn_fns[a->esz]);
2617}
2618
3a7be554 2619static bool trans_TRN2_z(DisasContext *s, arg_rrr_esz *a)
234b48e9
RH
2620{
2621 return do_zzz_data_ool(s, a, 1 << a->esz, trn_fns[a->esz]);
2622}
2623
3ca879ae
RH
2624/*
2625 *** SVE Permute Vector - Predicated Group
2626 */
2627
3a7be554 2628static bool trans_COMPACT(DisasContext *s, arg_rpr_esz *a)
3ca879ae
RH
2629{
2630 static gen_helper_gvec_3 * const fns[4] = {
2631 NULL, NULL, gen_helper_sve_compact_s, gen_helper_sve_compact_d
2632 };
2633 return do_zpz_ool(s, a, fns[a->esz]);
2634}
2635
ef23cb72
RH
2636/* Call the helper that computes the ARM LastActiveElement pseudocode
2637 * function, scaled by the element size. This includes the not found
2638 * indication; e.g. not found for esz=3 is -8.
2639 */
2640static void find_last_active(DisasContext *s, TCGv_i32 ret, int esz, int pg)
2641{
2642 /* Predicate sizes may be smaller and cannot use simd_desc. We cannot
2643 * round up, as we do elsewhere, because we need the exact size.
2644 */
2645 TCGv_ptr t_p = tcg_temp_new_ptr();
2646 TCGv_i32 t_desc;
2acbfbe4 2647 unsigned desc = 0;
ef23cb72 2648
2acbfbe4
RH
2649 desc = FIELD_DP32(desc, PREDDESC, OPRSZ, pred_full_reg_size(s));
2650 desc = FIELD_DP32(desc, PREDDESC, ESZ, esz);
ef23cb72
RH
2651
2652 tcg_gen_addi_ptr(t_p, cpu_env, pred_full_reg_offset(s, pg));
2653 t_desc = tcg_const_i32(desc);
2654
2655 gen_helper_sve_last_active_element(ret, t_p, t_desc);
2656
2657 tcg_temp_free_i32(t_desc);
2658 tcg_temp_free_ptr(t_p);
2659}
2660
2661/* Increment LAST to the offset of the next element in the vector,
2662 * wrapping around to 0.
2663 */
2664static void incr_last_active(DisasContext *s, TCGv_i32 last, int esz)
2665{
2666 unsigned vsz = vec_full_reg_size(s);
2667
2668 tcg_gen_addi_i32(last, last, 1 << esz);
2669 if (is_power_of_2(vsz)) {
2670 tcg_gen_andi_i32(last, last, vsz - 1);
2671 } else {
2672 TCGv_i32 max = tcg_const_i32(vsz);
2673 TCGv_i32 zero = tcg_const_i32(0);
2674 tcg_gen_movcond_i32(TCG_COND_GEU, last, last, max, zero, last);
2675 tcg_temp_free_i32(max);
2676 tcg_temp_free_i32(zero);
2677 }
2678}
2679
2680/* If LAST < 0, set LAST to the offset of the last element in the vector. */
2681static void wrap_last_active(DisasContext *s, TCGv_i32 last, int esz)
2682{
2683 unsigned vsz = vec_full_reg_size(s);
2684
2685 if (is_power_of_2(vsz)) {
2686 tcg_gen_andi_i32(last, last, vsz - 1);
2687 } else {
2688 TCGv_i32 max = tcg_const_i32(vsz - (1 << esz));
2689 TCGv_i32 zero = tcg_const_i32(0);
2690 tcg_gen_movcond_i32(TCG_COND_LT, last, last, zero, max, last);
2691 tcg_temp_free_i32(max);
2692 tcg_temp_free_i32(zero);
2693 }
2694}
2695
2696/* Load an unsigned element of ESZ from BASE+OFS. */
2697static TCGv_i64 load_esz(TCGv_ptr base, int ofs, int esz)
2698{
2699 TCGv_i64 r = tcg_temp_new_i64();
2700
2701 switch (esz) {
2702 case 0:
2703 tcg_gen_ld8u_i64(r, base, ofs);
2704 break;
2705 case 1:
2706 tcg_gen_ld16u_i64(r, base, ofs);
2707 break;
2708 case 2:
2709 tcg_gen_ld32u_i64(r, base, ofs);
2710 break;
2711 case 3:
2712 tcg_gen_ld_i64(r, base, ofs);
2713 break;
2714 default:
2715 g_assert_not_reached();
2716 }
2717 return r;
2718}
2719
2720/* Load an unsigned element of ESZ from RM[LAST]. */
2721static TCGv_i64 load_last_active(DisasContext *s, TCGv_i32 last,
2722 int rm, int esz)
2723{
2724 TCGv_ptr p = tcg_temp_new_ptr();
2725 TCGv_i64 r;
2726
2727 /* Convert offset into vector into offset into ENV.
2728 * The final adjustment for the vector register base
2729 * is added via constant offset to the load.
2730 */
2731#ifdef HOST_WORDS_BIGENDIAN
2732 /* Adjust for element ordering. See vec_reg_offset. */
2733 if (esz < 3) {
2734 tcg_gen_xori_i32(last, last, 8 - (1 << esz));
2735 }
2736#endif
2737 tcg_gen_ext_i32_ptr(p, last);
2738 tcg_gen_add_ptr(p, p, cpu_env);
2739
2740 r = load_esz(p, vec_full_reg_offset(s, rm), esz);
2741 tcg_temp_free_ptr(p);
2742
2743 return r;
2744}
2745
2746/* Compute CLAST for a Zreg. */
2747static bool do_clast_vector(DisasContext *s, arg_rprr_esz *a, bool before)
2748{
2749 TCGv_i32 last;
2750 TCGLabel *over;
2751 TCGv_i64 ele;
2752 unsigned vsz, esz = a->esz;
2753
2754 if (!sve_access_check(s)) {
2755 return true;
2756 }
2757
2758 last = tcg_temp_local_new_i32();
2759 over = gen_new_label();
2760
2761 find_last_active(s, last, esz, a->pg);
2762
2763 /* There is of course no movcond for a 2048-bit vector,
2764 * so we must branch over the actual store.
2765 */
2766 tcg_gen_brcondi_i32(TCG_COND_LT, last, 0, over);
2767
2768 if (!before) {
2769 incr_last_active(s, last, esz);
2770 }
2771
2772 ele = load_last_active(s, last, a->rm, esz);
2773 tcg_temp_free_i32(last);
2774
2775 vsz = vec_full_reg_size(s);
2776 tcg_gen_gvec_dup_i64(esz, vec_full_reg_offset(s, a->rd), vsz, vsz, ele);
2777 tcg_temp_free_i64(ele);
2778
2779 /* If this insn used MOVPRFX, we may need a second move. */
2780 if (a->rd != a->rn) {
2781 TCGLabel *done = gen_new_label();
2782 tcg_gen_br(done);
2783
2784 gen_set_label(over);
2785 do_mov_z(s, a->rd, a->rn);
2786
2787 gen_set_label(done);
2788 } else {
2789 gen_set_label(over);
2790 }
2791 return true;
2792}
2793
3a7be554 2794static bool trans_CLASTA_z(DisasContext *s, arg_rprr_esz *a)
ef23cb72
RH
2795{
2796 return do_clast_vector(s, a, false);
2797}
2798
3a7be554 2799static bool trans_CLASTB_z(DisasContext *s, arg_rprr_esz *a)
ef23cb72
RH
2800{
2801 return do_clast_vector(s, a, true);
2802}
2803
2804/* Compute CLAST for a scalar. */
2805static void do_clast_scalar(DisasContext *s, int esz, int pg, int rm,
2806 bool before, TCGv_i64 reg_val)
2807{
2808 TCGv_i32 last = tcg_temp_new_i32();
2809 TCGv_i64 ele, cmp, zero;
2810
2811 find_last_active(s, last, esz, pg);
2812
2813 /* Extend the original value of last prior to incrementing. */
2814 cmp = tcg_temp_new_i64();
2815 tcg_gen_ext_i32_i64(cmp, last);
2816
2817 if (!before) {
2818 incr_last_active(s, last, esz);
2819 }
2820
2821 /* The conceit here is that while last < 0 indicates not found, after
2822 * adjusting for cpu_env->vfp.zregs[rm], it is still a valid address
2823 * from which we can load garbage. We then discard the garbage with
2824 * a conditional move.
2825 */
2826 ele = load_last_active(s, last, rm, esz);
2827 tcg_temp_free_i32(last);
2828
2829 zero = tcg_const_i64(0);
2830 tcg_gen_movcond_i64(TCG_COND_GE, reg_val, cmp, zero, ele, reg_val);
2831
2832 tcg_temp_free_i64(zero);
2833 tcg_temp_free_i64(cmp);
2834 tcg_temp_free_i64(ele);
2835}
2836
2837/* Compute CLAST for a Vreg. */
2838static bool do_clast_fp(DisasContext *s, arg_rpr_esz *a, bool before)
2839{
2840 if (sve_access_check(s)) {
2841 int esz = a->esz;
2842 int ofs = vec_reg_offset(s, a->rd, 0, esz);
2843 TCGv_i64 reg = load_esz(cpu_env, ofs, esz);
2844
2845 do_clast_scalar(s, esz, a->pg, a->rn, before, reg);
2846 write_fp_dreg(s, a->rd, reg);
2847 tcg_temp_free_i64(reg);
2848 }
2849 return true;
2850}
2851
3a7be554 2852static bool trans_CLASTA_v(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2853{
2854 return do_clast_fp(s, a, false);
2855}
2856
3a7be554 2857static bool trans_CLASTB_v(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2858{
2859 return do_clast_fp(s, a, true);
2860}
2861
2862/* Compute CLAST for a Xreg. */
2863static bool do_clast_general(DisasContext *s, arg_rpr_esz *a, bool before)
2864{
2865 TCGv_i64 reg;
2866
2867 if (!sve_access_check(s)) {
2868 return true;
2869 }
2870
2871 reg = cpu_reg(s, a->rd);
2872 switch (a->esz) {
2873 case 0:
2874 tcg_gen_ext8u_i64(reg, reg);
2875 break;
2876 case 1:
2877 tcg_gen_ext16u_i64(reg, reg);
2878 break;
2879 case 2:
2880 tcg_gen_ext32u_i64(reg, reg);
2881 break;
2882 case 3:
2883 break;
2884 default:
2885 g_assert_not_reached();
2886 }
2887
2888 do_clast_scalar(s, a->esz, a->pg, a->rn, before, reg);
2889 return true;
2890}
2891
3a7be554 2892static bool trans_CLASTA_r(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2893{
2894 return do_clast_general(s, a, false);
2895}
2896
3a7be554 2897static bool trans_CLASTB_r(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2898{
2899 return do_clast_general(s, a, true);
2900}
2901
2902/* Compute LAST for a scalar. */
2903static TCGv_i64 do_last_scalar(DisasContext *s, int esz,
2904 int pg, int rm, bool before)
2905{
2906 TCGv_i32 last = tcg_temp_new_i32();
2907 TCGv_i64 ret;
2908
2909 find_last_active(s, last, esz, pg);
2910 if (before) {
2911 wrap_last_active(s, last, esz);
2912 } else {
2913 incr_last_active(s, last, esz);
2914 }
2915
2916 ret = load_last_active(s, last, rm, esz);
2917 tcg_temp_free_i32(last);
2918 return ret;
2919}
2920
2921/* Compute LAST for a Vreg. */
2922static bool do_last_fp(DisasContext *s, arg_rpr_esz *a, bool before)
2923{
2924 if (sve_access_check(s)) {
2925 TCGv_i64 val = do_last_scalar(s, a->esz, a->pg, a->rn, before);
2926 write_fp_dreg(s, a->rd, val);
2927 tcg_temp_free_i64(val);
2928 }
2929 return true;
2930}
2931
3a7be554 2932static bool trans_LASTA_v(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2933{
2934 return do_last_fp(s, a, false);
2935}
2936
3a7be554 2937static bool trans_LASTB_v(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2938{
2939 return do_last_fp(s, a, true);
2940}
2941
2942/* Compute LAST for a Xreg. */
2943static bool do_last_general(DisasContext *s, arg_rpr_esz *a, bool before)
2944{
2945 if (sve_access_check(s)) {
2946 TCGv_i64 val = do_last_scalar(s, a->esz, a->pg, a->rn, before);
2947 tcg_gen_mov_i64(cpu_reg(s, a->rd), val);
2948 tcg_temp_free_i64(val);
2949 }
2950 return true;
2951}
2952
3a7be554 2953static bool trans_LASTA_r(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2954{
2955 return do_last_general(s, a, false);
2956}
2957
3a7be554 2958static bool trans_LASTB_r(DisasContext *s, arg_rpr_esz *a)
ef23cb72
RH
2959{
2960 return do_last_general(s, a, true);
2961}
2962
3a7be554 2963static bool trans_CPY_m_r(DisasContext *s, arg_rpr_esz *a)
792a5578
RH
2964{
2965 if (sve_access_check(s)) {
2966 do_cpy_m(s, a->esz, a->rd, a->rd, a->pg, cpu_reg_sp(s, a->rn));
2967 }
2968 return true;
2969}
2970
3a7be554 2971static bool trans_CPY_m_v(DisasContext *s, arg_rpr_esz *a)
792a5578
RH
2972{
2973 if (sve_access_check(s)) {
2974 int ofs = vec_reg_offset(s, a->rn, 0, a->esz);
2975 TCGv_i64 t = load_esz(cpu_env, ofs, a->esz);
2976 do_cpy_m(s, a->esz, a->rd, a->rd, a->pg, t);
2977 tcg_temp_free_i64(t);
2978 }
2979 return true;
2980}
2981
3a7be554 2982static bool trans_REVB(DisasContext *s, arg_rpr_esz *a)
dae8fb90
RH
2983{
2984 static gen_helper_gvec_3 * const fns[4] = {
2985 NULL,
2986 gen_helper_sve_revb_h,
2987 gen_helper_sve_revb_s,
2988 gen_helper_sve_revb_d,
2989 };
2990 return do_zpz_ool(s, a, fns[a->esz]);
2991}
2992
3a7be554 2993static bool trans_REVH(DisasContext *s, arg_rpr_esz *a)
dae8fb90
RH
2994{
2995 static gen_helper_gvec_3 * const fns[4] = {
2996 NULL,
2997 NULL,
2998 gen_helper_sve_revh_s,
2999 gen_helper_sve_revh_d,
3000 };
3001 return do_zpz_ool(s, a, fns[a->esz]);
3002}
3003
3a7be554 3004static bool trans_REVW(DisasContext *s, arg_rpr_esz *a)
dae8fb90
RH
3005{
3006 return do_zpz_ool(s, a, a->esz == 3 ? gen_helper_sve_revw_d : NULL);
3007}
3008
3a7be554 3009static bool trans_RBIT(DisasContext *s, arg_rpr_esz *a)
dae8fb90
RH
3010{
3011 static gen_helper_gvec_3 * const fns[4] = {
3012 gen_helper_sve_rbit_b,
3013 gen_helper_sve_rbit_h,
3014 gen_helper_sve_rbit_s,
3015 gen_helper_sve_rbit_d,
3016 };
3017 return do_zpz_ool(s, a, fns[a->esz]);
3018}
3019
3a7be554 3020static bool trans_SPLICE(DisasContext *s, arg_rprr_esz *a)
b48ff240
RH
3021{
3022 if (sve_access_check(s)) {
36cbb7a8 3023 gen_gvec_ool_zzzp(s, gen_helper_sve_splice,
dd701faf 3024 a->rd, a->rn, a->rm, a->pg, a->esz);
b48ff240
RH
3025 }
3026 return true;
3027}
3028
75114792
SL
3029static bool trans_SPLICE_sve2(DisasContext *s, arg_rpr_esz *a)
3030{
3031 if (!dc_isar_feature(aa64_sve2, s)) {
3032 return false;
3033 }
3034 if (sve_access_check(s)) {
3035 gen_gvec_ool_zzzp(s, gen_helper_sve_splice,
3036 a->rd, a->rn, (a->rn + 1) % 32, a->pg, a->esz);
3037 }
3038 return true;
3039}
3040
757f9cff
RH
3041/*
3042 *** SVE Integer Compare - Vectors Group
3043 */
3044
3045static bool do_ppzz_flags(DisasContext *s, arg_rprr_esz *a,
3046 gen_helper_gvec_flags_4 *gen_fn)
3047{
3048 TCGv_ptr pd, zn, zm, pg;
3049 unsigned vsz;
3050 TCGv_i32 t;
3051
3052 if (gen_fn == NULL) {
3053 return false;
3054 }
3055 if (!sve_access_check(s)) {
3056 return true;
3057 }
3058
3059 vsz = vec_full_reg_size(s);
3060 t = tcg_const_i32(simd_desc(vsz, vsz, 0));
3061 pd = tcg_temp_new_ptr();
3062 zn = tcg_temp_new_ptr();
3063 zm = tcg_temp_new_ptr();
3064 pg = tcg_temp_new_ptr();
3065
3066 tcg_gen_addi_ptr(pd, cpu_env, pred_full_reg_offset(s, a->rd));
3067 tcg_gen_addi_ptr(zn, cpu_env, vec_full_reg_offset(s, a->rn));
3068 tcg_gen_addi_ptr(zm, cpu_env, vec_full_reg_offset(s, a->rm));
3069 tcg_gen_addi_ptr(pg, cpu_env, pred_full_reg_offset(s, a->pg));
3070
3071 gen_fn(t, pd, zn, zm, pg, t);
3072
3073 tcg_temp_free_ptr(pd);
3074 tcg_temp_free_ptr(zn);
3075 tcg_temp_free_ptr(zm);
3076 tcg_temp_free_ptr(pg);
3077
3078 do_pred_flags(t);
3079
3080 tcg_temp_free_i32(t);
3081 return true;
3082}
3083
3084#define DO_PPZZ(NAME, name) \
3a7be554 3085static bool trans_##NAME##_ppzz(DisasContext *s, arg_rprr_esz *a) \
757f9cff
RH
3086{ \
3087 static gen_helper_gvec_flags_4 * const fns[4] = { \
3088 gen_helper_sve_##name##_ppzz_b, gen_helper_sve_##name##_ppzz_h, \
3089 gen_helper_sve_##name##_ppzz_s, gen_helper_sve_##name##_ppzz_d, \
3090 }; \
3091 return do_ppzz_flags(s, a, fns[a->esz]); \
3092}
3093
3094DO_PPZZ(CMPEQ, cmpeq)
3095DO_PPZZ(CMPNE, cmpne)
3096DO_PPZZ(CMPGT, cmpgt)
3097DO_PPZZ(CMPGE, cmpge)
3098DO_PPZZ(CMPHI, cmphi)
3099DO_PPZZ(CMPHS, cmphs)
3100
3101#undef DO_PPZZ
3102
3103#define DO_PPZW(NAME, name) \
3a7be554 3104static bool trans_##NAME##_ppzw(DisasContext *s, arg_rprr_esz *a) \
757f9cff
RH
3105{ \
3106 static gen_helper_gvec_flags_4 * const fns[4] = { \
3107 gen_helper_sve_##name##_ppzw_b, gen_helper_sve_##name##_ppzw_h, \
3108 gen_helper_sve_##name##_ppzw_s, NULL \
3109 }; \
3110 return do_ppzz_flags(s, a, fns[a->esz]); \
3111}
3112
3113DO_PPZW(CMPEQ, cmpeq)
3114DO_PPZW(CMPNE, cmpne)
3115DO_PPZW(CMPGT, cmpgt)
3116DO_PPZW(CMPGE, cmpge)
3117DO_PPZW(CMPHI, cmphi)
3118DO_PPZW(CMPHS, cmphs)
3119DO_PPZW(CMPLT, cmplt)
3120DO_PPZW(CMPLE, cmple)
3121DO_PPZW(CMPLO, cmplo)
3122DO_PPZW(CMPLS, cmpls)
3123
3124#undef DO_PPZW
3125
38cadeba
RH
3126/*
3127 *** SVE Integer Compare - Immediate Groups
3128 */
3129
3130static bool do_ppzi_flags(DisasContext *s, arg_rpri_esz *a,
3131 gen_helper_gvec_flags_3 *gen_fn)
3132{
3133 TCGv_ptr pd, zn, pg;
3134 unsigned vsz;
3135 TCGv_i32 t;
3136
3137 if (gen_fn == NULL) {
3138 return false;
3139 }
3140 if (!sve_access_check(s)) {
3141 return true;
3142 }
3143
3144 vsz = vec_full_reg_size(s);
3145 t = tcg_const_i32(simd_desc(vsz, vsz, a->imm));
3146 pd = tcg_temp_new_ptr();
3147 zn = tcg_temp_new_ptr();
3148 pg = tcg_temp_new_ptr();
3149
3150 tcg_gen_addi_ptr(pd, cpu_env, pred_full_reg_offset(s, a->rd));
3151 tcg_gen_addi_ptr(zn, cpu_env, vec_full_reg_offset(s, a->rn));
3152 tcg_gen_addi_ptr(pg, cpu_env, pred_full_reg_offset(s, a->pg));
3153
3154 gen_fn(t, pd, zn, pg, t);
3155
3156 tcg_temp_free_ptr(pd);
3157 tcg_temp_free_ptr(zn);
3158 tcg_temp_free_ptr(pg);
3159
3160 do_pred_flags(t);
3161
3162 tcg_temp_free_i32(t);
3163 return true;
3164}
3165
3166#define DO_PPZI(NAME, name) \
3a7be554 3167static bool trans_##NAME##_ppzi(DisasContext *s, arg_rpri_esz *a) \
38cadeba
RH
3168{ \
3169 static gen_helper_gvec_flags_3 * const fns[4] = { \
3170 gen_helper_sve_##name##_ppzi_b, gen_helper_sve_##name##_ppzi_h, \
3171 gen_helper_sve_##name##_ppzi_s, gen_helper_sve_##name##_ppzi_d, \
3172 }; \
3173 return do_ppzi_flags(s, a, fns[a->esz]); \
3174}
3175
3176DO_PPZI(CMPEQ, cmpeq)
3177DO_PPZI(CMPNE, cmpne)
3178DO_PPZI(CMPGT, cmpgt)
3179DO_PPZI(CMPGE, cmpge)
3180DO_PPZI(CMPHI, cmphi)
3181DO_PPZI(CMPHS, cmphs)
3182DO_PPZI(CMPLT, cmplt)
3183DO_PPZI(CMPLE, cmple)
3184DO_PPZI(CMPLO, cmplo)
3185DO_PPZI(CMPLS, cmpls)
3186
3187#undef DO_PPZI
3188
35da316f
RH
3189/*
3190 *** SVE Partition Break Group
3191 */
3192
3193static bool do_brk3(DisasContext *s, arg_rprr_s *a,
3194 gen_helper_gvec_4 *fn, gen_helper_gvec_flags_4 *fn_s)
3195{
3196 if (!sve_access_check(s)) {
3197 return true;
3198 }
3199
3200 unsigned vsz = pred_full_reg_size(s);
3201
3202 /* Predicate sizes may be smaller and cannot use simd_desc. */
3203 TCGv_ptr d = tcg_temp_new_ptr();
3204 TCGv_ptr n = tcg_temp_new_ptr();
3205 TCGv_ptr m = tcg_temp_new_ptr();
3206 TCGv_ptr g = tcg_temp_new_ptr();
04c774a2 3207 TCGv_i32 t = tcg_const_i32(FIELD_DP32(0, PREDDESC, OPRSZ, vsz));
35da316f
RH
3208
3209 tcg_gen_addi_ptr(d, cpu_env, pred_full_reg_offset(s, a->rd));
3210 tcg_gen_addi_ptr(n, cpu_env, pred_full_reg_offset(s, a->rn));
3211 tcg_gen_addi_ptr(m, cpu_env, pred_full_reg_offset(s, a->rm));
3212 tcg_gen_addi_ptr(g, cpu_env, pred_full_reg_offset(s, a->pg));
3213
3214 if (a->s) {
3215 fn_s(t, d, n, m, g, t);
3216 do_pred_flags(t);
3217 } else {
3218 fn(d, n, m, g, t);
3219 }
3220 tcg_temp_free_ptr(d);
3221 tcg_temp_free_ptr(n);
3222 tcg_temp_free_ptr(m);
3223 tcg_temp_free_ptr(g);
3224 tcg_temp_free_i32(t);
3225 return true;
3226}
3227
3228static bool do_brk2(DisasContext *s, arg_rpr_s *a,
3229 gen_helper_gvec_3 *fn, gen_helper_gvec_flags_3 *fn_s)
3230{
3231 if (!sve_access_check(s)) {
3232 return true;
3233 }
3234
3235 unsigned vsz = pred_full_reg_size(s);
3236
3237 /* Predicate sizes may be smaller and cannot use simd_desc. */
3238 TCGv_ptr d = tcg_temp_new_ptr();
3239 TCGv_ptr n = tcg_temp_new_ptr();
3240 TCGv_ptr g = tcg_temp_new_ptr();
04c774a2 3241 TCGv_i32 t = tcg_const_i32(FIELD_DP32(0, PREDDESC, OPRSZ, vsz));
35da316f
RH
3242
3243 tcg_gen_addi_ptr(d, cpu_env, pred_full_reg_offset(s, a->rd));
3244 tcg_gen_addi_ptr(n, cpu_env, pred_full_reg_offset(s, a->rn));
3245 tcg_gen_addi_ptr(g, cpu_env, pred_full_reg_offset(s, a->pg));
3246
3247 if (a->s) {
3248 fn_s(t, d, n, g, t);
3249 do_pred_flags(t);
3250 } else {
3251 fn(d, n, g, t);
3252 }
3253 tcg_temp_free_ptr(d);
3254 tcg_temp_free_ptr(n);
3255 tcg_temp_free_ptr(g);
3256 tcg_temp_free_i32(t);
3257 return true;
3258}
3259
3a7be554 3260static bool trans_BRKPA(DisasContext *s, arg_rprr_s *a)
35da316f
RH
3261{
3262 return do_brk3(s, a, gen_helper_sve_brkpa, gen_helper_sve_brkpas);
3263}
3264
3a7be554 3265static bool trans_BRKPB(DisasContext *s, arg_rprr_s *a)
35da316f
RH
3266{
3267 return do_brk3(s, a, gen_helper_sve_brkpb, gen_helper_sve_brkpbs);
3268}
3269
3a7be554 3270static bool trans_BRKA_m(DisasContext *s, arg_rpr_s *a)
35da316f
RH
3271{
3272 return do_brk2(s, a, gen_helper_sve_brka_m, gen_helper_sve_brkas_m);
3273}
3274
3a7be554 3275static bool trans_BRKB_m(DisasContext *s, arg_rpr_s *a)
35da316f
RH
3276{
3277 return do_brk2(s, a, gen_helper_sve_brkb_m, gen_helper_sve_brkbs_m);
3278}
3279
3a7be554 3280static bool trans_BRKA_z(DisasContext *s, arg_rpr_s *a)
35da316f
RH
3281{
3282 return do_brk2(s, a, gen_helper_sve_brka_z, gen_helper_sve_brkas_z);
3283}
3284
3a7be554 3285static bool trans_BRKB_z(DisasContext *s, arg_rpr_s *a)
35da316f
RH
3286{
3287 return do_brk2(s, a, gen_helper_sve_brkb_z, gen_helper_sve_brkbs_z);
3288}
3289
3a7be554 3290static bool trans_BRKN(DisasContext *s, arg_rpr_s *a)
35da316f
RH
3291{
3292 return do_brk2(s, a, gen_helper_sve_brkn, gen_helper_sve_brkns);
3293}
3294
9ee3a611
RH
3295/*
3296 *** SVE Predicate Count Group
3297 */
3298
3299static void do_cntp(DisasContext *s, TCGv_i64 val, int esz, int pn, int pg)
3300{
3301 unsigned psz = pred_full_reg_size(s);
3302
3303 if (psz <= 8) {
3304 uint64_t psz_mask;
3305
3306 tcg_gen_ld_i64(val, cpu_env, pred_full_reg_offset(s, pn));
3307 if (pn != pg) {
3308 TCGv_i64 g = tcg_temp_new_i64();
3309 tcg_gen_ld_i64(g, cpu_env, pred_full_reg_offset(s, pg));
3310 tcg_gen_and_i64(val, val, g);
3311 tcg_temp_free_i64(g);
3312 }
3313
3314 /* Reduce the pred_esz_masks value simply to reduce the
3315 * size of the code generated here.
3316 */
3317 psz_mask = MAKE_64BIT_MASK(0, psz * 8);
3318 tcg_gen_andi_i64(val, val, pred_esz_masks[esz] & psz_mask);
3319
3320 tcg_gen_ctpop_i64(val, val);
3321 } else {
3322 TCGv_ptr t_pn = tcg_temp_new_ptr();
3323 TCGv_ptr t_pg = tcg_temp_new_ptr();
f556a201 3324 unsigned desc = 0;
9ee3a611
RH
3325 TCGv_i32 t_desc;
3326
f556a201
RH
3327 desc = FIELD_DP32(desc, PREDDESC, OPRSZ, psz);
3328 desc = FIELD_DP32(desc, PREDDESC, ESZ, esz);
9ee3a611
RH
3329
3330 tcg_gen_addi_ptr(t_pn, cpu_env, pred_full_reg_offset(s, pn));
3331 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, pg));
3332 t_desc = tcg_const_i32(desc);
3333
3334 gen_helper_sve_cntp(val, t_pn, t_pg, t_desc);
3335 tcg_temp_free_ptr(t_pn);
3336 tcg_temp_free_ptr(t_pg);
3337 tcg_temp_free_i32(t_desc);
3338 }
3339}
3340
3a7be554 3341static bool trans_CNTP(DisasContext *s, arg_CNTP *a)
9ee3a611
RH
3342{
3343 if (sve_access_check(s)) {
3344 do_cntp(s, cpu_reg(s, a->rd), a->esz, a->rn, a->pg);
3345 }
3346 return true;
3347}
3348
3a7be554 3349static bool trans_INCDECP_r(DisasContext *s, arg_incdec_pred *a)
9ee3a611
RH
3350{
3351 if (sve_access_check(s)) {
3352 TCGv_i64 reg = cpu_reg(s, a->rd);
3353 TCGv_i64 val = tcg_temp_new_i64();
3354
3355 do_cntp(s, val, a->esz, a->pg, a->pg);
3356 if (a->d) {
3357 tcg_gen_sub_i64(reg, reg, val);
3358 } else {
3359 tcg_gen_add_i64(reg, reg, val);
3360 }
3361 tcg_temp_free_i64(val);
3362 }
3363 return true;
3364}
3365
3a7be554 3366static bool trans_INCDECP_z(DisasContext *s, arg_incdec2_pred *a)
9ee3a611
RH
3367{
3368 if (a->esz == 0) {
3369 return false;
3370 }
3371 if (sve_access_check(s)) {
3372 unsigned vsz = vec_full_reg_size(s);
3373 TCGv_i64 val = tcg_temp_new_i64();
3374 GVecGen2sFn *gvec_fn = a->d ? tcg_gen_gvec_subs : tcg_gen_gvec_adds;
3375
3376 do_cntp(s, val, a->esz, a->pg, a->pg);
3377 gvec_fn(a->esz, vec_full_reg_offset(s, a->rd),
3378 vec_full_reg_offset(s, a->rn), val, vsz, vsz);
3379 }
3380 return true;
3381}
3382
3a7be554 3383static bool trans_SINCDECP_r_32(DisasContext *s, arg_incdec_pred *a)
9ee3a611
RH
3384{
3385 if (sve_access_check(s)) {
3386 TCGv_i64 reg = cpu_reg(s, a->rd);
3387 TCGv_i64 val = tcg_temp_new_i64();
3388
3389 do_cntp(s, val, a->esz, a->pg, a->pg);
3390 do_sat_addsub_32(reg, val, a->u, a->d);
3391 }
3392 return true;
3393}
3394
3a7be554 3395static bool trans_SINCDECP_r_64(DisasContext *s, arg_incdec_pred *a)
9ee3a611
RH
3396{
3397 if (sve_access_check(s)) {
3398 TCGv_i64 reg = cpu_reg(s, a->rd);
3399 TCGv_i64 val = tcg_temp_new_i64();
3400
3401 do_cntp(s, val, a->esz, a->pg, a->pg);
3402 do_sat_addsub_64(reg, val, a->u, a->d);
3403 }
3404 return true;
3405}
3406
3a7be554 3407static bool trans_SINCDECP_z(DisasContext *s, arg_incdec2_pred *a)
9ee3a611
RH
3408{
3409 if (a->esz == 0) {
3410 return false;
3411 }
3412 if (sve_access_check(s)) {
3413 TCGv_i64 val = tcg_temp_new_i64();
3414 do_cntp(s, val, a->esz, a->pg, a->pg);
3415 do_sat_addsub_vec(s, a->esz, a->rd, a->rn, val, a->u, a->d);
3416 }
3417 return true;
3418}
3419
caf1cefc
RH
3420/*
3421 *** SVE Integer Compare Scalars Group
3422 */
3423
3a7be554 3424static bool trans_CTERM(DisasContext *s, arg_CTERM *a)
caf1cefc
RH
3425{
3426 if (!sve_access_check(s)) {
3427 return true;
3428 }
3429
3430 TCGCond cond = (a->ne ? TCG_COND_NE : TCG_COND_EQ);
3431 TCGv_i64 rn = read_cpu_reg(s, a->rn, a->sf);
3432 TCGv_i64 rm = read_cpu_reg(s, a->rm, a->sf);
3433 TCGv_i64 cmp = tcg_temp_new_i64();
3434
3435 tcg_gen_setcond_i64(cond, cmp, rn, rm);
3436 tcg_gen_extrl_i64_i32(cpu_NF, cmp);
3437 tcg_temp_free_i64(cmp);
3438
3439 /* VF = !NF & !CF. */
3440 tcg_gen_xori_i32(cpu_VF, cpu_NF, 1);
3441 tcg_gen_andc_i32(cpu_VF, cpu_VF, cpu_CF);
3442
3443 /* Both NF and VF actually look at bit 31. */
3444 tcg_gen_neg_i32(cpu_NF, cpu_NF);
3445 tcg_gen_neg_i32(cpu_VF, cpu_VF);
3446 return true;
3447}
3448
3a7be554 3449static bool trans_WHILE(DisasContext *s, arg_WHILE *a)
caf1cefc 3450{
bbd0968c 3451 TCGv_i64 op0, op1, t0, t1, tmax;
caf1cefc
RH
3452 TCGv_i32 t2, t3;
3453 TCGv_ptr ptr;
e610906c
RH
3454 unsigned vsz = vec_full_reg_size(s);
3455 unsigned desc = 0;
caf1cefc 3456 TCGCond cond;
34688dbc
RH
3457 uint64_t maxval;
3458 /* Note that GE/HS has a->eq == 0 and GT/HI has a->eq == 1. */
3459 bool eq = a->eq == a->lt;
caf1cefc 3460
34688dbc
RH
3461 /* The greater-than conditions are all SVE2. */
3462 if (!a->lt && !dc_isar_feature(aa64_sve2, s)) {
3463 return false;
3464 }
bbd0968c
RH
3465 if (!sve_access_check(s)) {
3466 return true;
3467 }
3468
3469 op0 = read_cpu_reg(s, a->rn, 1);
3470 op1 = read_cpu_reg(s, a->rm, 1);
3471
caf1cefc
RH
3472 if (!a->sf) {
3473 if (a->u) {
3474 tcg_gen_ext32u_i64(op0, op0);
3475 tcg_gen_ext32u_i64(op1, op1);
3476 } else {
3477 tcg_gen_ext32s_i64(op0, op0);
3478 tcg_gen_ext32s_i64(op1, op1);
3479 }
3480 }
3481
3482 /* For the helper, compress the different conditions into a computation
3483 * of how many iterations for which the condition is true.
caf1cefc 3484 */
bbd0968c
RH
3485 t0 = tcg_temp_new_i64();
3486 t1 = tcg_temp_new_i64();
34688dbc
RH
3487
3488 if (a->lt) {
3489 tcg_gen_sub_i64(t0, op1, op0);
3490 if (a->u) {
3491 maxval = a->sf ? UINT64_MAX : UINT32_MAX;
3492 cond = eq ? TCG_COND_LEU : TCG_COND_LTU;
3493 } else {
3494 maxval = a->sf ? INT64_MAX : INT32_MAX;
3495 cond = eq ? TCG_COND_LE : TCG_COND_LT;
3496 }
3497 } else {
3498 tcg_gen_sub_i64(t0, op0, op1);
3499 if (a->u) {
3500 maxval = 0;
3501 cond = eq ? TCG_COND_GEU : TCG_COND_GTU;
3502 } else {
3503 maxval = a->sf ? INT64_MIN : INT32_MIN;
3504 cond = eq ? TCG_COND_GE : TCG_COND_GT;
3505 }
3506 }
caf1cefc 3507
bbd0968c 3508 tmax = tcg_const_i64(vsz >> a->esz);
34688dbc 3509 if (eq) {
caf1cefc
RH
3510 /* Equality means one more iteration. */
3511 tcg_gen_addi_i64(t0, t0, 1);
bbd0968c 3512
34688dbc
RH
3513 /*
3514 * For the less-than while, if op1 is maxval (and the only time
3515 * the addition above could overflow), then we produce an all-true
3516 * predicate by setting the count to the vector length. This is
3517 * because the pseudocode is described as an increment + compare
3518 * loop, and the maximum integer would always compare true.
3519 * Similarly, the greater-than while has the same issue with the
3520 * minimum integer due to the decrement + compare loop.
bbd0968c 3521 */
34688dbc 3522 tcg_gen_movi_i64(t1, maxval);
bbd0968c 3523 tcg_gen_movcond_i64(TCG_COND_EQ, t0, op1, t1, tmax, t0);
caf1cefc
RH
3524 }
3525
bbd0968c
RH
3526 /* Bound to the maximum. */
3527 tcg_gen_umin_i64(t0, t0, tmax);
3528 tcg_temp_free_i64(tmax);
3529
3530 /* Set the count to zero if the condition is false. */
caf1cefc
RH
3531 tcg_gen_movi_i64(t1, 0);
3532 tcg_gen_movcond_i64(cond, t0, op0, op1, t0, t1);
bbd0968c 3533 tcg_temp_free_i64(t1);
caf1cefc 3534
bbd0968c 3535 /* Since we're bounded, pass as a 32-bit type. */
caf1cefc
RH
3536 t2 = tcg_temp_new_i32();
3537 tcg_gen_extrl_i64_i32(t2, t0);
3538 tcg_temp_free_i64(t0);
bbd0968c
RH
3539
3540 /* Scale elements to bits. */
3541 tcg_gen_shli_i32(t2, t2, a->esz);
caf1cefc 3542
e610906c
RH
3543 desc = FIELD_DP32(desc, PREDDESC, OPRSZ, vsz / 8);
3544 desc = FIELD_DP32(desc, PREDDESC, ESZ, a->esz);
caf1cefc
RH
3545 t3 = tcg_const_i32(desc);
3546
3547 ptr = tcg_temp_new_ptr();
3548 tcg_gen_addi_ptr(ptr, cpu_env, pred_full_reg_offset(s, a->rd));
3549
34688dbc
RH
3550 if (a->lt) {
3551 gen_helper_sve_whilel(t2, ptr, t2, t3);
3552 } else {
3553 gen_helper_sve_whileg(t2, ptr, t2, t3);
3554 }
caf1cefc
RH
3555 do_pred_flags(t2);
3556
3557 tcg_temp_free_ptr(ptr);
3558 tcg_temp_free_i32(t2);
3559 tcg_temp_free_i32(t3);
3560 return true;
3561}
3562
14f6dad1
RH
3563static bool trans_WHILE_ptr(DisasContext *s, arg_WHILE_ptr *a)
3564{
3565 TCGv_i64 op0, op1, diff, t1, tmax;
3566 TCGv_i32 t2, t3;
3567 TCGv_ptr ptr;
3568 unsigned vsz = vec_full_reg_size(s);
3569 unsigned desc = 0;
3570
3571 if (!dc_isar_feature(aa64_sve2, s)) {
3572 return false;
3573 }
3574 if (!sve_access_check(s)) {
3575 return true;
3576 }
3577
3578 op0 = read_cpu_reg(s, a->rn, 1);
3579 op1 = read_cpu_reg(s, a->rm, 1);
3580
3581 tmax = tcg_const_i64(vsz);
3582 diff = tcg_temp_new_i64();
3583
3584 if (a->rw) {
3585 /* WHILERW */
3586 /* diff = abs(op1 - op0), noting that op0/1 are unsigned. */
3587 t1 = tcg_temp_new_i64();
3588 tcg_gen_sub_i64(diff, op0, op1);
3589 tcg_gen_sub_i64(t1, op1, op0);
3590 tcg_gen_movcond_i64(TCG_COND_GEU, diff, op0, op1, diff, t1);
3591 tcg_temp_free_i64(t1);
3592 /* Round down to a multiple of ESIZE. */
3593 tcg_gen_andi_i64(diff, diff, -1 << a->esz);
3594 /* If op1 == op0, diff == 0, and the condition is always true. */
3595 tcg_gen_movcond_i64(TCG_COND_EQ, diff, op0, op1, tmax, diff);
3596 } else {
3597 /* WHILEWR */
3598 tcg_gen_sub_i64(diff, op1, op0);
3599 /* Round down to a multiple of ESIZE. */
3600 tcg_gen_andi_i64(diff, diff, -1 << a->esz);
3601 /* If op0 >= op1, diff <= 0, the condition is always true. */
3602 tcg_gen_movcond_i64(TCG_COND_GEU, diff, op0, op1, tmax, diff);
3603 }
3604
3605 /* Bound to the maximum. */
3606 tcg_gen_umin_i64(diff, diff, tmax);
3607 tcg_temp_free_i64(tmax);
3608
3609 /* Since we're bounded, pass as a 32-bit type. */
3610 t2 = tcg_temp_new_i32();
3611 tcg_gen_extrl_i64_i32(t2, diff);
3612 tcg_temp_free_i64(diff);
3613
3614 desc = FIELD_DP32(desc, PREDDESC, OPRSZ, vsz / 8);
3615 desc = FIELD_DP32(desc, PREDDESC, ESZ, a->esz);
3616 t3 = tcg_const_i32(desc);
3617
3618 ptr = tcg_temp_new_ptr();
3619 tcg_gen_addi_ptr(ptr, cpu_env, pred_full_reg_offset(s, a->rd));
3620
3621 gen_helper_sve_whilel(t2, ptr, t2, t3);
3622 do_pred_flags(t2);
3623
3624 tcg_temp_free_ptr(ptr);
3625 tcg_temp_free_i32(t2);
3626 tcg_temp_free_i32(t3);
3627 return true;
3628}
3629
ed491961
RH
3630/*
3631 *** SVE Integer Wide Immediate - Unpredicated Group
3632 */
3633
3a7be554 3634static bool trans_FDUP(DisasContext *s, arg_FDUP *a)
ed491961
RH
3635{
3636 if (a->esz == 0) {
3637 return false;
3638 }
3639 if (sve_access_check(s)) {
3640 unsigned vsz = vec_full_reg_size(s);
3641 int dofs = vec_full_reg_offset(s, a->rd);
3642 uint64_t imm;
3643
3644 /* Decode the VFP immediate. */
3645 imm = vfp_expand_imm(a->esz, a->imm);
8711e71f 3646 tcg_gen_gvec_dup_imm(a->esz, dofs, vsz, vsz, imm);
ed491961
RH
3647 }
3648 return true;
3649}
3650
3a7be554 3651static bool trans_DUP_i(DisasContext *s, arg_DUP_i *a)
ed491961 3652{
3a7be554 3653 if (a->esz == 0 && extract32(s->insn, 13, 1)) {
ed491961
RH
3654 return false;
3655 }
3656 if (sve_access_check(s)) {
3657 unsigned vsz = vec_full_reg_size(s);
3658 int dofs = vec_full_reg_offset(s, a->rd);
3659
8711e71f 3660 tcg_gen_gvec_dup_imm(a->esz, dofs, vsz, vsz, a->imm);
ed491961
RH
3661 }
3662 return true;
3663}
3664
3a7be554 3665static bool trans_ADD_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d 3666{
3a7be554 3667 if (a->esz == 0 && extract32(s->insn, 13, 1)) {
6e6a157d
RH
3668 return false;
3669 }
3670 if (sve_access_check(s)) {
3671 unsigned vsz = vec_full_reg_size(s);
3672 tcg_gen_gvec_addi(a->esz, vec_full_reg_offset(s, a->rd),
3673 vec_full_reg_offset(s, a->rn), a->imm, vsz, vsz);
3674 }
3675 return true;
3676}
3677
3a7be554 3678static bool trans_SUB_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d
RH
3679{
3680 a->imm = -a->imm;
3a7be554 3681 return trans_ADD_zzi(s, a);
6e6a157d
RH
3682}
3683
3a7be554 3684static bool trans_SUBR_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d 3685{
53229a77 3686 static const TCGOpcode vecop_list[] = { INDEX_op_sub_vec, 0 };
6e6a157d
RH
3687 static const GVecGen2s op[4] = {
3688 { .fni8 = tcg_gen_vec_sub8_i64,
3689 .fniv = tcg_gen_sub_vec,
3690 .fno = gen_helper_sve_subri_b,
53229a77 3691 .opt_opc = vecop_list,
6e6a157d
RH
3692 .vece = MO_8,
3693 .scalar_first = true },
3694 { .fni8 = tcg_gen_vec_sub16_i64,
3695 .fniv = tcg_gen_sub_vec,
3696 .fno = gen_helper_sve_subri_h,
53229a77 3697 .opt_opc = vecop_list,
6e6a157d
RH
3698 .vece = MO_16,
3699 .scalar_first = true },
3700 { .fni4 = tcg_gen_sub_i32,
3701 .fniv = tcg_gen_sub_vec,
3702 .fno = gen_helper_sve_subri_s,
53229a77 3703 .opt_opc = vecop_list,
6e6a157d
RH
3704 .vece = MO_32,
3705 .scalar_first = true },
3706 { .fni8 = tcg_gen_sub_i64,
3707 .fniv = tcg_gen_sub_vec,
3708 .fno = gen_helper_sve_subri_d,
53229a77 3709 .opt_opc = vecop_list,
6e6a157d
RH
3710 .prefer_i64 = TCG_TARGET_REG_BITS == 64,
3711 .vece = MO_64,
3712 .scalar_first = true }
3713 };
3714
3a7be554 3715 if (a->esz == 0 && extract32(s->insn, 13, 1)) {
6e6a157d
RH
3716 return false;
3717 }
3718 if (sve_access_check(s)) {
3719 unsigned vsz = vec_full_reg_size(s);
3720 TCGv_i64 c = tcg_const_i64(a->imm);
3721 tcg_gen_gvec_2s(vec_full_reg_offset(s, a->rd),
3722 vec_full_reg_offset(s, a->rn),
3723 vsz, vsz, c, &op[a->esz]);
3724 tcg_temp_free_i64(c);
3725 }
3726 return true;
3727}
3728
3a7be554 3729static bool trans_MUL_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d
RH
3730{
3731 if (sve_access_check(s)) {
3732 unsigned vsz = vec_full_reg_size(s);
3733 tcg_gen_gvec_muli(a->esz, vec_full_reg_offset(s, a->rd),
3734 vec_full_reg_offset(s, a->rn), a->imm, vsz, vsz);
3735 }
3736 return true;
3737}
3738
3a7be554 3739static bool do_zzi_sat(DisasContext *s, arg_rri_esz *a, bool u, bool d)
6e6a157d 3740{
3a7be554 3741 if (a->esz == 0 && extract32(s->insn, 13, 1)) {
6e6a157d
RH
3742 return false;
3743 }
3744 if (sve_access_check(s)) {
3745 TCGv_i64 val = tcg_const_i64(a->imm);
3746 do_sat_addsub_vec(s, a->esz, a->rd, a->rn, val, u, d);
3747 tcg_temp_free_i64(val);
3748 }
3749 return true;
3750}
3751
3a7be554 3752static bool trans_SQADD_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d 3753{
3a7be554 3754 return do_zzi_sat(s, a, false, false);
6e6a157d
RH
3755}
3756
3a7be554 3757static bool trans_UQADD_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d 3758{
3a7be554 3759 return do_zzi_sat(s, a, true, false);
6e6a157d
RH
3760}
3761
3a7be554 3762static bool trans_SQSUB_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d 3763{
3a7be554 3764 return do_zzi_sat(s, a, false, true);
6e6a157d
RH
3765}
3766
3a7be554 3767static bool trans_UQSUB_zzi(DisasContext *s, arg_rri_esz *a)
6e6a157d 3768{
3a7be554 3769 return do_zzi_sat(s, a, true, true);
6e6a157d
RH
3770}
3771
3772static bool do_zzi_ool(DisasContext *s, arg_rri_esz *a, gen_helper_gvec_2i *fn)
3773{
3774 if (sve_access_check(s)) {
3775 unsigned vsz = vec_full_reg_size(s);
3776 TCGv_i64 c = tcg_const_i64(a->imm);
3777
3778 tcg_gen_gvec_2i_ool(vec_full_reg_offset(s, a->rd),
3779 vec_full_reg_offset(s, a->rn),
3780 c, vsz, vsz, 0, fn);
3781 tcg_temp_free_i64(c);
3782 }
3783 return true;
3784}
3785
3786#define DO_ZZI(NAME, name) \
3a7be554 3787static bool trans_##NAME##_zzi(DisasContext *s, arg_rri_esz *a) \
6e6a157d
RH
3788{ \
3789 static gen_helper_gvec_2i * const fns[4] = { \
3790 gen_helper_sve_##name##i_b, gen_helper_sve_##name##i_h, \
3791 gen_helper_sve_##name##i_s, gen_helper_sve_##name##i_d, \
3792 }; \
3793 return do_zzi_ool(s, a, fns[a->esz]); \
3794}
3795
3796DO_ZZI(SMAX, smax)
3797DO_ZZI(UMAX, umax)
3798DO_ZZI(SMIN, smin)
3799DO_ZZI(UMIN, umin)
3800
3801#undef DO_ZZI
3802
bc2bd697 3803static bool trans_DOT_zzzz(DisasContext *s, arg_DOT_zzzz *a)
d730ecaa 3804{
bc2bd697 3805 static gen_helper_gvec_4 * const fns[2][2] = {
d730ecaa
RH
3806 { gen_helper_gvec_sdot_b, gen_helper_gvec_sdot_h },
3807 { gen_helper_gvec_udot_b, gen_helper_gvec_udot_h }
3808 };
3809
3810 if (sve_access_check(s)) {
bc2bd697 3811 gen_gvec_ool_zzzz(s, fns[a->u][a->sz], a->rd, a->rn, a->rm, a->ra, 0);
d730ecaa
RH
3812 }
3813 return true;
3814}
3815
814d4c52
RH
3816/*
3817 * SVE Multiply - Indexed
3818 */
3819
0a82d963
RH
3820static bool do_zzxz_ool(DisasContext *s, arg_rrxr_esz *a,
3821 gen_helper_gvec_4 *fn)
16fcfdc7 3822{
0a82d963
RH
3823 if (fn == NULL) {
3824 return false;
3825 }
16fcfdc7 3826 if (sve_access_check(s)) {
0a82d963 3827 gen_gvec_ool_zzzz(s, fn, a->rd, a->rn, a->rm, a->ra, a->index);
16fcfdc7
RH
3828 }
3829 return true;
3830}
3831
0a82d963
RH
3832#define DO_RRXR(NAME, FUNC) \
3833 static bool NAME(DisasContext *s, arg_rrxr_esz *a) \
3834 { return do_zzxz_ool(s, a, FUNC); }
3835
3836DO_RRXR(trans_SDOT_zzxw_s, gen_helper_gvec_sdot_idx_b)
3837DO_RRXR(trans_SDOT_zzxw_d, gen_helper_gvec_sdot_idx_h)
3838DO_RRXR(trans_UDOT_zzxw_s, gen_helper_gvec_udot_idx_b)
3839DO_RRXR(trans_UDOT_zzxw_d, gen_helper_gvec_udot_idx_h)
3840
3841#undef DO_RRXR
16fcfdc7 3842
814d4c52
RH
3843static bool do_sve2_zzz_data(DisasContext *s, int rd, int rn, int rm, int data,
3844 gen_helper_gvec_3 *fn)
3845{
3846 if (fn == NULL || !dc_isar_feature(aa64_sve2, s)) {
3847 return false;
3848 }
3849 if (sve_access_check(s)) {
3850 unsigned vsz = vec_full_reg_size(s);
3851 tcg_gen_gvec_3_ool(vec_full_reg_offset(s, rd),
3852 vec_full_reg_offset(s, rn),
3853 vec_full_reg_offset(s, rm),
3854 vsz, vsz, data, fn);
3855 }
3856 return true;
3857}
3858
3859#define DO_SVE2_RRX(NAME, FUNC) \
3860 static bool NAME(DisasContext *s, arg_rrx_esz *a) \
3861 { return do_sve2_zzz_data(s, a->rd, a->rn, a->rm, a->index, FUNC); }
3862
3863DO_SVE2_RRX(trans_MUL_zzx_h, gen_helper_gvec_mul_idx_h)
3864DO_SVE2_RRX(trans_MUL_zzx_s, gen_helper_gvec_mul_idx_s)
3865DO_SVE2_RRX(trans_MUL_zzx_d, gen_helper_gvec_mul_idx_d)
3866
3867#undef DO_SVE2_RRX
3868
8a02aac7
RH
3869static bool do_sve2_zzzz_data(DisasContext *s, int rd, int rn, int rm, int ra,
3870 int data, gen_helper_gvec_4 *fn)
3871{
3872 if (fn == NULL || !dc_isar_feature(aa64_sve2, s)) {
3873 return false;
3874 }
3875 if (sve_access_check(s)) {
3876 unsigned vsz = vec_full_reg_size(s);
3877 tcg_gen_gvec_4_ool(vec_full_reg_offset(s, rd),
3878 vec_full_reg_offset(s, rn),
3879 vec_full_reg_offset(s, rm),
3880 vec_full_reg_offset(s, ra),
3881 vsz, vsz, data, fn);
3882 }
3883 return true;
3884}
3885
3886#define DO_SVE2_RRXR(NAME, FUNC) \
3887 static bool NAME(DisasContext *s, arg_rrxr_esz *a) \
3888 { return do_sve2_zzzz_data(s, a->rd, a->rn, a->rm, a->ra, a->index, FUNC); }
3889
3890DO_SVE2_RRXR(trans_MLA_zzxz_h, gen_helper_gvec_mla_idx_h)
3891DO_SVE2_RRXR(trans_MLA_zzxz_s, gen_helper_gvec_mla_idx_s)
3892DO_SVE2_RRXR(trans_MLA_zzxz_d, gen_helper_gvec_mla_idx_d)
3893
3894DO_SVE2_RRXR(trans_MLS_zzxz_h, gen_helper_gvec_mls_idx_h)
3895DO_SVE2_RRXR(trans_MLS_zzxz_s, gen_helper_gvec_mls_idx_s)
3896DO_SVE2_RRXR(trans_MLS_zzxz_d, gen_helper_gvec_mls_idx_d)
3897
75d6d5fc
RH
3898DO_SVE2_RRXR(trans_SQRDMLAH_zzxz_h, gen_helper_sve2_sqrdmlah_idx_h)
3899DO_SVE2_RRXR(trans_SQRDMLAH_zzxz_s, gen_helper_sve2_sqrdmlah_idx_s)
3900DO_SVE2_RRXR(trans_SQRDMLAH_zzxz_d, gen_helper_sve2_sqrdmlah_idx_d)
3901
3902DO_SVE2_RRXR(trans_SQRDMLSH_zzxz_h, gen_helper_sve2_sqrdmlsh_idx_h)
3903DO_SVE2_RRXR(trans_SQRDMLSH_zzxz_s, gen_helper_sve2_sqrdmlsh_idx_s)
3904DO_SVE2_RRXR(trans_SQRDMLSH_zzxz_d, gen_helper_sve2_sqrdmlsh_idx_d)
3905
8a02aac7
RH
3906#undef DO_SVE2_RRXR
3907
ca40a6e6
RH
3908/*
3909 *** SVE Floating Point Multiply-Add Indexed Group
3910 */
3911
0a82d963 3912static bool do_FMLA_zzxz(DisasContext *s, arg_rrxr_esz *a, bool sub)
ca40a6e6
RH
3913{
3914 static gen_helper_gvec_4_ptr * const fns[3] = {
3915 gen_helper_gvec_fmla_idx_h,
3916 gen_helper_gvec_fmla_idx_s,
3917 gen_helper_gvec_fmla_idx_d,
3918 };
3919
3920 if (sve_access_check(s)) {
3921 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 3922 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
ca40a6e6
RH
3923 tcg_gen_gvec_4_ptr(vec_full_reg_offset(s, a->rd),
3924 vec_full_reg_offset(s, a->rn),
3925 vec_full_reg_offset(s, a->rm),
3926 vec_full_reg_offset(s, a->ra),
0a82d963 3927 status, vsz, vsz, (a->index << 1) | sub,
ca40a6e6
RH
3928 fns[a->esz - 1]);
3929 tcg_temp_free_ptr(status);
3930 }
3931 return true;
3932}
3933
0a82d963
RH
3934static bool trans_FMLA_zzxz(DisasContext *s, arg_FMLA_zzxz *a)
3935{
3936 return do_FMLA_zzxz(s, a, false);
3937}
3938
3939static bool trans_FMLS_zzxz(DisasContext *s, arg_FMLA_zzxz *a)
3940{
3941 return do_FMLA_zzxz(s, a, true);
3942}
3943
ca40a6e6
RH
3944/*
3945 *** SVE Floating Point Multiply Indexed Group
3946 */
3947
3a7be554 3948static bool trans_FMUL_zzx(DisasContext *s, arg_FMUL_zzx *a)
ca40a6e6
RH
3949{
3950 static gen_helper_gvec_3_ptr * const fns[3] = {
3951 gen_helper_gvec_fmul_idx_h,
3952 gen_helper_gvec_fmul_idx_s,
3953 gen_helper_gvec_fmul_idx_d,
3954 };
3955
3956 if (sve_access_check(s)) {
3957 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 3958 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
ca40a6e6
RH
3959 tcg_gen_gvec_3_ptr(vec_full_reg_offset(s, a->rd),
3960 vec_full_reg_offset(s, a->rn),
3961 vec_full_reg_offset(s, a->rm),
3962 status, vsz, vsz, a->index, fns[a->esz - 1]);
3963 tcg_temp_free_ptr(status);
3964 }
3965 return true;
3966}
3967
23fbe79f
RH
3968/*
3969 *** SVE Floating Point Fast Reduction Group
3970 */
3971
3972typedef void gen_helper_fp_reduce(TCGv_i64, TCGv_ptr, TCGv_ptr,
3973 TCGv_ptr, TCGv_i32);
3974
3975static void do_reduce(DisasContext *s, arg_rpr_esz *a,
3976 gen_helper_fp_reduce *fn)
3977{
3978 unsigned vsz = vec_full_reg_size(s);
3979 unsigned p2vsz = pow2ceil(vsz);
c648c9b7 3980 TCGv_i32 t_desc = tcg_const_i32(simd_desc(vsz, vsz, p2vsz));
23fbe79f
RH
3981 TCGv_ptr t_zn, t_pg, status;
3982 TCGv_i64 temp;
3983
3984 temp = tcg_temp_new_i64();
3985 t_zn = tcg_temp_new_ptr();
3986 t_pg = tcg_temp_new_ptr();
3987
3988 tcg_gen_addi_ptr(t_zn, cpu_env, vec_full_reg_offset(s, a->rn));
3989 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, a->pg));
cdfb22bb 3990 status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
23fbe79f
RH
3991
3992 fn(temp, t_zn, t_pg, status, t_desc);
3993 tcg_temp_free_ptr(t_zn);
3994 tcg_temp_free_ptr(t_pg);
3995 tcg_temp_free_ptr(status);
3996 tcg_temp_free_i32(t_desc);
3997
3998 write_fp_dreg(s, a->rd, temp);
3999 tcg_temp_free_i64(temp);
4000}
4001
4002#define DO_VPZ(NAME, name) \
3a7be554 4003static bool trans_##NAME(DisasContext *s, arg_rpr_esz *a) \
23fbe79f
RH
4004{ \
4005 static gen_helper_fp_reduce * const fns[3] = { \
4006 gen_helper_sve_##name##_h, \
4007 gen_helper_sve_##name##_s, \
4008 gen_helper_sve_##name##_d, \
4009 }; \
4010 if (a->esz == 0) { \
4011 return false; \
4012 } \
4013 if (sve_access_check(s)) { \
4014 do_reduce(s, a, fns[a->esz - 1]); \
4015 } \
4016 return true; \
4017}
4018
4019DO_VPZ(FADDV, faddv)
4020DO_VPZ(FMINNMV, fminnmv)
4021DO_VPZ(FMAXNMV, fmaxnmv)
4022DO_VPZ(FMINV, fminv)
4023DO_VPZ(FMAXV, fmaxv)
4024
3887c038
RH
4025/*
4026 *** SVE Floating Point Unary Operations - Unpredicated Group
4027 */
4028
4029static void do_zz_fp(DisasContext *s, arg_rr_esz *a, gen_helper_gvec_2_ptr *fn)
4030{
4031 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4032 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
3887c038
RH
4033
4034 tcg_gen_gvec_2_ptr(vec_full_reg_offset(s, a->rd),
4035 vec_full_reg_offset(s, a->rn),
4036 status, vsz, vsz, 0, fn);
4037 tcg_temp_free_ptr(status);
4038}
4039
3a7be554 4040static bool trans_FRECPE(DisasContext *s, arg_rr_esz *a)
3887c038
RH
4041{
4042 static gen_helper_gvec_2_ptr * const fns[3] = {
4043 gen_helper_gvec_frecpe_h,
4044 gen_helper_gvec_frecpe_s,
4045 gen_helper_gvec_frecpe_d,
4046 };
4047 if (a->esz == 0) {
4048 return false;
4049 }
4050 if (sve_access_check(s)) {
4051 do_zz_fp(s, a, fns[a->esz - 1]);
4052 }
4053 return true;
4054}
4055
3a7be554 4056static bool trans_FRSQRTE(DisasContext *s, arg_rr_esz *a)
3887c038
RH
4057{
4058 static gen_helper_gvec_2_ptr * const fns[3] = {
4059 gen_helper_gvec_frsqrte_h,
4060 gen_helper_gvec_frsqrte_s,
4061 gen_helper_gvec_frsqrte_d,
4062 };
4063 if (a->esz == 0) {
4064 return false;
4065 }
4066 if (sve_access_check(s)) {
4067 do_zz_fp(s, a, fns[a->esz - 1]);
4068 }
4069 return true;
4070}
4071
4d2e2a03
RH
4072/*
4073 *** SVE Floating Point Compare with Zero Group
4074 */
4075
4076static void do_ppz_fp(DisasContext *s, arg_rpr_esz *a,
4077 gen_helper_gvec_3_ptr *fn)
4078{
4079 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4080 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
4d2e2a03
RH
4081
4082 tcg_gen_gvec_3_ptr(pred_full_reg_offset(s, a->rd),
4083 vec_full_reg_offset(s, a->rn),
4084 pred_full_reg_offset(s, a->pg),
4085 status, vsz, vsz, 0, fn);
4086 tcg_temp_free_ptr(status);
4087}
4088
4089#define DO_PPZ(NAME, name) \
3a7be554 4090static bool trans_##NAME(DisasContext *s, arg_rpr_esz *a) \
4d2e2a03
RH
4091{ \
4092 static gen_helper_gvec_3_ptr * const fns[3] = { \
4093 gen_helper_sve_##name##_h, \
4094 gen_helper_sve_##name##_s, \
4095 gen_helper_sve_##name##_d, \
4096 }; \
4097 if (a->esz == 0) { \
4098 return false; \
4099 } \
4100 if (sve_access_check(s)) { \
4101 do_ppz_fp(s, a, fns[a->esz - 1]); \
4102 } \
4103 return true; \
4104}
4105
4106DO_PPZ(FCMGE_ppz0, fcmge0)
4107DO_PPZ(FCMGT_ppz0, fcmgt0)
4108DO_PPZ(FCMLE_ppz0, fcmle0)
4109DO_PPZ(FCMLT_ppz0, fcmlt0)
4110DO_PPZ(FCMEQ_ppz0, fcmeq0)
4111DO_PPZ(FCMNE_ppz0, fcmne0)
4112
4113#undef DO_PPZ
4114
67fcd9ad
RH
4115/*
4116 *** SVE floating-point trig multiply-add coefficient
4117 */
4118
3a7be554 4119static bool trans_FTMAD(DisasContext *s, arg_FTMAD *a)
67fcd9ad
RH
4120{
4121 static gen_helper_gvec_3_ptr * const fns[3] = {
4122 gen_helper_sve_ftmad_h,
4123 gen_helper_sve_ftmad_s,
4124 gen_helper_sve_ftmad_d,
4125 };
4126
4127 if (a->esz == 0) {
4128 return false;
4129 }
4130 if (sve_access_check(s)) {
4131 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4132 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
67fcd9ad
RH
4133 tcg_gen_gvec_3_ptr(vec_full_reg_offset(s, a->rd),
4134 vec_full_reg_offset(s, a->rn),
4135 vec_full_reg_offset(s, a->rm),
4136 status, vsz, vsz, a->imm, fns[a->esz - 1]);
4137 tcg_temp_free_ptr(status);
4138 }
4139 return true;
4140}
4141
7f9ddf64
RH
4142/*
4143 *** SVE Floating Point Accumulating Reduction Group
4144 */
4145
3a7be554 4146static bool trans_FADDA(DisasContext *s, arg_rprr_esz *a)
7f9ddf64
RH
4147{
4148 typedef void fadda_fn(TCGv_i64, TCGv_i64, TCGv_ptr,
4149 TCGv_ptr, TCGv_ptr, TCGv_i32);
4150 static fadda_fn * const fns[3] = {
4151 gen_helper_sve_fadda_h,
4152 gen_helper_sve_fadda_s,
4153 gen_helper_sve_fadda_d,
4154 };
4155 unsigned vsz = vec_full_reg_size(s);
4156 TCGv_ptr t_rm, t_pg, t_fpst;
4157 TCGv_i64 t_val;
4158 TCGv_i32 t_desc;
4159
4160 if (a->esz == 0) {
4161 return false;
4162 }
4163 if (!sve_access_check(s)) {
4164 return true;
4165 }
4166
4167 t_val = load_esz(cpu_env, vec_reg_offset(s, a->rn, 0, a->esz), a->esz);
4168 t_rm = tcg_temp_new_ptr();
4169 t_pg = tcg_temp_new_ptr();
4170 tcg_gen_addi_ptr(t_rm, cpu_env, vec_full_reg_offset(s, a->rm));
4171 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, a->pg));
cdfb22bb 4172 t_fpst = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
7f9ddf64
RH
4173 t_desc = tcg_const_i32(simd_desc(vsz, vsz, 0));
4174
4175 fns[a->esz - 1](t_val, t_val, t_rm, t_pg, t_fpst, t_desc);
4176
4177 tcg_temp_free_i32(t_desc);
4178 tcg_temp_free_ptr(t_fpst);
4179 tcg_temp_free_ptr(t_pg);
4180 tcg_temp_free_ptr(t_rm);
4181
4182 write_fp_dreg(s, a->rd, t_val);
4183 tcg_temp_free_i64(t_val);
4184 return true;
4185}
4186
29b80469
RH
4187/*
4188 *** SVE Floating Point Arithmetic - Unpredicated Group
4189 */
4190
4191static bool do_zzz_fp(DisasContext *s, arg_rrr_esz *a,
4192 gen_helper_gvec_3_ptr *fn)
4193{
4194 if (fn == NULL) {
4195 return false;
4196 }
4197 if (sve_access_check(s)) {
4198 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4199 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
29b80469
RH
4200 tcg_gen_gvec_3_ptr(vec_full_reg_offset(s, a->rd),
4201 vec_full_reg_offset(s, a->rn),
4202 vec_full_reg_offset(s, a->rm),
4203 status, vsz, vsz, 0, fn);
4204 tcg_temp_free_ptr(status);
4205 }
4206 return true;
4207}
4208
4209
4210#define DO_FP3(NAME, name) \
3a7be554 4211static bool trans_##NAME(DisasContext *s, arg_rrr_esz *a) \
29b80469
RH
4212{ \
4213 static gen_helper_gvec_3_ptr * const fns[4] = { \
4214 NULL, gen_helper_gvec_##name##_h, \
4215 gen_helper_gvec_##name##_s, gen_helper_gvec_##name##_d \
4216 }; \
4217 return do_zzz_fp(s, a, fns[a->esz]); \
4218}
4219
4220DO_FP3(FADD_zzz, fadd)
4221DO_FP3(FSUB_zzz, fsub)
4222DO_FP3(FMUL_zzz, fmul)
4223DO_FP3(FTSMUL, ftsmul)
4224DO_FP3(FRECPS, recps)
4225DO_FP3(FRSQRTS, rsqrts)
4226
4227#undef DO_FP3
4228
ec3b87c2
RH
4229/*
4230 *** SVE Floating Point Arithmetic - Predicated Group
4231 */
4232
4233static bool do_zpzz_fp(DisasContext *s, arg_rprr_esz *a,
4234 gen_helper_gvec_4_ptr *fn)
4235{
4236 if (fn == NULL) {
4237 return false;
4238 }
4239 if (sve_access_check(s)) {
4240 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4241 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
ec3b87c2
RH
4242 tcg_gen_gvec_4_ptr(vec_full_reg_offset(s, a->rd),
4243 vec_full_reg_offset(s, a->rn),
4244 vec_full_reg_offset(s, a->rm),
4245 pred_full_reg_offset(s, a->pg),
4246 status, vsz, vsz, 0, fn);
4247 tcg_temp_free_ptr(status);
4248 }
4249 return true;
4250}
4251
4252#define DO_FP3(NAME, name) \
3a7be554 4253static bool trans_##NAME(DisasContext *s, arg_rprr_esz *a) \
ec3b87c2
RH
4254{ \
4255 static gen_helper_gvec_4_ptr * const fns[4] = { \
4256 NULL, gen_helper_sve_##name##_h, \
4257 gen_helper_sve_##name##_s, gen_helper_sve_##name##_d \
4258 }; \
4259 return do_zpzz_fp(s, a, fns[a->esz]); \
4260}
4261
4262DO_FP3(FADD_zpzz, fadd)
4263DO_FP3(FSUB_zpzz, fsub)
4264DO_FP3(FMUL_zpzz, fmul)
4265DO_FP3(FMIN_zpzz, fmin)
4266DO_FP3(FMAX_zpzz, fmax)
4267DO_FP3(FMINNM_zpzz, fminnum)
4268DO_FP3(FMAXNM_zpzz, fmaxnum)
4269DO_FP3(FABD, fabd)
4270DO_FP3(FSCALE, fscalbn)
4271DO_FP3(FDIV, fdiv)
4272DO_FP3(FMULX, fmulx)
4273
4274#undef DO_FP3
8092c6a3 4275
cc48affe
RH
4276typedef void gen_helper_sve_fp2scalar(TCGv_ptr, TCGv_ptr, TCGv_ptr,
4277 TCGv_i64, TCGv_ptr, TCGv_i32);
4278
4279static void do_fp_scalar(DisasContext *s, int zd, int zn, int pg, bool is_fp16,
4280 TCGv_i64 scalar, gen_helper_sve_fp2scalar *fn)
4281{
4282 unsigned vsz = vec_full_reg_size(s);
4283 TCGv_ptr t_zd, t_zn, t_pg, status;
4284 TCGv_i32 desc;
4285
4286 t_zd = tcg_temp_new_ptr();
4287 t_zn = tcg_temp_new_ptr();
4288 t_pg = tcg_temp_new_ptr();
4289 tcg_gen_addi_ptr(t_zd, cpu_env, vec_full_reg_offset(s, zd));
4290 tcg_gen_addi_ptr(t_zn, cpu_env, vec_full_reg_offset(s, zn));
4291 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, pg));
4292
cdfb22bb 4293 status = fpstatus_ptr(is_fp16 ? FPST_FPCR_F16 : FPST_FPCR);
cc48affe
RH
4294 desc = tcg_const_i32(simd_desc(vsz, vsz, 0));
4295 fn(t_zd, t_zn, t_pg, scalar, status, desc);
4296
4297 tcg_temp_free_i32(desc);
4298 tcg_temp_free_ptr(status);
4299 tcg_temp_free_ptr(t_pg);
4300 tcg_temp_free_ptr(t_zn);
4301 tcg_temp_free_ptr(t_zd);
4302}
4303
4304static void do_fp_imm(DisasContext *s, arg_rpri_esz *a, uint64_t imm,
4305 gen_helper_sve_fp2scalar *fn)
4306{
4307 TCGv_i64 temp = tcg_const_i64(imm);
4308 do_fp_scalar(s, a->rd, a->rn, a->pg, a->esz == MO_16, temp, fn);
4309 tcg_temp_free_i64(temp);
4310}
4311
4312#define DO_FP_IMM(NAME, name, const0, const1) \
3a7be554 4313static bool trans_##NAME##_zpzi(DisasContext *s, arg_rpri_esz *a) \
cc48affe
RH
4314{ \
4315 static gen_helper_sve_fp2scalar * const fns[3] = { \
4316 gen_helper_sve_##name##_h, \
4317 gen_helper_sve_##name##_s, \
4318 gen_helper_sve_##name##_d \
4319 }; \
4320 static uint64_t const val[3][2] = { \
4321 { float16_##const0, float16_##const1 }, \
4322 { float32_##const0, float32_##const1 }, \
4323 { float64_##const0, float64_##const1 }, \
4324 }; \
4325 if (a->esz == 0) { \
4326 return false; \
4327 } \
4328 if (sve_access_check(s)) { \
4329 do_fp_imm(s, a, val[a->esz - 1][a->imm], fns[a->esz - 1]); \
4330 } \
4331 return true; \
4332}
4333
cc48affe
RH
4334DO_FP_IMM(FADD, fadds, half, one)
4335DO_FP_IMM(FSUB, fsubs, half, one)
4336DO_FP_IMM(FMUL, fmuls, half, two)
4337DO_FP_IMM(FSUBR, fsubrs, half, one)
4338DO_FP_IMM(FMAXNM, fmaxnms, zero, one)
4339DO_FP_IMM(FMINNM, fminnms, zero, one)
4340DO_FP_IMM(FMAX, fmaxs, zero, one)
4341DO_FP_IMM(FMIN, fmins, zero, one)
4342
4343#undef DO_FP_IMM
4344
abfdefd5
RH
4345static bool do_fp_cmp(DisasContext *s, arg_rprr_esz *a,
4346 gen_helper_gvec_4_ptr *fn)
4347{
4348 if (fn == NULL) {
4349 return false;
4350 }
4351 if (sve_access_check(s)) {
4352 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4353 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
abfdefd5
RH
4354 tcg_gen_gvec_4_ptr(pred_full_reg_offset(s, a->rd),
4355 vec_full_reg_offset(s, a->rn),
4356 vec_full_reg_offset(s, a->rm),
4357 pred_full_reg_offset(s, a->pg),
4358 status, vsz, vsz, 0, fn);
4359 tcg_temp_free_ptr(status);
4360 }
4361 return true;
4362}
4363
4364#define DO_FPCMP(NAME, name) \
3a7be554 4365static bool trans_##NAME##_ppzz(DisasContext *s, arg_rprr_esz *a) \
abfdefd5
RH
4366{ \
4367 static gen_helper_gvec_4_ptr * const fns[4] = { \
4368 NULL, gen_helper_sve_##name##_h, \
4369 gen_helper_sve_##name##_s, gen_helper_sve_##name##_d \
4370 }; \
4371 return do_fp_cmp(s, a, fns[a->esz]); \
4372}
4373
4374DO_FPCMP(FCMGE, fcmge)
4375DO_FPCMP(FCMGT, fcmgt)
4376DO_FPCMP(FCMEQ, fcmeq)
4377DO_FPCMP(FCMNE, fcmne)
4378DO_FPCMP(FCMUO, fcmuo)
4379DO_FPCMP(FACGE, facge)
4380DO_FPCMP(FACGT, facgt)
4381
4382#undef DO_FPCMP
4383
3a7be554 4384static bool trans_FCADD(DisasContext *s, arg_FCADD *a)
76a9d9cd
RH
4385{
4386 static gen_helper_gvec_4_ptr * const fns[3] = {
4387 gen_helper_sve_fcadd_h,
4388 gen_helper_sve_fcadd_s,
4389 gen_helper_sve_fcadd_d
4390 };
4391
4392 if (a->esz == 0) {
4393 return false;
4394 }
4395 if (sve_access_check(s)) {
4396 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4397 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
76a9d9cd
RH
4398 tcg_gen_gvec_4_ptr(vec_full_reg_offset(s, a->rd),
4399 vec_full_reg_offset(s, a->rn),
4400 vec_full_reg_offset(s, a->rm),
4401 pred_full_reg_offset(s, a->pg),
4402 status, vsz, vsz, a->rot, fns[a->esz - 1]);
4403 tcg_temp_free_ptr(status);
4404 }
4405 return true;
4406}
4407
08975da9
RH
4408static bool do_fmla(DisasContext *s, arg_rprrr_esz *a,
4409 gen_helper_gvec_5_ptr *fn)
6ceabaad 4410{
08975da9 4411 if (a->esz == 0) {
6ceabaad
RH
4412 return false;
4413 }
08975da9
RH
4414 if (sve_access_check(s)) {
4415 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4416 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
08975da9
RH
4417 tcg_gen_gvec_5_ptr(vec_full_reg_offset(s, a->rd),
4418 vec_full_reg_offset(s, a->rn),
4419 vec_full_reg_offset(s, a->rm),
4420 vec_full_reg_offset(s, a->ra),
4421 pred_full_reg_offset(s, a->pg),
4422 status, vsz, vsz, 0, fn);
4423 tcg_temp_free_ptr(status);
6ceabaad 4424 }
6ceabaad
RH
4425 return true;
4426}
4427
4428#define DO_FMLA(NAME, name) \
3a7be554 4429static bool trans_##NAME(DisasContext *s, arg_rprrr_esz *a) \
6ceabaad 4430{ \
08975da9 4431 static gen_helper_gvec_5_ptr * const fns[4] = { \
6ceabaad
RH
4432 NULL, gen_helper_sve_##name##_h, \
4433 gen_helper_sve_##name##_s, gen_helper_sve_##name##_d \
4434 }; \
4435 return do_fmla(s, a, fns[a->esz]); \
4436}
4437
4438DO_FMLA(FMLA_zpzzz, fmla_zpzzz)
4439DO_FMLA(FMLS_zpzzz, fmls_zpzzz)
4440DO_FMLA(FNMLA_zpzzz, fnmla_zpzzz)
4441DO_FMLA(FNMLS_zpzzz, fnmls_zpzzz)
4442
4443#undef DO_FMLA
4444
3a7be554 4445static bool trans_FCMLA_zpzzz(DisasContext *s, arg_FCMLA_zpzzz *a)
05f48bab 4446{
08975da9
RH
4447 static gen_helper_gvec_5_ptr * const fns[4] = {
4448 NULL,
05f48bab
RH
4449 gen_helper_sve_fcmla_zpzzz_h,
4450 gen_helper_sve_fcmla_zpzzz_s,
4451 gen_helper_sve_fcmla_zpzzz_d,
4452 };
4453
4454 if (a->esz == 0) {
4455 return false;
4456 }
4457 if (sve_access_check(s)) {
4458 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4459 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
08975da9
RH
4460 tcg_gen_gvec_5_ptr(vec_full_reg_offset(s, a->rd),
4461 vec_full_reg_offset(s, a->rn),
4462 vec_full_reg_offset(s, a->rm),
4463 vec_full_reg_offset(s, a->ra),
4464 pred_full_reg_offset(s, a->pg),
4465 status, vsz, vsz, a->rot, fns[a->esz]);
4466 tcg_temp_free_ptr(status);
05f48bab
RH
4467 }
4468 return true;
4469}
4470
3a7be554 4471static bool trans_FCMLA_zzxz(DisasContext *s, arg_FCMLA_zzxz *a)
18fc2405 4472{
636ddeb1 4473 static gen_helper_gvec_4_ptr * const fns[2] = {
18fc2405
RH
4474 gen_helper_gvec_fcmlah_idx,
4475 gen_helper_gvec_fcmlas_idx,
4476 };
4477
4478 tcg_debug_assert(a->esz == 1 || a->esz == 2);
4479 tcg_debug_assert(a->rd == a->ra);
4480 if (sve_access_check(s)) {
4481 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4482 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
636ddeb1 4483 tcg_gen_gvec_4_ptr(vec_full_reg_offset(s, a->rd),
18fc2405
RH
4484 vec_full_reg_offset(s, a->rn),
4485 vec_full_reg_offset(s, a->rm),
636ddeb1 4486 vec_full_reg_offset(s, a->ra),
18fc2405
RH
4487 status, vsz, vsz,
4488 a->index * 4 + a->rot,
4489 fns[a->esz - 1]);
4490 tcg_temp_free_ptr(status);
4491 }
4492 return true;
4493}
4494
8092c6a3
RH
4495/*
4496 *** SVE Floating Point Unary Operations Predicated Group
4497 */
4498
4499static bool do_zpz_ptr(DisasContext *s, int rd, int rn, int pg,
4500 bool is_fp16, gen_helper_gvec_3_ptr *fn)
4501{
4502 if (sve_access_check(s)) {
4503 unsigned vsz = vec_full_reg_size(s);
cdfb22bb 4504 TCGv_ptr status = fpstatus_ptr(is_fp16 ? FPST_FPCR_F16 : FPST_FPCR);
8092c6a3
RH
4505 tcg_gen_gvec_3_ptr(vec_full_reg_offset(s, rd),
4506 vec_full_reg_offset(s, rn),
4507 pred_full_reg_offset(s, pg),
4508 status, vsz, vsz, 0, fn);
4509 tcg_temp_free_ptr(status);
4510 }
4511 return true;
4512}
4513
3a7be554 4514static bool trans_FCVT_sh(DisasContext *s, arg_rpr_esz *a)
46d33d1e 4515{
e4ab5124 4516 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvt_sh);
46d33d1e
RH
4517}
4518
3a7be554 4519static bool trans_FCVT_hs(DisasContext *s, arg_rpr_esz *a)
46d33d1e
RH
4520{
4521 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvt_hs);
4522}
4523
3a7be554 4524static bool trans_FCVT_dh(DisasContext *s, arg_rpr_esz *a)
46d33d1e 4525{
e4ab5124 4526 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvt_dh);
46d33d1e
RH
4527}
4528
3a7be554 4529static bool trans_FCVT_hd(DisasContext *s, arg_rpr_esz *a)
46d33d1e
RH
4530{
4531 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvt_hd);
4532}
4533
3a7be554 4534static bool trans_FCVT_ds(DisasContext *s, arg_rpr_esz *a)
46d33d1e
RH
4535{
4536 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvt_ds);
4537}
4538
3a7be554 4539static bool trans_FCVT_sd(DisasContext *s, arg_rpr_esz *a)
46d33d1e
RH
4540{
4541 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvt_sd);
4542}
4543
3a7be554 4544static bool trans_FCVTZS_hh(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4545{
4546 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_fcvtzs_hh);
4547}
4548
3a7be554 4549static bool trans_FCVTZU_hh(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4550{
4551 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_fcvtzu_hh);
4552}
4553
3a7be554 4554static bool trans_FCVTZS_hs(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4555{
4556 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_fcvtzs_hs);
4557}
4558
3a7be554 4559static bool trans_FCVTZU_hs(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4560{
4561 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_fcvtzu_hs);
4562}
4563
3a7be554 4564static bool trans_FCVTZS_hd(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4565{
4566 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_fcvtzs_hd);
4567}
4568
3a7be554 4569static bool trans_FCVTZU_hd(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4570{
4571 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_fcvtzu_hd);
4572}
4573
3a7be554 4574static bool trans_FCVTZS_ss(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4575{
4576 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzs_ss);
4577}
4578
3a7be554 4579static bool trans_FCVTZU_ss(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4580{
4581 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzu_ss);
4582}
4583
3a7be554 4584static bool trans_FCVTZS_sd(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4585{
4586 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzs_sd);
4587}
4588
3a7be554 4589static bool trans_FCVTZU_sd(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4590{
4591 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzu_sd);
4592}
4593
3a7be554 4594static bool trans_FCVTZS_ds(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4595{
4596 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzs_ds);
4597}
4598
3a7be554 4599static bool trans_FCVTZU_ds(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4600{
4601 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzu_ds);
4602}
4603
3a7be554 4604static bool trans_FCVTZS_dd(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4605{
4606 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzs_dd);
4607}
4608
3a7be554 4609static bool trans_FCVTZU_dd(DisasContext *s, arg_rpr_esz *a)
df4de1af
RH
4610{
4611 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_fcvtzu_dd);
4612}
4613
cda3c753
RH
4614static gen_helper_gvec_3_ptr * const frint_fns[3] = {
4615 gen_helper_sve_frint_h,
4616 gen_helper_sve_frint_s,
4617 gen_helper_sve_frint_d
4618};
4619
3a7be554 4620static bool trans_FRINTI(DisasContext *s, arg_rpr_esz *a)
cda3c753
RH
4621{
4622 if (a->esz == 0) {
4623 return false;
4624 }
4625 return do_zpz_ptr(s, a->rd, a->rn, a->pg, a->esz == MO_16,
4626 frint_fns[a->esz - 1]);
4627}
4628
3a7be554 4629static bool trans_FRINTX(DisasContext *s, arg_rpr_esz *a)
cda3c753
RH
4630{
4631 static gen_helper_gvec_3_ptr * const fns[3] = {
4632 gen_helper_sve_frintx_h,
4633 gen_helper_sve_frintx_s,
4634 gen_helper_sve_frintx_d
4635 };
4636 if (a->esz == 0) {
4637 return false;
4638 }
4639 return do_zpz_ptr(s, a->rd, a->rn, a->pg, a->esz == MO_16, fns[a->esz - 1]);
4640}
4641
4642static bool do_frint_mode(DisasContext *s, arg_rpr_esz *a, int mode)
4643{
4644 if (a->esz == 0) {
4645 return false;
4646 }
4647 if (sve_access_check(s)) {
4648 unsigned vsz = vec_full_reg_size(s);
4649 TCGv_i32 tmode = tcg_const_i32(mode);
cdfb22bb 4650 TCGv_ptr status = fpstatus_ptr(a->esz == MO_16 ? FPST_FPCR_F16 : FPST_FPCR);
cda3c753
RH
4651
4652 gen_helper_set_rmode(tmode, tmode, status);
4653
4654 tcg_gen_gvec_3_ptr(vec_full_reg_offset(s, a->rd),
4655 vec_full_reg_offset(s, a->rn),
4656 pred_full_reg_offset(s, a->pg),
4657 status, vsz, vsz, 0, frint_fns[a->esz - 1]);
4658
4659 gen_helper_set_rmode(tmode, tmode, status);
4660 tcg_temp_free_i32(tmode);
4661 tcg_temp_free_ptr(status);
4662 }
4663 return true;
4664}
4665
3a7be554 4666static bool trans_FRINTN(DisasContext *s, arg_rpr_esz *a)
cda3c753
RH
4667{
4668 return do_frint_mode(s, a, float_round_nearest_even);
4669}
4670
3a7be554 4671static bool trans_FRINTP(DisasContext *s, arg_rpr_esz *a)
cda3c753
RH
4672{
4673 return do_frint_mode(s, a, float_round_up);
4674}
4675
3a7be554 4676static bool trans_FRINTM(DisasContext *s, arg_rpr_esz *a)
cda3c753
RH
4677{
4678 return do_frint_mode(s, a, float_round_down);
4679}
4680
3a7be554 4681static bool trans_FRINTZ(DisasContext *s, arg_rpr_esz *a)
cda3c753
RH
4682{
4683 return do_frint_mode(s, a, float_round_to_zero);
4684}
4685
3a7be554 4686static bool trans_FRINTA(DisasContext *s, arg_rpr_esz *a)
cda3c753
RH
4687{
4688 return do_frint_mode(s, a, float_round_ties_away);
4689}
4690
3a7be554 4691static bool trans_FRECPX(DisasContext *s, arg_rpr_esz *a)
ec5b375b
RH
4692{
4693 static gen_helper_gvec_3_ptr * const fns[3] = {
4694 gen_helper_sve_frecpx_h,
4695 gen_helper_sve_frecpx_s,
4696 gen_helper_sve_frecpx_d
4697 };
4698 if (a->esz == 0) {
4699 return false;
4700 }
4701 return do_zpz_ptr(s, a->rd, a->rn, a->pg, a->esz == MO_16, fns[a->esz - 1]);
4702}
4703
3a7be554 4704static bool trans_FSQRT(DisasContext *s, arg_rpr_esz *a)
ec5b375b
RH
4705{
4706 static gen_helper_gvec_3_ptr * const fns[3] = {
4707 gen_helper_sve_fsqrt_h,
4708 gen_helper_sve_fsqrt_s,
4709 gen_helper_sve_fsqrt_d
4710 };
4711 if (a->esz == 0) {
4712 return false;
4713 }
4714 return do_zpz_ptr(s, a->rd, a->rn, a->pg, a->esz == MO_16, fns[a->esz - 1]);
4715}
4716
3a7be554 4717static bool trans_SCVTF_hh(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4718{
4719 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_scvt_hh);
4720}
4721
3a7be554 4722static bool trans_SCVTF_sh(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4723{
4724 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_scvt_sh);
4725}
4726
3a7be554 4727static bool trans_SCVTF_dh(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4728{
4729 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_scvt_dh);
4730}
4731
3a7be554 4732static bool trans_SCVTF_ss(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4733{
4734 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_scvt_ss);
4735}
4736
3a7be554 4737static bool trans_SCVTF_ds(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4738{
4739 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_scvt_ds);
4740}
4741
3a7be554 4742static bool trans_SCVTF_sd(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4743{
4744 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_scvt_sd);
4745}
4746
3a7be554 4747static bool trans_SCVTF_dd(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4748{
4749 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_scvt_dd);
4750}
4751
3a7be554 4752static bool trans_UCVTF_hh(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4753{
4754 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_ucvt_hh);
4755}
4756
3a7be554 4757static bool trans_UCVTF_sh(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4758{
4759 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_ucvt_sh);
4760}
4761
3a7be554 4762static bool trans_UCVTF_dh(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4763{
4764 return do_zpz_ptr(s, a->rd, a->rn, a->pg, true, gen_helper_sve_ucvt_dh);
4765}
4766
3a7be554 4767static bool trans_UCVTF_ss(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4768{
4769 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_ucvt_ss);
4770}
4771
3a7be554 4772static bool trans_UCVTF_ds(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4773{
4774 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_ucvt_ds);
4775}
4776
3a7be554 4777static bool trans_UCVTF_sd(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4778{
4779 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_ucvt_sd);
4780}
4781
3a7be554 4782static bool trans_UCVTF_dd(DisasContext *s, arg_rpr_esz *a)
8092c6a3
RH
4783{
4784 return do_zpz_ptr(s, a->rd, a->rn, a->pg, false, gen_helper_sve_ucvt_dd);
4785}
4786
d1822297
RH
4787/*
4788 *** SVE Memory - 32-bit Gather and Unsized Contiguous Group
4789 */
4790
4791/* Subroutine loading a vector register at VOFS of LEN bytes.
4792 * The load should begin at the address Rn + IMM.
4793 */
4794
19f2acc9 4795static void do_ldr(DisasContext *s, uint32_t vofs, int len, int rn, int imm)
d1822297 4796{
19f2acc9
RH
4797 int len_align = QEMU_ALIGN_DOWN(len, 8);
4798 int len_remain = len % 8;
4799 int nparts = len / 8 + ctpop8(len_remain);
d1822297 4800 int midx = get_mem_index(s);
b2aa8879 4801 TCGv_i64 dirty_addr, clean_addr, t0, t1;
d1822297 4802
b2aa8879
RH
4803 dirty_addr = tcg_temp_new_i64();
4804 tcg_gen_addi_i64(dirty_addr, cpu_reg_sp(s, rn), imm);
33e74c31 4805 clean_addr = gen_mte_checkN(s, dirty_addr, false, rn != 31, len);
b2aa8879 4806 tcg_temp_free_i64(dirty_addr);
d1822297 4807
b2aa8879
RH
4808 /*
4809 * Note that unpredicated load/store of vector/predicate registers
d1822297 4810 * are defined as a stream of bytes, which equates to little-endian
b2aa8879 4811 * operations on larger quantities.
d1822297
RH
4812 * Attempt to keep code expansion to a minimum by limiting the
4813 * amount of unrolling done.
4814 */
4815 if (nparts <= 4) {
4816 int i;
4817
b2aa8879 4818 t0 = tcg_temp_new_i64();
d1822297 4819 for (i = 0; i < len_align; i += 8) {
b2aa8879 4820 tcg_gen_qemu_ld_i64(t0, clean_addr, midx, MO_LEQ);
d1822297 4821 tcg_gen_st_i64(t0, cpu_env, vofs + i);
d8227b09 4822 tcg_gen_addi_i64(clean_addr, clean_addr, 8);
d1822297 4823 }
b2aa8879 4824 tcg_temp_free_i64(t0);
d1822297
RH
4825 } else {
4826 TCGLabel *loop = gen_new_label();
4827 TCGv_ptr tp, i = tcg_const_local_ptr(0);
4828
b2aa8879
RH
4829 /* Copy the clean address into a local temp, live across the loop. */
4830 t0 = clean_addr;
4b4dc975 4831 clean_addr = new_tmp_a64_local(s);
b2aa8879 4832 tcg_gen_mov_i64(clean_addr, t0);
d1822297 4833
b2aa8879 4834 gen_set_label(loop);
d1822297 4835
b2aa8879
RH
4836 t0 = tcg_temp_new_i64();
4837 tcg_gen_qemu_ld_i64(t0, clean_addr, midx, MO_LEQ);
4838 tcg_gen_addi_i64(clean_addr, clean_addr, 8);
d1822297 4839
b2aa8879 4840 tp = tcg_temp_new_ptr();
d1822297
RH
4841 tcg_gen_add_ptr(tp, cpu_env, i);
4842 tcg_gen_addi_ptr(i, i, 8);
4843 tcg_gen_st_i64(t0, tp, vofs);
4844 tcg_temp_free_ptr(tp);
b2aa8879 4845 tcg_temp_free_i64(t0);
d1822297
RH
4846
4847 tcg_gen_brcondi_ptr(TCG_COND_LTU, i, len_align, loop);
4848 tcg_temp_free_ptr(i);
4849 }
4850
b2aa8879
RH
4851 /*
4852 * Predicate register loads can be any multiple of 2.
d1822297
RH
4853 * Note that we still store the entire 64-bit unit into cpu_env.
4854 */
4855 if (len_remain) {
b2aa8879 4856 t0 = tcg_temp_new_i64();
d1822297
RH
4857 switch (len_remain) {
4858 case 2:
4859 case 4:
4860 case 8:
b2aa8879
RH
4861 tcg_gen_qemu_ld_i64(t0, clean_addr, midx,
4862 MO_LE | ctz32(len_remain));
d1822297
RH
4863 break;
4864
4865 case 6:
4866 t1 = tcg_temp_new_i64();
b2aa8879
RH
4867 tcg_gen_qemu_ld_i64(t0, clean_addr, midx, MO_LEUL);
4868 tcg_gen_addi_i64(clean_addr, clean_addr, 4);
4869 tcg_gen_qemu_ld_i64(t1, clean_addr, midx, MO_LEUW);
d1822297
RH
4870 tcg_gen_deposit_i64(t0, t0, t1, 32, 32);
4871 tcg_temp_free_i64(t1);
4872 break;
4873
4874 default:
4875 g_assert_not_reached();
4876 }
4877 tcg_gen_st_i64(t0, cpu_env, vofs + len_align);
b2aa8879 4878 tcg_temp_free_i64(t0);
d1822297 4879 }
d1822297
RH
4880}
4881
5047c204 4882/* Similarly for stores. */
19f2acc9 4883static void do_str(DisasContext *s, uint32_t vofs, int len, int rn, int imm)
5047c204 4884{
19f2acc9
RH
4885 int len_align = QEMU_ALIGN_DOWN(len, 8);
4886 int len_remain = len % 8;
4887 int nparts = len / 8 + ctpop8(len_remain);
5047c204 4888 int midx = get_mem_index(s);
bba87d0a 4889 TCGv_i64 dirty_addr, clean_addr, t0;
5047c204 4890
bba87d0a
RH
4891 dirty_addr = tcg_temp_new_i64();
4892 tcg_gen_addi_i64(dirty_addr, cpu_reg_sp(s, rn), imm);
33e74c31 4893 clean_addr = gen_mte_checkN(s, dirty_addr, false, rn != 31, len);
bba87d0a 4894 tcg_temp_free_i64(dirty_addr);
5047c204
RH
4895
4896 /* Note that unpredicated load/store of vector/predicate registers
4897 * are defined as a stream of bytes, which equates to little-endian
4898 * operations on larger quantities. There is no nice way to force
4899 * a little-endian store for aarch64_be-linux-user out of line.
4900 *
4901 * Attempt to keep code expansion to a minimum by limiting the
4902 * amount of unrolling done.
4903 */
4904 if (nparts <= 4) {
4905 int i;
4906
bba87d0a 4907 t0 = tcg_temp_new_i64();
5047c204
RH
4908 for (i = 0; i < len_align; i += 8) {
4909 tcg_gen_ld_i64(t0, cpu_env, vofs + i);
bba87d0a 4910 tcg_gen_qemu_st_i64(t0, clean_addr, midx, MO_LEQ);
d8227b09 4911 tcg_gen_addi_i64(clean_addr, clean_addr, 8);
5047c204 4912 }
bba87d0a 4913 tcg_temp_free_i64(t0);
5047c204
RH
4914 } else {
4915 TCGLabel *loop = gen_new_label();
bba87d0a 4916 TCGv_ptr tp, i = tcg_const_local_ptr(0);
5047c204 4917
bba87d0a
RH
4918 /* Copy the clean address into a local temp, live across the loop. */
4919 t0 = clean_addr;
4b4dc975 4920 clean_addr = new_tmp_a64_local(s);
bba87d0a 4921 tcg_gen_mov_i64(clean_addr, t0);
5047c204 4922
bba87d0a 4923 gen_set_label(loop);
5047c204 4924
bba87d0a
RH
4925 t0 = tcg_temp_new_i64();
4926 tp = tcg_temp_new_ptr();
4927 tcg_gen_add_ptr(tp, cpu_env, i);
4928 tcg_gen_ld_i64(t0, tp, vofs);
5047c204 4929 tcg_gen_addi_ptr(i, i, 8);
bba87d0a
RH
4930 tcg_temp_free_ptr(tp);
4931
4932 tcg_gen_qemu_st_i64(t0, clean_addr, midx, MO_LEQ);
4933 tcg_gen_addi_i64(clean_addr, clean_addr, 8);
4934 tcg_temp_free_i64(t0);
5047c204
RH
4935
4936 tcg_gen_brcondi_ptr(TCG_COND_LTU, i, len_align, loop);
4937 tcg_temp_free_ptr(i);
4938 }
4939
4940 /* Predicate register stores can be any multiple of 2. */
4941 if (len_remain) {
bba87d0a 4942 t0 = tcg_temp_new_i64();
5047c204 4943 tcg_gen_ld_i64(t0, cpu_env, vofs + len_align);
5047c204
RH
4944
4945 switch (len_remain) {
4946 case 2:
4947 case 4:
4948 case 8:
bba87d0a
RH
4949 tcg_gen_qemu_st_i64(t0, clean_addr, midx,
4950 MO_LE | ctz32(len_remain));
5047c204
RH
4951 break;
4952
4953 case 6:
bba87d0a
RH
4954 tcg_gen_qemu_st_i64(t0, clean_addr, midx, MO_LEUL);
4955 tcg_gen_addi_i64(clean_addr, clean_addr, 4);
5047c204 4956 tcg_gen_shri_i64(t0, t0, 32);
bba87d0a 4957 tcg_gen_qemu_st_i64(t0, clean_addr, midx, MO_LEUW);
5047c204
RH
4958 break;
4959
4960 default:
4961 g_assert_not_reached();
4962 }
bba87d0a 4963 tcg_temp_free_i64(t0);
5047c204 4964 }
5047c204
RH
4965}
4966
3a7be554 4967static bool trans_LDR_zri(DisasContext *s, arg_rri *a)
d1822297
RH
4968{
4969 if (sve_access_check(s)) {
4970 int size = vec_full_reg_size(s);
4971 int off = vec_full_reg_offset(s, a->rd);
4972 do_ldr(s, off, size, a->rn, a->imm * size);
4973 }
4974 return true;
4975}
4976
3a7be554 4977static bool trans_LDR_pri(DisasContext *s, arg_rri *a)
d1822297
RH
4978{
4979 if (sve_access_check(s)) {
4980 int size = pred_full_reg_size(s);
4981 int off = pred_full_reg_offset(s, a->rd);
4982 do_ldr(s, off, size, a->rn, a->imm * size);
4983 }
4984 return true;
4985}
c4e7c493 4986
3a7be554 4987static bool trans_STR_zri(DisasContext *s, arg_rri *a)
5047c204
RH
4988{
4989 if (sve_access_check(s)) {
4990 int size = vec_full_reg_size(s);
4991 int off = vec_full_reg_offset(s, a->rd);
4992 do_str(s, off, size, a->rn, a->imm * size);
4993 }
4994 return true;
4995}
4996
3a7be554 4997static bool trans_STR_pri(DisasContext *s, arg_rri *a)
5047c204
RH
4998{
4999 if (sve_access_check(s)) {
5000 int size = pred_full_reg_size(s);
5001 int off = pred_full_reg_offset(s, a->rd);
5002 do_str(s, off, size, a->rn, a->imm * size);
5003 }
5004 return true;
5005}
5006
c4e7c493
RH
5007/*
5008 *** SVE Memory - Contiguous Load Group
5009 */
5010
5011/* The memory mode of the dtype. */
14776ab5 5012static const MemOp dtype_mop[16] = {
c4e7c493
RH
5013 MO_UB, MO_UB, MO_UB, MO_UB,
5014 MO_SL, MO_UW, MO_UW, MO_UW,
5015 MO_SW, MO_SW, MO_UL, MO_UL,
5016 MO_SB, MO_SB, MO_SB, MO_Q
5017};
5018
5019#define dtype_msz(x) (dtype_mop[x] & MO_SIZE)
5020
5021/* The vector element size of dtype. */
5022static const uint8_t dtype_esz[16] = {
5023 0, 1, 2, 3,
5024 3, 1, 2, 3,
5025 3, 2, 2, 3,
5026 3, 2, 1, 3
5027};
5028
5029static void do_mem_zpa(DisasContext *s, int zt, int pg, TCGv_i64 addr,
206adacf
RH
5030 int dtype, uint32_t mte_n, bool is_write,
5031 gen_helper_gvec_mem *fn)
c4e7c493
RH
5032{
5033 unsigned vsz = vec_full_reg_size(s);
5034 TCGv_ptr t_pg;
500d0484 5035 TCGv_i32 t_desc;
206adacf 5036 int desc = 0;
c4e7c493 5037
206adacf
RH
5038 /*
5039 * For e.g. LD4, there are not enough arguments to pass all 4
c4e7c493
RH
5040 * registers as pointers, so encode the regno into the data field.
5041 * For consistency, do this even for LD1.
5042 */
9473d0ec 5043 if (s->mte_active[0]) {
206adacf
RH
5044 int msz = dtype_msz(dtype);
5045
5046 desc = FIELD_DP32(desc, MTEDESC, MIDX, get_mem_index(s));
5047 desc = FIELD_DP32(desc, MTEDESC, TBI, s->tbid);
5048 desc = FIELD_DP32(desc, MTEDESC, TCMA, s->tcma);
5049 desc = FIELD_DP32(desc, MTEDESC, WRITE, is_write);
28f32503 5050 desc = FIELD_DP32(desc, MTEDESC, SIZEM1, (mte_n << msz) - 1);
206adacf 5051 desc <<= SVE_MTEDESC_SHIFT;
9473d0ec
RH
5052 } else {
5053 addr = clean_data_tbi(s, addr);
206adacf 5054 }
9473d0ec 5055
206adacf 5056 desc = simd_desc(vsz, vsz, zt | desc);
500d0484 5057 t_desc = tcg_const_i32(desc);
c4e7c493
RH
5058 t_pg = tcg_temp_new_ptr();
5059
5060 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, pg));
500d0484 5061 fn(cpu_env, t_pg, addr, t_desc);
c4e7c493
RH
5062
5063 tcg_temp_free_ptr(t_pg);
500d0484 5064 tcg_temp_free_i32(t_desc);
c4e7c493
RH
5065}
5066
5067static void do_ld_zpa(DisasContext *s, int zt, int pg,
5068 TCGv_i64 addr, int dtype, int nreg)
5069{
206adacf
RH
5070 static gen_helper_gvec_mem * const fns[2][2][16][4] = {
5071 { /* mte inactive, little-endian */
5072 { { gen_helper_sve_ld1bb_r, gen_helper_sve_ld2bb_r,
7d0a57a2 5073 gen_helper_sve_ld3bb_r, gen_helper_sve_ld4bb_r },
206adacf
RH
5074 { gen_helper_sve_ld1bhu_r, NULL, NULL, NULL },
5075 { gen_helper_sve_ld1bsu_r, NULL, NULL, NULL },
5076 { gen_helper_sve_ld1bdu_r, NULL, NULL, NULL },
5077
5078 { gen_helper_sve_ld1sds_le_r, NULL, NULL, NULL },
5079 { gen_helper_sve_ld1hh_le_r, gen_helper_sve_ld2hh_le_r,
5080 gen_helper_sve_ld3hh_le_r, gen_helper_sve_ld4hh_le_r },
5081 { gen_helper_sve_ld1hsu_le_r, NULL, NULL, NULL },
5082 { gen_helper_sve_ld1hdu_le_r, NULL, NULL, NULL },
5083
5084 { gen_helper_sve_ld1hds_le_r, NULL, NULL, NULL },
5085 { gen_helper_sve_ld1hss_le_r, NULL, NULL, NULL },
5086 { gen_helper_sve_ld1ss_le_r, gen_helper_sve_ld2ss_le_r,
5087 gen_helper_sve_ld3ss_le_r, gen_helper_sve_ld4ss_le_r },
5088 { gen_helper_sve_ld1sdu_le_r, NULL, NULL, NULL },
5089
5090 { gen_helper_sve_ld1bds_r, NULL, NULL, NULL },
5091 { gen_helper_sve_ld1bss_r, NULL, NULL, NULL },
5092 { gen_helper_sve_ld1bhs_r, NULL, NULL, NULL },
5093 { gen_helper_sve_ld1dd_le_r, gen_helper_sve_ld2dd_le_r,
5094 gen_helper_sve_ld3dd_le_r, gen_helper_sve_ld4dd_le_r } },
5095
5096 /* mte inactive, big-endian */
5097 { { gen_helper_sve_ld1bb_r, gen_helper_sve_ld2bb_r,
5098 gen_helper_sve_ld3bb_r, gen_helper_sve_ld4bb_r },
5099 { gen_helper_sve_ld1bhu_r, NULL, NULL, NULL },
5100 { gen_helper_sve_ld1bsu_r, NULL, NULL, NULL },
5101 { gen_helper_sve_ld1bdu_r, NULL, NULL, NULL },
5102
5103 { gen_helper_sve_ld1sds_be_r, NULL, NULL, NULL },
5104 { gen_helper_sve_ld1hh_be_r, gen_helper_sve_ld2hh_be_r,
5105 gen_helper_sve_ld3hh_be_r, gen_helper_sve_ld4hh_be_r },
5106 { gen_helper_sve_ld1hsu_be_r, NULL, NULL, NULL },
5107 { gen_helper_sve_ld1hdu_be_r, NULL, NULL, NULL },
5108
5109 { gen_helper_sve_ld1hds_be_r, NULL, NULL, NULL },
5110 { gen_helper_sve_ld1hss_be_r, NULL, NULL, NULL },
5111 { gen_helper_sve_ld1ss_be_r, gen_helper_sve_ld2ss_be_r,
5112 gen_helper_sve_ld3ss_be_r, gen_helper_sve_ld4ss_be_r },
5113 { gen_helper_sve_ld1sdu_be_r, NULL, NULL, NULL },
5114
5115 { gen_helper_sve_ld1bds_r, NULL, NULL, NULL },
5116 { gen_helper_sve_ld1bss_r, NULL, NULL, NULL },
5117 { gen_helper_sve_ld1bhs_r, NULL, NULL, NULL },
5118 { gen_helper_sve_ld1dd_be_r, gen_helper_sve_ld2dd_be_r,
5119 gen_helper_sve_ld3dd_be_r, gen_helper_sve_ld4dd_be_r } } },
5120
5121 { /* mte active, little-endian */
5122 { { gen_helper_sve_ld1bb_r_mte,
5123 gen_helper_sve_ld2bb_r_mte,
5124 gen_helper_sve_ld3bb_r_mte,
5125 gen_helper_sve_ld4bb_r_mte },
5126 { gen_helper_sve_ld1bhu_r_mte, NULL, NULL, NULL },
5127 { gen_helper_sve_ld1bsu_r_mte, NULL, NULL, NULL },
5128 { gen_helper_sve_ld1bdu_r_mte, NULL, NULL, NULL },
5129
5130 { gen_helper_sve_ld1sds_le_r_mte, NULL, NULL, NULL },
5131 { gen_helper_sve_ld1hh_le_r_mte,
5132 gen_helper_sve_ld2hh_le_r_mte,
5133 gen_helper_sve_ld3hh_le_r_mte,
5134 gen_helper_sve_ld4hh_le_r_mte },
5135 { gen_helper_sve_ld1hsu_le_r_mte, NULL, NULL, NULL },
5136 { gen_helper_sve_ld1hdu_le_r_mte, NULL, NULL, NULL },
5137
5138 { gen_helper_sve_ld1hds_le_r_mte, NULL, NULL, NULL },
5139 { gen_helper_sve_ld1hss_le_r_mte, NULL, NULL, NULL },
5140 { gen_helper_sve_ld1ss_le_r_mte,
5141 gen_helper_sve_ld2ss_le_r_mte,
5142 gen_helper_sve_ld3ss_le_r_mte,
5143 gen_helper_sve_ld4ss_le_r_mte },
5144 { gen_helper_sve_ld1sdu_le_r_mte, NULL, NULL, NULL },
5145
5146 { gen_helper_sve_ld1bds_r_mte, NULL, NULL, NULL },
5147 { gen_helper_sve_ld1bss_r_mte, NULL, NULL, NULL },
5148 { gen_helper_sve_ld1bhs_r_mte, NULL, NULL, NULL },
5149 { gen_helper_sve_ld1dd_le_r_mte,
5150 gen_helper_sve_ld2dd_le_r_mte,
5151 gen_helper_sve_ld3dd_le_r_mte,
5152 gen_helper_sve_ld4dd_le_r_mte } },
5153
5154 /* mte active, big-endian */
5155 { { gen_helper_sve_ld1bb_r_mte,
5156 gen_helper_sve_ld2bb_r_mte,
5157 gen_helper_sve_ld3bb_r_mte,
5158 gen_helper_sve_ld4bb_r_mte },
5159 { gen_helper_sve_ld1bhu_r_mte, NULL, NULL, NULL },
5160 { gen_helper_sve_ld1bsu_r_mte, NULL, NULL, NULL },
5161 { gen_helper_sve_ld1bdu_r_mte, NULL, NULL, NULL },
5162
5163 { gen_helper_sve_ld1sds_be_r_mte, NULL, NULL, NULL },
5164 { gen_helper_sve_ld1hh_be_r_mte,
5165 gen_helper_sve_ld2hh_be_r_mte,
5166 gen_helper_sve_ld3hh_be_r_mte,
5167 gen_helper_sve_ld4hh_be_r_mte },
5168 { gen_helper_sve_ld1hsu_be_r_mte, NULL, NULL, NULL },
5169 { gen_helper_sve_ld1hdu_be_r_mte, NULL, NULL, NULL },
5170
5171 { gen_helper_sve_ld1hds_be_r_mte, NULL, NULL, NULL },
5172 { gen_helper_sve_ld1hss_be_r_mte, NULL, NULL, NULL },
5173 { gen_helper_sve_ld1ss_be_r_mte,
5174 gen_helper_sve_ld2ss_be_r_mte,
5175 gen_helper_sve_ld3ss_be_r_mte,
5176 gen_helper_sve_ld4ss_be_r_mte },
5177 { gen_helper_sve_ld1sdu_be_r_mte, NULL, NULL, NULL },
5178
5179 { gen_helper_sve_ld1bds_r_mte, NULL, NULL, NULL },
5180 { gen_helper_sve_ld1bss_r_mte, NULL, NULL, NULL },
5181 { gen_helper_sve_ld1bhs_r_mte, NULL, NULL, NULL },
5182 { gen_helper_sve_ld1dd_be_r_mte,
5183 gen_helper_sve_ld2dd_be_r_mte,
5184 gen_helper_sve_ld3dd_be_r_mte,
5185 gen_helper_sve_ld4dd_be_r_mte } } },
c4e7c493 5186 };
206adacf
RH
5187 gen_helper_gvec_mem *fn
5188 = fns[s->mte_active[0]][s->be_data == MO_BE][dtype][nreg];
c4e7c493 5189
206adacf
RH
5190 /*
5191 * While there are holes in the table, they are not
c4e7c493
RH
5192 * accessible via the instruction encoding.
5193 */
5194 assert(fn != NULL);
206adacf 5195 do_mem_zpa(s, zt, pg, addr, dtype, nreg, false, fn);
c4e7c493
RH
5196}
5197
3a7be554 5198static bool trans_LD_zprr(DisasContext *s, arg_rprr_load *a)
c4e7c493
RH
5199{
5200 if (a->rm == 31) {
5201 return false;
5202 }
5203 if (sve_access_check(s)) {
5204 TCGv_i64 addr = new_tmp_a64(s);
50ef1cbf 5205 tcg_gen_shli_i64(addr, cpu_reg(s, a->rm), dtype_msz(a->dtype));
c4e7c493
RH
5206 tcg_gen_add_i64(addr, addr, cpu_reg_sp(s, a->rn));
5207 do_ld_zpa(s, a->rd, a->pg, addr, a->dtype, a->nreg);
5208 }
5209 return true;
5210}
5211
3a7be554 5212static bool trans_LD_zpri(DisasContext *s, arg_rpri_load *a)
c4e7c493
RH
5213{
5214 if (sve_access_check(s)) {
5215 int vsz = vec_full_reg_size(s);
5216 int elements = vsz >> dtype_esz[a->dtype];
5217 TCGv_i64 addr = new_tmp_a64(s);
5218
5219 tcg_gen_addi_i64(addr, cpu_reg_sp(s, a->rn),
5220 (a->imm * elements * (a->nreg + 1))
5221 << dtype_msz(a->dtype));
5222 do_ld_zpa(s, a->rd, a->pg, addr, a->dtype, a->nreg);
5223 }
5224 return true;
5225}
e2654d75 5226
3a7be554 5227static bool trans_LDFF1_zprr(DisasContext *s, arg_rprr_load *a)
e2654d75 5228{
aa13f7c3
RH
5229 static gen_helper_gvec_mem * const fns[2][2][16] = {
5230 { /* mte inactive, little-endian */
5231 { gen_helper_sve_ldff1bb_r,
5232 gen_helper_sve_ldff1bhu_r,
5233 gen_helper_sve_ldff1bsu_r,
5234 gen_helper_sve_ldff1bdu_r,
5235
5236 gen_helper_sve_ldff1sds_le_r,
5237 gen_helper_sve_ldff1hh_le_r,
5238 gen_helper_sve_ldff1hsu_le_r,
5239 gen_helper_sve_ldff1hdu_le_r,
5240
5241 gen_helper_sve_ldff1hds_le_r,
5242 gen_helper_sve_ldff1hss_le_r,
5243 gen_helper_sve_ldff1ss_le_r,
5244 gen_helper_sve_ldff1sdu_le_r,
5245
5246 gen_helper_sve_ldff1bds_r,
5247 gen_helper_sve_ldff1bss_r,
5248 gen_helper_sve_ldff1bhs_r,
5249 gen_helper_sve_ldff1dd_le_r },
5250
5251 /* mte inactive, big-endian */
5252 { gen_helper_sve_ldff1bb_r,
5253 gen_helper_sve_ldff1bhu_r,
5254 gen_helper_sve_ldff1bsu_r,
5255 gen_helper_sve_ldff1bdu_r,
5256
5257 gen_helper_sve_ldff1sds_be_r,
5258 gen_helper_sve_ldff1hh_be_r,
5259 gen_helper_sve_ldff1hsu_be_r,
5260 gen_helper_sve_ldff1hdu_be_r,
5261
5262 gen_helper_sve_ldff1hds_be_r,
5263 gen_helper_sve_ldff1hss_be_r,
5264 gen_helper_sve_ldff1ss_be_r,
5265 gen_helper_sve_ldff1sdu_be_r,
5266
5267 gen_helper_sve_ldff1bds_r,
5268 gen_helper_sve_ldff1bss_r,
5269 gen_helper_sve_ldff1bhs_r,
5270 gen_helper_sve_ldff1dd_be_r } },
5271
5272 { /* mte active, little-endian */
5273 { gen_helper_sve_ldff1bb_r_mte,
5274 gen_helper_sve_ldff1bhu_r_mte,
5275 gen_helper_sve_ldff1bsu_r_mte,
5276 gen_helper_sve_ldff1bdu_r_mte,
5277
5278 gen_helper_sve_ldff1sds_le_r_mte,
5279 gen_helper_sve_ldff1hh_le_r_mte,
5280 gen_helper_sve_ldff1hsu_le_r_mte,
5281 gen_helper_sve_ldff1hdu_le_r_mte,
5282
5283 gen_helper_sve_ldff1hds_le_r_mte,
5284 gen_helper_sve_ldff1hss_le_r_mte,
5285 gen_helper_sve_ldff1ss_le_r_mte,
5286 gen_helper_sve_ldff1sdu_le_r_mte,
5287
5288 gen_helper_sve_ldff1bds_r_mte,
5289 gen_helper_sve_ldff1bss_r_mte,
5290 gen_helper_sve_ldff1bhs_r_mte,
5291 gen_helper_sve_ldff1dd_le_r_mte },
5292
5293 /* mte active, big-endian */
5294 { gen_helper_sve_ldff1bb_r_mte,
5295 gen_helper_sve_ldff1bhu_r_mte,
5296 gen_helper_sve_ldff1bsu_r_mte,
5297 gen_helper_sve_ldff1bdu_r_mte,
5298
5299 gen_helper_sve_ldff1sds_be_r_mte,
5300 gen_helper_sve_ldff1hh_be_r_mte,
5301 gen_helper_sve_ldff1hsu_be_r_mte,
5302 gen_helper_sve_ldff1hdu_be_r_mte,
5303
5304 gen_helper_sve_ldff1hds_be_r_mte,
5305 gen_helper_sve_ldff1hss_be_r_mte,
5306 gen_helper_sve_ldff1ss_be_r_mte,
5307 gen_helper_sve_ldff1sdu_be_r_mte,
5308
5309 gen_helper_sve_ldff1bds_r_mte,
5310 gen_helper_sve_ldff1bss_r_mte,
5311 gen_helper_sve_ldff1bhs_r_mte,
5312 gen_helper_sve_ldff1dd_be_r_mte } },
e2654d75
RH
5313 };
5314
5315 if (sve_access_check(s)) {
5316 TCGv_i64 addr = new_tmp_a64(s);
5317 tcg_gen_shli_i64(addr, cpu_reg(s, a->rm), dtype_msz(a->dtype));
5318 tcg_gen_add_i64(addr, addr, cpu_reg_sp(s, a->rn));
aa13f7c3
RH
5319 do_mem_zpa(s, a->rd, a->pg, addr, a->dtype, 1, false,
5320 fns[s->mte_active[0]][s->be_data == MO_BE][a->dtype]);
e2654d75
RH
5321 }
5322 return true;
5323}
5324
3a7be554 5325static bool trans_LDNF1_zpri(DisasContext *s, arg_rpri_load *a)
e2654d75 5326{
aa13f7c3
RH
5327 static gen_helper_gvec_mem * const fns[2][2][16] = {
5328 { /* mte inactive, little-endian */
5329 { gen_helper_sve_ldnf1bb_r,
5330 gen_helper_sve_ldnf1bhu_r,
5331 gen_helper_sve_ldnf1bsu_r,
5332 gen_helper_sve_ldnf1bdu_r,
5333
5334 gen_helper_sve_ldnf1sds_le_r,
5335 gen_helper_sve_ldnf1hh_le_r,
5336 gen_helper_sve_ldnf1hsu_le_r,
5337 gen_helper_sve_ldnf1hdu_le_r,
5338
5339 gen_helper_sve_ldnf1hds_le_r,
5340 gen_helper_sve_ldnf1hss_le_r,
5341 gen_helper_sve_ldnf1ss_le_r,
5342 gen_helper_sve_ldnf1sdu_le_r,
5343
5344 gen_helper_sve_ldnf1bds_r,
5345 gen_helper_sve_ldnf1bss_r,
5346 gen_helper_sve_ldnf1bhs_r,
5347 gen_helper_sve_ldnf1dd_le_r },
5348
5349 /* mte inactive, big-endian */
5350 { gen_helper_sve_ldnf1bb_r,
5351 gen_helper_sve_ldnf1bhu_r,
5352 gen_helper_sve_ldnf1bsu_r,
5353 gen_helper_sve_ldnf1bdu_r,
5354
5355 gen_helper_sve_ldnf1sds_be_r,
5356 gen_helper_sve_ldnf1hh_be_r,
5357 gen_helper_sve_ldnf1hsu_be_r,
5358 gen_helper_sve_ldnf1hdu_be_r,
5359
5360 gen_helper_sve_ldnf1hds_be_r,
5361 gen_helper_sve_ldnf1hss_be_r,
5362 gen_helper_sve_ldnf1ss_be_r,
5363 gen_helper_sve_ldnf1sdu_be_r,
5364
5365 gen_helper_sve_ldnf1bds_r,
5366 gen_helper_sve_ldnf1bss_r,
5367 gen_helper_sve_ldnf1bhs_r,
5368 gen_helper_sve_ldnf1dd_be_r } },
5369
5370 { /* mte inactive, little-endian */
5371 { gen_helper_sve_ldnf1bb_r_mte,
5372 gen_helper_sve_ldnf1bhu_r_mte,
5373 gen_helper_sve_ldnf1bsu_r_mte,
5374 gen_helper_sve_ldnf1bdu_r_mte,
5375
5376 gen_helper_sve_ldnf1sds_le_r_mte,
5377 gen_helper_sve_ldnf1hh_le_r_mte,
5378 gen_helper_sve_ldnf1hsu_le_r_mte,
5379 gen_helper_sve_ldnf1hdu_le_r_mte,
5380
5381 gen_helper_sve_ldnf1hds_le_r_mte,
5382 gen_helper_sve_ldnf1hss_le_r_mte,
5383 gen_helper_sve_ldnf1ss_le_r_mte,
5384 gen_helper_sve_ldnf1sdu_le_r_mte,
5385
5386 gen_helper_sve_ldnf1bds_r_mte,
5387 gen_helper_sve_ldnf1bss_r_mte,
5388 gen_helper_sve_ldnf1bhs_r_mte,
5389 gen_helper_sve_ldnf1dd_le_r_mte },
5390
5391 /* mte inactive, big-endian */
5392 { gen_helper_sve_ldnf1bb_r_mte,
5393 gen_helper_sve_ldnf1bhu_r_mte,
5394 gen_helper_sve_ldnf1bsu_r_mte,
5395 gen_helper_sve_ldnf1bdu_r_mte,
5396
5397 gen_helper_sve_ldnf1sds_be_r_mte,
5398 gen_helper_sve_ldnf1hh_be_r_mte,
5399 gen_helper_sve_ldnf1hsu_be_r_mte,
5400 gen_helper_sve_ldnf1hdu_be_r_mte,
5401
5402 gen_helper_sve_ldnf1hds_be_r_mte,
5403 gen_helper_sve_ldnf1hss_be_r_mte,
5404 gen_helper_sve_ldnf1ss_be_r_mte,
5405 gen_helper_sve_ldnf1sdu_be_r_mte,
5406
5407 gen_helper_sve_ldnf1bds_r_mte,
5408 gen_helper_sve_ldnf1bss_r_mte,
5409 gen_helper_sve_ldnf1bhs_r_mte,
5410 gen_helper_sve_ldnf1dd_be_r_mte } },
e2654d75
RH
5411 };
5412
5413 if (sve_access_check(s)) {
5414 int vsz = vec_full_reg_size(s);
5415 int elements = vsz >> dtype_esz[a->dtype];
5416 int off = (a->imm * elements) << dtype_msz(a->dtype);
5417 TCGv_i64 addr = new_tmp_a64(s);
5418
5419 tcg_gen_addi_i64(addr, cpu_reg_sp(s, a->rn), off);
aa13f7c3
RH
5420 do_mem_zpa(s, a->rd, a->pg, addr, a->dtype, 1, false,
5421 fns[s->mte_active[0]][s->be_data == MO_BE][a->dtype]);
e2654d75
RH
5422 }
5423 return true;
5424}
1a039c7e 5425
05abe304
RH
5426static void do_ldrq(DisasContext *s, int zt, int pg, TCGv_i64 addr, int msz)
5427{
7d0a57a2
RH
5428 static gen_helper_gvec_mem * const fns[2][4] = {
5429 { gen_helper_sve_ld1bb_r, gen_helper_sve_ld1hh_le_r,
5430 gen_helper_sve_ld1ss_le_r, gen_helper_sve_ld1dd_le_r },
5431 { gen_helper_sve_ld1bb_r, gen_helper_sve_ld1hh_be_r,
5432 gen_helper_sve_ld1ss_be_r, gen_helper_sve_ld1dd_be_r },
05abe304
RH
5433 };
5434 unsigned vsz = vec_full_reg_size(s);
5435 TCGv_ptr t_pg;
500d0484
RH
5436 TCGv_i32 t_desc;
5437 int desc, poff;
05abe304
RH
5438
5439 /* Load the first quadword using the normal predicated load helpers. */
ba080b86 5440 desc = simd_desc(16, 16, zt);
500d0484 5441 t_desc = tcg_const_i32(desc);
2a99ab2b
RH
5442
5443 poff = pred_full_reg_offset(s, pg);
5444 if (vsz > 16) {
5445 /*
5446 * Zero-extend the first 16 bits of the predicate into a temporary.
5447 * This avoids triggering an assert making sure we don't have bits
5448 * set within a predicate beyond VQ, but we have lowered VQ to 1
5449 * for this load operation.
5450 */
5451 TCGv_i64 tmp = tcg_temp_new_i64();
5452#ifdef HOST_WORDS_BIGENDIAN
5453 poff += 6;
5454#endif
5455 tcg_gen_ld16u_i64(tmp, cpu_env, poff);
5456
5457 poff = offsetof(CPUARMState, vfp.preg_tmp);
5458 tcg_gen_st_i64(tmp, cpu_env, poff);
5459 tcg_temp_free_i64(tmp);
5460 }
5461
05abe304 5462 t_pg = tcg_temp_new_ptr();
2a99ab2b 5463 tcg_gen_addi_ptr(t_pg, cpu_env, poff);
05abe304 5464
500d0484 5465 fns[s->be_data == MO_BE][msz](cpu_env, t_pg, addr, t_desc);
05abe304
RH
5466
5467 tcg_temp_free_ptr(t_pg);
500d0484 5468 tcg_temp_free_i32(t_desc);
05abe304
RH
5469
5470 /* Replicate that first quadword. */
5471 if (vsz > 16) {
5472 unsigned dofs = vec_full_reg_offset(s, zt);
5473 tcg_gen_gvec_dup_mem(4, dofs + 16, dofs, vsz - 16, vsz - 16);
5474 }
5475}
5476
3a7be554 5477static bool trans_LD1RQ_zprr(DisasContext *s, arg_rprr_load *a)
05abe304
RH
5478{
5479 if (a->rm == 31) {
5480 return false;
5481 }
5482 if (sve_access_check(s)) {
5483 int msz = dtype_msz(a->dtype);
5484 TCGv_i64 addr = new_tmp_a64(s);
5485 tcg_gen_shli_i64(addr, cpu_reg(s, a->rm), msz);
5486 tcg_gen_add_i64(addr, addr, cpu_reg_sp(s, a->rn));
5487 do_ldrq(s, a->rd, a->pg, addr, msz);
5488 }
5489 return true;
5490}
5491
3a7be554 5492static bool trans_LD1RQ_zpri(DisasContext *s, arg_rpri_load *a)
05abe304
RH
5493{
5494 if (sve_access_check(s)) {
5495 TCGv_i64 addr = new_tmp_a64(s);
5496 tcg_gen_addi_i64(addr, cpu_reg_sp(s, a->rn), a->imm * 16);
5497 do_ldrq(s, a->rd, a->pg, addr, dtype_msz(a->dtype));
5498 }
5499 return true;
5500}
5501
68459864 5502/* Load and broadcast element. */
3a7be554 5503static bool trans_LD1R_zpri(DisasContext *s, arg_rpri_load *a)
68459864 5504{
68459864
RH
5505 unsigned vsz = vec_full_reg_size(s);
5506 unsigned psz = pred_full_reg_size(s);
5507 unsigned esz = dtype_esz[a->dtype];
d0e372b0 5508 unsigned msz = dtype_msz(a->dtype);
c0ed9166 5509 TCGLabel *over;
4ac430e1 5510 TCGv_i64 temp, clean_addr;
68459864 5511
c0ed9166
RH
5512 if (!sve_access_check(s)) {
5513 return true;
5514 }
5515
5516 over = gen_new_label();
5517
68459864
RH
5518 /* If the guarding predicate has no bits set, no load occurs. */
5519 if (psz <= 8) {
5520 /* Reduce the pred_esz_masks value simply to reduce the
5521 * size of the code generated here.
5522 */
5523 uint64_t psz_mask = MAKE_64BIT_MASK(0, psz * 8);
5524 temp = tcg_temp_new_i64();
5525 tcg_gen_ld_i64(temp, cpu_env, pred_full_reg_offset(s, a->pg));
5526 tcg_gen_andi_i64(temp, temp, pred_esz_masks[esz] & psz_mask);
5527 tcg_gen_brcondi_i64(TCG_COND_EQ, temp, 0, over);
5528 tcg_temp_free_i64(temp);
5529 } else {
5530 TCGv_i32 t32 = tcg_temp_new_i32();
5531 find_last_active(s, t32, esz, a->pg);
5532 tcg_gen_brcondi_i32(TCG_COND_LT, t32, 0, over);
5533 tcg_temp_free_i32(t32);
5534 }
5535
5536 /* Load the data. */
5537 temp = tcg_temp_new_i64();
d0e372b0 5538 tcg_gen_addi_i64(temp, cpu_reg_sp(s, a->rn), a->imm << msz);
4ac430e1
RH
5539 clean_addr = gen_mte_check1(s, temp, false, true, msz);
5540
5541 tcg_gen_qemu_ld_i64(temp, clean_addr, get_mem_index(s),
0ca0f872 5542 finalize_memop(s, dtype_mop[a->dtype]));
68459864
RH
5543
5544 /* Broadcast to *all* elements. */
5545 tcg_gen_gvec_dup_i64(esz, vec_full_reg_offset(s, a->rd),
5546 vsz, vsz, temp);
5547 tcg_temp_free_i64(temp);
5548
5549 /* Zero the inactive elements. */
5550 gen_set_label(over);
60245996 5551 return do_movz_zpz(s, a->rd, a->rd, a->pg, esz, false);
68459864
RH
5552}
5553
1a039c7e
RH
5554static void do_st_zpa(DisasContext *s, int zt, int pg, TCGv_i64 addr,
5555 int msz, int esz, int nreg)
5556{
71b9f394
RH
5557 static gen_helper_gvec_mem * const fn_single[2][2][4][4] = {
5558 { { { gen_helper_sve_st1bb_r,
5559 gen_helper_sve_st1bh_r,
5560 gen_helper_sve_st1bs_r,
5561 gen_helper_sve_st1bd_r },
5562 { NULL,
5563 gen_helper_sve_st1hh_le_r,
5564 gen_helper_sve_st1hs_le_r,
5565 gen_helper_sve_st1hd_le_r },
5566 { NULL, NULL,
5567 gen_helper_sve_st1ss_le_r,
5568 gen_helper_sve_st1sd_le_r },
5569 { NULL, NULL, NULL,
5570 gen_helper_sve_st1dd_le_r } },
5571 { { gen_helper_sve_st1bb_r,
5572 gen_helper_sve_st1bh_r,
5573 gen_helper_sve_st1bs_r,
5574 gen_helper_sve_st1bd_r },
5575 { NULL,
5576 gen_helper_sve_st1hh_be_r,
5577 gen_helper_sve_st1hs_be_r,
5578 gen_helper_sve_st1hd_be_r },
5579 { NULL, NULL,
5580 gen_helper_sve_st1ss_be_r,
5581 gen_helper_sve_st1sd_be_r },
5582 { NULL, NULL, NULL,
5583 gen_helper_sve_st1dd_be_r } } },
5584
5585 { { { gen_helper_sve_st1bb_r_mte,
5586 gen_helper_sve_st1bh_r_mte,
5587 gen_helper_sve_st1bs_r_mte,
5588 gen_helper_sve_st1bd_r_mte },
5589 { NULL,
5590 gen_helper_sve_st1hh_le_r_mte,
5591 gen_helper_sve_st1hs_le_r_mte,
5592 gen_helper_sve_st1hd_le_r_mte },
5593 { NULL, NULL,
5594 gen_helper_sve_st1ss_le_r_mte,
5595 gen_helper_sve_st1sd_le_r_mte },
5596 { NULL, NULL, NULL,
5597 gen_helper_sve_st1dd_le_r_mte } },
5598 { { gen_helper_sve_st1bb_r_mte,
5599 gen_helper_sve_st1bh_r_mte,
5600 gen_helper_sve_st1bs_r_mte,
5601 gen_helper_sve_st1bd_r_mte },
5602 { NULL,
5603 gen_helper_sve_st1hh_be_r_mte,
5604 gen_helper_sve_st1hs_be_r_mte,
5605 gen_helper_sve_st1hd_be_r_mte },
5606 { NULL, NULL,
5607 gen_helper_sve_st1ss_be_r_mte,
5608 gen_helper_sve_st1sd_be_r_mte },
5609 { NULL, NULL, NULL,
5610 gen_helper_sve_st1dd_be_r_mte } } },
1a039c7e 5611 };
71b9f394
RH
5612 static gen_helper_gvec_mem * const fn_multiple[2][2][3][4] = {
5613 { { { gen_helper_sve_st2bb_r,
5614 gen_helper_sve_st2hh_le_r,
5615 gen_helper_sve_st2ss_le_r,
5616 gen_helper_sve_st2dd_le_r },
5617 { gen_helper_sve_st3bb_r,
5618 gen_helper_sve_st3hh_le_r,
5619 gen_helper_sve_st3ss_le_r,
5620 gen_helper_sve_st3dd_le_r },
5621 { gen_helper_sve_st4bb_r,
5622 gen_helper_sve_st4hh_le_r,
5623 gen_helper_sve_st4ss_le_r,
5624 gen_helper_sve_st4dd_le_r } },
5625 { { gen_helper_sve_st2bb_r,
5626 gen_helper_sve_st2hh_be_r,
5627 gen_helper_sve_st2ss_be_r,
5628 gen_helper_sve_st2dd_be_r },
5629 { gen_helper_sve_st3bb_r,
5630 gen_helper_sve_st3hh_be_r,
5631 gen_helper_sve_st3ss_be_r,
5632 gen_helper_sve_st3dd_be_r },
5633 { gen_helper_sve_st4bb_r,
5634 gen_helper_sve_st4hh_be_r,
5635 gen_helper_sve_st4ss_be_r,
5636 gen_helper_sve_st4dd_be_r } } },
5637 { { { gen_helper_sve_st2bb_r_mte,
5638 gen_helper_sve_st2hh_le_r_mte,
5639 gen_helper_sve_st2ss_le_r_mte,
5640 gen_helper_sve_st2dd_le_r_mte },
5641 { gen_helper_sve_st3bb_r_mte,
5642 gen_helper_sve_st3hh_le_r_mte,
5643 gen_helper_sve_st3ss_le_r_mte,
5644 gen_helper_sve_st3dd_le_r_mte },
5645 { gen_helper_sve_st4bb_r_mte,
5646 gen_helper_sve_st4hh_le_r_mte,
5647 gen_helper_sve_st4ss_le_r_mte,
5648 gen_helper_sve_st4dd_le_r_mte } },
5649 { { gen_helper_sve_st2bb_r_mte,
5650 gen_helper_sve_st2hh_be_r_mte,
5651 gen_helper_sve_st2ss_be_r_mte,
5652 gen_helper_sve_st2dd_be_r_mte },
5653 { gen_helper_sve_st3bb_r_mte,
5654 gen_helper_sve_st3hh_be_r_mte,
5655 gen_helper_sve_st3ss_be_r_mte,
5656 gen_helper_sve_st3dd_be_r_mte },
5657 { gen_helper_sve_st4bb_r_mte,
5658 gen_helper_sve_st4hh_be_r_mte,
5659 gen_helper_sve_st4ss_be_r_mte,
5660 gen_helper_sve_st4dd_be_r_mte } } },
1a039c7e
RH
5661 };
5662 gen_helper_gvec_mem *fn;
28d57f2d 5663 int be = s->be_data == MO_BE;
1a039c7e
RH
5664
5665 if (nreg == 0) {
5666 /* ST1 */
71b9f394
RH
5667 fn = fn_single[s->mte_active[0]][be][msz][esz];
5668 nreg = 1;
1a039c7e
RH
5669 } else {
5670 /* ST2, ST3, ST4 -- msz == esz, enforced by encoding */
5671 assert(msz == esz);
71b9f394 5672 fn = fn_multiple[s->mte_active[0]][be][nreg - 1][msz];
1a039c7e
RH
5673 }
5674 assert(fn != NULL);
71b9f394 5675 do_mem_zpa(s, zt, pg, addr, msz_dtype(s, msz), nreg, true, fn);
1a039c7e
RH
5676}
5677
3a7be554 5678static bool trans_ST_zprr(DisasContext *s, arg_rprr_store *a)
1a039c7e
RH
5679{
5680 if (a->rm == 31 || a->msz > a->esz) {
5681 return false;
5682 }
5683 if (sve_access_check(s)) {
5684 TCGv_i64 addr = new_tmp_a64(s);
50ef1cbf 5685 tcg_gen_shli_i64(addr, cpu_reg(s, a->rm), a->msz);
1a039c7e
RH
5686 tcg_gen_add_i64(addr, addr, cpu_reg_sp(s, a->rn));
5687 do_st_zpa(s, a->rd, a->pg, addr, a->msz, a->esz, a->nreg);
5688 }
5689 return true;
5690}
5691
3a7be554 5692static bool trans_ST_zpri(DisasContext *s, arg_rpri_store *a)
1a039c7e
RH
5693{
5694 if (a->msz > a->esz) {
5695 return false;
5696 }
5697 if (sve_access_check(s)) {
5698 int vsz = vec_full_reg_size(s);
5699 int elements = vsz >> a->esz;
5700 TCGv_i64 addr = new_tmp_a64(s);
5701
5702 tcg_gen_addi_i64(addr, cpu_reg_sp(s, a->rn),
5703 (a->imm * elements * (a->nreg + 1)) << a->msz);
5704 do_st_zpa(s, a->rd, a->pg, addr, a->msz, a->esz, a->nreg);
5705 }
5706 return true;
5707}
f6dbf62a
RH
5708
5709/*
5710 *** SVE gather loads / scatter stores
5711 */
5712
500d0484 5713static void do_mem_zpz(DisasContext *s, int zt, int pg, int zm,
d28d12f0 5714 int scale, TCGv_i64 scalar, int msz, bool is_write,
500d0484 5715 gen_helper_gvec_mem_scatter *fn)
f6dbf62a
RH
5716{
5717 unsigned vsz = vec_full_reg_size(s);
f6dbf62a
RH
5718 TCGv_ptr t_zm = tcg_temp_new_ptr();
5719 TCGv_ptr t_pg = tcg_temp_new_ptr();
5720 TCGv_ptr t_zt = tcg_temp_new_ptr();
500d0484 5721 TCGv_i32 t_desc;
d28d12f0 5722 int desc = 0;
500d0484 5723
d28d12f0
RH
5724 if (s->mte_active[0]) {
5725 desc = FIELD_DP32(desc, MTEDESC, MIDX, get_mem_index(s));
5726 desc = FIELD_DP32(desc, MTEDESC, TBI, s->tbid);
5727 desc = FIELD_DP32(desc, MTEDESC, TCMA, s->tcma);
5728 desc = FIELD_DP32(desc, MTEDESC, WRITE, is_write);
28f32503 5729 desc = FIELD_DP32(desc, MTEDESC, SIZEM1, (1 << msz) - 1);
d28d12f0
RH
5730 desc <<= SVE_MTEDESC_SHIFT;
5731 }
cdecb3fc 5732 desc = simd_desc(vsz, vsz, desc | scale);
500d0484 5733 t_desc = tcg_const_i32(desc);
f6dbf62a
RH
5734
5735 tcg_gen_addi_ptr(t_pg, cpu_env, pred_full_reg_offset(s, pg));
5736 tcg_gen_addi_ptr(t_zm, cpu_env, vec_full_reg_offset(s, zm));
5737 tcg_gen_addi_ptr(t_zt, cpu_env, vec_full_reg_offset(s, zt));
500d0484 5738 fn(cpu_env, t_zt, t_pg, t_zm, scalar, t_desc);
f6dbf62a
RH
5739
5740 tcg_temp_free_ptr(t_zt);
5741 tcg_temp_free_ptr(t_zm);
5742 tcg_temp_free_ptr(t_pg);
500d0484 5743 tcg_temp_free_i32(t_desc);
f6dbf62a
RH
5744}
5745
d28d12f0
RH
5746/* Indexed by [mte][be][ff][xs][u][msz]. */
5747static gen_helper_gvec_mem_scatter * const
5748gather_load_fn32[2][2][2][2][2][3] = {
5749 { /* MTE Inactive */
5750 { /* Little-endian */
5751 { { { gen_helper_sve_ldbss_zsu,
5752 gen_helper_sve_ldhss_le_zsu,
5753 NULL, },
5754 { gen_helper_sve_ldbsu_zsu,
5755 gen_helper_sve_ldhsu_le_zsu,
5756 gen_helper_sve_ldss_le_zsu, } },
5757 { { gen_helper_sve_ldbss_zss,
5758 gen_helper_sve_ldhss_le_zss,
5759 NULL, },
5760 { gen_helper_sve_ldbsu_zss,
5761 gen_helper_sve_ldhsu_le_zss,
5762 gen_helper_sve_ldss_le_zss, } } },
5763
5764 /* First-fault */
5765 { { { gen_helper_sve_ldffbss_zsu,
5766 gen_helper_sve_ldffhss_le_zsu,
5767 NULL, },
5768 { gen_helper_sve_ldffbsu_zsu,
5769 gen_helper_sve_ldffhsu_le_zsu,
5770 gen_helper_sve_ldffss_le_zsu, } },
5771 { { gen_helper_sve_ldffbss_zss,
5772 gen_helper_sve_ldffhss_le_zss,
5773 NULL, },
5774 { gen_helper_sve_ldffbsu_zss,
5775 gen_helper_sve_ldffhsu_le_zss,
5776 gen_helper_sve_ldffss_le_zss, } } } },
5777
5778 { /* Big-endian */
5779 { { { gen_helper_sve_ldbss_zsu,
5780 gen_helper_sve_ldhss_be_zsu,
5781 NULL, },
5782 { gen_helper_sve_ldbsu_zsu,
5783 gen_helper_sve_ldhsu_be_zsu,
5784 gen_helper_sve_ldss_be_zsu, } },
5785 { { gen_helper_sve_ldbss_zss,
5786 gen_helper_sve_ldhss_be_zss,
5787 NULL, },
5788 { gen_helper_sve_ldbsu_zss,
5789 gen_helper_sve_ldhsu_be_zss,
5790 gen_helper_sve_ldss_be_zss, } } },
5791
5792 /* First-fault */
5793 { { { gen_helper_sve_ldffbss_zsu,
5794 gen_helper_sve_ldffhss_be_zsu,
5795 NULL, },
5796 { gen_helper_sve_ldffbsu_zsu,
5797 gen_helper_sve_ldffhsu_be_zsu,
5798 gen_helper_sve_ldffss_be_zsu, } },
5799 { { gen_helper_sve_ldffbss_zss,
5800 gen_helper_sve_ldffhss_be_zss,
5801 NULL, },
5802 { gen_helper_sve_ldffbsu_zss,
5803 gen_helper_sve_ldffhsu_be_zss,
5804 gen_helper_sve_ldffss_be_zss, } } } } },
5805 { /* MTE Active */
5806 { /* Little-endian */
5807 { { { gen_helper_sve_ldbss_zsu_mte,
5808 gen_helper_sve_ldhss_le_zsu_mte,
5809 NULL, },
5810 { gen_helper_sve_ldbsu_zsu_mte,
5811 gen_helper_sve_ldhsu_le_zsu_mte,
5812 gen_helper_sve_ldss_le_zsu_mte, } },
5813 { { gen_helper_sve_ldbss_zss_mte,
5814 gen_helper_sve_ldhss_le_zss_mte,
5815 NULL, },
5816 { gen_helper_sve_ldbsu_zss_mte,
5817 gen_helper_sve_ldhsu_le_zss_mte,
5818 gen_helper_sve_ldss_le_zss_mte, } } },
5819
5820 /* First-fault */
5821 { { { gen_helper_sve_ldffbss_zsu_mte,
5822 gen_helper_sve_ldffhss_le_zsu_mte,
5823 NULL, },
5824 { gen_helper_sve_ldffbsu_zsu_mte,
5825 gen_helper_sve_ldffhsu_le_zsu_mte,
5826 gen_helper_sve_ldffss_le_zsu_mte, } },
5827 { { gen_helper_sve_ldffbss_zss_mte,
5828 gen_helper_sve_ldffhss_le_zss_mte,
5829 NULL, },
5830 { gen_helper_sve_ldffbsu_zss_mte,
5831 gen_helper_sve_ldffhsu_le_zss_mte,
5832 gen_helper_sve_ldffss_le_zss_mte, } } } },
5833
5834 { /* Big-endian */
5835 { { { gen_helper_sve_ldbss_zsu_mte,
5836 gen_helper_sve_ldhss_be_zsu_mte,
5837 NULL, },
5838 { gen_helper_sve_ldbsu_zsu_mte,
5839 gen_helper_sve_ldhsu_be_zsu_mte,
5840 gen_helper_sve_ldss_be_zsu_mte, } },
5841 { { gen_helper_sve_ldbss_zss_mte,
5842 gen_helper_sve_ldhss_be_zss_mte,
5843 NULL, },
5844 { gen_helper_sve_ldbsu_zss_mte,
5845 gen_helper_sve_ldhsu_be_zss_mte,
5846 gen_helper_sve_ldss_be_zss_mte, } } },
5847
5848 /* First-fault */
5849 { { { gen_helper_sve_ldffbss_zsu_mte,
5850 gen_helper_sve_ldffhss_be_zsu_mte,
5851 NULL, },
5852 { gen_helper_sve_ldffbsu_zsu_mte,
5853 gen_helper_sve_ldffhsu_be_zsu_mte,
5854 gen_helper_sve_ldffss_be_zsu_mte, } },
5855 { { gen_helper_sve_ldffbss_zss_mte,
5856 gen_helper_sve_ldffhss_be_zss_mte,
5857 NULL, },
5858 { gen_helper_sve_ldffbsu_zss_mte,
5859 gen_helper_sve_ldffhsu_be_zss_mte,
5860 gen_helper_sve_ldffss_be_zss_mte, } } } } },
673e9fa6
RH
5861};
5862
5863/* Note that we overload xs=2 to indicate 64-bit offset. */
d28d12f0
RH
5864static gen_helper_gvec_mem_scatter * const
5865gather_load_fn64[2][2][2][3][2][4] = {
5866 { /* MTE Inactive */
5867 { /* Little-endian */
5868 { { { gen_helper_sve_ldbds_zsu,
5869 gen_helper_sve_ldhds_le_zsu,
5870 gen_helper_sve_ldsds_le_zsu,
5871 NULL, },
5872 { gen_helper_sve_ldbdu_zsu,
5873 gen_helper_sve_ldhdu_le_zsu,
5874 gen_helper_sve_ldsdu_le_zsu,
5875 gen_helper_sve_lddd_le_zsu, } },
5876 { { gen_helper_sve_ldbds_zss,
5877 gen_helper_sve_ldhds_le_zss,
5878 gen_helper_sve_ldsds_le_zss,
5879 NULL, },
5880 { gen_helper_sve_ldbdu_zss,
5881 gen_helper_sve_ldhdu_le_zss,
5882 gen_helper_sve_ldsdu_le_zss,
5883 gen_helper_sve_lddd_le_zss, } },
5884 { { gen_helper_sve_ldbds_zd,
5885 gen_helper_sve_ldhds_le_zd,
5886 gen_helper_sve_ldsds_le_zd,
5887 NULL, },
5888 { gen_helper_sve_ldbdu_zd,
5889 gen_helper_sve_ldhdu_le_zd,
5890 gen_helper_sve_ldsdu_le_zd,
5891 gen_helper_sve_lddd_le_zd, } } },
5892
5893 /* First-fault */
5894 { { { gen_helper_sve_ldffbds_zsu,
5895 gen_helper_sve_ldffhds_le_zsu,
5896 gen_helper_sve_ldffsds_le_zsu,
5897 NULL, },
5898 { gen_helper_sve_ldffbdu_zsu,
5899 gen_helper_sve_ldffhdu_le_zsu,
5900 gen_helper_sve_ldffsdu_le_zsu,
5901 gen_helper_sve_ldffdd_le_zsu, } },
5902 { { gen_helper_sve_ldffbds_zss,
5903 gen_helper_sve_ldffhds_le_zss,
5904 gen_helper_sve_ldffsds_le_zss,
5905 NULL, },
5906 { gen_helper_sve_ldffbdu_zss,
5907 gen_helper_sve_ldffhdu_le_zss,
5908 gen_helper_sve_ldffsdu_le_zss,
5909 gen_helper_sve_ldffdd_le_zss, } },
5910 { { gen_helper_sve_ldffbds_zd,
5911 gen_helper_sve_ldffhds_le_zd,
5912 gen_helper_sve_ldffsds_le_zd,
5913 NULL, },
5914 { gen_helper_sve_ldffbdu_zd,
5915 gen_helper_sve_ldffhdu_le_zd,
5916 gen_helper_sve_ldffsdu_le_zd,
5917 gen_helper_sve_ldffdd_le_zd, } } } },
5918 { /* Big-endian */
5919 { { { gen_helper_sve_ldbds_zsu,
5920 gen_helper_sve_ldhds_be_zsu,
5921 gen_helper_sve_ldsds_be_zsu,
5922 NULL, },
5923 { gen_helper_sve_ldbdu_zsu,
5924 gen_helper_sve_ldhdu_be_zsu,
5925 gen_helper_sve_ldsdu_be_zsu,
5926 gen_helper_sve_lddd_be_zsu, } },
5927 { { gen_helper_sve_ldbds_zss,
5928 gen_helper_sve_ldhds_be_zss,
5929 gen_helper_sve_ldsds_be_zss,
5930 NULL, },
5931 { gen_helper_sve_ldbdu_zss,
5932 gen_helper_sve_ldhdu_be_zss,
5933 gen_helper_sve_ldsdu_be_zss,
5934 gen_helper_sve_lddd_be_zss, } },
5935 { { gen_helper_sve_ldbds_zd,
5936 gen_helper_sve_ldhds_be_zd,
5937 gen_helper_sve_ldsds_be_zd,
5938 NULL, },
5939 { gen_helper_sve_ldbdu_zd,
5940 gen_helper_sve_ldhdu_be_zd,
5941 gen_helper_sve_ldsdu_be_zd,
5942 gen_helper_sve_lddd_be_zd, } } },
5943
5944 /* First-fault */
5945 { { { gen_helper_sve_ldffbds_zsu,
5946 gen_helper_sve_ldffhds_be_zsu,
5947 gen_helper_sve_ldffsds_be_zsu,
5948 NULL, },
5949 { gen_helper_sve_ldffbdu_zsu,
5950 gen_helper_sve_ldffhdu_be_zsu,
5951 gen_helper_sve_ldffsdu_be_zsu,
5952 gen_helper_sve_ldffdd_be_zsu, } },
5953 { { gen_helper_sve_ldffbds_zss,
5954 gen_helper_sve_ldffhds_be_zss,
5955 gen_helper_sve_ldffsds_be_zss,
5956 NULL, },
5957 { gen_helper_sve_ldffbdu_zss,
5958 gen_helper_sve_ldffhdu_be_zss,
5959 gen_helper_sve_ldffsdu_be_zss,
5960 gen_helper_sve_ldffdd_be_zss, } },
5961 { { gen_helper_sve_ldffbds_zd,
5962 gen_helper_sve_ldffhds_be_zd,
5963 gen_helper_sve_ldffsds_be_zd,
5964 NULL, },
5965 { gen_helper_sve_ldffbdu_zd,
5966 gen_helper_sve_ldffhdu_be_zd,
5967 gen_helper_sve_ldffsdu_be_zd,
5968 gen_helper_sve_ldffdd_be_zd, } } } } },
5969 { /* MTE Active */
5970 { /* Little-endian */
5971 { { { gen_helper_sve_ldbds_zsu_mte,
5972 gen_helper_sve_ldhds_le_zsu_mte,
5973 gen_helper_sve_ldsds_le_zsu_mte,
5974 NULL, },
5975 { gen_helper_sve_ldbdu_zsu_mte,
5976 gen_helper_sve_ldhdu_le_zsu_mte,
5977 gen_helper_sve_ldsdu_le_zsu_mte,
5978 gen_helper_sve_lddd_le_zsu_mte, } },
5979 { { gen_helper_sve_ldbds_zss_mte,
5980 gen_helper_sve_ldhds_le_zss_mte,
5981 gen_helper_sve_ldsds_le_zss_mte,
5982 NULL, },
5983 { gen_helper_sve_ldbdu_zss_mte,
5984 gen_helper_sve_ldhdu_le_zss_mte,
5985 gen_helper_sve_ldsdu_le_zss_mte,
5986 gen_helper_sve_lddd_le_zss_mte, } },
5987 { { gen_helper_sve_ldbds_zd_mte,
5988 gen_helper_sve_ldhds_le_zd_mte,
5989 gen_helper_sve_ldsds_le_zd_mte,
5990 NULL, },
5991 { gen_helper_sve_ldbdu_zd_mte,
5992 gen_helper_sve_ldhdu_le_zd_mte,
5993 gen_helper_sve_ldsdu_le_zd_mte,
5994 gen_helper_sve_lddd_le_zd_mte, } } },
5995
5996 /* First-fault */
5997 { { { gen_helper_sve_ldffbds_zsu_mte,
5998 gen_helper_sve_ldffhds_le_zsu_mte,
5999 gen_helper_sve_ldffsds_le_zsu_mte,
6000 NULL, },
6001 { gen_helper_sve_ldffbdu_zsu_mte,
6002 gen_helper_sve_ldffhdu_le_zsu_mte,
6003 gen_helper_sve_ldffsdu_le_zsu_mte,
6004 gen_helper_sve_ldffdd_le_zsu_mte, } },
6005 { { gen_helper_sve_ldffbds_zss_mte,
6006 gen_helper_sve_ldffhds_le_zss_mte,
6007 gen_helper_sve_ldffsds_le_zss_mte,
6008 NULL, },
6009 { gen_helper_sve_ldffbdu_zss_mte,
6010 gen_helper_sve_ldffhdu_le_zss_mte,
6011 gen_helper_sve_ldffsdu_le_zss_mte,
6012 gen_helper_sve_ldffdd_le_zss_mte, } },
6013 { { gen_helper_sve_ldffbds_zd_mte,
6014 gen_helper_sve_ldffhds_le_zd_mte,
6015 gen_helper_sve_ldffsds_le_zd_mte,
6016 NULL, },
6017 { gen_helper_sve_ldffbdu_zd_mte,
6018 gen_helper_sve_ldffhdu_le_zd_mte,
6019 gen_helper_sve_ldffsdu_le_zd_mte,
6020 gen_helper_sve_ldffdd_le_zd_mte, } } } },
6021 { /* Big-endian */
6022 { { { gen_helper_sve_ldbds_zsu_mte,
6023 gen_helper_sve_ldhds_be_zsu_mte,
6024 gen_helper_sve_ldsds_be_zsu_mte,
6025 NULL, },
6026 { gen_helper_sve_ldbdu_zsu_mte,
6027 gen_helper_sve_ldhdu_be_zsu_mte,
6028 gen_helper_sve_ldsdu_be_zsu_mte,
6029 gen_helper_sve_lddd_be_zsu_mte, } },
6030 { { gen_helper_sve_ldbds_zss_mte,
6031 gen_helper_sve_ldhds_be_zss_mte,
6032 gen_helper_sve_ldsds_be_zss_mte,
6033 NULL, },
6034 { gen_helper_sve_ldbdu_zss_mte,
6035 gen_helper_sve_ldhdu_be_zss_mte,
6036 gen_helper_sve_ldsdu_be_zss_mte,
6037 gen_helper_sve_lddd_be_zss_mte, } },
6038 { { gen_helper_sve_ldbds_zd_mte,
6039 gen_helper_sve_ldhds_be_zd_mte,
6040 gen_helper_sve_ldsds_be_zd_mte,
6041 NULL, },
6042 { gen_helper_sve_ldbdu_zd_mte,
6043 gen_helper_sve_ldhdu_be_zd_mte,
6044 gen_helper_sve_ldsdu_be_zd_mte,
6045 gen_helper_sve_lddd_be_zd_mte, } } },
6046
6047 /* First-fault */
6048 { { { gen_helper_sve_ldffbds_zsu_mte,
6049 gen_helper_sve_ldffhds_be_zsu_mte,
6050 gen_helper_sve_ldffsds_be_zsu_mte,
6051 NULL, },
6052 { gen_helper_sve_ldffbdu_zsu_mte,
6053 gen_helper_sve_ldffhdu_be_zsu_mte,
6054 gen_helper_sve_ldffsdu_be_zsu_mte,
6055 gen_helper_sve_ldffdd_be_zsu_mte, } },
6056 { { gen_helper_sve_ldffbds_zss_mte,
6057 gen_helper_sve_ldffhds_be_zss_mte,
6058 gen_helper_sve_ldffsds_be_zss_mte,
6059 NULL, },
6060 { gen_helper_sve_ldffbdu_zss_mte,
6061 gen_helper_sve_ldffhdu_be_zss_mte,
6062 gen_helper_sve_ldffsdu_be_zss_mte,
6063 gen_helper_sve_ldffdd_be_zss_mte, } },
6064 { { gen_helper_sve_ldffbds_zd_mte,
6065 gen_helper_sve_ldffhds_be_zd_mte,
6066 gen_helper_sve_ldffsds_be_zd_mte,
6067 NULL, },
6068 { gen_helper_sve_ldffbdu_zd_mte,
6069 gen_helper_sve_ldffhdu_be_zd_mte,
6070 gen_helper_sve_ldffsdu_be_zd_mte,
6071 gen_helper_sve_ldffdd_be_zd_mte, } } } } },
673e9fa6
RH
6072};
6073
3a7be554 6074static bool trans_LD1_zprz(DisasContext *s, arg_LD1_zprz *a)
673e9fa6
RH
6075{
6076 gen_helper_gvec_mem_scatter *fn = NULL;
d28d12f0
RH
6077 bool be = s->be_data == MO_BE;
6078 bool mte = s->mte_active[0];
673e9fa6
RH
6079
6080 if (!sve_access_check(s)) {
6081 return true;
6082 }
6083
6084 switch (a->esz) {
6085 case MO_32:
d28d12f0 6086 fn = gather_load_fn32[mte][be][a->ff][a->xs][a->u][a->msz];
673e9fa6
RH
6087 break;
6088 case MO_64:
d28d12f0 6089 fn = gather_load_fn64[mte][be][a->ff][a->xs][a->u][a->msz];
673e9fa6
RH
6090 break;
6091 }
6092 assert(fn != NULL);
6093
6094 do_mem_zpz(s, a->rd, a->pg, a->rm, a->scale * a->msz,
d28d12f0 6095 cpu_reg_sp(s, a->rn), a->msz, false, fn);
673e9fa6
RH
6096 return true;
6097}
6098
3a7be554 6099static bool trans_LD1_zpiz(DisasContext *s, arg_LD1_zpiz *a)
673e9fa6
RH
6100{
6101 gen_helper_gvec_mem_scatter *fn = NULL;
d28d12f0
RH
6102 bool be = s->be_data == MO_BE;
6103 bool mte = s->mte_active[0];
673e9fa6
RH
6104 TCGv_i64 imm;
6105
6106 if (a->esz < a->msz || (a->esz == a->msz && !a->u)) {
6107 return false;
6108 }
6109 if (!sve_access_check(s)) {
6110 return true;
6111 }
6112
6113 switch (a->esz) {
6114 case MO_32:
d28d12f0 6115 fn = gather_load_fn32[mte][be][a->ff][0][a->u][a->msz];
673e9fa6
RH
6116 break;
6117 case MO_64:
d28d12f0 6118 fn = gather_load_fn64[mte][be][a->ff][2][a->u][a->msz];
673e9fa6
RH
6119 break;
6120 }
6121 assert(fn != NULL);
6122
6123 /* Treat LD1_zpiz (zn[x] + imm) the same way as LD1_zprz (rn + zm[x])
6124 * by loading the immediate into the scalar parameter.
6125 */
6126 imm = tcg_const_i64(a->imm << a->msz);
d28d12f0 6127 do_mem_zpz(s, a->rd, a->pg, a->rn, 0, imm, a->msz, false, fn);
673e9fa6
RH
6128 tcg_temp_free_i64(imm);
6129 return true;
6130}
6131
cf327449
SL
6132static bool trans_LDNT1_zprz(DisasContext *s, arg_LD1_zprz *a)
6133{
6134 if (!dc_isar_feature(aa64_sve2, s)) {
6135 return false;
6136 }
6137 return trans_LD1_zprz(s, a);
6138}
6139
d28d12f0
RH
6140/* Indexed by [mte][be][xs][msz]. */
6141static gen_helper_gvec_mem_scatter * const scatter_store_fn32[2][2][2][3] = {
6142 { /* MTE Inactive */
6143 { /* Little-endian */
6144 { gen_helper_sve_stbs_zsu,
6145 gen_helper_sve_sths_le_zsu,
6146 gen_helper_sve_stss_le_zsu, },
6147 { gen_helper_sve_stbs_zss,
6148 gen_helper_sve_sths_le_zss,
6149 gen_helper_sve_stss_le_zss, } },
6150 { /* Big-endian */
6151 { gen_helper_sve_stbs_zsu,
6152 gen_helper_sve_sths_be_zsu,
6153 gen_helper_sve_stss_be_zsu, },
6154 { gen_helper_sve_stbs_zss,
6155 gen_helper_sve_sths_be_zss,
6156 gen_helper_sve_stss_be_zss, } } },
6157 { /* MTE Active */
6158 { /* Little-endian */
6159 { gen_helper_sve_stbs_zsu_mte,
6160 gen_helper_sve_sths_le_zsu_mte,
6161 gen_helper_sve_stss_le_zsu_mte, },
6162 { gen_helper_sve_stbs_zss_mte,
6163 gen_helper_sve_sths_le_zss_mte,
6164 gen_helper_sve_stss_le_zss_mte, } },
6165 { /* Big-endian */
6166 { gen_helper_sve_stbs_zsu_mte,
6167 gen_helper_sve_sths_be_zsu_mte,
6168 gen_helper_sve_stss_be_zsu_mte, },
6169 { gen_helper_sve_stbs_zss_mte,
6170 gen_helper_sve_sths_be_zss_mte,
6171 gen_helper_sve_stss_be_zss_mte, } } },
408ecde9
RH
6172};
6173
6174/* Note that we overload xs=2 to indicate 64-bit offset. */
d28d12f0
RH
6175static gen_helper_gvec_mem_scatter * const scatter_store_fn64[2][2][3][4] = {
6176 { /* MTE Inactive */
6177 { /* Little-endian */
6178 { gen_helper_sve_stbd_zsu,
6179 gen_helper_sve_sthd_le_zsu,
6180 gen_helper_sve_stsd_le_zsu,
6181 gen_helper_sve_stdd_le_zsu, },
6182 { gen_helper_sve_stbd_zss,
6183 gen_helper_sve_sthd_le_zss,
6184 gen_helper_sve_stsd_le_zss,
6185 gen_helper_sve_stdd_le_zss, },
6186 { gen_helper_sve_stbd_zd,
6187 gen_helper_sve_sthd_le_zd,
6188 gen_helper_sve_stsd_le_zd,
6189 gen_helper_sve_stdd_le_zd, } },
6190 { /* Big-endian */
6191 { gen_helper_sve_stbd_zsu,
6192 gen_helper_sve_sthd_be_zsu,
6193 gen_helper_sve_stsd_be_zsu,
6194 gen_helper_sve_stdd_be_zsu, },
6195 { gen_helper_sve_stbd_zss,
6196 gen_helper_sve_sthd_be_zss,
6197 gen_helper_sve_stsd_be_zss,
6198 gen_helper_sve_stdd_be_zss, },
6199 { gen_helper_sve_stbd_zd,
6200 gen_helper_sve_sthd_be_zd,
6201 gen_helper_sve_stsd_be_zd,
6202 gen_helper_sve_stdd_be_zd, } } },
6203 { /* MTE Inactive */
6204 { /* Little-endian */
6205 { gen_helper_sve_stbd_zsu_mte,
6206 gen_helper_sve_sthd_le_zsu_mte,
6207 gen_helper_sve_stsd_le_zsu_mte,
6208 gen_helper_sve_stdd_le_zsu_mte, },
6209 { gen_helper_sve_stbd_zss_mte,
6210 gen_helper_sve_sthd_le_zss_mte,
6211 gen_helper_sve_stsd_le_zss_mte,
6212 gen_helper_sve_stdd_le_zss_mte, },
6213 { gen_helper_sve_stbd_zd_mte,
6214 gen_helper_sve_sthd_le_zd_mte,
6215 gen_helper_sve_stsd_le_zd_mte,
6216 gen_helper_sve_stdd_le_zd_mte, } },
6217 { /* Big-endian */
6218 { gen_helper_sve_stbd_zsu_mte,
6219 gen_helper_sve_sthd_be_zsu_mte,
6220 gen_helper_sve_stsd_be_zsu_mte,
6221 gen_helper_sve_stdd_be_zsu_mte, },
6222 { gen_helper_sve_stbd_zss_mte,
6223 gen_helper_sve_sthd_be_zss_mte,
6224 gen_helper_sve_stsd_be_zss_mte,
6225 gen_helper_sve_stdd_be_zss_mte, },
6226 { gen_helper_sve_stbd_zd_mte,
6227 gen_helper_sve_sthd_be_zd_mte,
6228 gen_helper_sve_stsd_be_zd_mte,
6229 gen_helper_sve_stdd_be_zd_mte, } } },
408ecde9
RH
6230};
6231
3a7be554 6232static bool trans_ST1_zprz(DisasContext *s, arg_ST1_zprz *a)
f6dbf62a 6233{
f6dbf62a 6234 gen_helper_gvec_mem_scatter *fn;
d28d12f0
RH
6235 bool be = s->be_data == MO_BE;
6236 bool mte = s->mte_active[0];
f6dbf62a
RH
6237
6238 if (a->esz < a->msz || (a->msz == 0 && a->scale)) {
6239 return false;
6240 }
6241 if (!sve_access_check(s)) {
6242 return true;
6243 }
6244 switch (a->esz) {
6245 case MO_32:
d28d12f0 6246 fn = scatter_store_fn32[mte][be][a->xs][a->msz];
f6dbf62a
RH
6247 break;
6248 case MO_64:
d28d12f0 6249 fn = scatter_store_fn64[mte][be][a->xs][a->msz];
f6dbf62a
RH
6250 break;
6251 default:
6252 g_assert_not_reached();
6253 }
6254 do_mem_zpz(s, a->rd, a->pg, a->rm, a->scale * a->msz,
d28d12f0 6255 cpu_reg_sp(s, a->rn), a->msz, true, fn);
f6dbf62a
RH
6256 return true;
6257}
dec6cf6b 6258
3a7be554 6259static bool trans_ST1_zpiz(DisasContext *s, arg_ST1_zpiz *a)
408ecde9
RH
6260{
6261 gen_helper_gvec_mem_scatter *fn = NULL;
d28d12f0
RH
6262 bool be = s->be_data == MO_BE;
6263 bool mte = s->mte_active[0];
408ecde9
RH
6264 TCGv_i64 imm;
6265
6266 if (a->esz < a->msz) {
6267 return false;
6268 }
6269 if (!sve_access_check(s)) {
6270 return true;
6271 }
6272
6273 switch (a->esz) {
6274 case MO_32:
d28d12f0 6275 fn = scatter_store_fn32[mte][be][0][a->msz];
408ecde9
RH
6276 break;
6277 case MO_64:
d28d12f0 6278 fn = scatter_store_fn64[mte][be][2][a->msz];
408ecde9
RH
6279 break;
6280 }
6281 assert(fn != NULL);
6282
6283 /* Treat ST1_zpiz (zn[x] + imm) the same way as ST1_zprz (rn + zm[x])
6284 * by loading the immediate into the scalar parameter.
6285 */
6286 imm = tcg_const_i64(a->imm << a->msz);
d28d12f0 6287 do_mem_zpz(s, a->rd, a->pg, a->rn, 0, imm, a->msz, true, fn);
408ecde9
RH
6288 tcg_temp_free_i64(imm);
6289 return true;
6290}
6291
6ebca45f
SL
6292static bool trans_STNT1_zprz(DisasContext *s, arg_ST1_zprz *a)
6293{
6294 if (!dc_isar_feature(aa64_sve2, s)) {
6295 return false;
6296 }
6297 return trans_ST1_zprz(s, a);
6298}
6299
dec6cf6b
RH
6300/*
6301 * Prefetches
6302 */
6303
3a7be554 6304static bool trans_PRF(DisasContext *s, arg_PRF *a)
dec6cf6b
RH
6305{
6306 /* Prefetch is a nop within QEMU. */
2f95a3b0 6307 (void)sve_access_check(s);
dec6cf6b
RH
6308 return true;
6309}
6310
3a7be554 6311static bool trans_PRF_rr(DisasContext *s, arg_PRF_rr *a)
dec6cf6b
RH
6312{
6313 if (a->rm == 31) {
6314 return false;
6315 }
6316 /* Prefetch is a nop within QEMU. */
2f95a3b0 6317 (void)sve_access_check(s);
dec6cf6b
RH
6318 return true;
6319}
a2103582
RH
6320
6321/*
6322 * Move Prefix
6323 *
6324 * TODO: The implementation so far could handle predicated merging movprfx.
6325 * The helper functions as written take an extra source register to
6326 * use in the operation, but the result is only written when predication
6327 * succeeds. For unpredicated movprfx, we need to rearrange the helpers
6328 * to allow the final write back to the destination to be unconditional.
6329 * For predicated zeroing movprfx, we need to rearrange the helpers to
6330 * allow the final write back to zero inactives.
6331 *
6332 * In the meantime, just emit the moves.
6333 */
6334
3a7be554 6335static bool trans_MOVPRFX(DisasContext *s, arg_MOVPRFX *a)
a2103582
RH
6336{
6337 return do_mov_z(s, a->rd, a->rn);
6338}
6339
3a7be554 6340static bool trans_MOVPRFX_m(DisasContext *s, arg_rpr_esz *a)
a2103582
RH
6341{
6342 if (sve_access_check(s)) {
6343 do_sel_z(s, a->rd, a->rn, a->rd, a->pg, a->esz);
6344 }
6345 return true;
6346}
6347
3a7be554 6348static bool trans_MOVPRFX_z(DisasContext *s, arg_rpr_esz *a)
a2103582 6349{
60245996 6350 return do_movz_zpz(s, a->rd, a->rn, a->pg, a->esz, false);
a2103582 6351}
5dad1ba5
RH
6352
6353/*
6354 * SVE2 Integer Multiply - Unpredicated
6355 */
6356
6357static bool trans_MUL_zzz(DisasContext *s, arg_rrr_esz *a)
6358{
6359 if (!dc_isar_feature(aa64_sve2, s)) {
6360 return false;
6361 }
6362 if (sve_access_check(s)) {
6363 gen_gvec_fn_zzz(s, tcg_gen_gvec_mul, a->esz, a->rd, a->rn, a->rm);
6364 }
6365 return true;
6366}
6367
6368static bool do_sve2_zzz_ool(DisasContext *s, arg_rrr_esz *a,
6369 gen_helper_gvec_3 *fn)
6370{
6371 if (fn == NULL || !dc_isar_feature(aa64_sve2, s)) {
6372 return false;
6373 }
6374 if (sve_access_check(s)) {
6375 gen_gvec_ool_zzz(s, fn, a->rd, a->rn, a->rm, 0);
6376 }
6377 return true;
6378}
6379
6380static bool trans_SMULH_zzz(DisasContext *s, arg_rrr_esz *a)
6381{
6382 static gen_helper_gvec_3 * const fns[4] = {
6383 gen_helper_gvec_smulh_b, gen_helper_gvec_smulh_h,
6384 gen_helper_gvec_smulh_s, gen_helper_gvec_smulh_d,
6385 };
6386 return do_sve2_zzz_ool(s, a, fns[a->esz]);
6387}
6388
6389static bool trans_UMULH_zzz(DisasContext *s, arg_rrr_esz *a)
6390{
6391 static gen_helper_gvec_3 * const fns[4] = {
6392 gen_helper_gvec_umulh_b, gen_helper_gvec_umulh_h,
6393 gen_helper_gvec_umulh_s, gen_helper_gvec_umulh_d,
6394 };
6395 return do_sve2_zzz_ool(s, a, fns[a->esz]);
6396}
6397
6398static bool trans_PMUL_zzz(DisasContext *s, arg_rrr_esz *a)
6399{
6400 return do_sve2_zzz_ool(s, a, gen_helper_gvec_pmul_b);
6401}
d4b1e59d
RH
6402
6403/*
6404 * SVE2 Integer - Predicated
6405 */
6406
6407static bool do_sve2_zpzz_ool(DisasContext *s, arg_rprr_esz *a,
6408 gen_helper_gvec_4 *fn)
6409{
6410 if (!dc_isar_feature(aa64_sve2, s)) {
6411 return false;
6412 }
6413 return do_zpzz_ool(s, a, fn);
6414}
6415
6416static bool trans_SADALP_zpzz(DisasContext *s, arg_rprr_esz *a)
6417{
6418 static gen_helper_gvec_4 * const fns[3] = {
6419 gen_helper_sve2_sadalp_zpzz_h,
6420 gen_helper_sve2_sadalp_zpzz_s,
6421 gen_helper_sve2_sadalp_zpzz_d,
6422 };
6423 if (a->esz == 0) {
6424 return false;
6425 }
6426 return do_sve2_zpzz_ool(s, a, fns[a->esz - 1]);
6427}
6428
6429static bool trans_UADALP_zpzz(DisasContext *s, arg_rprr_esz *a)
6430{
6431 static gen_helper_gvec_4 * const fns[3] = {
6432 gen_helper_sve2_uadalp_zpzz_h,
6433 gen_helper_sve2_uadalp_zpzz_s,
6434 gen_helper_sve2_uadalp_zpzz_d,
6435 };
6436 if (a->esz == 0) {
6437 return false;
6438 }
6439 return do_sve2_zpzz_ool(s, a, fns[a->esz - 1]);
6440}
db366da8
RH
6441
6442/*
6443 * SVE2 integer unary operations (predicated)
6444 */
6445
6446static bool do_sve2_zpz_ool(DisasContext *s, arg_rpr_esz *a,
6447 gen_helper_gvec_3 *fn)
6448{
6449 if (!dc_isar_feature(aa64_sve2, s)) {
6450 return false;
6451 }
6452 return do_zpz_ool(s, a, fn);
6453}
6454
6455static bool trans_URECPE(DisasContext *s, arg_rpr_esz *a)
6456{
6457 if (a->esz != 2) {
6458 return false;
6459 }
6460 return do_sve2_zpz_ool(s, a, gen_helper_sve2_urecpe_s);
6461}
6462
6463static bool trans_URSQRTE(DisasContext *s, arg_rpr_esz *a)
6464{
6465 if (a->esz != 2) {
6466 return false;
6467 }
6468 return do_sve2_zpz_ool(s, a, gen_helper_sve2_ursqrte_s);
6469}
6470
6471static bool trans_SQABS(DisasContext *s, arg_rpr_esz *a)
6472{
6473 static gen_helper_gvec_3 * const fns[4] = {
6474 gen_helper_sve2_sqabs_b, gen_helper_sve2_sqabs_h,
6475 gen_helper_sve2_sqabs_s, gen_helper_sve2_sqabs_d,
6476 };
6477 return do_sve2_zpz_ool(s, a, fns[a->esz]);
6478}
6479
6480static bool trans_SQNEG(DisasContext *s, arg_rpr_esz *a)
6481{
6482 static gen_helper_gvec_3 * const fns[4] = {
6483 gen_helper_sve2_sqneg_b, gen_helper_sve2_sqneg_h,
6484 gen_helper_sve2_sqneg_s, gen_helper_sve2_sqneg_d,
6485 };
6486 return do_sve2_zpz_ool(s, a, fns[a->esz]);
6487}
45d9503d
RH
6488
6489#define DO_SVE2_ZPZZ(NAME, name) \
6490static bool trans_##NAME(DisasContext *s, arg_rprr_esz *a) \
6491{ \
6492 static gen_helper_gvec_4 * const fns[4] = { \
6493 gen_helper_sve2_##name##_zpzz_b, gen_helper_sve2_##name##_zpzz_h, \
6494 gen_helper_sve2_##name##_zpzz_s, gen_helper_sve2_##name##_zpzz_d, \
6495 }; \
6496 return do_sve2_zpzz_ool(s, a, fns[a->esz]); \
6497}
6498
6499DO_SVE2_ZPZZ(SQSHL, sqshl)
6500DO_SVE2_ZPZZ(SQRSHL, sqrshl)
6501DO_SVE2_ZPZZ(SRSHL, srshl)
6502
6503DO_SVE2_ZPZZ(UQSHL, uqshl)
6504DO_SVE2_ZPZZ(UQRSHL, uqrshl)
6505DO_SVE2_ZPZZ(URSHL, urshl)
a47dc220
RH
6506
6507DO_SVE2_ZPZZ(SHADD, shadd)
6508DO_SVE2_ZPZZ(SRHADD, srhadd)
6509DO_SVE2_ZPZZ(SHSUB, shsub)
6510
6511DO_SVE2_ZPZZ(UHADD, uhadd)
6512DO_SVE2_ZPZZ(URHADD, urhadd)
6513DO_SVE2_ZPZZ(UHSUB, uhsub)
8597dc8b
RH
6514
6515DO_SVE2_ZPZZ(ADDP, addp)
6516DO_SVE2_ZPZZ(SMAXP, smaxp)
6517DO_SVE2_ZPZZ(UMAXP, umaxp)
6518DO_SVE2_ZPZZ(SMINP, sminp)
6519DO_SVE2_ZPZZ(UMINP, uminp)
4f07fbeb
RH
6520
6521DO_SVE2_ZPZZ(SQADD_zpzz, sqadd)
6522DO_SVE2_ZPZZ(UQADD_zpzz, uqadd)
6523DO_SVE2_ZPZZ(SQSUB_zpzz, sqsub)
6524DO_SVE2_ZPZZ(UQSUB_zpzz, uqsub)
6525DO_SVE2_ZPZZ(SUQADD, suqadd)
6526DO_SVE2_ZPZZ(USQADD, usqadd)
0ce1dda8
RH
6527
6528/*
6529 * SVE2 Widening Integer Arithmetic
6530 */
6531
6532static bool do_sve2_zzw_ool(DisasContext *s, arg_rrr_esz *a,
6533 gen_helper_gvec_3 *fn, int data)
6534{
6535 if (fn == NULL || !dc_isar_feature(aa64_sve2, s)) {
6536 return false;
6537 }
6538 if (sve_access_check(s)) {
6539 unsigned vsz = vec_full_reg_size(s);
6540 tcg_gen_gvec_3_ool(vec_full_reg_offset(s, a->rd),
6541 vec_full_reg_offset(s, a->rn),
6542 vec_full_reg_offset(s, a->rm),
6543 vsz, vsz, data, fn);
6544 }
6545 return true;
6546}
6547
6548#define DO_SVE2_ZZZ_TB(NAME, name, SEL1, SEL2) \
6549static bool trans_##NAME(DisasContext *s, arg_rrr_esz *a) \
6550{ \
6551 static gen_helper_gvec_3 * const fns[4] = { \
6552 NULL, gen_helper_sve2_##name##_h, \
6553 gen_helper_sve2_##name##_s, gen_helper_sve2_##name##_d, \
6554 }; \
6555 return do_sve2_zzw_ool(s, a, fns[a->esz], (SEL2 << 1) | SEL1); \
6556}
6557
6558DO_SVE2_ZZZ_TB(SADDLB, saddl, false, false)
6559DO_SVE2_ZZZ_TB(SSUBLB, ssubl, false, false)
6560DO_SVE2_ZZZ_TB(SABDLB, sabdl, false, false)
6561
6562DO_SVE2_ZZZ_TB(UADDLB, uaddl, false, false)
6563DO_SVE2_ZZZ_TB(USUBLB, usubl, false, false)
6564DO_SVE2_ZZZ_TB(UABDLB, uabdl, false, false)
6565
6566DO_SVE2_ZZZ_TB(SADDLT, saddl, true, true)
6567DO_SVE2_ZZZ_TB(SSUBLT, ssubl, true, true)
6568DO_SVE2_ZZZ_TB(SABDLT, sabdl, true, true)
6569
6570DO_SVE2_ZZZ_TB(UADDLT, uaddl, true, true)
6571DO_SVE2_ZZZ_TB(USUBLT, usubl, true, true)
6572DO_SVE2_ZZZ_TB(UABDLT, uabdl, true, true)
daec426b
RH
6573
6574DO_SVE2_ZZZ_TB(SADDLBT, saddl, false, true)
6575DO_SVE2_ZZZ_TB(SSUBLBT, ssubl, false, true)
6576DO_SVE2_ZZZ_TB(SSUBLTB, ssubl, true, false)
81fccf09 6577
69ccc099
RH
6578DO_SVE2_ZZZ_TB(SQDMULLB_zzz, sqdmull_zzz, false, false)
6579DO_SVE2_ZZZ_TB(SQDMULLT_zzz, sqdmull_zzz, true, true)
6580
6581DO_SVE2_ZZZ_TB(SMULLB_zzz, smull_zzz, false, false)
6582DO_SVE2_ZZZ_TB(SMULLT_zzz, smull_zzz, true, true)
6583
6584DO_SVE2_ZZZ_TB(UMULLB_zzz, umull_zzz, false, false)
6585DO_SVE2_ZZZ_TB(UMULLT_zzz, umull_zzz, true, true)
6586
2df3ca55
RH
6587static bool do_eor_tb(DisasContext *s, arg_rrr_esz *a, bool sel1)
6588{
6589 static gen_helper_gvec_3 * const fns[4] = {
6590 gen_helper_sve2_eoril_b, gen_helper_sve2_eoril_h,
6591 gen_helper_sve2_eoril_s, gen_helper_sve2_eoril_d,
6592 };
6593 return do_sve2_zzw_ool(s, a, fns[a->esz], (!sel1 << 1) | sel1);
6594}
6595
6596static bool trans_EORBT(DisasContext *s, arg_rrr_esz *a)
6597{
6598 return do_eor_tb(s, a, false);
6599}
6600
6601static bool trans_EORTB(DisasContext *s, arg_rrr_esz *a)
6602{
6603 return do_eor_tb(s, a, true);
6604}
6605
e3a56131
RH
6606static bool do_trans_pmull(DisasContext *s, arg_rrr_esz *a, bool sel)
6607{
6608 static gen_helper_gvec_3 * const fns[4] = {
6609 gen_helper_gvec_pmull_q, gen_helper_sve2_pmull_h,
6610 NULL, gen_helper_sve2_pmull_d,
6611 };
6612 if (a->esz == 0 && !dc_isar_feature(aa64_sve2_pmull128, s)) {
6613 return false;
6614 }
6615 return do_sve2_zzw_ool(s, a, fns[a->esz], sel);
6616}
6617
6618static bool trans_PMULLB(DisasContext *s, arg_rrr_esz *a)
6619{
6620 return do_trans_pmull(s, a, false);
6621}
6622
6623static bool trans_PMULLT(DisasContext *s, arg_rrr_esz *a)
6624{
6625 return do_trans_pmull(s, a, true);
6626}
6627
81fccf09
RH
6628#define DO_SVE2_ZZZ_WTB(NAME, name, SEL2) \
6629static bool trans_##NAME(DisasContext *s, arg_rrr_esz *a) \
6630{ \
6631 static gen_helper_gvec_3 * const fns[4] = { \
6632 NULL, gen_helper_sve2_##name##_h, \
6633 gen_helper_sve2_##name##_s, gen_helper_sve2_##name##_d, \
6634 }; \
6635 return do_sve2_zzw_ool(s, a, fns[a->esz], SEL2); \
6636}
6637
6638DO_SVE2_ZZZ_WTB(SADDWB, saddw, false)
6639DO_SVE2_ZZZ_WTB(SADDWT, saddw, true)
6640DO_SVE2_ZZZ_WTB(SSUBWB, ssubw, false)
6641DO_SVE2_ZZZ_WTB(SSUBWT, ssubw, true)
6642
6643DO_SVE2_ZZZ_WTB(UADDWB, uaddw, false)
6644DO_SVE2_ZZZ_WTB(UADDWT, uaddw, true)
6645DO_SVE2_ZZZ_WTB(USUBWB, usubw, false)
6646DO_SVE2_ZZZ_WTB(USUBWT, usubw, true)
4269fef1
RH
6647
6648static void gen_sshll_vec(unsigned vece, TCGv_vec d, TCGv_vec n, int64_t imm)
6649{
6650 int top = imm & 1;
6651 int shl = imm >> 1;
6652 int halfbits = 4 << vece;
6653
6654 if (top) {
6655 if (shl == halfbits) {
6656 TCGv_vec t = tcg_temp_new_vec_matching(d);
6657 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(halfbits, halfbits));
6658 tcg_gen_and_vec(vece, d, n, t);
6659 tcg_temp_free_vec(t);
6660 } else {
6661 tcg_gen_sari_vec(vece, d, n, halfbits);
6662 tcg_gen_shli_vec(vece, d, d, shl);
6663 }
6664 } else {
6665 tcg_gen_shli_vec(vece, d, n, halfbits);
6666 tcg_gen_sari_vec(vece, d, d, halfbits - shl);
6667 }
6668}
6669
6670static void gen_ushll_i64(unsigned vece, TCGv_i64 d, TCGv_i64 n, int imm)
6671{
6672 int halfbits = 4 << vece;
6673 int top = imm & 1;
6674 int shl = (imm >> 1);
6675 int shift;
6676 uint64_t mask;
6677
6678 mask = MAKE_64BIT_MASK(0, halfbits);
6679 mask <<= shl;
6680 mask = dup_const(vece, mask);
6681
6682 shift = shl - top * halfbits;
6683 if (shift < 0) {
6684 tcg_gen_shri_i64(d, n, -shift);
6685 } else {
6686 tcg_gen_shli_i64(d, n, shift);
6687 }
6688 tcg_gen_andi_i64(d, d, mask);
6689}
6690
6691static void gen_ushll16_i64(TCGv_i64 d, TCGv_i64 n, int64_t imm)
6692{
6693 gen_ushll_i64(MO_16, d, n, imm);
6694}
6695
6696static void gen_ushll32_i64(TCGv_i64 d, TCGv_i64 n, int64_t imm)
6697{
6698 gen_ushll_i64(MO_32, d, n, imm);
6699}
6700
6701static void gen_ushll64_i64(TCGv_i64 d, TCGv_i64 n, int64_t imm)
6702{
6703 gen_ushll_i64(MO_64, d, n, imm);
6704}
6705
6706static void gen_ushll_vec(unsigned vece, TCGv_vec d, TCGv_vec n, int64_t imm)
6707{
6708 int halfbits = 4 << vece;
6709 int top = imm & 1;
6710 int shl = imm >> 1;
6711
6712 if (top) {
6713 if (shl == halfbits) {
6714 TCGv_vec t = tcg_temp_new_vec_matching(d);
6715 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(halfbits, halfbits));
6716 tcg_gen_and_vec(vece, d, n, t);
6717 tcg_temp_free_vec(t);
6718 } else {
6719 tcg_gen_shri_vec(vece, d, n, halfbits);
6720 tcg_gen_shli_vec(vece, d, d, shl);
6721 }
6722 } else {
6723 if (shl == 0) {
6724 TCGv_vec t = tcg_temp_new_vec_matching(d);
6725 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(0, halfbits));
6726 tcg_gen_and_vec(vece, d, n, t);
6727 tcg_temp_free_vec(t);
6728 } else {
6729 tcg_gen_shli_vec(vece, d, n, halfbits);
6730 tcg_gen_shri_vec(vece, d, d, halfbits - shl);
6731 }
6732 }
6733}
6734
6735static bool do_sve2_shll_tb(DisasContext *s, arg_rri_esz *a,
6736 bool sel, bool uns)
6737{
6738 static const TCGOpcode sshll_list[] = {
6739 INDEX_op_shli_vec, INDEX_op_sari_vec, 0
6740 };
6741 static const TCGOpcode ushll_list[] = {
6742 INDEX_op_shli_vec, INDEX_op_shri_vec, 0
6743 };
6744 static const GVecGen2i ops[2][3] = {
6745 { { .fniv = gen_sshll_vec,
6746 .opt_opc = sshll_list,
6747 .fno = gen_helper_sve2_sshll_h,
6748 .vece = MO_16 },
6749 { .fniv = gen_sshll_vec,
6750 .opt_opc = sshll_list,
6751 .fno = gen_helper_sve2_sshll_s,
6752 .vece = MO_32 },
6753 { .fniv = gen_sshll_vec,
6754 .opt_opc = sshll_list,
6755 .fno = gen_helper_sve2_sshll_d,
6756 .vece = MO_64 } },
6757 { { .fni8 = gen_ushll16_i64,
6758 .fniv = gen_ushll_vec,
6759 .opt_opc = ushll_list,
6760 .fno = gen_helper_sve2_ushll_h,
6761 .vece = MO_16 },
6762 { .fni8 = gen_ushll32_i64,
6763 .fniv = gen_ushll_vec,
6764 .opt_opc = ushll_list,
6765 .fno = gen_helper_sve2_ushll_s,
6766 .vece = MO_32 },
6767 { .fni8 = gen_ushll64_i64,
6768 .fniv = gen_ushll_vec,
6769 .opt_opc = ushll_list,
6770 .fno = gen_helper_sve2_ushll_d,
6771 .vece = MO_64 } },
6772 };
6773
6774 if (a->esz < 0 || a->esz > 2 || !dc_isar_feature(aa64_sve2, s)) {
6775 return false;
6776 }
6777 if (sve_access_check(s)) {
6778 unsigned vsz = vec_full_reg_size(s);
6779 tcg_gen_gvec_2i(vec_full_reg_offset(s, a->rd),
6780 vec_full_reg_offset(s, a->rn),
6781 vsz, vsz, (a->imm << 1) | sel,
6782 &ops[uns][a->esz]);
6783 }
6784 return true;
6785}
6786
6787static bool trans_SSHLLB(DisasContext *s, arg_rri_esz *a)
6788{
6789 return do_sve2_shll_tb(s, a, false, false);
6790}
6791
6792static bool trans_SSHLLT(DisasContext *s, arg_rri_esz *a)
6793{
6794 return do_sve2_shll_tb(s, a, true, false);
6795}
6796
6797static bool trans_USHLLB(DisasContext *s, arg_rri_esz *a)
6798{
6799 return do_sve2_shll_tb(s, a, false, true);
6800}
6801
6802static bool trans_USHLLT(DisasContext *s, arg_rri_esz *a)
6803{
6804 return do_sve2_shll_tb(s, a, true, true);
6805}
cb9c33b8
RH
6806
6807static bool trans_BEXT(DisasContext *s, arg_rrr_esz *a)
6808{
6809 static gen_helper_gvec_3 * const fns[4] = {
6810 gen_helper_sve2_bext_b, gen_helper_sve2_bext_h,
6811 gen_helper_sve2_bext_s, gen_helper_sve2_bext_d,
6812 };
6813 if (!dc_isar_feature(aa64_sve2_bitperm, s)) {
6814 return false;
6815 }
6816 return do_sve2_zzw_ool(s, a, fns[a->esz], 0);
6817}
6818
6819static bool trans_BDEP(DisasContext *s, arg_rrr_esz *a)
6820{
6821 static gen_helper_gvec_3 * const fns[4] = {
6822 gen_helper_sve2_bdep_b, gen_helper_sve2_bdep_h,
6823 gen_helper_sve2_bdep_s, gen_helper_sve2_bdep_d,
6824 };
6825 if (!dc_isar_feature(aa64_sve2_bitperm, s)) {
6826 return false;
6827 }
6828 return do_sve2_zzw_ool(s, a, fns[a->esz], 0);
6829}
6830
6831static bool trans_BGRP(DisasContext *s, arg_rrr_esz *a)
6832{
6833 static gen_helper_gvec_3 * const fns[4] = {
6834 gen_helper_sve2_bgrp_b, gen_helper_sve2_bgrp_h,
6835 gen_helper_sve2_bgrp_s, gen_helper_sve2_bgrp_d,
6836 };
6837 if (!dc_isar_feature(aa64_sve2_bitperm, s)) {
6838 return false;
6839 }
6840 return do_sve2_zzw_ool(s, a, fns[a->esz], 0);
6841}
ed4a6387
RH
6842
6843static bool do_cadd(DisasContext *s, arg_rrr_esz *a, bool sq, bool rot)
6844{
6845 static gen_helper_gvec_3 * const fns[2][4] = {
6846 { gen_helper_sve2_cadd_b, gen_helper_sve2_cadd_h,
6847 gen_helper_sve2_cadd_s, gen_helper_sve2_cadd_d },
6848 { gen_helper_sve2_sqcadd_b, gen_helper_sve2_sqcadd_h,
6849 gen_helper_sve2_sqcadd_s, gen_helper_sve2_sqcadd_d },
6850 };
6851 return do_sve2_zzw_ool(s, a, fns[sq][a->esz], rot);
6852}
6853
6854static bool trans_CADD_rot90(DisasContext *s, arg_rrr_esz *a)
6855{
6856 return do_cadd(s, a, false, false);
6857}
6858
6859static bool trans_CADD_rot270(DisasContext *s, arg_rrr_esz *a)
6860{
6861 return do_cadd(s, a, false, true);
6862}
6863
6864static bool trans_SQCADD_rot90(DisasContext *s, arg_rrr_esz *a)
6865{
6866 return do_cadd(s, a, true, false);
6867}
6868
6869static bool trans_SQCADD_rot270(DisasContext *s, arg_rrr_esz *a)
6870{
6871 return do_cadd(s, a, true, true);
6872}
38650638
RH
6873
6874static bool do_sve2_zzzz_ool(DisasContext *s, arg_rrrr_esz *a,
6875 gen_helper_gvec_4 *fn, int data)
6876{
6877 if (fn == NULL || !dc_isar_feature(aa64_sve2, s)) {
6878 return false;
6879 }
6880 if (sve_access_check(s)) {
6881 gen_gvec_ool_zzzz(s, fn, a->rd, a->rn, a->rm, a->ra, data);
6882 }
6883 return true;
6884}
6885
6886static bool do_abal(DisasContext *s, arg_rrrr_esz *a, bool uns, bool sel)
6887{
6888 static gen_helper_gvec_4 * const fns[2][4] = {
6889 { NULL, gen_helper_sve2_sabal_h,
6890 gen_helper_sve2_sabal_s, gen_helper_sve2_sabal_d },
6891 { NULL, gen_helper_sve2_uabal_h,
6892 gen_helper_sve2_uabal_s, gen_helper_sve2_uabal_d },
6893 };
6894 return do_sve2_zzzz_ool(s, a, fns[uns][a->esz], sel);
6895}
6896
6897static bool trans_SABALB(DisasContext *s, arg_rrrr_esz *a)
6898{
6899 return do_abal(s, a, false, false);
6900}
6901
6902static bool trans_SABALT(DisasContext *s, arg_rrrr_esz *a)
6903{
6904 return do_abal(s, a, false, true);
6905}
6906
6907static bool trans_UABALB(DisasContext *s, arg_rrrr_esz *a)
6908{
6909 return do_abal(s, a, true, false);
6910}
6911
6912static bool trans_UABALT(DisasContext *s, arg_rrrr_esz *a)
6913{
6914 return do_abal(s, a, true, true);
6915}
b8295dfb
RH
6916
6917static bool do_adcl(DisasContext *s, arg_rrrr_esz *a, bool sel)
6918{
6919 static gen_helper_gvec_4 * const fns[2] = {
6920 gen_helper_sve2_adcl_s,
6921 gen_helper_sve2_adcl_d,
6922 };
6923 /*
6924 * Note that in this case the ESZ field encodes both size and sign.
6925 * Split out 'subtract' into bit 1 of the data field for the helper.
6926 */
6927 return do_sve2_zzzz_ool(s, a, fns[a->esz & 1], (a->esz & 2) | sel);
6928}
6929
6930static bool trans_ADCLB(DisasContext *s, arg_rrrr_esz *a)
6931{
6932 return do_adcl(s, a, false);
6933}
6934
6935static bool trans_ADCLT(DisasContext *s, arg_rrrr_esz *a)
6936{
6937 return do_adcl(s, a, true);
6938}
a7e3a90e
RH
6939
6940static bool do_sve2_fn2i(DisasContext *s, arg_rri_esz *a, GVecGen2iFn *fn)
6941{
6942 if (a->esz < 0 || !dc_isar_feature(aa64_sve2, s)) {
6943 return false;
6944 }
6945 if (sve_access_check(s)) {
6946 unsigned vsz = vec_full_reg_size(s);
6947 unsigned rd_ofs = vec_full_reg_offset(s, a->rd);
6948 unsigned rn_ofs = vec_full_reg_offset(s, a->rn);
6949 fn(a->esz, rd_ofs, rn_ofs, a->imm, vsz, vsz);
6950 }
6951 return true;
6952}
6953
6954static bool trans_SSRA(DisasContext *s, arg_rri_esz *a)
6955{
6956 return do_sve2_fn2i(s, a, gen_gvec_ssra);
6957}
6958
6959static bool trans_USRA(DisasContext *s, arg_rri_esz *a)
6960{
6961 return do_sve2_fn2i(s, a, gen_gvec_usra);
6962}
6963
6964static bool trans_SRSRA(DisasContext *s, arg_rri_esz *a)
6965{
6966 return do_sve2_fn2i(s, a, gen_gvec_srsra);
6967}
6968
6969static bool trans_URSRA(DisasContext *s, arg_rri_esz *a)
6970{
6971 return do_sve2_fn2i(s, a, gen_gvec_ursra);
6972}
fc12b46a
RH
6973
6974static bool trans_SRI(DisasContext *s, arg_rri_esz *a)
6975{
6976 return do_sve2_fn2i(s, a, gen_gvec_sri);
6977}
6978
6979static bool trans_SLI(DisasContext *s, arg_rri_esz *a)
6980{
6981 return do_sve2_fn2i(s, a, gen_gvec_sli);
6982}
289a1797
RH
6983
6984static bool do_sve2_fn_zzz(DisasContext *s, arg_rrr_esz *a, GVecGen3Fn *fn)
6985{
6986 if (!dc_isar_feature(aa64_sve2, s)) {
6987 return false;
6988 }
6989 if (sve_access_check(s)) {
6990 gen_gvec_fn_zzz(s, fn, a->esz, a->rd, a->rn, a->rm);
6991 }
6992 return true;
6993}
6994
6995static bool trans_SABA(DisasContext *s, arg_rrr_esz *a)
6996{
6997 return do_sve2_fn_zzz(s, a, gen_gvec_saba);
6998}
6999
7000static bool trans_UABA(DisasContext *s, arg_rrr_esz *a)
7001{
7002 return do_sve2_fn_zzz(s, a, gen_gvec_uaba);
7003}
5ff2838d
RH
7004
7005static bool do_sve2_narrow_extract(DisasContext *s, arg_rri_esz *a,
7006 const GVecGen2 ops[3])
7007{
7008 if (a->esz < 0 || a->esz > MO_32 || a->imm != 0 ||
7009 !dc_isar_feature(aa64_sve2, s)) {
7010 return false;
7011 }
7012 if (sve_access_check(s)) {
7013 unsigned vsz = vec_full_reg_size(s);
7014 tcg_gen_gvec_2(vec_full_reg_offset(s, a->rd),
7015 vec_full_reg_offset(s, a->rn),
7016 vsz, vsz, &ops[a->esz]);
7017 }
7018 return true;
7019}
7020
7021static const TCGOpcode sqxtn_list[] = {
7022 INDEX_op_shli_vec, INDEX_op_smin_vec, INDEX_op_smax_vec, 0
7023};
7024
7025static void gen_sqxtnb_vec(unsigned vece, TCGv_vec d, TCGv_vec n)
7026{
7027 TCGv_vec t = tcg_temp_new_vec_matching(d);
7028 int halfbits = 4 << vece;
7029 int64_t mask = (1ull << halfbits) - 1;
7030 int64_t min = -1ull << (halfbits - 1);
7031 int64_t max = -min - 1;
7032
7033 tcg_gen_dupi_vec(vece, t, min);
7034 tcg_gen_smax_vec(vece, d, n, t);
7035 tcg_gen_dupi_vec(vece, t, max);
7036 tcg_gen_smin_vec(vece, d, d, t);
7037 tcg_gen_dupi_vec(vece, t, mask);
7038 tcg_gen_and_vec(vece, d, d, t);
7039 tcg_temp_free_vec(t);
7040}
7041
7042static bool trans_SQXTNB(DisasContext *s, arg_rri_esz *a)
7043{
7044 static const GVecGen2 ops[3] = {
7045 { .fniv = gen_sqxtnb_vec,
7046 .opt_opc = sqxtn_list,
7047 .fno = gen_helper_sve2_sqxtnb_h,
7048 .vece = MO_16 },
7049 { .fniv = gen_sqxtnb_vec,
7050 .opt_opc = sqxtn_list,
7051 .fno = gen_helper_sve2_sqxtnb_s,
7052 .vece = MO_32 },
7053 { .fniv = gen_sqxtnb_vec,
7054 .opt_opc = sqxtn_list,
7055 .fno = gen_helper_sve2_sqxtnb_d,
7056 .vece = MO_64 },
7057 };
7058 return do_sve2_narrow_extract(s, a, ops);
7059}
7060
7061static void gen_sqxtnt_vec(unsigned vece, TCGv_vec d, TCGv_vec n)
7062{
7063 TCGv_vec t = tcg_temp_new_vec_matching(d);
7064 int halfbits = 4 << vece;
7065 int64_t mask = (1ull << halfbits) - 1;
7066 int64_t min = -1ull << (halfbits - 1);
7067 int64_t max = -min - 1;
7068
7069 tcg_gen_dupi_vec(vece, t, min);
7070 tcg_gen_smax_vec(vece, n, n, t);
7071 tcg_gen_dupi_vec(vece, t, max);
7072 tcg_gen_smin_vec(vece, n, n, t);
7073 tcg_gen_shli_vec(vece, n, n, halfbits);
7074 tcg_gen_dupi_vec(vece, t, mask);
7075 tcg_gen_bitsel_vec(vece, d, t, d, n);
7076 tcg_temp_free_vec(t);
7077}
7078
7079static bool trans_SQXTNT(DisasContext *s, arg_rri_esz *a)
7080{
7081 static const GVecGen2 ops[3] = {
7082 { .fniv = gen_sqxtnt_vec,
7083 .opt_opc = sqxtn_list,
7084 .load_dest = true,
7085 .fno = gen_helper_sve2_sqxtnt_h,
7086 .vece = MO_16 },
7087 { .fniv = gen_sqxtnt_vec,
7088 .opt_opc = sqxtn_list,
7089 .load_dest = true,
7090 .fno = gen_helper_sve2_sqxtnt_s,
7091 .vece = MO_32 },
7092 { .fniv = gen_sqxtnt_vec,
7093 .opt_opc = sqxtn_list,
7094 .load_dest = true,
7095 .fno = gen_helper_sve2_sqxtnt_d,
7096 .vece = MO_64 },
7097 };
7098 return do_sve2_narrow_extract(s, a, ops);
7099}
7100
7101static const TCGOpcode uqxtn_list[] = {
7102 INDEX_op_shli_vec, INDEX_op_umin_vec, 0
7103};
7104
7105static void gen_uqxtnb_vec(unsigned vece, TCGv_vec d, TCGv_vec n)
7106{
7107 TCGv_vec t = tcg_temp_new_vec_matching(d);
7108 int halfbits = 4 << vece;
7109 int64_t max = (1ull << halfbits) - 1;
7110
7111 tcg_gen_dupi_vec(vece, t, max);
7112 tcg_gen_umin_vec(vece, d, n, t);
7113 tcg_temp_free_vec(t);
7114}
7115
7116static bool trans_UQXTNB(DisasContext *s, arg_rri_esz *a)
7117{
7118 static const GVecGen2 ops[3] = {
7119 { .fniv = gen_uqxtnb_vec,
7120 .opt_opc = uqxtn_list,
7121 .fno = gen_helper_sve2_uqxtnb_h,
7122 .vece = MO_16 },
7123 { .fniv = gen_uqxtnb_vec,
7124 .opt_opc = uqxtn_list,
7125 .fno = gen_helper_sve2_uqxtnb_s,
7126 .vece = MO_32 },
7127 { .fniv = gen_uqxtnb_vec,
7128 .opt_opc = uqxtn_list,
7129 .fno = gen_helper_sve2_uqxtnb_d,
7130 .vece = MO_64 },
7131 };
7132 return do_sve2_narrow_extract(s, a, ops);
7133}
7134
7135static void gen_uqxtnt_vec(unsigned vece, TCGv_vec d, TCGv_vec n)
7136{
7137 TCGv_vec t = tcg_temp_new_vec_matching(d);
7138 int halfbits = 4 << vece;
7139 int64_t max = (1ull << halfbits) - 1;
7140
7141 tcg_gen_dupi_vec(vece, t, max);
7142 tcg_gen_umin_vec(vece, n, n, t);
7143 tcg_gen_shli_vec(vece, n, n, halfbits);
7144 tcg_gen_bitsel_vec(vece, d, t, d, n);
7145 tcg_temp_free_vec(t);
7146}
7147
7148static bool trans_UQXTNT(DisasContext *s, arg_rri_esz *a)
7149{
7150 static const GVecGen2 ops[3] = {
7151 { .fniv = gen_uqxtnt_vec,
7152 .opt_opc = uqxtn_list,
7153 .load_dest = true,
7154 .fno = gen_helper_sve2_uqxtnt_h,
7155 .vece = MO_16 },
7156 { .fniv = gen_uqxtnt_vec,
7157 .opt_opc = uqxtn_list,
7158 .load_dest = true,
7159 .fno = gen_helper_sve2_uqxtnt_s,
7160 .vece = MO_32 },
7161 { .fniv = gen_uqxtnt_vec,
7162 .opt_opc = uqxtn_list,
7163 .load_dest = true,
7164 .fno = gen_helper_sve2_uqxtnt_d,
7165 .vece = MO_64 },
7166 };
7167 return do_sve2_narrow_extract(s, a, ops);
7168}
7169
7170static const TCGOpcode sqxtun_list[] = {
7171 INDEX_op_shli_vec, INDEX_op_umin_vec, INDEX_op_smax_vec, 0
7172};
7173
7174static void gen_sqxtunb_vec(unsigned vece, TCGv_vec d, TCGv_vec n)
7175{
7176 TCGv_vec t = tcg_temp_new_vec_matching(d);
7177 int halfbits = 4 << vece;
7178 int64_t max = (1ull << halfbits) - 1;
7179
7180 tcg_gen_dupi_vec(vece, t, 0);
7181 tcg_gen_smax_vec(vece, d, n, t);
7182 tcg_gen_dupi_vec(vece, t, max);
7183 tcg_gen_umin_vec(vece, d, d, t);
7184 tcg_temp_free_vec(t);
7185}
7186
7187static bool trans_SQXTUNB(DisasContext *s, arg_rri_esz *a)
7188{
7189 static const GVecGen2 ops[3] = {
7190 { .fniv = gen_sqxtunb_vec,
7191 .opt_opc = sqxtun_list,
7192 .fno = gen_helper_sve2_sqxtunb_h,
7193 .vece = MO_16 },
7194 { .fniv = gen_sqxtunb_vec,
7195 .opt_opc = sqxtun_list,
7196 .fno = gen_helper_sve2_sqxtunb_s,
7197 .vece = MO_32 },
7198 { .fniv = gen_sqxtunb_vec,
7199 .opt_opc = sqxtun_list,
7200 .fno = gen_helper_sve2_sqxtunb_d,
7201 .vece = MO_64 },
7202 };
7203 return do_sve2_narrow_extract(s, a, ops);
7204}
7205
7206static void gen_sqxtunt_vec(unsigned vece, TCGv_vec d, TCGv_vec n)
7207{
7208 TCGv_vec t = tcg_temp_new_vec_matching(d);
7209 int halfbits = 4 << vece;
7210 int64_t max = (1ull << halfbits) - 1;
7211
7212 tcg_gen_dupi_vec(vece, t, 0);
7213 tcg_gen_smax_vec(vece, n, n, t);
7214 tcg_gen_dupi_vec(vece, t, max);
7215 tcg_gen_umin_vec(vece, n, n, t);
7216 tcg_gen_shli_vec(vece, n, n, halfbits);
7217 tcg_gen_bitsel_vec(vece, d, t, d, n);
7218 tcg_temp_free_vec(t);
7219}
7220
7221static bool trans_SQXTUNT(DisasContext *s, arg_rri_esz *a)
7222{
7223 static const GVecGen2 ops[3] = {
7224 { .fniv = gen_sqxtunt_vec,
7225 .opt_opc = sqxtun_list,
7226 .load_dest = true,
7227 .fno = gen_helper_sve2_sqxtunt_h,
7228 .vece = MO_16 },
7229 { .fniv = gen_sqxtunt_vec,
7230 .opt_opc = sqxtun_list,
7231 .load_dest = true,
7232 .fno = gen_helper_sve2_sqxtunt_s,
7233 .vece = MO_32 },
7234 { .fniv = gen_sqxtunt_vec,
7235 .opt_opc = sqxtun_list,
7236 .load_dest = true,
7237 .fno = gen_helper_sve2_sqxtunt_d,
7238 .vece = MO_64 },
7239 };
7240 return do_sve2_narrow_extract(s, a, ops);
46d111b2
RH
7241}
7242
7243static bool do_sve2_shr_narrow(DisasContext *s, arg_rri_esz *a,
7244 const GVecGen2i ops[3])
7245{
7246 if (a->esz < 0 || a->esz > MO_32 || !dc_isar_feature(aa64_sve2, s)) {
7247 return false;
7248 }
7249 assert(a->imm > 0 && a->imm <= (8 << a->esz));
7250 if (sve_access_check(s)) {
7251 unsigned vsz = vec_full_reg_size(s);
7252 tcg_gen_gvec_2i(vec_full_reg_offset(s, a->rd),
7253 vec_full_reg_offset(s, a->rn),
7254 vsz, vsz, a->imm, &ops[a->esz]);
7255 }
7256 return true;
7257}
7258
7259static void gen_shrnb_i64(unsigned vece, TCGv_i64 d, TCGv_i64 n, int shr)
7260{
7261 int halfbits = 4 << vece;
7262 uint64_t mask = dup_const(vece, MAKE_64BIT_MASK(0, halfbits));
7263
7264 tcg_gen_shri_i64(d, n, shr);
7265 tcg_gen_andi_i64(d, d, mask);
7266}
7267
7268static void gen_shrnb16_i64(TCGv_i64 d, TCGv_i64 n, int64_t shr)
7269{
7270 gen_shrnb_i64(MO_16, d, n, shr);
7271}
7272
7273static void gen_shrnb32_i64(TCGv_i64 d, TCGv_i64 n, int64_t shr)
7274{
7275 gen_shrnb_i64(MO_32, d, n, shr);
7276}
7277
7278static void gen_shrnb64_i64(TCGv_i64 d, TCGv_i64 n, int64_t shr)
7279{
7280 gen_shrnb_i64(MO_64, d, n, shr);
7281}
7282
7283static void gen_shrnb_vec(unsigned vece, TCGv_vec d, TCGv_vec n, int64_t shr)
7284{
7285 TCGv_vec t = tcg_temp_new_vec_matching(d);
7286 int halfbits = 4 << vece;
7287 uint64_t mask = MAKE_64BIT_MASK(0, halfbits);
7288
7289 tcg_gen_shri_vec(vece, n, n, shr);
7290 tcg_gen_dupi_vec(vece, t, mask);
7291 tcg_gen_and_vec(vece, d, n, t);
7292 tcg_temp_free_vec(t);
7293}
7294
7295static bool trans_SHRNB(DisasContext *s, arg_rri_esz *a)
7296{
7297 static const TCGOpcode vec_list[] = { INDEX_op_shri_vec, 0 };
7298 static const GVecGen2i ops[3] = {
7299 { .fni8 = gen_shrnb16_i64,
7300 .fniv = gen_shrnb_vec,
7301 .opt_opc = vec_list,
7302 .fno = gen_helper_sve2_shrnb_h,
7303 .vece = MO_16 },
7304 { .fni8 = gen_shrnb32_i64,
7305 .fniv = gen_shrnb_vec,
7306 .opt_opc = vec_list,
7307 .fno = gen_helper_sve2_shrnb_s,
7308 .vece = MO_32 },
7309 { .fni8 = gen_shrnb64_i64,
7310 .fniv = gen_shrnb_vec,
7311 .opt_opc = vec_list,
7312 .fno = gen_helper_sve2_shrnb_d,
7313 .vece = MO_64 },
7314 };
7315 return do_sve2_shr_narrow(s, a, ops);
7316}
7317
7318static void gen_shrnt_i64(unsigned vece, TCGv_i64 d, TCGv_i64 n, int shr)
7319{
7320 int halfbits = 4 << vece;
7321 uint64_t mask = dup_const(vece, MAKE_64BIT_MASK(0, halfbits));
7322
7323 tcg_gen_shli_i64(n, n, halfbits - shr);
7324 tcg_gen_andi_i64(n, n, ~mask);
7325 tcg_gen_andi_i64(d, d, mask);
7326 tcg_gen_or_i64(d, d, n);
7327}
7328
7329static void gen_shrnt16_i64(TCGv_i64 d, TCGv_i64 n, int64_t shr)
7330{
7331 gen_shrnt_i64(MO_16, d, n, shr);
7332}
7333
7334static void gen_shrnt32_i64(TCGv_i64 d, TCGv_i64 n, int64_t shr)
7335{
7336 gen_shrnt_i64(MO_32, d, n, shr);
7337}
7338
7339static void gen_shrnt64_i64(TCGv_i64 d, TCGv_i64 n, int64_t shr)
7340{
7341 tcg_gen_shri_i64(n, n, shr);
7342 tcg_gen_deposit_i64(d, d, n, 32, 32);
7343}
7344
7345static void gen_shrnt_vec(unsigned vece, TCGv_vec d, TCGv_vec n, int64_t shr)
7346{
7347 TCGv_vec t = tcg_temp_new_vec_matching(d);
7348 int halfbits = 4 << vece;
7349 uint64_t mask = MAKE_64BIT_MASK(0, halfbits);
7350
7351 tcg_gen_shli_vec(vece, n, n, halfbits - shr);
7352 tcg_gen_dupi_vec(vece, t, mask);
7353 tcg_gen_bitsel_vec(vece, d, t, d, n);
7354 tcg_temp_free_vec(t);
7355}
7356
7357static bool trans_SHRNT(DisasContext *s, arg_rri_esz *a)
7358{
7359 static const TCGOpcode vec_list[] = { INDEX_op_shli_vec, 0 };
7360 static const GVecGen2i ops[3] = {
7361 { .fni8 = gen_shrnt16_i64,
7362 .fniv = gen_shrnt_vec,
7363 .opt_opc = vec_list,
7364 .load_dest = true,
7365 .fno = gen_helper_sve2_shrnt_h,
7366 .vece = MO_16 },
7367 { .fni8 = gen_shrnt32_i64,
7368 .fniv = gen_shrnt_vec,
7369 .opt_opc = vec_list,
7370 .load_dest = true,
7371 .fno = gen_helper_sve2_shrnt_s,
7372 .vece = MO_32 },
7373 { .fni8 = gen_shrnt64_i64,
7374 .fniv = gen_shrnt_vec,
7375 .opt_opc = vec_list,
7376 .load_dest = true,
7377 .fno = gen_helper_sve2_shrnt_d,
7378 .vece = MO_64 },
7379 };
7380 return do_sve2_shr_narrow(s, a, ops);
7381}
7382
7383static bool trans_RSHRNB(DisasContext *s, arg_rri_esz *a)
7384{
7385 static const GVecGen2i ops[3] = {
7386 { .fno = gen_helper_sve2_rshrnb_h },
7387 { .fno = gen_helper_sve2_rshrnb_s },
7388 { .fno = gen_helper_sve2_rshrnb_d },
7389 };
7390 return do_sve2_shr_narrow(s, a, ops);
7391}
7392
7393static bool trans_RSHRNT(DisasContext *s, arg_rri_esz *a)
7394{
7395 static const GVecGen2i ops[3] = {
7396 { .fno = gen_helper_sve2_rshrnt_h },
7397 { .fno = gen_helper_sve2_rshrnt_s },
7398 { .fno = gen_helper_sve2_rshrnt_d },
7399 };
7400 return do_sve2_shr_narrow(s, a, ops);
81fd3e6e
RH
7401}
7402
7403static void gen_sqshrunb_vec(unsigned vece, TCGv_vec d,
7404 TCGv_vec n, int64_t shr)
7405{
7406 TCGv_vec t = tcg_temp_new_vec_matching(d);
7407 int halfbits = 4 << vece;
7408
7409 tcg_gen_sari_vec(vece, n, n, shr);
7410 tcg_gen_dupi_vec(vece, t, 0);
7411 tcg_gen_smax_vec(vece, n, n, t);
7412 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(0, halfbits));
7413 tcg_gen_umin_vec(vece, d, n, t);
7414 tcg_temp_free_vec(t);
7415}
7416
7417static bool trans_SQSHRUNB(DisasContext *s, arg_rri_esz *a)
7418{
7419 static const TCGOpcode vec_list[] = {
7420 INDEX_op_sari_vec, INDEX_op_smax_vec, INDEX_op_umin_vec, 0
7421 };
7422 static const GVecGen2i ops[3] = {
7423 { .fniv = gen_sqshrunb_vec,
7424 .opt_opc = vec_list,
7425 .fno = gen_helper_sve2_sqshrunb_h,
7426 .vece = MO_16 },
7427 { .fniv = gen_sqshrunb_vec,
7428 .opt_opc = vec_list,
7429 .fno = gen_helper_sve2_sqshrunb_s,
7430 .vece = MO_32 },
7431 { .fniv = gen_sqshrunb_vec,
7432 .opt_opc = vec_list,
7433 .fno = gen_helper_sve2_sqshrunb_d,
7434 .vece = MO_64 },
7435 };
7436 return do_sve2_shr_narrow(s, a, ops);
7437}
7438
7439static void gen_sqshrunt_vec(unsigned vece, TCGv_vec d,
7440 TCGv_vec n, int64_t shr)
7441{
7442 TCGv_vec t = tcg_temp_new_vec_matching(d);
7443 int halfbits = 4 << vece;
7444
7445 tcg_gen_sari_vec(vece, n, n, shr);
7446 tcg_gen_dupi_vec(vece, t, 0);
7447 tcg_gen_smax_vec(vece, n, n, t);
7448 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(0, halfbits));
7449 tcg_gen_umin_vec(vece, n, n, t);
7450 tcg_gen_shli_vec(vece, n, n, halfbits);
7451 tcg_gen_bitsel_vec(vece, d, t, d, n);
7452 tcg_temp_free_vec(t);
7453}
7454
7455static bool trans_SQSHRUNT(DisasContext *s, arg_rri_esz *a)
7456{
7457 static const TCGOpcode vec_list[] = {
7458 INDEX_op_shli_vec, INDEX_op_sari_vec,
7459 INDEX_op_smax_vec, INDEX_op_umin_vec, 0
7460 };
7461 static const GVecGen2i ops[3] = {
7462 { .fniv = gen_sqshrunt_vec,
7463 .opt_opc = vec_list,
7464 .load_dest = true,
7465 .fno = gen_helper_sve2_sqshrunt_h,
7466 .vece = MO_16 },
7467 { .fniv = gen_sqshrunt_vec,
7468 .opt_opc = vec_list,
7469 .load_dest = true,
7470 .fno = gen_helper_sve2_sqshrunt_s,
7471 .vece = MO_32 },
7472 { .fniv = gen_sqshrunt_vec,
7473 .opt_opc = vec_list,
7474 .load_dest = true,
7475 .fno = gen_helper_sve2_sqshrunt_d,
7476 .vece = MO_64 },
7477 };
7478 return do_sve2_shr_narrow(s, a, ops);
7479}
7480
7481static bool trans_SQRSHRUNB(DisasContext *s, arg_rri_esz *a)
7482{
7483 static const GVecGen2i ops[3] = {
7484 { .fno = gen_helper_sve2_sqrshrunb_h },
7485 { .fno = gen_helper_sve2_sqrshrunb_s },
7486 { .fno = gen_helper_sve2_sqrshrunb_d },
7487 };
7488 return do_sve2_shr_narrow(s, a, ops);
7489}
7490
7491static bool trans_SQRSHRUNT(DisasContext *s, arg_rri_esz *a)
7492{
7493 static const GVecGen2i ops[3] = {
7494 { .fno = gen_helper_sve2_sqrshrunt_h },
7495 { .fno = gen_helper_sve2_sqrshrunt_s },
7496 { .fno = gen_helper_sve2_sqrshrunt_d },
7497 };
7498 return do_sve2_shr_narrow(s, a, ops);
c13418da
RH
7499}
7500
743bb147
RH
7501static void gen_sqshrnb_vec(unsigned vece, TCGv_vec d,
7502 TCGv_vec n, int64_t shr)
7503{
7504 TCGv_vec t = tcg_temp_new_vec_matching(d);
7505 int halfbits = 4 << vece;
7506 int64_t max = MAKE_64BIT_MASK(0, halfbits - 1);
7507 int64_t min = -max - 1;
7508
7509 tcg_gen_sari_vec(vece, n, n, shr);
7510 tcg_gen_dupi_vec(vece, t, min);
7511 tcg_gen_smax_vec(vece, n, n, t);
7512 tcg_gen_dupi_vec(vece, t, max);
7513 tcg_gen_smin_vec(vece, n, n, t);
7514 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(0, halfbits));
7515 tcg_gen_and_vec(vece, d, n, t);
7516 tcg_temp_free_vec(t);
7517}
7518
7519static bool trans_SQSHRNB(DisasContext *s, arg_rri_esz *a)
7520{
7521 static const TCGOpcode vec_list[] = {
7522 INDEX_op_sari_vec, INDEX_op_smax_vec, INDEX_op_smin_vec, 0
7523 };
7524 static const GVecGen2i ops[3] = {
7525 { .fniv = gen_sqshrnb_vec,
7526 .opt_opc = vec_list,
7527 .fno = gen_helper_sve2_sqshrnb_h,
7528 .vece = MO_16 },
7529 { .fniv = gen_sqshrnb_vec,
7530 .opt_opc = vec_list,
7531 .fno = gen_helper_sve2_sqshrnb_s,
7532 .vece = MO_32 },
7533 { .fniv = gen_sqshrnb_vec,
7534 .opt_opc = vec_list,
7535 .fno = gen_helper_sve2_sqshrnb_d,
7536 .vece = MO_64 },
7537 };
7538 return do_sve2_shr_narrow(s, a, ops);
7539}
7540
7541static void gen_sqshrnt_vec(unsigned vece, TCGv_vec d,
7542 TCGv_vec n, int64_t shr)
7543{
7544 TCGv_vec t = tcg_temp_new_vec_matching(d);
7545 int halfbits = 4 << vece;
7546 int64_t max = MAKE_64BIT_MASK(0, halfbits - 1);
7547 int64_t min = -max - 1;
7548
7549 tcg_gen_sari_vec(vece, n, n, shr);
7550 tcg_gen_dupi_vec(vece, t, min);
7551 tcg_gen_smax_vec(vece, n, n, t);
7552 tcg_gen_dupi_vec(vece, t, max);
7553 tcg_gen_smin_vec(vece, n, n, t);
7554 tcg_gen_shli_vec(vece, n, n, halfbits);
7555 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(0, halfbits));
7556 tcg_gen_bitsel_vec(vece, d, t, d, n);
7557 tcg_temp_free_vec(t);
7558}
7559
7560static bool trans_SQSHRNT(DisasContext *s, arg_rri_esz *a)
7561{
7562 static const TCGOpcode vec_list[] = {
7563 INDEX_op_shli_vec, INDEX_op_sari_vec,
7564 INDEX_op_smax_vec, INDEX_op_smin_vec, 0
7565 };
7566 static const GVecGen2i ops[3] = {
7567 { .fniv = gen_sqshrnt_vec,
7568 .opt_opc = vec_list,
7569 .load_dest = true,
7570 .fno = gen_helper_sve2_sqshrnt_h,
7571 .vece = MO_16 },
7572 { .fniv = gen_sqshrnt_vec,
7573 .opt_opc = vec_list,
7574 .load_dest = true,
7575 .fno = gen_helper_sve2_sqshrnt_s,
7576 .vece = MO_32 },
7577 { .fniv = gen_sqshrnt_vec,
7578 .opt_opc = vec_list,
7579 .load_dest = true,
7580 .fno = gen_helper_sve2_sqshrnt_d,
7581 .vece = MO_64 },
7582 };
7583 return do_sve2_shr_narrow(s, a, ops);
7584}
7585
7586static bool trans_SQRSHRNB(DisasContext *s, arg_rri_esz *a)
7587{
7588 static const GVecGen2i ops[3] = {
7589 { .fno = gen_helper_sve2_sqrshrnb_h },
7590 { .fno = gen_helper_sve2_sqrshrnb_s },
7591 { .fno = gen_helper_sve2_sqrshrnb_d },
7592 };
7593 return do_sve2_shr_narrow(s, a, ops);
7594}
7595
7596static bool trans_SQRSHRNT(DisasContext *s, arg_rri_esz *a)
7597{
7598 static const GVecGen2i ops[3] = {
7599 { .fno = gen_helper_sve2_sqrshrnt_h },
7600 { .fno = gen_helper_sve2_sqrshrnt_s },
7601 { .fno = gen_helper_sve2_sqrshrnt_d },
7602 };
7603 return do_sve2_shr_narrow(s, a, ops);
7604}
7605
c13418da
RH
7606static void gen_uqshrnb_vec(unsigned vece, TCGv_vec d,
7607 TCGv_vec n, int64_t shr)
7608{
7609 TCGv_vec t = tcg_temp_new_vec_matching(d);
7610 int halfbits = 4 << vece;
7611
7612 tcg_gen_shri_vec(vece, n, n, shr);
7613 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(0, halfbits));
7614 tcg_gen_umin_vec(vece, d, n, t);
7615 tcg_temp_free_vec(t);
7616}
7617
7618static bool trans_UQSHRNB(DisasContext *s, arg_rri_esz *a)
7619{
7620 static const TCGOpcode vec_list[] = {
7621 INDEX_op_shri_vec, INDEX_op_umin_vec, 0
7622 };
7623 static const GVecGen2i ops[3] = {
7624 { .fniv = gen_uqshrnb_vec,
7625 .opt_opc = vec_list,
7626 .fno = gen_helper_sve2_uqshrnb_h,
7627 .vece = MO_16 },
7628 { .fniv = gen_uqshrnb_vec,
7629 .opt_opc = vec_list,
7630 .fno = gen_helper_sve2_uqshrnb_s,
7631 .vece = MO_32 },
7632 { .fniv = gen_uqshrnb_vec,
7633 .opt_opc = vec_list,
7634 .fno = gen_helper_sve2_uqshrnb_d,
7635 .vece = MO_64 },
7636 };
7637 return do_sve2_shr_narrow(s, a, ops);
7638}
7639
7640static void gen_uqshrnt_vec(unsigned vece, TCGv_vec d,
7641 TCGv_vec n, int64_t shr)
7642{
7643 TCGv_vec t = tcg_temp_new_vec_matching(d);
7644 int halfbits = 4 << vece;
7645
7646 tcg_gen_shri_vec(vece, n, n, shr);
7647 tcg_gen_dupi_vec(vece, t, MAKE_64BIT_MASK(0, halfbits));
7648 tcg_gen_umin_vec(vece, n, n, t);
7649 tcg_gen_shli_vec(vece, n, n, halfbits);
7650 tcg_gen_bitsel_vec(vece, d, t, d, n);
7651 tcg_temp_free_vec(t);
7652}
7653
7654static bool trans_UQSHRNT(DisasContext *s, arg_rri_esz *a)
7655{
7656 static const TCGOpcode vec_list[] = {
7657 INDEX_op_shli_vec, INDEX_op_shri_vec, INDEX_op_umin_vec, 0
7658 };
7659 static const GVecGen2i ops[3] = {
7660 { .fniv = gen_uqshrnt_vec,
7661 .opt_opc = vec_list,
7662 .load_dest = true,
7663 .fno = gen_helper_sve2_uqshrnt_h,
7664 .vece = MO_16 },
7665 { .fniv = gen_uqshrnt_vec,
7666 .opt_opc = vec_list,
7667 .load_dest = true,
7668 .fno = gen_helper_sve2_uqshrnt_s,
7669 .vece = MO_32 },
7670 { .fniv = gen_uqshrnt_vec,
7671 .opt_opc = vec_list,
7672 .load_dest = true,
7673 .fno = gen_helper_sve2_uqshrnt_d,
7674 .vece = MO_64 },
7675 };
7676 return do_sve2_shr_narrow(s, a, ops);
7677}
7678
7679static bool trans_UQRSHRNB(DisasContext *s, arg_rri_esz *a)
7680{
7681 static const GVecGen2i ops[3] = {
7682 { .fno = gen_helper_sve2_uqrshrnb_h },
7683 { .fno = gen_helper_sve2_uqrshrnb_s },
7684 { .fno = gen_helper_sve2_uqrshrnb_d },
7685 };
7686 return do_sve2_shr_narrow(s, a, ops);
7687}
7688
7689static bool trans_UQRSHRNT(DisasContext *s, arg_rri_esz *a)
7690{
7691 static const GVecGen2i ops[3] = {
7692 { .fno = gen_helper_sve2_uqrshrnt_h },
7693 { .fno = gen_helper_sve2_uqrshrnt_s },
7694 { .fno = gen_helper_sve2_uqrshrnt_d },
7695 };
7696 return do_sve2_shr_narrow(s, a, ops);
5ff2838d 7697}
b87dbeeb 7698
40d5ea50
SL
7699#define DO_SVE2_ZZZ_NARROW(NAME, name) \
7700static bool trans_##NAME(DisasContext *s, arg_rrr_esz *a) \
7701{ \
7702 static gen_helper_gvec_3 * const fns[4] = { \
7703 NULL, gen_helper_sve2_##name##_h, \
7704 gen_helper_sve2_##name##_s, gen_helper_sve2_##name##_d, \
7705 }; \
7706 return do_sve2_zzz_ool(s, a, fns[a->esz]); \
7707}
7708
7709DO_SVE2_ZZZ_NARROW(ADDHNB, addhnb)
7710DO_SVE2_ZZZ_NARROW(ADDHNT, addhnt)
0ea3ff02
SL
7711DO_SVE2_ZZZ_NARROW(RADDHNB, raddhnb)
7712DO_SVE2_ZZZ_NARROW(RADDHNT, raddhnt)
40d5ea50 7713
c3cd6766
SL
7714DO_SVE2_ZZZ_NARROW(SUBHNB, subhnb)
7715DO_SVE2_ZZZ_NARROW(SUBHNT, subhnt)
e9443d10
SL
7716DO_SVE2_ZZZ_NARROW(RSUBHNB, rsubhnb)
7717DO_SVE2_ZZZ_NARROW(RSUBHNT, rsubhnt)
c3cd6766 7718
e0ae6ec3
SL
7719static bool do_sve2_ppzz_flags(DisasContext *s, arg_rprr_esz *a,
7720 gen_helper_gvec_flags_4 *fn)
7721{
7722 if (!dc_isar_feature(aa64_sve2, s)) {
7723 return false;
7724 }
7725 return do_ppzz_flags(s, a, fn);
7726}
7727
7728#define DO_SVE2_PPZZ_MATCH(NAME, name) \
7729static bool trans_##NAME(DisasContext *s, arg_rprr_esz *a) \
7730{ \
7731 static gen_helper_gvec_flags_4 * const fns[4] = { \
7732 gen_helper_sve2_##name##_ppzz_b, gen_helper_sve2_##name##_ppzz_h, \
7733 NULL, NULL \
7734 }; \
7735 return do_sve2_ppzz_flags(s, a, fns[a->esz]); \
7736}
7737
7738DO_SVE2_PPZZ_MATCH(MATCH, match)
7739DO_SVE2_PPZZ_MATCH(NMATCH, nmatch)
7740
7d47ac94
SL
7741static bool trans_HISTCNT(DisasContext *s, arg_rprr_esz *a)
7742{
7743 static gen_helper_gvec_4 * const fns[2] = {
7744 gen_helper_sve2_histcnt_s, gen_helper_sve2_histcnt_d
7745 };
7746 if (a->esz < 2) {
7747 return false;
7748 }
7749 return do_sve2_zpzz_ool(s, a, fns[a->esz - 2]);
7750}
7751
7752static bool trans_HISTSEG(DisasContext *s, arg_rrr_esz *a)
7753{
7754 if (a->esz != 0) {
7755 return false;
7756 }
7757 return do_sve2_zzz_ool(s, a, gen_helper_sve2_histseg);
7758}
7759
b87dbeeb
SL
7760static bool do_sve2_zpzz_fp(DisasContext *s, arg_rprr_esz *a,
7761 gen_helper_gvec_4_ptr *fn)
7762{
7763 if (!dc_isar_feature(aa64_sve2, s)) {
7764 return false;
7765 }
7766 return do_zpzz_fp(s, a, fn);
7767}
7768
7769#define DO_SVE2_ZPZZ_FP(NAME, name) \
7770static bool trans_##NAME(DisasContext *s, arg_rprr_esz *a) \
7771{ \
7772 static gen_helper_gvec_4_ptr * const fns[4] = { \
7773 NULL, gen_helper_sve2_##name##_zpzz_h, \
7774 gen_helper_sve2_##name##_zpzz_s, gen_helper_sve2_##name##_zpzz_d \
7775 }; \
7776 return do_sve2_zpzz_fp(s, a, fns[a->esz]); \
7777}
7778
7779DO_SVE2_ZPZZ_FP(FADDP, faddp)
7780DO_SVE2_ZPZZ_FP(FMAXNMP, fmaxnmp)
7781DO_SVE2_ZPZZ_FP(FMINNMP, fminnmp)
7782DO_SVE2_ZPZZ_FP(FMAXP, fmaxp)
7783DO_SVE2_ZPZZ_FP(FMINP, fminp)
bfc9307e
RH
7784
7785/*
7786 * SVE Integer Multiply-Add (unpredicated)
7787 */
7788
4f26756b
SL
7789static bool trans_FMMLA(DisasContext *s, arg_rrrr_esz *a)
7790{
7791 gen_helper_gvec_4_ptr *fn;
7792
7793 switch (a->esz) {
7794 case MO_32:
7795 if (!dc_isar_feature(aa64_sve_f32mm, s)) {
7796 return false;
7797 }
7798 fn = gen_helper_fmmla_s;
7799 break;
7800 case MO_64:
7801 if (!dc_isar_feature(aa64_sve_f64mm, s)) {
7802 return false;
7803 }
7804 fn = gen_helper_fmmla_d;
7805 break;
7806 default:
7807 return false;
7808 }
7809
7810 if (sve_access_check(s)) {
7811 unsigned vsz = vec_full_reg_size(s);
7812 TCGv_ptr status = fpstatus_ptr(FPST_FPCR);
7813 tcg_gen_gvec_4_ptr(vec_full_reg_offset(s, a->rd),
7814 vec_full_reg_offset(s, a->rn),
7815 vec_full_reg_offset(s, a->rm),
7816 vec_full_reg_offset(s, a->ra),
7817 status, vsz, vsz, 0, fn);
7818 tcg_temp_free_ptr(status);
7819 }
7820 return true;
7821}
7822
bfc9307e
RH
7823static bool do_sqdmlal_zzzw(DisasContext *s, arg_rrrr_esz *a,
7824 bool sel1, bool sel2)
7825{
7826 static gen_helper_gvec_4 * const fns[] = {
7827 NULL, gen_helper_sve2_sqdmlal_zzzw_h,
7828 gen_helper_sve2_sqdmlal_zzzw_s, gen_helper_sve2_sqdmlal_zzzw_d,
7829 };
7830 return do_sve2_zzzz_ool(s, a, fns[a->esz], (sel2 << 1) | sel1);
7831}
7832
7833static bool do_sqdmlsl_zzzw(DisasContext *s, arg_rrrr_esz *a,
7834 bool sel1, bool sel2)
7835{
7836 static gen_helper_gvec_4 * const fns[] = {
7837 NULL, gen_helper_sve2_sqdmlsl_zzzw_h,
7838 gen_helper_sve2_sqdmlsl_zzzw_s, gen_helper_sve2_sqdmlsl_zzzw_d,
7839 };
7840 return do_sve2_zzzz_ool(s, a, fns[a->esz], (sel2 << 1) | sel1);
7841}
7842
7843static bool trans_SQDMLALB_zzzw(DisasContext *s, arg_rrrr_esz *a)
7844{
7845 return do_sqdmlal_zzzw(s, a, false, false);
7846}
7847
7848static bool trans_SQDMLALT_zzzw(DisasContext *s, arg_rrrr_esz *a)
7849{
7850 return do_sqdmlal_zzzw(s, a, true, true);
7851}
7852
7853static bool trans_SQDMLALBT(DisasContext *s, arg_rrrr_esz *a)
7854{
7855 return do_sqdmlal_zzzw(s, a, false, true);
7856}
7857
7858static bool trans_SQDMLSLB_zzzw(DisasContext *s, arg_rrrr_esz *a)
7859{
7860 return do_sqdmlsl_zzzw(s, a, false, false);
7861}
7862
7863static bool trans_SQDMLSLT_zzzw(DisasContext *s, arg_rrrr_esz *a)
7864{
7865 return do_sqdmlsl_zzzw(s, a, true, true);
7866}
7867
7868static bool trans_SQDMLSLBT(DisasContext *s, arg_rrrr_esz *a)
7869{
7870 return do_sqdmlsl_zzzw(s, a, false, true);
7871}
ab3ddf31
RH
7872
7873static bool trans_SQRDMLAH_zzzz(DisasContext *s, arg_rrrr_esz *a)
7874{
7875 static gen_helper_gvec_4 * const fns[] = {
7876 gen_helper_sve2_sqrdmlah_b, gen_helper_sve2_sqrdmlah_h,
7877 gen_helper_sve2_sqrdmlah_s, gen_helper_sve2_sqrdmlah_d,
7878 };
7879 return do_sve2_zzzz_ool(s, a, fns[a->esz], 0);
7880}
7881
7882static bool trans_SQRDMLSH_zzzz(DisasContext *s, arg_rrrr_esz *a)
7883{
7884 static gen_helper_gvec_4 * const fns[] = {
7885 gen_helper_sve2_sqrdmlsh_b, gen_helper_sve2_sqrdmlsh_h,
7886 gen_helper_sve2_sqrdmlsh_s, gen_helper_sve2_sqrdmlsh_d,
7887 };
7888 return do_sve2_zzzz_ool(s, a, fns[a->esz], 0);
7889}
45a32e80
RH
7890
7891static bool do_smlal_zzzw(DisasContext *s, arg_rrrr_esz *a, bool sel)
7892{
7893 static gen_helper_gvec_4 * const fns[] = {
7894 NULL, gen_helper_sve2_smlal_zzzw_h,
7895 gen_helper_sve2_smlal_zzzw_s, gen_helper_sve2_smlal_zzzw_d,
7896 };
7897 return do_sve2_zzzz_ool(s, a, fns[a->esz], sel);
7898}
7899
7900static bool trans_SMLALB_zzzw(DisasContext *s, arg_rrrr_esz *a)
7901{
7902 return do_smlal_zzzw(s, a, false);
7903}
7904
7905static bool trans_SMLALT_zzzw(DisasContext *s, arg_rrrr_esz *a)
7906{
7907 return do_smlal_zzzw(s, a, true);
7908}
7909
7910static bool do_umlal_zzzw(DisasContext *s, arg_rrrr_esz *a, bool sel)
7911{
7912 static gen_helper_gvec_4 * const fns[] = {
7913 NULL, gen_helper_sve2_umlal_zzzw_h,
7914 gen_helper_sve2_umlal_zzzw_s, gen_helper_sve2_umlal_zzzw_d,
7915 };
7916 return do_sve2_zzzz_ool(s, a, fns[a->esz], sel);
7917}
7918
7919static bool trans_UMLALB_zzzw(DisasContext *s, arg_rrrr_esz *a)
7920{
7921 return do_umlal_zzzw(s, a, false);
7922}
7923
7924static bool trans_UMLALT_zzzw(DisasContext *s, arg_rrrr_esz *a)
7925{
7926 return do_umlal_zzzw(s, a, true);
7927}
7928
7929static bool do_smlsl_zzzw(DisasContext *s, arg_rrrr_esz *a, bool sel)
7930{
7931 static gen_helper_gvec_4 * const fns[] = {
7932 NULL, gen_helper_sve2_smlsl_zzzw_h,
7933 gen_helper_sve2_smlsl_zzzw_s, gen_helper_sve2_smlsl_zzzw_d,
7934 };
7935 return do_sve2_zzzz_ool(s, a, fns[a->esz], sel);
7936}
7937
7938static bool trans_SMLSLB_zzzw(DisasContext *s, arg_rrrr_esz *a)
7939{
7940 return do_smlsl_zzzw(s, a, false);
7941}
7942
7943static bool trans_SMLSLT_zzzw(DisasContext *s, arg_rrrr_esz *a)
7944{
7945 return do_smlsl_zzzw(s, a, true);
7946}
7947
7948static bool do_umlsl_zzzw(DisasContext *s, arg_rrrr_esz *a, bool sel)
7949{
7950 static gen_helper_gvec_4 * const fns[] = {
7951 NULL, gen_helper_sve2_umlsl_zzzw_h,
7952 gen_helper_sve2_umlsl_zzzw_s, gen_helper_sve2_umlsl_zzzw_d,
7953 };
7954 return do_sve2_zzzz_ool(s, a, fns[a->esz], sel);
7955}
7956
7957static bool trans_UMLSLB_zzzw(DisasContext *s, arg_rrrr_esz *a)
7958{
7959 return do_umlsl_zzzw(s, a, false);
7960}
7961
7962static bool trans_UMLSLT_zzzw(DisasContext *s, arg_rrrr_esz *a)
7963{
7964 return do_umlsl_zzzw(s, a, true);
7965}
d782d3ca
RH
7966
7967static bool trans_CMLA_zzzz(DisasContext *s, arg_CMLA_zzzz *a)
7968{
7969 static gen_helper_gvec_4 * const fns[] = {
7970 gen_helper_sve2_cmla_zzzz_b, gen_helper_sve2_cmla_zzzz_h,
7971 gen_helper_sve2_cmla_zzzz_s, gen_helper_sve2_cmla_zzzz_d,
7972 };
7973
7974 if (!dc_isar_feature(aa64_sve2, s)) {
7975 return false;
7976 }
7977 if (sve_access_check(s)) {
7978 gen_gvec_ool_zzzz(s, fns[a->esz], a->rd, a->rn, a->rm, a->ra, a->rot);
7979 }
7980 return true;
7981}
7982
7983static bool trans_SQRDCMLAH_zzzz(DisasContext *s, arg_SQRDCMLAH_zzzz *a)
7984{
7985 static gen_helper_gvec_4 * const fns[] = {
7986 gen_helper_sve2_sqrdcmlah_zzzz_b, gen_helper_sve2_sqrdcmlah_zzzz_h,
7987 gen_helper_sve2_sqrdcmlah_zzzz_s, gen_helper_sve2_sqrdcmlah_zzzz_d,
7988 };
7989
7990 if (!dc_isar_feature(aa64_sve2, s)) {
7991 return false;
7992 }
7993 if (sve_access_check(s)) {
7994 gen_gvec_ool_zzzz(s, fns[a->esz], a->rd, a->rn, a->rm, a->ra, a->rot);
7995 }
7996 return true;
7997}