xref: /freebsd/contrib/llvm-project/llvm/lib/Target/AArch64/Utils/AArch64BaseInfo.h (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
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 
getWRegFromXReg(unsigned Reg)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 
getXRegFromWReg(unsigned Reg)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 
getXRegFromXRegTuple(unsigned RegTuple)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 
getBRegFromDReg(unsigned Reg)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 
getDRegFromBReg(unsigned Reg)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 
atomicBarrierDroppedOnZero(unsigned Opcode)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 
getCondCodeName(CondCode Code)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 
getInvertedCondCode(CondCode Code)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.
getNZCVToSatisfyCondCode(CondCode Code)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 
SysAliasSysAlias343   constexpr SysAlias(const char *N, uint16_t E) : Name(N), Encoding(E) {}
SysAliasSysAlias344   constexpr SysAlias(const char *N, uint16_t E, FeatureBitset F)
345       : Name(N), Encoding(E), FeaturesRequired(F) {}
346 
haveFeaturesSysAlias347   bool haveFeatures(FeatureBitset ActiveFeatures) const {
348     return ActiveFeatures[llvm::AArch64::FeatureAll] ||
349            (FeaturesRequired & ActiveFeatures) == FeaturesRequired;
350   }
351 
getRequiredFeaturesSysAlias352   FeatureBitset getRequiredFeatures() const { return FeaturesRequired; }
353 };
354 
355 struct SysAliasReg : SysAlias {
356   bool NeedsReg;
SysAliasRegSysAliasReg357   constexpr SysAliasReg(const char *N, uint16_t E, bool R)
358       : SysAlias(N, E), NeedsReg(R) {}
SysAliasRegSysAliasReg359   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;
SysAliasImmSysAliasImm365   constexpr SysAliasImm(const char *N, uint16_t E, uint16_t I)
366       : SysAlias(N, E), ImmValue(I) {}
SysAliasImmSysAliasImm367   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.
getNumElementsFromSVEPredPattern(unsigned Pattern)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>
getSVEPredPatternFromNumElements(unsigned MinNumElts)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 AArch64SME {
595 enum ToggleCondition : unsigned {
596   Always,
597   IfCallerIsStreaming,
598   IfCallerIsNonStreaming
599 };
600 }
601 
602 namespace AArch64SE {
603     enum ShiftExtSpecifiers {
604         Invalid = -1,
605         LSL,
606         MSL,
607         LSR,
608         ASR,
609         ROR,
610 
611         UXTB,
612         UXTH,
613         UXTW,
614         UXTX,
615 
616         SXTB,
617         SXTH,
618         SXTW,
619         SXTX
620     };
621 }
622 
623 namespace AArch64Layout {
624     enum VectorLayout {
625         Invalid = -1,
626         VL_8B,
627         VL_4H,
628         VL_2S,
629         VL_1D,
630 
631         VL_16B,
632         VL_8H,
633         VL_4S,
634         VL_2D,
635 
636         // Bare layout for the 128-bit vector
637         // (only show ".b", ".h", ".s", ".d" without vector number)
638         VL_B,
639         VL_H,
640         VL_S,
641         VL_D
642     };
643 }
644 
645 inline static const char *
AArch64VectorLayoutToString(AArch64Layout::VectorLayout Layout)646 AArch64VectorLayoutToString(AArch64Layout::VectorLayout Layout) {
647   switch (Layout) {
648   case AArch64Layout::VL_8B:  return ".8b";
649   case AArch64Layout::VL_4H:  return ".4h";
650   case AArch64Layout::VL_2S:  return ".2s";
651   case AArch64Layout::VL_1D:  return ".1d";
652   case AArch64Layout::VL_16B:  return ".16b";
653   case AArch64Layout::VL_8H:  return ".8h";
654   case AArch64Layout::VL_4S:  return ".4s";
655   case AArch64Layout::VL_2D:  return ".2d";
656   case AArch64Layout::VL_B:  return ".b";
657   case AArch64Layout::VL_H:  return ".h";
658   case AArch64Layout::VL_S:  return ".s";
659   case AArch64Layout::VL_D:  return ".d";
660   default: llvm_unreachable("Unknown Vector Layout");
661   }
662 }
663 
664 inline static AArch64Layout::VectorLayout
AArch64StringToVectorLayout(StringRef LayoutStr)665 AArch64StringToVectorLayout(StringRef LayoutStr) {
666   return StringSwitch<AArch64Layout::VectorLayout>(LayoutStr)
667              .Case(".8b", AArch64Layout::VL_8B)
668              .Case(".4h", AArch64Layout::VL_4H)
669              .Case(".2s", AArch64Layout::VL_2S)
670              .Case(".1d", AArch64Layout::VL_1D)
671              .Case(".16b", AArch64Layout::VL_16B)
672              .Case(".8h", AArch64Layout::VL_8H)
673              .Case(".4s", AArch64Layout::VL_4S)
674              .Case(".2d", AArch64Layout::VL_2D)
675              .Case(".b", AArch64Layout::VL_B)
676              .Case(".h", AArch64Layout::VL_H)
677              .Case(".s", AArch64Layout::VL_S)
678              .Case(".d", AArch64Layout::VL_D)
679              .Default(AArch64Layout::Invalid);
680 }
681 
682 namespace AArch64SysReg {
683   struct SysReg {
684     const char *Name;
685     const char *AltName;
686     unsigned Encoding;
687     bool Readable;
688     bool Writeable;
689     FeatureBitset FeaturesRequired;
690 
haveFeaturesSysReg691     bool haveFeatures(FeatureBitset ActiveFeatures) const {
692       return ActiveFeatures[llvm::AArch64::FeatureAll] ||
693              (FeaturesRequired & ActiveFeatures) == FeaturesRequired;
694     }
695   };
696 
697   #define GET_SYSREG_DECL
698   #include "AArch64GenSystemOperands.inc"
699 
700   const SysReg *lookupSysRegByName(StringRef);
701   const SysReg *lookupSysRegByEncoding(uint16_t);
702 
703   uint32_t parseGenericRegister(StringRef Name);
704   std::string genericRegisterString(uint32_t Bits);
705 }
706 
707 namespace AArch64TLBI {
708   struct TLBI : SysAliasReg {
709     using SysAliasReg::SysAliasReg;
710   };
711   #define GET_TLBITable_DECL
712   #include "AArch64GenSystemOperands.inc"
713 }
714 
715 namespace AArch64PRCTX {
716   struct PRCTX : SysAliasReg {
717     using SysAliasReg::SysAliasReg;
718   };
719   #define GET_PRCTX_DECL
720   #include "AArch64GenSystemOperands.inc"
721 }
722 
723 namespace AArch64II {
724 /// Target Operand Flag enum.
725 enum TOF {
726   //===------------------------------------------------------------------===//
727   // AArch64 Specific MachineOperand flags.
728 
729   MO_NO_FLAG,
730 
731   MO_FRAGMENT = 0x7,
732 
733   /// MO_PAGE - A symbol operand with this flag represents the pc-relative
734   /// offset of the 4K page containing the symbol.  This is used with the
735   /// ADRP instruction.
736   MO_PAGE = 1,
737 
738   /// MO_PAGEOFF - A symbol operand with this flag represents the offset of
739   /// that symbol within a 4K page.  This offset is added to the page address
740   /// to produce the complete address.
741   MO_PAGEOFF = 2,
742 
743   /// MO_G3 - A symbol operand with this flag (granule 3) represents the high
744   /// 16-bits of a 64-bit address, used in a MOVZ or MOVK instruction
745   MO_G3 = 3,
746 
747   /// MO_G2 - A symbol operand with this flag (granule 2) represents the bits
748   /// 32-47 of a 64-bit address, used in a MOVZ or MOVK instruction
749   MO_G2 = 4,
750 
751   /// MO_G1 - A symbol operand with this flag (granule 1) represents the bits
752   /// 16-31 of a 64-bit address, used in a MOVZ or MOVK instruction
753   MO_G1 = 5,
754 
755   /// MO_G0 - A symbol operand with this flag (granule 0) represents the bits
756   /// 0-15 of a 64-bit address, used in a MOVZ or MOVK instruction
757   MO_G0 = 6,
758 
759   /// MO_HI12 - This flag indicates that a symbol operand represents the bits
760   /// 13-24 of a 64-bit address, used in a arithmetic immediate-shifted-left-
761   /// by-12-bits instruction.
762   MO_HI12 = 7,
763 
764   /// MO_COFFSTUB - On a symbol operand "FOO", this indicates that the
765   /// reference is actually to the ".refptr.FOO" symbol.  This is used for
766   /// stub symbols on windows.
767   MO_COFFSTUB = 0x8,
768 
769   /// MO_GOT - This flag indicates that a symbol operand represents the
770   /// address of the GOT entry for the symbol, rather than the address of
771   /// the symbol itself.
772   MO_GOT = 0x10,
773 
774   /// MO_NC - Indicates whether the linker is expected to check the symbol
775   /// reference for overflow. For example in an ADRP/ADD pair of relocations
776   /// the ADRP usually does check, but not the ADD.
777   MO_NC = 0x20,
778 
779   /// MO_TLS - Indicates that the operand being accessed is some kind of
780   /// thread-local symbol. On Darwin, only one type of thread-local access
781   /// exists (pre linker-relaxation), but on ELF the TLSModel used for the
782   /// referee will affect interpretation.
783   MO_TLS = 0x40,
784 
785   /// MO_DLLIMPORT - On a symbol operand, this represents that the reference
786   /// to the symbol is for an import stub.  This is used for DLL import
787   /// storage class indication on Windows.
788   MO_DLLIMPORT = 0x80,
789 
790   /// MO_S - Indicates that the bits of the symbol operand represented by
791   /// MO_G0 etc are signed.
792   MO_S = 0x100,
793 
794   /// MO_PREL - Indicates that the bits of the symbol operand represented by
795   /// MO_G0 etc are PC relative.
796   MO_PREL = 0x200,
797 
798   /// MO_TAGGED - With MO_PAGE, indicates that the page includes a memory tag
799   /// in bits 56-63.
800   /// On a FrameIndex operand, indicates that the underlying memory is tagged
801   /// with an unknown tag value (MTE); this needs to be lowered either to an
802   /// SP-relative load or store instruction (which do not check tags), or to
803   /// an LDG instruction to obtain the tag value.
804   MO_TAGGED = 0x400,
805 
806   /// MO_ARM64EC_CALLMANGLE - Operand refers to the Arm64EC-mangled version
807   /// of a symbol, not the original. For dllimport symbols, this means it
808   /// uses "__imp_aux".  For other symbols, this means it uses the mangled
809   /// ("#" prefix for C) name.
810   MO_ARM64EC_CALLMANGLE = 0x800,
811 };
812 } // end namespace AArch64II
813 
814 //===----------------------------------------------------------------------===//
815 // v8.3a Pointer Authentication
816 //
817 
818 namespace AArch64PACKey {
819 enum ID : uint8_t {
820   IA = 0,
821   IB = 1,
822   DA = 2,
823   DB = 3,
824   LAST = DB
825 };
826 } // namespace AArch64PACKey
827 
828 /// Return 2-letter identifier string for numeric key ID.
AArch64PACKeyIDToString(AArch64PACKey::ID KeyID)829 inline static StringRef AArch64PACKeyIDToString(AArch64PACKey::ID KeyID) {
830   switch (KeyID) {
831   case AArch64PACKey::IA:
832     return StringRef("ia");
833   case AArch64PACKey::IB:
834     return StringRef("ib");
835   case AArch64PACKey::DA:
836     return StringRef("da");
837   case AArch64PACKey::DB:
838     return StringRef("db");
839   }
840   llvm_unreachable("Unhandled AArch64PACKey::ID enum");
841 }
842 
843 /// Return numeric key ID for 2-letter identifier string.
844 inline static std::optional<AArch64PACKey::ID>
AArch64StringToPACKeyID(StringRef Name)845 AArch64StringToPACKeyID(StringRef Name) {
846   if (Name == "ia")
847     return AArch64PACKey::IA;
848   if (Name == "ib")
849     return AArch64PACKey::IB;
850   if (Name == "da")
851     return AArch64PACKey::DA;
852   if (Name == "db")
853     return AArch64PACKey::DB;
854   return std::nullopt;
855 }
856 
857 namespace AArch64 {
858 // The number of bits in a SVE register is architecturally defined
859 // to be a multiple of this value.  If <M x t> has this number of bits,
860 // a <n x M x t> vector can be stored in a SVE register without any
861 // redundant bits.  If <M x t> has this number of bits divided by P,
862 // a <n x M x t> vector is stored in a SVE register by placing index i
863 // in index i*P of a <n x (M*P) x t> vector.  The other elements of the
864 // <n x (M*P) x t> vector (such as index 1) are undefined.
865 static constexpr unsigned SVEBitsPerBlock = 128;
866 static constexpr unsigned SVEMaxBitsPerVector = 2048;
867 } // end namespace AArch64
868 } // end namespace llvm
869 
870 #endif
871