1 //===-- AArch64BaseInfo.h - Top level definitions for AArch64 ---*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file contains small standalone helper functions and enum definitions for
10 // the AArch64 target useful for the compiler back-end and the MC libraries.
11 // As such, it deliberately does not include references to LLVM core
12 // code gen types, passes, etc..
13 //
14 //===----------------------------------------------------------------------===//
15 
16 #ifndef LLVM_LIB_TARGET_AARCH64_UTILS_AARCH64BASEINFO_H
17 #define LLVM_LIB_TARGET_AARCH64_UTILS_AARCH64BASEINFO_H
18 
19 // FIXME: Is it easiest to fix this layering violation by moving the .inc
20 // #includes from AArch64MCTargetDesc.h to here?
21 #include "MCTargetDesc/AArch64MCTargetDesc.h" // For AArch64::X0 and friends.
22 #include "llvm/ADT/BitmaskEnum.h"
23 #include "llvm/ADT/STLExtras.h"
24 #include "llvm/ADT/StringSwitch.h"
25 #include "llvm/Support/ErrorHandling.h"
26 #include "llvm/TargetParser/SubtargetFeature.h"
27 
28 namespace llvm {
29 
30 inline static unsigned getWRegFromXReg(unsigned Reg) {
31   switch (Reg) {
32   case AArch64::X0: return AArch64::W0;
33   case AArch64::X1: return AArch64::W1;
34   case AArch64::X2: return AArch64::W2;
35   case AArch64::X3: return AArch64::W3;
36   case AArch64::X4: return AArch64::W4;
37   case AArch64::X5: return AArch64::W5;
38   case AArch64::X6: return AArch64::W6;
39   case AArch64::X7: return AArch64::W7;
40   case AArch64::X8: return AArch64::W8;
41   case AArch64::X9: return AArch64::W9;
42   case AArch64::X10: return AArch64::W10;
43   case AArch64::X11: return AArch64::W11;
44   case AArch64::X12: return AArch64::W12;
45   case AArch64::X13: return AArch64::W13;
46   case AArch64::X14: return AArch64::W14;
47   case AArch64::X15: return AArch64::W15;
48   case AArch64::X16: return AArch64::W16;
49   case AArch64::X17: return AArch64::W17;
50   case AArch64::X18: return AArch64::W18;
51   case AArch64::X19: return AArch64::W19;
52   case AArch64::X20: return AArch64::W20;
53   case AArch64::X21: return AArch64::W21;
54   case AArch64::X22: return AArch64::W22;
55   case AArch64::X23: return AArch64::W23;
56   case AArch64::X24: return AArch64::W24;
57   case AArch64::X25: return AArch64::W25;
58   case AArch64::X26: return AArch64::W26;
59   case AArch64::X27: return AArch64::W27;
60   case AArch64::X28: return AArch64::W28;
61   case AArch64::FP: return AArch64::W29;
62   case AArch64::LR: return AArch64::W30;
63   case AArch64::SP: return AArch64::WSP;
64   case AArch64::XZR: return AArch64::WZR;
65   }
66   // For anything else, return it unchanged.
67   return Reg;
68 }
69 
70 inline static unsigned getXRegFromWReg(unsigned Reg) {
71   switch (Reg) {
72   case AArch64::W0: return AArch64::X0;
73   case AArch64::W1: return AArch64::X1;
74   case AArch64::W2: return AArch64::X2;
75   case AArch64::W3: return AArch64::X3;
76   case AArch64::W4: return AArch64::X4;
77   case AArch64::W5: return AArch64::X5;
78   case AArch64::W6: return AArch64::X6;
79   case AArch64::W7: return AArch64::X7;
80   case AArch64::W8: return AArch64::X8;
81   case AArch64::W9: return AArch64::X9;
82   case AArch64::W10: return AArch64::X10;
83   case AArch64::W11: return AArch64::X11;
84   case AArch64::W12: return AArch64::X12;
85   case AArch64::W13: return AArch64::X13;
86   case AArch64::W14: return AArch64::X14;
87   case AArch64::W15: return AArch64::X15;
88   case AArch64::W16: return AArch64::X16;
89   case AArch64::W17: return AArch64::X17;
90   case AArch64::W18: return AArch64::X18;
91   case AArch64::W19: return AArch64::X19;
92   case AArch64::W20: return AArch64::X20;
93   case AArch64::W21: return AArch64::X21;
94   case AArch64::W22: return AArch64::X22;
95   case AArch64::W23: return AArch64::X23;
96   case AArch64::W24: return AArch64::X24;
97   case AArch64::W25: return AArch64::X25;
98   case AArch64::W26: return AArch64::X26;
99   case AArch64::W27: return AArch64::X27;
100   case AArch64::W28: return AArch64::X28;
101   case AArch64::W29: return AArch64::FP;
102   case AArch64::W30: return AArch64::LR;
103   case AArch64::WSP: return AArch64::SP;
104   case AArch64::WZR: return AArch64::XZR;
105   }
106   // For anything else, return it unchanged.
107   return Reg;
108 }
109 
110 inline static unsigned getXRegFromXRegTuple(unsigned RegTuple) {
111   switch (RegTuple) {
112   case AArch64::X0_X1_X2_X3_X4_X5_X6_X7: return AArch64::X0;
113   case AArch64::X2_X3_X4_X5_X6_X7_X8_X9: return AArch64::X2;
114   case AArch64::X4_X5_X6_X7_X8_X9_X10_X11: return AArch64::X4;
115   case AArch64::X6_X7_X8_X9_X10_X11_X12_X13: return AArch64::X6;
116   case AArch64::X8_X9_X10_X11_X12_X13_X14_X15: return AArch64::X8;
117   case AArch64::X10_X11_X12_X13_X14_X15_X16_X17: return AArch64::X10;
118   case AArch64::X12_X13_X14_X15_X16_X17_X18_X19: return AArch64::X12;
119   case AArch64::X14_X15_X16_X17_X18_X19_X20_X21: return AArch64::X14;
120   case AArch64::X16_X17_X18_X19_X20_X21_X22_X23: return AArch64::X16;
121   case AArch64::X18_X19_X20_X21_X22_X23_X24_X25: return AArch64::X18;
122   case AArch64::X20_X21_X22_X23_X24_X25_X26_X27: return AArch64::X20;
123   case AArch64::X22_X23_X24_X25_X26_X27_X28_FP: return AArch64::X22;
124   }
125   // For anything else, return it unchanged.
126   return RegTuple;
127 }
128 
129 static inline unsigned getBRegFromDReg(unsigned Reg) {
130   switch (Reg) {
131   case AArch64::D0:  return AArch64::B0;
132   case AArch64::D1:  return AArch64::B1;
133   case AArch64::D2:  return AArch64::B2;
134   case AArch64::D3:  return AArch64::B3;
135   case AArch64::D4:  return AArch64::B4;
136   case AArch64::D5:  return AArch64::B5;
137   case AArch64::D6:  return AArch64::B6;
138   case AArch64::D7:  return AArch64::B7;
139   case AArch64::D8:  return AArch64::B8;
140   case AArch64::D9:  return AArch64::B9;
141   case AArch64::D10: return AArch64::B10;
142   case AArch64::D11: return AArch64::B11;
143   case AArch64::D12: return AArch64::B12;
144   case AArch64::D13: return AArch64::B13;
145   case AArch64::D14: return AArch64::B14;
146   case AArch64::D15: return AArch64::B15;
147   case AArch64::D16: return AArch64::B16;
148   case AArch64::D17: return AArch64::B17;
149   case AArch64::D18: return AArch64::B18;
150   case AArch64::D19: return AArch64::B19;
151   case AArch64::D20: return AArch64::B20;
152   case AArch64::D21: return AArch64::B21;
153   case AArch64::D22: return AArch64::B22;
154   case AArch64::D23: return AArch64::B23;
155   case AArch64::D24: return AArch64::B24;
156   case AArch64::D25: return AArch64::B25;
157   case AArch64::D26: return AArch64::B26;
158   case AArch64::D27: return AArch64::B27;
159   case AArch64::D28: return AArch64::B28;
160   case AArch64::D29: return AArch64::B29;
161   case AArch64::D30: return AArch64::B30;
162   case AArch64::D31: return AArch64::B31;
163   }
164   // For anything else, return it unchanged.
165   return Reg;
166 }
167 
168 
169 static inline unsigned getDRegFromBReg(unsigned Reg) {
170   switch (Reg) {
171   case AArch64::B0:  return AArch64::D0;
172   case AArch64::B1:  return AArch64::D1;
173   case AArch64::B2:  return AArch64::D2;
174   case AArch64::B3:  return AArch64::D3;
175   case AArch64::B4:  return AArch64::D4;
176   case AArch64::B5:  return AArch64::D5;
177   case AArch64::B6:  return AArch64::D6;
178   case AArch64::B7:  return AArch64::D7;
179   case AArch64::B8:  return AArch64::D8;
180   case AArch64::B9:  return AArch64::D9;
181   case AArch64::B10: return AArch64::D10;
182   case AArch64::B11: return AArch64::D11;
183   case AArch64::B12: return AArch64::D12;
184   case AArch64::B13: return AArch64::D13;
185   case AArch64::B14: return AArch64::D14;
186   case AArch64::B15: return AArch64::D15;
187   case AArch64::B16: return AArch64::D16;
188   case AArch64::B17: return AArch64::D17;
189   case AArch64::B18: return AArch64::D18;
190   case AArch64::B19: return AArch64::D19;
191   case AArch64::B20: return AArch64::D20;
192   case AArch64::B21: return AArch64::D21;
193   case AArch64::B22: return AArch64::D22;
194   case AArch64::B23: return AArch64::D23;
195   case AArch64::B24: return AArch64::D24;
196   case AArch64::B25: return AArch64::D25;
197   case AArch64::B26: return AArch64::D26;
198   case AArch64::B27: return AArch64::D27;
199   case AArch64::B28: return AArch64::D28;
200   case AArch64::B29: return AArch64::D29;
201   case AArch64::B30: return AArch64::D30;
202   case AArch64::B31: return AArch64::D31;
203   }
204   // For anything else, return it unchanged.
205   return Reg;
206 }
207 
208 static inline bool atomicBarrierDroppedOnZero(unsigned Opcode) {
209   switch (Opcode) {
210   case AArch64::LDADDAB:   case AArch64::LDADDAH:
211   case AArch64::LDADDAW:   case AArch64::LDADDAX:
212   case AArch64::LDADDALB:  case AArch64::LDADDALH:
213   case AArch64::LDADDALW:  case AArch64::LDADDALX:
214   case AArch64::LDCLRAB:   case AArch64::LDCLRAH:
215   case AArch64::LDCLRAW:   case AArch64::LDCLRAX:
216   case AArch64::LDCLRALB:  case AArch64::LDCLRALH:
217   case AArch64::LDCLRALW:  case AArch64::LDCLRALX:
218   case AArch64::LDEORAB:   case AArch64::LDEORAH:
219   case AArch64::LDEORAW:   case AArch64::LDEORAX:
220   case AArch64::LDEORALB:  case AArch64::LDEORALH:
221   case AArch64::LDEORALW:  case AArch64::LDEORALX:
222   case AArch64::LDSETAB:   case AArch64::LDSETAH:
223   case AArch64::LDSETAW:   case AArch64::LDSETAX:
224   case AArch64::LDSETALB:  case AArch64::LDSETALH:
225   case AArch64::LDSETALW:  case AArch64::LDSETALX:
226   case AArch64::LDSMAXAB:  case AArch64::LDSMAXAH:
227   case AArch64::LDSMAXAW:  case AArch64::LDSMAXAX:
228   case AArch64::LDSMAXALB: case AArch64::LDSMAXALH:
229   case AArch64::LDSMAXALW: case AArch64::LDSMAXALX:
230   case AArch64::LDSMINAB:  case AArch64::LDSMINAH:
231   case AArch64::LDSMINAW:  case AArch64::LDSMINAX:
232   case AArch64::LDSMINALB: case AArch64::LDSMINALH:
233   case AArch64::LDSMINALW: case AArch64::LDSMINALX:
234   case AArch64::LDUMAXAB:  case AArch64::LDUMAXAH:
235   case AArch64::LDUMAXAW:  case AArch64::LDUMAXAX:
236   case AArch64::LDUMAXALB: case AArch64::LDUMAXALH:
237   case AArch64::LDUMAXALW: case AArch64::LDUMAXALX:
238   case AArch64::LDUMINAB:  case AArch64::LDUMINAH:
239   case AArch64::LDUMINAW:  case AArch64::LDUMINAX:
240   case AArch64::LDUMINALB: case AArch64::LDUMINALH:
241   case AArch64::LDUMINALW: case AArch64::LDUMINALX:
242   case AArch64::SWPAB:     case AArch64::SWPAH:
243   case AArch64::SWPAW:     case AArch64::SWPAX:
244   case AArch64::SWPALB:    case AArch64::SWPALH:
245   case AArch64::SWPALW:    case AArch64::SWPALX:
246     return true;
247   }
248   return false;
249 }
250 
251 namespace AArch64CC {
252 
253 // The CondCodes constants map directly to the 4-bit encoding of the condition
254 // field for predicated instructions.
255 enum CondCode {  // Meaning (integer)          Meaning (floating-point)
256   EQ = 0x0,      // Equal                      Equal
257   NE = 0x1,      // Not equal                  Not equal, or unordered
258   HS = 0x2,      // Unsigned higher or same    >, ==, or unordered
259   LO = 0x3,      // Unsigned lower             Less than
260   MI = 0x4,      // Minus, negative            Less than
261   PL = 0x5,      // Plus, positive or zero     >, ==, or unordered
262   VS = 0x6,      // Overflow                   Unordered
263   VC = 0x7,      // No overflow                Not unordered
264   HI = 0x8,      // Unsigned higher            Greater than, or unordered
265   LS = 0x9,      // Unsigned lower or same     Less than or equal
266   GE = 0xa,      // Greater than or equal      Greater than or equal
267   LT = 0xb,      // Less than                  Less than, or unordered
268   GT = 0xc,      // Greater than               Greater than
269   LE = 0xd,      // Less than or equal         <, ==, or unordered
270   AL = 0xe,      // Always (unconditional)     Always (unconditional)
271   NV = 0xf,      // Always (unconditional)     Always (unconditional)
272   // Note the NV exists purely to disassemble 0b1111. Execution is "always".
273   Invalid,
274 
275   // Common aliases used for SVE.
276   ANY_ACTIVE   = NE, // (!Z)
277   FIRST_ACTIVE = MI, // ( N)
278   LAST_ACTIVE  = LO, // (!C)
279   NONE_ACTIVE  = EQ  // ( Z)
280 };
281 
282 inline static const char *getCondCodeName(CondCode Code) {
283   switch (Code) {
284   default: llvm_unreachable("Unknown condition code");
285   case EQ:  return "eq";
286   case NE:  return "ne";
287   case HS:  return "hs";
288   case LO:  return "lo";
289   case MI:  return "mi";
290   case PL:  return "pl";
291   case VS:  return "vs";
292   case VC:  return "vc";
293   case HI:  return "hi";
294   case LS:  return "ls";
295   case GE:  return "ge";
296   case LT:  return "lt";
297   case GT:  return "gt";
298   case LE:  return "le";
299   case AL:  return "al";
300   case NV:  return "nv";
301   }
302 }
303 
304 inline static CondCode getInvertedCondCode(CondCode Code) {
305   // To reverse a condition it's necessary to only invert the low bit:
306 
307   return static_cast<CondCode>(static_cast<unsigned>(Code) ^ 0x1);
308 }
309 
310 /// Given a condition code, return NZCV flags that would satisfy that condition.
311 /// The flag bits are in the format expected by the ccmp instructions.
312 /// Note that many different flag settings can satisfy a given condition code,
313 /// this function just returns one of them.
314 inline static unsigned getNZCVToSatisfyCondCode(CondCode Code) {
315   // NZCV flags encoded as expected by ccmp instructions, ARMv8 ISA 5.5.7.
316   enum { N = 8, Z = 4, C = 2, V = 1 };
317   switch (Code) {
318   default: llvm_unreachable("Unknown condition code");
319   case EQ: return Z; // Z == 1
320   case NE: return 0; // Z == 0
321   case HS: return C; // C == 1
322   case LO: return 0; // C == 0
323   case MI: return N; // N == 1
324   case PL: return 0; // N == 0
325   case VS: return V; // V == 1
326   case VC: return 0; // V == 0
327   case HI: return C; // C == 1 && Z == 0
328   case LS: return 0; // C == 0 || Z == 1
329   case GE: return 0; // N == V
330   case LT: return N; // N != V
331   case GT: return 0; // Z == 0 && N == V
332   case LE: return Z; // Z == 1 || N != V
333   }
334 }
335 
336 } // end namespace AArch64CC
337 
338 struct SysAlias {
339   const char *Name;
340   uint16_t Encoding;
341   FeatureBitset FeaturesRequired;
342 
343   constexpr SysAlias(const char *N, uint16_t E) : Name(N), Encoding(E) {}
344   constexpr SysAlias(const char *N, uint16_t E, FeatureBitset F)
345       : Name(N), Encoding(E), FeaturesRequired(F) {}
346 
347   bool haveFeatures(FeatureBitset ActiveFeatures) const {
348     return ActiveFeatures[llvm::AArch64::FeatureAll] ||
349            (FeaturesRequired & ActiveFeatures) == FeaturesRequired;
350   }
351 
352   FeatureBitset getRequiredFeatures() const { return FeaturesRequired; }
353 };
354 
355 struct SysAliasReg : SysAlias {
356   bool NeedsReg;
357   constexpr SysAliasReg(const char *N, uint16_t E, bool R)
358       : SysAlias(N, E), NeedsReg(R) {}
359   constexpr SysAliasReg(const char *N, uint16_t E, bool R, FeatureBitset F)
360       : SysAlias(N, E, F), NeedsReg(R) {}
361 };
362 
363 struct SysAliasImm : SysAlias {
364   uint16_t ImmValue;
365   constexpr SysAliasImm(const char *N, uint16_t E, uint16_t I)
366       : SysAlias(N, E), ImmValue(I) {}
367   constexpr SysAliasImm(const char *N, uint16_t E, uint16_t I, FeatureBitset F)
368       : SysAlias(N, E, F), ImmValue(I) {}
369 };
370 
371 namespace AArch64SVCR {
372   struct SVCR : SysAlias{
373     using SysAlias::SysAlias;
374   };
375   #define GET_SVCR_DECL
376   #include "AArch64GenSystemOperands.inc"
377 }
378 
379 namespace AArch64AT{
380   struct AT : SysAlias {
381     using SysAlias::SysAlias;
382   };
383   #define GET_AT_DECL
384   #include "AArch64GenSystemOperands.inc"
385 }
386 
387 namespace AArch64DB {
388   struct DB : SysAlias {
389     using SysAlias::SysAlias;
390   };
391   #define GET_DB_DECL
392   #include "AArch64GenSystemOperands.inc"
393 }
394 
395 namespace AArch64DBnXS {
396   struct DBnXS : SysAliasImm {
397     using SysAliasImm::SysAliasImm;
398   };
399   #define GET_DBNXS_DECL
400   #include "AArch64GenSystemOperands.inc"
401 }
402 
403 namespace  AArch64DC {
404   struct DC : SysAlias {
405     using SysAlias::SysAlias;
406   };
407   #define GET_DC_DECL
408   #include "AArch64GenSystemOperands.inc"
409 }
410 
411 namespace  AArch64IC {
412   struct IC : SysAliasReg {
413     using SysAliasReg::SysAliasReg;
414   };
415   #define GET_IC_DECL
416   #include "AArch64GenSystemOperands.inc"
417 }
418 
419 namespace  AArch64ISB {
420   struct ISB : SysAlias {
421     using SysAlias::SysAlias;
422   };
423   #define GET_ISB_DECL
424   #include "AArch64GenSystemOperands.inc"
425 }
426 
427 namespace  AArch64TSB {
428   struct TSB : SysAlias {
429     using SysAlias::SysAlias;
430   };
431   #define GET_TSB_DECL
432   #include "AArch64GenSystemOperands.inc"
433 }
434 
435 namespace AArch64PRFM {
436   struct PRFM : SysAlias {
437     using SysAlias::SysAlias;
438   };
439   #define GET_PRFM_DECL
440   #include "AArch64GenSystemOperands.inc"
441 }
442 
443 namespace AArch64SVEPRFM {
444   struct SVEPRFM : SysAlias {
445     using SysAlias::SysAlias;
446   };
447 #define GET_SVEPRFM_DECL
448 #include "AArch64GenSystemOperands.inc"
449 }
450 
451 namespace AArch64RPRFM {
452 struct RPRFM : SysAlias {
453   using SysAlias::SysAlias;
454 };
455 #define GET_RPRFM_DECL
456 #include "AArch64GenSystemOperands.inc"
457 } // namespace AArch64RPRFM
458 
459 namespace AArch64SVEPredPattern {
460   struct SVEPREDPAT {
461     const char *Name;
462     uint16_t Encoding;
463   };
464 #define GET_SVEPREDPAT_DECL
465 #include "AArch64GenSystemOperands.inc"
466 }
467 
468 namespace AArch64SVEVecLenSpecifier {
469   struct SVEVECLENSPECIFIER {
470     const char *Name;
471     uint16_t Encoding;
472   };
473 #define GET_SVEVECLENSPECIFIER_DECL
474 #include "AArch64GenSystemOperands.inc"
475 } // namespace AArch64SVEVecLenSpecifier
476 
477 /// Return the number of active elements for VL1 to VL256 predicate pattern,
478 /// zero for all other patterns.
479 inline unsigned getNumElementsFromSVEPredPattern(unsigned Pattern) {
480   switch (Pattern) {
481   default:
482     return 0;
483   case AArch64SVEPredPattern::vl1:
484   case AArch64SVEPredPattern::vl2:
485   case AArch64SVEPredPattern::vl3:
486   case AArch64SVEPredPattern::vl4:
487   case AArch64SVEPredPattern::vl5:
488   case AArch64SVEPredPattern::vl6:
489   case AArch64SVEPredPattern::vl7:
490   case AArch64SVEPredPattern::vl8:
491     return Pattern;
492   case AArch64SVEPredPattern::vl16:
493     return 16;
494   case AArch64SVEPredPattern::vl32:
495     return 32;
496   case AArch64SVEPredPattern::vl64:
497     return 64;
498   case AArch64SVEPredPattern::vl128:
499     return 128;
500   case AArch64SVEPredPattern::vl256:
501     return 256;
502   }
503 }
504 
505 /// Return specific VL predicate pattern based on the number of elements.
506 inline std::optional<unsigned>
507 getSVEPredPatternFromNumElements(unsigned MinNumElts) {
508   switch (MinNumElts) {
509   default:
510     return std::nullopt;
511   case 1:
512   case 2:
513   case 3:
514   case 4:
515   case 5:
516   case 6:
517   case 7:
518   case 8:
519     return MinNumElts;
520   case 16:
521     return AArch64SVEPredPattern::vl16;
522   case 32:
523     return AArch64SVEPredPattern::vl32;
524   case 64:
525     return AArch64SVEPredPattern::vl64;
526   case 128:
527     return AArch64SVEPredPattern::vl128;
528   case 256:
529     return AArch64SVEPredPattern::vl256;
530   }
531 }
532 
533 /// An enum to describe what types of loops we should attempt to tail-fold:
534 ///   Disabled:    None
535 ///   Reductions:  Loops containing reductions
536 ///   Recurrences: Loops with first-order recurrences, i.e. that would
537 ///                  require a SVE splice instruction
538 ///   Reverse:     Reverse loops
539 ///   Simple:      Loops that are not reversed and don't contain reductions
540 ///                  or first-order recurrences.
541 ///   All:         All
542 enum class TailFoldingOpts : uint8_t {
543   Disabled = 0x00,
544   Simple = 0x01,
545   Reductions = 0x02,
546   Recurrences = 0x04,
547   Reverse = 0x08,
548   All = Reductions | Recurrences | Simple | Reverse
549 };
550 
551 LLVM_DECLARE_ENUM_AS_BITMASK(TailFoldingOpts,
552                              /* LargestValue */ (long)TailFoldingOpts::Reverse);
553 
554 namespace AArch64ExactFPImm {
555   struct ExactFPImm {
556     const char *Name;
557     int Enum;
558     const char *Repr;
559   };
560 #define GET_EXACTFPIMM_DECL
561 #include "AArch64GenSystemOperands.inc"
562 }
563 
564 namespace AArch64PState {
565   struct PStateImm0_15 : SysAlias{
566     using SysAlias::SysAlias;
567   };
568   #define GET_PSTATEIMM0_15_DECL
569   #include "AArch64GenSystemOperands.inc"
570 
571   struct PStateImm0_1 : SysAlias{
572     using SysAlias::SysAlias;
573   };
574   #define GET_PSTATEIMM0_1_DECL
575   #include "AArch64GenSystemOperands.inc"
576 }
577 
578 namespace AArch64PSBHint {
579   struct PSB : SysAlias {
580     using SysAlias::SysAlias;
581   };
582   #define GET_PSB_DECL
583   #include "AArch64GenSystemOperands.inc"
584 }
585 
586 namespace AArch64BTIHint {
587   struct BTI : SysAlias {
588     using SysAlias::SysAlias;
589   };
590   #define GET_BTI_DECL
591   #include "AArch64GenSystemOperands.inc"
592 }
593 
594 namespace AArch64SE {
595     enum ShiftExtSpecifiers {
596         Invalid = -1,
597         LSL,
598         MSL,
599         LSR,
600         ASR,
601         ROR,
602 
603         UXTB,
604         UXTH,
605         UXTW,
606         UXTX,
607 
608         SXTB,
609         SXTH,
610         SXTW,
611         SXTX
612     };
613 }
614 
615 namespace AArch64Layout {
616     enum VectorLayout {
617         Invalid = -1,
618         VL_8B,
619         VL_4H,
620         VL_2S,
621         VL_1D,
622 
623         VL_16B,
624         VL_8H,
625         VL_4S,
626         VL_2D,
627 
628         // Bare layout for the 128-bit vector
629         // (only show ".b", ".h", ".s", ".d" without vector number)
630         VL_B,
631         VL_H,
632         VL_S,
633         VL_D
634     };
635 }
636 
637 inline static const char *
638 AArch64VectorLayoutToString(AArch64Layout::VectorLayout Layout) {
639   switch (Layout) {
640   case AArch64Layout::VL_8B:  return ".8b";
641   case AArch64Layout::VL_4H:  return ".4h";
642   case AArch64Layout::VL_2S:  return ".2s";
643   case AArch64Layout::VL_1D:  return ".1d";
644   case AArch64Layout::VL_16B:  return ".16b";
645   case AArch64Layout::VL_8H:  return ".8h";
646   case AArch64Layout::VL_4S:  return ".4s";
647   case AArch64Layout::VL_2D:  return ".2d";
648   case AArch64Layout::VL_B:  return ".b";
649   case AArch64Layout::VL_H:  return ".h";
650   case AArch64Layout::VL_S:  return ".s";
651   case AArch64Layout::VL_D:  return ".d";
652   default: llvm_unreachable("Unknown Vector Layout");
653   }
654 }
655 
656 inline static AArch64Layout::VectorLayout
657 AArch64StringToVectorLayout(StringRef LayoutStr) {
658   return StringSwitch<AArch64Layout::VectorLayout>(LayoutStr)
659              .Case(".8b", AArch64Layout::VL_8B)
660              .Case(".4h", AArch64Layout::VL_4H)
661              .Case(".2s", AArch64Layout::VL_2S)
662              .Case(".1d", AArch64Layout::VL_1D)
663              .Case(".16b", AArch64Layout::VL_16B)
664              .Case(".8h", AArch64Layout::VL_8H)
665              .Case(".4s", AArch64Layout::VL_4S)
666              .Case(".2d", AArch64Layout::VL_2D)
667              .Case(".b", AArch64Layout::VL_B)
668              .Case(".h", AArch64Layout::VL_H)
669              .Case(".s", AArch64Layout::VL_S)
670              .Case(".d", AArch64Layout::VL_D)
671              .Default(AArch64Layout::Invalid);
672 }
673 
674 namespace AArch64SysReg {
675   struct SysReg {
676     const char *Name;
677     const char *AltName;
678     unsigned Encoding;
679     bool Readable;
680     bool Writeable;
681     FeatureBitset FeaturesRequired;
682 
683     bool haveFeatures(FeatureBitset ActiveFeatures) const {
684       return ActiveFeatures[llvm::AArch64::FeatureAll] ||
685              (FeaturesRequired & ActiveFeatures) == FeaturesRequired;
686     }
687   };
688 
689   #define GET_SYSREG_DECL
690   #include "AArch64GenSystemOperands.inc"
691 
692   const SysReg *lookupSysRegByName(StringRef);
693   const SysReg *lookupSysRegByEncoding(uint16_t);
694 
695   uint32_t parseGenericRegister(StringRef Name);
696   std::string genericRegisterString(uint32_t Bits);
697 }
698 
699 namespace AArch64TLBI {
700   struct TLBI : SysAliasReg {
701     using SysAliasReg::SysAliasReg;
702   };
703   #define GET_TLBITable_DECL
704   #include "AArch64GenSystemOperands.inc"
705 }
706 
707 namespace AArch64PRCTX {
708   struct PRCTX : SysAliasReg {
709     using SysAliasReg::SysAliasReg;
710   };
711   #define GET_PRCTX_DECL
712   #include "AArch64GenSystemOperands.inc"
713 }
714 
715 namespace AArch64II {
716   /// Target Operand Flag enum.
717   enum TOF {
718     //===------------------------------------------------------------------===//
719     // AArch64 Specific MachineOperand flags.
720 
721     MO_NO_FLAG,
722 
723     MO_FRAGMENT = 0x7,
724 
725     /// MO_PAGE - A symbol operand with this flag represents the pc-relative
726     /// offset of the 4K page containing the symbol.  This is used with the
727     /// ADRP instruction.
728     MO_PAGE = 1,
729 
730     /// MO_PAGEOFF - A symbol operand with this flag represents the offset of
731     /// that symbol within a 4K page.  This offset is added to the page address
732     /// to produce the complete address.
733     MO_PAGEOFF = 2,
734 
735     /// MO_G3 - A symbol operand with this flag (granule 3) represents the high
736     /// 16-bits of a 64-bit address, used in a MOVZ or MOVK instruction
737     MO_G3 = 3,
738 
739     /// MO_G2 - A symbol operand with this flag (granule 2) represents the bits
740     /// 32-47 of a 64-bit address, used in a MOVZ or MOVK instruction
741     MO_G2 = 4,
742 
743     /// MO_G1 - A symbol operand with this flag (granule 1) represents the bits
744     /// 16-31 of a 64-bit address, used in a MOVZ or MOVK instruction
745     MO_G1 = 5,
746 
747     /// MO_G0 - A symbol operand with this flag (granule 0) represents the bits
748     /// 0-15 of a 64-bit address, used in a MOVZ or MOVK instruction
749     MO_G0 = 6,
750 
751     /// MO_HI12 - This flag indicates that a symbol operand represents the bits
752     /// 13-24 of a 64-bit address, used in a arithmetic immediate-shifted-left-
753     /// by-12-bits instruction.
754     MO_HI12 = 7,
755 
756     /// MO_COFFSTUB - On a symbol operand "FOO", this indicates that the
757     /// reference is actually to the ".refptr.FOO" symbol.  This is used for
758     /// stub symbols on windows.
759     MO_COFFSTUB = 0x8,
760 
761     /// MO_GOT - This flag indicates that a symbol operand represents the
762     /// address of the GOT entry for the symbol, rather than the address of
763     /// the symbol itself.
764     MO_GOT = 0x10,
765 
766     /// MO_NC - Indicates whether the linker is expected to check the symbol
767     /// reference for overflow. For example in an ADRP/ADD pair of relocations
768     /// the ADRP usually does check, but not the ADD.
769     MO_NC = 0x20,
770 
771     /// MO_TLS - Indicates that the operand being accessed is some kind of
772     /// thread-local symbol. On Darwin, only one type of thread-local access
773     /// exists (pre linker-relaxation), but on ELF the TLSModel used for the
774     /// referee will affect interpretation.
775     MO_TLS = 0x40,
776 
777     /// MO_DLLIMPORT - On a symbol operand, this represents that the reference
778     /// to the symbol is for an import stub.  This is used for DLL import
779     /// storage class indication on Windows.
780     MO_DLLIMPORT = 0x80,
781 
782     /// MO_S - Indicates that the bits of the symbol operand represented by
783     /// MO_G0 etc are signed.
784     MO_S = 0x100,
785 
786     /// MO_PREL - Indicates that the bits of the symbol operand represented by
787     /// MO_G0 etc are PC relative.
788     MO_PREL = 0x200,
789 
790     /// MO_TAGGED - With MO_PAGE, indicates that the page includes a memory tag
791     /// in bits 56-63.
792     /// On a FrameIndex operand, indicates that the underlying memory is tagged
793     /// with an unknown tag value (MTE); this needs to be lowered either to an
794     /// SP-relative load or store instruction (which do not check tags), or to
795     /// an LDG instruction to obtain the tag value.
796     MO_TAGGED = 0x400,
797 
798     /// MO_DLLIMPORTAUX - Symbol refers to "auxilliary" import stub. On
799     /// Arm64EC, there are two kinds of import stubs used for DLL import of
800     /// functions: MO_DLLIMPORT refers to natively callable Arm64 code, and
801     /// MO_DLLIMPORTAUX refers to the original address which can be compared
802     /// for equality.
803     MO_DLLIMPORTAUX = 0x800,
804   };
805 } // end namespace AArch64II
806 
807 //===----------------------------------------------------------------------===//
808 // v8.3a Pointer Authentication
809 //
810 
811 namespace AArch64PACKey {
812 enum ID : uint8_t {
813   IA = 0,
814   IB = 1,
815   DA = 2,
816   DB = 3,
817   LAST = DB
818 };
819 } // namespace AArch64PACKey
820 
821 /// Return 2-letter identifier string for numeric key ID.
822 inline static StringRef AArch64PACKeyIDToString(AArch64PACKey::ID KeyID) {
823   switch (KeyID) {
824   case AArch64PACKey::IA:
825     return StringRef("ia");
826   case AArch64PACKey::IB:
827     return StringRef("ib");
828   case AArch64PACKey::DA:
829     return StringRef("da");
830   case AArch64PACKey::DB:
831     return StringRef("db");
832   }
833 }
834 
835 /// Return numeric key ID for 2-letter identifier string.
836 inline static std::optional<AArch64PACKey::ID>
837 AArch64StringToPACKeyID(StringRef Name) {
838   if (Name == "ia")
839     return AArch64PACKey::IA;
840   if (Name == "ib")
841     return AArch64PACKey::IB;
842   if (Name == "da")
843     return AArch64PACKey::DA;
844   if (Name == "db")
845     return AArch64PACKey::DB;
846   return std::nullopt;
847 }
848 
849 namespace AArch64 {
850 // The number of bits in a SVE register is architecturally defined
851 // to be a multiple of this value.  If <M x t> has this number of bits,
852 // a <n x M x t> vector can be stored in a SVE register without any
853 // redundant bits.  If <M x t> has this number of bits divided by P,
854 // a <n x M x t> vector is stored in a SVE register by placing index i
855 // in index i*P of a <n x (M*P) x t> vector.  The other elements of the
856 // <n x (M*P) x t> vector (such as index 1) are undefined.
857 static constexpr unsigned SVEBitsPerBlock = 128;
858 static constexpr unsigned SVEMaxBitsPerVector = 2048;
859 } // end namespace AArch64
860 } // end namespace llvm
861 
862 #endif
863