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