1 //===----------- TargetParser.cpp - Target Parser -------------------------===//
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 #include "llvm/ADT/STLExtras.h"
10 #include "llvm/ADT/StringExtras.h"
11 #include "llvm/Support/AArch64TargetParser.h"
12 #include "llvm/Support/ARMBuildAttributes.h"
13 #include "llvm/Support/FormatVariadic.h"
14 #include "llvm/Support/TargetParser.h"
15 #include "gtest/gtest.h"
16 #include <string>
17 
18 using namespace llvm;
19 
20 namespace {
21 const char *ARMArch[] = {
22     "armv2",       "armv2a",         "armv3",       "armv3m",    "armv4",
23     "armv4t",      "armv5",          "armv5t",      "armv5e",    "armv5te",
24     "armv5tej",    "armv6",          "armv6j",      "armv6k",    "armv6hl",
25     "armv6t2",     "armv6kz",        "armv6z",      "armv6zk",   "armv6-m",
26     "armv6m",      "armv6sm",        "armv6s-m",    "armv7-a",   "armv7",
27     "armv7a",      "armv7ve",        "armv7hl",     "armv7l",    "armv7-r",
28     "armv7r",      "armv7-m",        "armv7m",      "armv7k",    "armv7s",
29     "armv7e-m",    "armv7em",        "armv8-a",     "armv8",     "armv8a",
30     "armv8l",      "armv8.1-a",      "armv8.1a",    "armv8.2-a", "armv8.2a",
31     "armv8.3-a",   "armv8.3a",       "armv8.4-a",   "armv8.4a",  "armv8.5-a",
32     "armv8.5a",    "armv8.6-a",      "armv8.6a",    "armv8.7-a", "armv8.7a",
33     "armv8.8-a",   "armv8.8a",       "armv8-r",     "armv8r",    "armv8-m.base",
34     "armv8m.base", "armv8-m.main",   "armv8m.main", "iwmmxt",    "iwmmxt2",
35     "xscale",      "armv8.1-m.main", "armv9-a",     "armv9",     "armv9a",
36     "armv9.1-a",   "armv9.1a",       "armv9.2-a",   "armv9.2a",
37 };
38 
39 template <ARM::ISAKind ISAKind>
40 std::string FormatExtensionFlags(uint64_t Flags) {
41   std::vector<StringRef> Features;
42 
43   if (ISAKind == ARM::ISAKind::AARCH64) {
44     // AEK_NONE is not meant to be shown to the user so the target parser
45     // does not recognise it. It is relevant here though.
46     if (Flags & AArch64::AEK_NONE)
47       Features.push_back("none");
48     AArch64::getExtensionFeatures(Flags, Features);
49   } else {
50     if (Flags & ARM::AEK_NONE)
51       Features.push_back("none");
52     ARM::getExtensionFeatures(Flags, Features);
53   }
54 
55   // The target parser also includes every extension you don't have.
56   // E.g. if AEK_CRC is not set then it adds "-crc". Not useful here.
57   Features.erase(std::remove_if(Features.begin(), Features.end(),
58                                 [](StringRef extension) {
59                                   return extension.startswith("-");
60                                 }),
61                  Features.end());
62 
63   return llvm::join(Features, ", ");
64 }
65 
66 template <ARM::ISAKind ISAKind>
67 testing::AssertionResult
68 AssertSameExtensionFlags(const char *m_expr, const char *n_expr,
69                          uint64_t ExpectedFlags, uint64_t GotFlags) {
70   if (ExpectedFlags == GotFlags)
71     return testing::AssertionSuccess();
72 
73   return testing::AssertionFailure() << llvm::formatv(
74              "Expected extension flags: {0} ({1:x})\n"
75              "     Got extension flags: {2} ({3:x})\n",
76              FormatExtensionFlags<ISAKind>(ExpectedFlags), ExpectedFlags,
77              FormatExtensionFlags<ISAKind>(GotFlags), GotFlags);
78 }
79 
80 struct ARMCPUTestParams {
81   ARMCPUTestParams(StringRef CPUName, StringRef ExpectedArch,
82                    StringRef ExpectedFPU, uint64_t ExpectedFlags,
83                    StringRef CPUAttr)
84       : CPUName(CPUName), ExpectedArch(ExpectedArch), ExpectedFPU(ExpectedFPU),
85         ExpectedFlags(ExpectedFlags), CPUAttr(CPUAttr) {}
86 
87   friend std::ostream &operator<<(std::ostream &os,
88                                   const ARMCPUTestParams &params) {
89     return os << "\"" << params.CPUName.str() << "\", \""
90               << params.ExpectedArch.str() << "\", \""
91               << params.ExpectedFPU.str() << "\", 0x" << std::hex
92               << params.ExpectedFlags << ", \"" << params.CPUAttr.str() << "\"";
93   }
94 
95   StringRef CPUName;
96   StringRef ExpectedArch;
97   StringRef ExpectedFPU;
98   uint64_t ExpectedFlags;
99   StringRef CPUAttr;
100 };
101 
102 class ARMCPUTestFixture : public ::testing::TestWithParam<ARMCPUTestParams> {};
103 
104 TEST_P(ARMCPUTestFixture, ARMCPUTests) {
105   auto params = GetParam();
106 
107   ARM::ArchKind AK = ARM::parseCPUArch(params.CPUName);
108   EXPECT_EQ(params.ExpectedArch, ARM::getArchName(AK));
109 
110   unsigned FPUKind = ARM::getDefaultFPU(params.CPUName, AK);
111   EXPECT_EQ(params.ExpectedFPU, ARM::getFPUName(FPUKind));
112 
113   uint64_t default_extensions = ARM::getDefaultExtensions(params.CPUName, AK);
114   EXPECT_PRED_FORMAT2(AssertSameExtensionFlags<ARM::ISAKind::ARM>,
115                       params.ExpectedFlags, default_extensions);
116 
117   EXPECT_EQ(params.CPUAttr, ARM::getCPUAttr(AK));
118 }
119 
120 // Note that we include ARM::AEK_NONE even when there are other extensions
121 // we expect. This is because the default extensions for a CPU are the sum
122 // of the default extensions for its architecture and for the CPU.
123 // So if a CPU has no extra extensions, it adds AEK_NONE.
124 INSTANTIATE_TEST_SUITE_P(
125     ARMCPUTestsPart1, ARMCPUTestFixture,
126     ::testing::Values(
127         ARMCPUTestParams("invalid", "invalid", "invalid", ARM::AEK_NONE, ""),
128         ARMCPUTestParams("generic", "invalid", "none", ARM::AEK_NONE, ""),
129 
130         ARMCPUTestParams("arm8", "armv4", "none", ARM::AEK_NONE, "4"),
131         ARMCPUTestParams("arm810", "armv4", "none", ARM::AEK_NONE, "4"),
132         ARMCPUTestParams("strongarm", "armv4", "none", ARM::AEK_NONE, "4"),
133         ARMCPUTestParams("strongarm110", "armv4", "none", ARM::AEK_NONE, "4"),
134         ARMCPUTestParams("strongarm1100", "armv4", "none", ARM::AEK_NONE, "4"),
135         ARMCPUTestParams("strongarm1110", "armv4", "none", ARM::AEK_NONE, "4"),
136         ARMCPUTestParams("arm7tdmi", "armv4t", "none", ARM::AEK_NONE, "4T"),
137         ARMCPUTestParams("arm7tdmi-s", "armv4t", "none", ARM::AEK_NONE, "4T"),
138         ARMCPUTestParams("arm710t", "armv4t", "none", ARM::AEK_NONE, "4T"),
139         ARMCPUTestParams("arm720t", "armv4t", "none", ARM::AEK_NONE, "4T"),
140         ARMCPUTestParams("arm9", "armv4t", "none", ARM::AEK_NONE, "4T"),
141         ARMCPUTestParams("arm9tdmi", "armv4t", "none", ARM::AEK_NONE, "4T"),
142         ARMCPUTestParams("arm920", "armv4t", "none", ARM::AEK_NONE, "4T"),
143         ARMCPUTestParams("arm920t", "armv4t", "none", ARM::AEK_NONE, "4T"),
144         ARMCPUTestParams("arm922t", "armv4t", "none", ARM::AEK_NONE, "4T"),
145         ARMCPUTestParams("arm940t", "armv4t", "none", ARM::AEK_NONE, "4T"),
146         ARMCPUTestParams("ep9312", "armv4t", "none", ARM::AEK_NONE, "4T"),
147         ARMCPUTestParams("arm10tdmi", "armv5t", "none", ARM::AEK_NONE, "5T"),
148         ARMCPUTestParams("arm1020t", "armv5t", "none", ARM::AEK_NONE, "5T"),
149         ARMCPUTestParams("arm9e", "armv5te", "none",
150                          ARM::AEK_NONE | ARM::AEK_DSP, "5TE"),
151         ARMCPUTestParams("arm946e-s", "armv5te", "none",
152                          ARM::AEK_NONE | ARM::AEK_DSP, "5TE"),
153         ARMCPUTestParams("arm966e-s", "armv5te", "none",
154                          ARM::AEK_NONE | ARM::AEK_DSP, "5TE"),
155         ARMCPUTestParams("arm968e-s", "armv5te", "none",
156                          ARM::AEK_NONE | ARM::AEK_DSP, "5TE"),
157         ARMCPUTestParams("arm10e", "armv5te", "none",
158                          ARM::AEK_NONE | ARM::AEK_DSP, "5TE"),
159         ARMCPUTestParams("arm1020e", "armv5te", "none",
160                          ARM::AEK_NONE | ARM::AEK_DSP, "5TE"),
161         ARMCPUTestParams("arm1022e", "armv5te", "none",
162                          ARM::AEK_NONE | ARM::AEK_DSP, "5TE"),
163         ARMCPUTestParams("arm926ej-s", "armv5tej", "none",
164                          ARM::AEK_NONE | ARM::AEK_DSP, "5TEJ"),
165         ARMCPUTestParams("arm1136j-s", "armv6", "none",
166                          ARM::AEK_NONE | ARM::AEK_DSP, "6"),
167         ARMCPUTestParams("arm1136jf-s", "armv6", "vfpv2",
168                          ARM::AEK_NONE | ARM::AEK_DSP, "6"),
169         ARMCPUTestParams("arm1176jz-s", "armv6kz", "none",
170                          ARM::AEK_NONE | ARM::AEK_SEC | ARM::AEK_DSP, "6KZ"),
171         ARMCPUTestParams("mpcore", "armv6k", "vfpv2",
172                          ARM::AEK_NONE | ARM::AEK_DSP, "6K"),
173         ARMCPUTestParams("mpcorenovfp", "armv6k", "none",
174                          ARM::AEK_NONE | ARM::AEK_DSP, "6K"),
175         ARMCPUTestParams("arm1176jzf-s", "armv6kz", "vfpv2",
176                          ARM::AEK_NONE | ARM::AEK_SEC | ARM::AEK_DSP, "6KZ"),
177         ARMCPUTestParams("arm1156t2-s", "armv6t2", "none",
178                          ARM::AEK_NONE | ARM::AEK_DSP, "6T2"),
179         ARMCPUTestParams("arm1156t2f-s", "armv6t2", "vfpv2",
180                          ARM::AEK_NONE | ARM::AEK_DSP, "6T2"),
181         ARMCPUTestParams("cortex-m0", "armv6-m", "none", ARM::AEK_NONE, "6-M"),
182         ARMCPUTestParams("cortex-m0plus", "armv6-m", "none", ARM::AEK_NONE,
183                          "6-M"),
184         ARMCPUTestParams("cortex-m1", "armv6-m", "none", ARM::AEK_NONE, "6-M"),
185         ARMCPUTestParams("sc000", "armv6-m", "none", ARM::AEK_NONE, "6-M"),
186         ARMCPUTestParams("cortex-a5", "armv7-a", "neon-vfpv4",
187                          ARM::AEK_MP | ARM::AEK_SEC | ARM::AEK_DSP, "7-A"),
188         ARMCPUTestParams("cortex-a7", "armv7-a", "neon-vfpv4",
189                          ARM::AEK_HWDIVTHUMB | ARM::AEK_HWDIVARM | ARM::AEK_MP |
190                              ARM::AEK_SEC | ARM::AEK_VIRT | ARM::AEK_DSP,
191                          "7-A"),
192         ARMCPUTestParams("cortex-a8", "armv7-a", "neon",
193                          ARM::AEK_SEC | ARM::AEK_DSP, "7-A")));
194 
195 // gtest in llvm has a limit of 50 test cases when using ::Values so we split
196 // them into 2 blocks
197 INSTANTIATE_TEST_SUITE_P(
198     ARMCPUTestsPart2, ARMCPUTestFixture,
199     ::testing::Values(
200         ARMCPUTestParams("cortex-a9", "armv7-a", "neon-fp16",
201                          ARM::AEK_MP | ARM::AEK_SEC | ARM::AEK_DSP, "7-A"),
202         ARMCPUTestParams("cortex-a12", "armv7-a", "neon-vfpv4",
203                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
204                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
205                              ARM::AEK_DSP,
206                          "7-A"),
207         ARMCPUTestParams("cortex-a15", "armv7-a", "neon-vfpv4",
208                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
209                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
210                              ARM::AEK_DSP,
211                          "7-A"),
212         ARMCPUTestParams("cortex-a17", "armv7-a", "neon-vfpv4",
213                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
214                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
215                              ARM::AEK_DSP,
216                          "7-A"),
217         ARMCPUTestParams("krait", "armv7-a", "neon-vfpv4",
218                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
219                          "7-A"),
220         ARMCPUTestParams("cortex-r4", "armv7-r", "none",
221                          ARM::AEK_NONE | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
222                          "7-R"),
223         ARMCPUTestParams("cortex-r4f", "armv7-r", "vfpv3-d16",
224                          ARM::AEK_NONE | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
225                          "7-R"),
226         ARMCPUTestParams("cortex-r5", "armv7-r", "vfpv3-d16",
227                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
228                              ARM::AEK_DSP,
229                          "7-R"),
230         ARMCPUTestParams("cortex-r7", "armv7-r", "vfpv3-d16-fp16",
231                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
232                              ARM::AEK_DSP,
233                          "7-R"),
234         ARMCPUTestParams("cortex-r8", "armv7-r", "vfpv3-d16-fp16",
235                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
236                              ARM::AEK_DSP,
237                          "7-R"),
238         ARMCPUTestParams("cortex-r52", "armv8-r", "neon-fp-armv8",
239                          ARM::AEK_NONE | ARM::AEK_CRC | ARM::AEK_MP |
240                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
241                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
242                          "8-R"),
243         ARMCPUTestParams("sc300", "armv7-m", "none",
244                          ARM::AEK_NONE | ARM::AEK_HWDIVTHUMB, "7-M"),
245         ARMCPUTestParams("cortex-m3", "armv7-m", "none",
246                          ARM::AEK_NONE | ARM::AEK_HWDIVTHUMB, "7-M"),
247         ARMCPUTestParams("cortex-m4", "armv7e-m", "fpv4-sp-d16",
248                          ARM::AEK_NONE | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
249                          "7E-M"),
250         ARMCPUTestParams("cortex-m7", "armv7e-m", "fpv5-d16",
251                          ARM::AEK_NONE | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
252                          "7E-M"),
253         ARMCPUTestParams("cortex-a32", "armv8-a", "crypto-neon-fp-armv8",
254                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
255                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
256                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
257                          "8-A"),
258         ARMCPUTestParams("cortex-a35", "armv8-a", "crypto-neon-fp-armv8",
259                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
260                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
261                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
262                          "8-A"),
263         ARMCPUTestParams("cortex-a53", "armv8-a", "crypto-neon-fp-armv8",
264                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
265                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
266                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
267                          "8-A"),
268         ARMCPUTestParams("cortex-a55", "armv8.2-a", "crypto-neon-fp-armv8",
269                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
270                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
271                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
272                              ARM::AEK_FP16 | ARM::AEK_RAS | ARM::AEK_DOTPROD,
273                          "8.2-A"),
274         ARMCPUTestParams("cortex-a57", "armv8-a", "crypto-neon-fp-armv8",
275                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
276                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
277                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
278                          "8-A"),
279         ARMCPUTestParams("cortex-a72", "armv8-a", "crypto-neon-fp-armv8",
280                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
281                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
282                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
283                          "8-A"),
284         ARMCPUTestParams("cortex-a73", "armv8-a", "crypto-neon-fp-armv8",
285                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
286                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
287                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
288                          "8-A"),
289         ARMCPUTestParams("cortex-a75", "armv8.2-a", "crypto-neon-fp-armv8",
290                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
291                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
292                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
293                              ARM::AEK_FP16 | ARM::AEK_RAS | ARM::AEK_DOTPROD,
294                          "8.2-A"),
295         ARMCPUTestParams("cortex-a76", "armv8.2-a", "crypto-neon-fp-armv8",
296                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
297                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
298                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
299                              ARM::AEK_FP16 | ARM::AEK_RAS | ARM::AEK_DOTPROD,
300                          "8.2-A"),
301         ARMCPUTestParams("cortex-a76ae", "armv8.2-a", "crypto-neon-fp-armv8",
302                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
303                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
304                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
305                              ARM::AEK_FP16 | ARM::AEK_RAS | ARM::AEK_DOTPROD,
306                          "8.2-A"),
307         ARMCPUTestParams("cortex-a78c", "armv8.2-a", "crypto-neon-fp-armv8",
308                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
309                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
310                              ARM::AEK_DSP | ARM::AEK_CRC | ARM::AEK_RAS |
311                              ARM::AEK_FP16 | ARM::AEK_DOTPROD,
312                          "8.2-A"),
313         ARMCPUTestParams("cortex-a710", "armv9-a", "neon-fp-armv8",
314                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
315                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
316                              ARM::AEK_DSP | ARM::AEK_CRC | ARM::AEK_RAS |
317                              ARM::AEK_DOTPROD | ARM::AEK_FP16FML |
318                              ARM::AEK_BF16 | ARM::AEK_I8MM | ARM::AEK_SB,
319                          "9-A"),
320         ARMCPUTestParams("cortex-a77", "armv8.2-a", "crypto-neon-fp-armv8",
321                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
322                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
323                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
324                              ARM::AEK_FP16 | ARM::AEK_RAS | ARM::AEK_DOTPROD,
325                          "8.2-A"),
326         ARMCPUTestParams("cortex-a78", "armv8.2-a", "crypto-neon-fp-armv8",
327                          ARM::AEK_DOTPROD | ARM::AEK_FP16 | ARM::AEK_SEC |
328                              ARM::AEK_MP | ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
329                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_CRC |
330                              ARM::AEK_RAS,
331                          "8.2-A"),
332         ARMCPUTestParams("cortex-x1", "armv8.2-a", "crypto-neon-fp-armv8",
333                          ARM::AEK_RAS | ARM::AEK_FP16 | ARM::AEK_DOTPROD |
334                              ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
335                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
336                              ARM::AEK_DSP | ARM::AEK_CRC,
337                          "8.2-A"),
338         ARMCPUTestParams("cortex-x1c", "armv8.2-a", "crypto-neon-fp-armv8",
339                          ARM::AEK_RAS | ARM::AEK_FP16 | ARM::AEK_DOTPROD |
340                              ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
341                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
342                              ARM::AEK_DSP | ARM::AEK_CRC,
343                          "8.2-A"),
344         ARMCPUTestParams("neoverse-n1", "armv8.2-a", "crypto-neon-fp-armv8",
345                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
346                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
347                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
348                              ARM::AEK_FP16 | ARM::AEK_RAS | ARM::AEK_DOTPROD,
349                          "8.2-A"),
350         ARMCPUTestParams("neoverse-n2", "armv8.5-a", "crypto-neon-fp-armv8",
351                          ARM::AEK_CRC | ARM::AEK_HWDIVTHUMB |
352                              ARM::AEK_HWDIVARM | ARM::AEK_MP | ARM::AEK_SEC |
353                              ARM::AEK_VIRT | ARM::AEK_DSP | ARM::AEK_BF16 |
354                              ARM::AEK_DOTPROD | ARM::AEK_RAS | ARM::AEK_I8MM |
355                              ARM::AEK_SB,
356                          "8.5-A"),
357         ARMCPUTestParams("neoverse-v1", "armv8.4-a", "crypto-neon-fp-armv8",
358                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
359                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
360                              ARM::AEK_DSP | ARM::AEK_CRC | ARM::AEK_RAS |
361                              ARM::AEK_FP16 | ARM::AEK_BF16 | ARM::AEK_DOTPROD,
362                          "8.4-A"),
363         ARMCPUTestParams("cyclone", "armv8-a", "crypto-neon-fp-armv8",
364                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
365                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
366                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
367                          "8-A"),
368         ARMCPUTestParams("exynos-m3", "armv8-a", "crypto-neon-fp-armv8",
369                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
370                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
371                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
372                          "8-A"),
373         ARMCPUTestParams("exynos-m4", "armv8.2-a", "crypto-neon-fp-armv8",
374                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
375                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
376                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
377                              ARM::AEK_DOTPROD | ARM::AEK_FP16 | ARM::AEK_RAS,
378                          "8.2-A"),
379         ARMCPUTestParams("exynos-m5", "armv8.2-a", "crypto-neon-fp-armv8",
380                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
381                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
382                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP |
383                              ARM::AEK_DOTPROD | ARM::AEK_FP16 | ARM::AEK_RAS,
384                          "8.2-A"),
385         ARMCPUTestParams("cortex-m23", "armv8-m.base", "none",
386                          ARM::AEK_NONE | ARM::AEK_HWDIVTHUMB, "8-M.Baseline"),
387         ARMCPUTestParams("cortex-m33", "armv8-m.main", "fpv5-sp-d16",
388                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "8-M.Mainline"),
389         ARMCPUTestParams("cortex-m35p", "armv8-m.main", "fpv5-sp-d16",
390                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "8-M.Mainline"),
391         ARMCPUTestParams("cortex-m55", "armv8.1-m.main",
392                          "fp-armv8-fullfp16-d16",
393                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_SIMD |
394                              ARM::AEK_FP | ARM::AEK_RAS | ARM::AEK_LOB |
395                              ARM::AEK_FP16,
396                          "8.1-M.Mainline"),
397         ARMCPUTestParams("iwmmxt", "iwmmxt", "none", ARM::AEK_NONE, "iwmmxt"),
398         ARMCPUTestParams("xscale", "xscale", "none", ARM::AEK_NONE, "xscale"),
399         ARMCPUTestParams("swift", "armv7s", "neon-vfpv4",
400                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
401                          "7-S")));
402 
403 static constexpr unsigned NumARMCPUArchs = 88;
404 
405 TEST(TargetParserTest, testARMCPUArchList) {
406   SmallVector<StringRef, NumARMCPUArchs> List;
407   ARM::fillValidCPUArchList(List);
408 
409   // No list exists for these in this test suite, so ensure all are
410   // valid, and match the expected 'magic' count.
411   EXPECT_EQ(List.size(), NumARMCPUArchs);
412   for(StringRef CPU : List) {
413     EXPECT_NE(ARM::parseCPUArch(CPU), ARM::ArchKind::INVALID);
414   }
415 }
416 
417 TEST(TargetParserTest, testInvalidARMArch) {
418   auto InvalidArchStrings = {"armv", "armv99", "noarm"};
419   for (const char* InvalidArch : InvalidArchStrings)
420     EXPECT_EQ(ARM::parseArch(InvalidArch), ARM::ArchKind::INVALID);
421 }
422 
423 bool testARMArch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
424                  unsigned ArchAttr) {
425   ARM::ArchKind AK = ARM::parseArch(Arch);
426   bool Result = (AK != ARM::ArchKind::INVALID);
427   Result &= ARM::getDefaultCPU(Arch).equals(DefaultCPU);
428   Result &= ARM::getSubArch(AK).equals(SubArch);
429   Result &= (ARM::getArchAttr(AK) == ArchAttr);
430   return Result;
431 }
432 
433 TEST(TargetParserTest, testARMArch) {
434   EXPECT_TRUE(
435       testARMArch("armv2", "generic", "v2", ARMBuildAttrs::CPUArch::Pre_v4));
436   EXPECT_TRUE(
437       testARMArch("armv2a", "generic", "v2a", ARMBuildAttrs::CPUArch::Pre_v4));
438   EXPECT_TRUE(
439       testARMArch("armv3", "generic", "v3", ARMBuildAttrs::CPUArch::Pre_v4));
440   EXPECT_TRUE(
441       testARMArch("armv3m", "generic", "v3m", ARMBuildAttrs::CPUArch::Pre_v4));
442   EXPECT_TRUE(
443       testARMArch("armv4", "strongarm", "v4",
444                           ARMBuildAttrs::CPUArch::v4));
445   EXPECT_TRUE(
446       testARMArch("armv4t", "arm7tdmi", "v4t",
447                           ARMBuildAttrs::CPUArch::v4T));
448   EXPECT_TRUE(
449       testARMArch("armv5t", "arm10tdmi", "v5",
450                           ARMBuildAttrs::CPUArch::v5T));
451   EXPECT_TRUE(
452       testARMArch("armv5te", "arm1022e", "v5e",
453                           ARMBuildAttrs::CPUArch::v5TE));
454   EXPECT_TRUE(
455       testARMArch("armv5tej", "arm926ej-s", "v5e",
456                           ARMBuildAttrs::CPUArch::v5TEJ));
457   EXPECT_TRUE(
458       testARMArch("armv6", "arm1136jf-s", "v6",
459                           ARMBuildAttrs::CPUArch::v6));
460   EXPECT_TRUE(
461       testARMArch("armv6k", "mpcore", "v6k",
462                           ARMBuildAttrs::CPUArch::v6K));
463   EXPECT_TRUE(
464       testARMArch("armv6t2", "arm1156t2-s", "v6t2",
465                           ARMBuildAttrs::CPUArch::v6T2));
466   EXPECT_TRUE(
467       testARMArch("armv6kz", "arm1176jzf-s", "v6kz",
468                           ARMBuildAttrs::CPUArch::v6KZ));
469   EXPECT_TRUE(
470       testARMArch("armv6-m", "cortex-m0", "v6m",
471                           ARMBuildAttrs::CPUArch::v6_M));
472   EXPECT_TRUE(
473       testARMArch("armv7-a", "generic", "v7",
474                           ARMBuildAttrs::CPUArch::v7));
475   EXPECT_TRUE(
476       testARMArch("armv7ve", "generic", "v7ve",
477                           ARMBuildAttrs::CPUArch::v7));
478   EXPECT_TRUE(
479       testARMArch("armv7-r", "cortex-r4", "v7r",
480                           ARMBuildAttrs::CPUArch::v7));
481   EXPECT_TRUE(
482       testARMArch("armv7-m", "cortex-m3", "v7m",
483                           ARMBuildAttrs::CPUArch::v7));
484   EXPECT_TRUE(
485       testARMArch("armv7e-m", "cortex-m4", "v7em",
486                           ARMBuildAttrs::CPUArch::v7E_M));
487   EXPECT_TRUE(
488       testARMArch("armv8-a", "generic", "v8",
489                           ARMBuildAttrs::CPUArch::v8_A));
490   EXPECT_TRUE(
491       testARMArch("armv8.1-a", "generic", "v8.1a",
492                           ARMBuildAttrs::CPUArch::v8_A));
493   EXPECT_TRUE(
494       testARMArch("armv8.2-a", "generic", "v8.2a",
495                           ARMBuildAttrs::CPUArch::v8_A));
496   EXPECT_TRUE(
497       testARMArch("armv8.3-a", "generic", "v8.3a",
498                           ARMBuildAttrs::CPUArch::v8_A));
499   EXPECT_TRUE(
500       testARMArch("armv8.4-a", "generic", "v8.4a",
501                           ARMBuildAttrs::CPUArch::v8_A));
502   EXPECT_TRUE(
503       testARMArch("armv8.5-a", "generic", "v8.5a",
504                           ARMBuildAttrs::CPUArch::v8_A));
505   EXPECT_TRUE(
506       testARMArch("armv8.6-a", "generic", "v8.6a",
507                           ARMBuildAttrs::CPUArch::v8_A));
508   EXPECT_TRUE(
509       testARMArch("armv8.7-a", "generic", "v8.7a",
510                           ARMBuildAttrs::CPUArch::v8_A));
511   EXPECT_TRUE(testARMArch("armv8.8-a", "generic", "v8.8a",
512                           ARMBuildAttrs::CPUArch::v8_A));
513   EXPECT_TRUE(
514       testARMArch("armv9-a", "generic", "v9a",
515                           ARMBuildAttrs::CPUArch::v8_A));
516   EXPECT_TRUE(
517       testARMArch("armv9.1-a", "generic", "v9.1a",
518                           ARMBuildAttrs::CPUArch::v8_A));
519   EXPECT_TRUE(
520       testARMArch("armv9.2-a", "generic", "v9.2a",
521                           ARMBuildAttrs::CPUArch::v8_A));
522   EXPECT_TRUE(
523       testARMArch("armv8-r", "cortex-r52", "v8r",
524                           ARMBuildAttrs::CPUArch::v8_R));
525   EXPECT_TRUE(
526       testARMArch("armv8-m.base", "generic", "v8m.base",
527                           ARMBuildAttrs::CPUArch::v8_M_Base));
528   EXPECT_TRUE(
529       testARMArch("armv8-m.main", "generic", "v8m.main",
530                           ARMBuildAttrs::CPUArch::v8_M_Main));
531   EXPECT_TRUE(
532       testARMArch("armv8.1-m.main", "generic", "v8.1m.main",
533                           ARMBuildAttrs::CPUArch::v8_1_M_Main));
534   EXPECT_TRUE(
535       testARMArch("iwmmxt", "iwmmxt", "",
536                           ARMBuildAttrs::CPUArch::v5TE));
537   EXPECT_TRUE(
538       testARMArch("iwmmxt2", "generic", "",
539                           ARMBuildAttrs::CPUArch::v5TE));
540   EXPECT_TRUE(
541       testARMArch("xscale", "xscale", "v5e",
542                           ARMBuildAttrs::CPUArch::v5TE));
543   EXPECT_TRUE(
544       testARMArch("armv7s", "swift", "v7s",
545                           ARMBuildAttrs::CPUArch::v7));
546   EXPECT_TRUE(
547       testARMArch("armv7k", "generic", "v7k",
548                           ARMBuildAttrs::CPUArch::v7));
549 }
550 
551 bool testARMExtension(StringRef CPUName,ARM::ArchKind ArchKind, StringRef ArchExt) {
552   return ARM::getDefaultExtensions(CPUName, ArchKind) &
553          ARM::parseArchExt(ArchExt);
554 }
555 
556 TEST(TargetParserTest, testARMExtension) {
557   EXPECT_FALSE(testARMExtension("strongarm", ARM::ArchKind::INVALID, "dsp"));
558   EXPECT_FALSE(testARMExtension("arm7tdmi", ARM::ArchKind::INVALID, "dsp"));
559   EXPECT_FALSE(testARMExtension("arm10tdmi",
560                                 ARM::ArchKind::INVALID, "simd"));
561   EXPECT_FALSE(testARMExtension("arm1022e", ARM::ArchKind::INVALID, "simd"));
562   EXPECT_FALSE(testARMExtension("arm926ej-s",
563                                 ARM::ArchKind::INVALID, "simd"));
564   EXPECT_FALSE(testARMExtension("arm1136jf-s",
565                                 ARM::ArchKind::INVALID, "crypto"));
566   EXPECT_FALSE(testARMExtension("arm1156t2-s",
567                                 ARM::ArchKind::INVALID, "crypto"));
568   EXPECT_FALSE(testARMExtension("arm1176jzf-s",
569                                 ARM::ArchKind::INVALID, "crypto"));
570   EXPECT_FALSE(testARMExtension("cortex-m0",
571                                 ARM::ArchKind::INVALID, "crypto"));
572   EXPECT_FALSE(testARMExtension("cortex-a8",
573                                 ARM::ArchKind::INVALID, "crypto"));
574   EXPECT_FALSE(testARMExtension("cortex-r4",
575                                 ARM::ArchKind::INVALID, "crypto"));
576   EXPECT_FALSE(testARMExtension("cortex-m3",
577                                 ARM::ArchKind::INVALID, "crypto"));
578   EXPECT_FALSE(testARMExtension("cortex-a53",
579                                 ARM::ArchKind::INVALID, "ras"));
580   EXPECT_FALSE(testARMExtension("cortex-a53",
581                                 ARM::ArchKind::INVALID, "fp16"));
582   EXPECT_TRUE(testARMExtension("cortex-a55",
583                                 ARM::ArchKind::INVALID, "fp16"));
584   EXPECT_FALSE(testARMExtension("cortex-a55",
585                                 ARM::ArchKind::INVALID, "fp16fml"));
586   EXPECT_TRUE(testARMExtension("cortex-a75",
587                                 ARM::ArchKind::INVALID, "fp16"));
588   EXPECT_FALSE(testARMExtension("cortex-a75",
589                                 ARM::ArchKind::INVALID, "fp16fml"));
590   EXPECT_FALSE(testARMExtension("cortex-r52",
591                                 ARM::ArchKind::INVALID, "ras"));
592   EXPECT_FALSE(testARMExtension("iwmmxt", ARM::ArchKind::INVALID, "crc"));
593   EXPECT_FALSE(testARMExtension("xscale", ARM::ArchKind::INVALID, "crc"));
594   EXPECT_FALSE(testARMExtension("swift", ARM::ArchKind::INVALID, "crc"));
595 
596   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2, "thumb"));
597   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2A, "thumb"));
598   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3, "thumb"));
599   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3M, "thumb"));
600   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4, "dsp"));
601   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4T, "dsp"));
602   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5T, "simd"));
603   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TE, "simd"));
604   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TEJ, "simd"));
605   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6, "crypto"));
606   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6K, "crypto"));
607   EXPECT_FALSE(testARMExtension("generic",
608                                 ARM::ArchKind::ARMV6T2, "crypto"));
609   EXPECT_FALSE(testARMExtension("generic",
610                                 ARM::ArchKind::ARMV6KZ, "crypto"));
611   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6M, "crypto"));
612   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7A, "crypto"));
613   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7R, "crypto"));
614   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7M, "crypto"));
615   EXPECT_FALSE(testARMExtension("generic",
616                                 ARM::ArchKind::ARMV7EM, "crypto"));
617   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8A, "ras"));
618   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_1A, "ras"));
619   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "profile"));
620   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16"));
621   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16fml"));
622   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16"));
623   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16fml"));
624   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16"));
625   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16fml"));
626   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8R, "ras"));
627   EXPECT_FALSE(testARMExtension("generic",
628                                 ARM::ArchKind::ARMV8MBaseline, "crc"));
629   EXPECT_FALSE(testARMExtension("generic",
630                                 ARM::ArchKind::ARMV8MMainline, "crc"));
631   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT, "crc"));
632   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT2, "crc"));
633   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::XSCALE, "crc"));
634   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7S, "crypto"));
635   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7K, "crypto"));
636 }
637 
638 TEST(TargetParserTest, ARMFPUVersion) {
639   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
640        FK <= ARM::FPUKind::FK_LAST;
641        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
642     if (FK == ARM::FK_LAST || ARM::getFPUName(FK) == "invalid" ||
643         ARM::getFPUName(FK) == "none" || ARM::getFPUName(FK) == "softvfp")
644       EXPECT_EQ(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
645     else
646       EXPECT_NE(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
647 }
648 
649 TEST(TargetParserTest, ARMFPUNeonSupportLevel) {
650   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
651        FK <= ARM::FPUKind::FK_LAST;
652        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
653     if (FK == ARM::FK_LAST ||
654         ARM::getFPUName(FK).find("neon") == std::string::npos)
655       EXPECT_EQ(ARM::NeonSupportLevel::None,
656                  ARM::getFPUNeonSupportLevel(FK));
657     else
658       EXPECT_NE(ARM::NeonSupportLevel::None,
659                 ARM::getFPUNeonSupportLevel(FK));
660 }
661 
662 TEST(TargetParserTest, ARMFPURestriction) {
663   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
664        FK <= ARM::FPUKind::FK_LAST;
665        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1)) {
666     if (FK == ARM::FK_LAST ||
667         (ARM::getFPUName(FK).find("d16") == std::string::npos &&
668          ARM::getFPUName(FK).find("vfpv3xd") == std::string::npos))
669       EXPECT_EQ(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
670     else
671       EXPECT_NE(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
672   }
673 }
674 
675 TEST(TargetParserTest, ARMExtensionFeatures) {
676   std::map<uint64_t, std::vector<StringRef>> Extensions;
677 
678   for (auto &Ext : ARM::ARCHExtNames) {
679     if (Ext.Feature && Ext.NegFeature)
680       Extensions[Ext.ID] = { StringRef(Ext.Feature),
681                              StringRef(Ext.NegFeature) };
682   }
683 
684   Extensions[ARM::AEK_HWDIVARM]   = { "+hwdiv-arm", "-hwdiv-arm" };
685   Extensions[ARM::AEK_HWDIVTHUMB] = { "+hwdiv",     "-hwdiv" };
686 
687   std::vector<StringRef> Features;
688 
689   EXPECT_FALSE(ARM::getExtensionFeatures(ARM::AEK_INVALID, Features));
690 
691   for (auto &E : Extensions) {
692     // test +extension
693     Features.clear();
694     ARM::getExtensionFeatures(E.first, Features);
695     EXPECT_TRUE(llvm::is_contained(Features, E.second.at(0)));
696     EXPECT_EQ(Extensions.size(), Features.size());
697 
698     // test -extension
699     Features.clear();
700     ARM::getExtensionFeatures(~E.first, Features);
701     EXPECT_TRUE(llvm::is_contained(Features, E.second.at(1)));
702     EXPECT_EQ(Extensions.size(), Features.size());
703   }
704 }
705 
706 TEST(TargetParserTest, ARMFPUFeatures) {
707   std::vector<StringRef> Features;
708   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
709        FK <= ARM::FPUKind::FK_LAST;
710        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1)) {
711     if (FK == ARM::FK_INVALID || FK >= ARM::FK_LAST)
712       EXPECT_FALSE(ARM::getFPUFeatures(FK, Features));
713     else
714       EXPECT_TRUE(ARM::getFPUFeatures(FK, Features));
715   }
716 }
717 
718 TEST(TargetParserTest, ARMArchExtFeature) {
719   const char *ArchExt[][4] = {{"crc", "nocrc", "+crc", "-crc"},
720                               {"crypto", "nocrypto", "+crypto", "-crypto"},
721                               {"dsp", "nodsp", "+dsp", "-dsp"},
722                               {"fp", "nofp", nullptr, nullptr},
723                               {"idiv", "noidiv", nullptr, nullptr},
724                               {"mp", "nomp", nullptr, nullptr},
725                               {"simd", "nosimd", nullptr, nullptr},
726                               {"sec", "nosec", nullptr, nullptr},
727                               {"virt", "novirt", nullptr, nullptr},
728                               {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
729                               {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
730                               {"ras", "noras", "+ras", "-ras"},
731                               {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
732                               {"os", "noos", nullptr, nullptr},
733                               {"iwmmxt", "noiwmmxt", nullptr, nullptr},
734                               {"iwmmxt2", "noiwmmxt2", nullptr, nullptr},
735                               {"maverick", "maverick", nullptr, nullptr},
736                               {"xscale", "noxscale", nullptr, nullptr},
737                               {"sb", "nosb", "+sb", "-sb"},
738                               {"i8mm", "noi8mm", "+i8mm", "-i8mm"},
739                               {"mve", "nomve", "+mve", "-mve"},
740                               {"mve.fp", "nomve.fp", "+mve.fp", "-mve.fp"}};
741 
742   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
743     EXPECT_EQ(StringRef(ArchExt[i][2]), ARM::getArchExtFeature(ArchExt[i][0]));
744     EXPECT_EQ(StringRef(ArchExt[i][3]), ARM::getArchExtFeature(ArchExt[i][1]));
745   }
746 }
747 
748 static bool
749 testArchExtDependency(const char *ArchExt,
750                       const std::initializer_list<const char *> &Expected) {
751   std::vector<StringRef> Features;
752   unsigned FPUID;
753 
754   if (!ARM::appendArchExtFeatures("", ARM::ArchKind::ARMV8_1MMainline, ArchExt,
755                                   Features, FPUID))
756     return false;
757 
758   return llvm::all_of(Expected, [&](StringRef Ext) {
759     return llvm::is_contained(Features, Ext);
760   });
761 }
762 
763 TEST(TargetParserTest, ARMArchExtDependencies) {
764   EXPECT_TRUE(testArchExtDependency("mve", {"+mve", "+dsp"}));
765   EXPECT_TRUE(testArchExtDependency("mve.fp", {"+mve.fp", "+mve", "+dsp"}));
766   EXPECT_TRUE(testArchExtDependency("nodsp", {"-dsp", "-mve", "-mve.fp"}));
767   EXPECT_TRUE(testArchExtDependency("nomve", {"-mve", "-mve.fp"}));
768 }
769 
770 TEST(TargetParserTest, ARMparseHWDiv) {
771   const char *hwdiv[] = {"thumb", "arm", "arm,thumb", "thumb,arm"};
772 
773   for (unsigned i = 0; i < array_lengthof(hwdiv); i++)
774     EXPECT_NE(ARM::AEK_INVALID, ARM::parseHWDiv((StringRef)hwdiv[i]));
775 }
776 
777 TEST(TargetParserTest, ARMparseArchEndianAndISA) {
778   const char *Arch[] = {
779       "v2",        "v2a",    "v3",    "v3m",    "v4",       "v4t",
780       "v5",        "v5t",    "v5e",   "v5te",   "v5tej",    "v6",
781       "v6j",       "v6k",    "v6hl",  "v6t2",   "v6kz",     "v6z",
782       "v6zk",      "v6-m",   "v6m",   "v6sm",   "v6s-m",    "v7-a",
783       "v7",        "v7a",    "v7ve",  "v7hl",   "v7l",      "v7-r",
784       "v7r",       "v7-m",   "v7m",   "v7k",    "v7s",      "v7e-m",
785       "v7em",      "v8-a",   "v8",    "v8a",    "v8l",      "v8.1-a",
786       "v8.1a",     "v8.2-a", "v8.2a", "v8.3-a", "v8.3a",    "v8.4-a",
787       "v8.4a",     "v8.5-a", "v8.5a", "v8.6-a", "v8.6a",    "v8.7-a",
788       "v8.7a",     "v8.8-a", "v8.8a", "v8-r",   "v8m.base", "v8m.main",
789       "v8.1m.main"};
790 
791   for (unsigned i = 0; i < array_lengthof(Arch); i++) {
792     std::string arm_1 = "armeb" + (std::string)(Arch[i]);
793     std::string arm_2 = "arm" + (std::string)(Arch[i]) + "eb";
794     std::string arm_3 = "arm" + (std::string)(Arch[i]);
795     std::string thumb_1 = "thumbeb" + (std::string)(Arch[i]);
796     std::string thumb_2 = "thumb" + (std::string)(Arch[i]) + "eb";
797     std::string thumb_3 = "thumb" + (std::string)(Arch[i]);
798 
799     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_1));
800     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_2));
801     EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(arm_3));
802 
803     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_1));
804     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_2));
805     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_3));
806     if (i >= 4) {
807       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_1));
808       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_2));
809       EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(thumb_3));
810 
811       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_1));
812       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_2));
813       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_3));
814     }
815   }
816 
817   EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian("aarch64"));
818   EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian("arm64_32"));
819   EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian("aarch64_be"));
820 
821   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64"));
822   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64_be"));
823   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64"));
824   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64_be"));
825   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64_32"));
826   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64_32"));
827 }
828 
829 TEST(TargetParserTest, ARMparseArchProfile) {
830   for (unsigned i = 0; i < array_lengthof(ARMArch); i++) {
831     switch (ARM::parseArch(ARMArch[i])) {
832     case ARM::ArchKind::ARMV6M:
833     case ARM::ArchKind::ARMV7M:
834     case ARM::ArchKind::ARMV7EM:
835     case ARM::ArchKind::ARMV8MMainline:
836     case ARM::ArchKind::ARMV8MBaseline:
837     case ARM::ArchKind::ARMV8_1MMainline:
838       EXPECT_EQ(ARM::ProfileKind::M, ARM::parseArchProfile(ARMArch[i]));
839       break;
840     case ARM::ArchKind::ARMV7R:
841     case ARM::ArchKind::ARMV8R:
842       EXPECT_EQ(ARM::ProfileKind::R, ARM::parseArchProfile(ARMArch[i]));
843       break;
844     case ARM::ArchKind::ARMV7A:
845     case ARM::ArchKind::ARMV7VE:
846     case ARM::ArchKind::ARMV7K:
847     case ARM::ArchKind::ARMV8A:
848     case ARM::ArchKind::ARMV8_1A:
849     case ARM::ArchKind::ARMV8_2A:
850     case ARM::ArchKind::ARMV8_3A:
851     case ARM::ArchKind::ARMV8_4A:
852     case ARM::ArchKind::ARMV8_5A:
853     case ARM::ArchKind::ARMV8_6A:
854     case ARM::ArchKind::ARMV8_7A:
855     case ARM::ArchKind::ARMV8_8A:
856     case ARM::ArchKind::ARMV9A:
857     case ARM::ArchKind::ARMV9_1A:
858     case ARM::ArchKind::ARMV9_2A:
859     case ARM::ArchKind::ARMV9_3A:
860       EXPECT_EQ(ARM::ProfileKind::A, ARM::parseArchProfile(ARMArch[i]));
861       break;
862     default:
863       EXPECT_EQ(ARM::ProfileKind::INVALID, ARM::parseArchProfile(ARMArch[i]));
864       break;
865     }
866   }
867 }
868 
869 TEST(TargetParserTest, ARMparseArchVersion) {
870   for (unsigned i = 0; i < array_lengthof(ARMArch); i++)
871     if (((std::string)ARMArch[i]).substr(0, 4) == "armv")
872       EXPECT_EQ((ARMArch[i][4] - 48u), ARM::parseArchVersion(ARMArch[i]));
873     else
874       EXPECT_EQ(5u, ARM::parseArchVersion(ARMArch[i]));
875 }
876 
877 class AArch64CPUTestFixture
878     : public ::testing::TestWithParam<ARMCPUTestParams> {};
879 
880 TEST_P(AArch64CPUTestFixture, testAArch64CPU) {
881   ARMCPUTestParams params = GetParam();
882 
883   AArch64::ArchKind AK = AArch64::parseCPUArch(params.CPUName);
884   EXPECT_EQ(params.ExpectedArch, AArch64::getArchName(AK));
885 
886   uint64_t default_extensions =
887       AArch64::getDefaultExtensions(params.CPUName, AK);
888   EXPECT_PRED_FORMAT2(AssertSameExtensionFlags<ARM::ISAKind::AARCH64>,
889                       params.ExpectedFlags, default_extensions);
890 
891   unsigned FPUKind = AArch64::getDefaultFPU(params.CPUName, AK);
892   EXPECT_EQ(params.ExpectedFPU, ARM::getFPUName(FPUKind));
893 
894   EXPECT_EQ(params.CPUAttr, AArch64::getCPUAttr(AK));
895 }
896 
897 INSTANTIATE_TEST_SUITE_P(
898     AArch64CPUTests, AArch64CPUTestFixture,
899     ::testing::Values(
900         ARMCPUTestParams("invalid", "invalid", "invalid", AArch64::AEK_NONE,
901                          ""),
902         ARMCPUTestParams("generic", "invalid", "none", AArch64::AEK_NONE, ""),
903 
904         ARMCPUTestParams("cortex-a34", "armv8-a", "crypto-neon-fp-armv8",
905                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
906                              AArch64::AEK_FP | AArch64::AEK_SIMD,
907                          "8-A"),
908         ARMCPUTestParams("cortex-a35", "armv8-a", "crypto-neon-fp-armv8",
909                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
910                              AArch64::AEK_FP | AArch64::AEK_SIMD,
911                          "8-A"),
912         ARMCPUTestParams("cortex-a53", "armv8-a", "crypto-neon-fp-armv8",
913                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
914                              AArch64::AEK_FP | AArch64::AEK_SIMD,
915                          "8-A"),
916         ARMCPUTestParams("cortex-a55", "armv8.2-a", "crypto-neon-fp-armv8",
917                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
918                              AArch64::AEK_FP | AArch64::AEK_SIMD |
919                              AArch64::AEK_RAS | AArch64::AEK_LSE |
920                              AArch64::AEK_RDM | AArch64::AEK_FP16 |
921                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC,
922                          "8.2-A"),
923         ARMCPUTestParams("cortex-a510", "armv9-a", "neon-fp-armv8",
924                          AArch64::AEK_CRC | AArch64::AEK_FP |
925                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
926                              AArch64::AEK_LSE | AArch64::AEK_RDM |
927                              AArch64::AEK_RCPC | AArch64::AEK_DOTPROD |
928                              AArch64::AEK_BF16 | AArch64::AEK_I8MM |
929                              AArch64::AEK_SVE | AArch64::AEK_SVE2 |
930                              AArch64::AEK_SVE2BITPERM | AArch64::AEK_PAUTH |
931                              AArch64::AEK_MTE | AArch64::AEK_SSBS |
932                              AArch64::AEK_FP16FML | AArch64::AEK_SB,
933                          "9-A"),
934         ARMCPUTestParams("cortex-a57", "armv8-a", "crypto-neon-fp-armv8",
935                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
936                              AArch64::AEK_FP | AArch64::AEK_SIMD,
937                          "8-A"),
938         ARMCPUTestParams("cortex-a65", "armv8.2-a", "crypto-neon-fp-armv8",
939                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
940                              AArch64::AEK_DOTPROD | AArch64::AEK_FP |
941                              AArch64::AEK_FP16 | AArch64::AEK_LSE |
942                              AArch64::AEK_RAS | AArch64::AEK_RCPC |
943                              AArch64::AEK_RDM | AArch64::AEK_SIMD |
944                              AArch64::AEK_SSBS,
945                          "8.2-A"),
946         ARMCPUTestParams("cortex-a65ae", "armv8.2-a", "crypto-neon-fp-armv8",
947                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
948                              AArch64::AEK_DOTPROD | AArch64::AEK_FP |
949                              AArch64::AEK_FP16 | AArch64::AEK_LSE |
950                              AArch64::AEK_RAS | AArch64::AEK_RCPC |
951                              AArch64::AEK_RDM | AArch64::AEK_SIMD |
952                              AArch64::AEK_SSBS,
953                          "8.2-A"),
954         ARMCPUTestParams("cortex-a72", "armv8-a", "crypto-neon-fp-armv8",
955                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
956                              AArch64::AEK_FP | AArch64::AEK_SIMD,
957                          "8-A"),
958         ARMCPUTestParams("cortex-a73", "armv8-a", "crypto-neon-fp-armv8",
959                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
960                              AArch64::AEK_FP | AArch64::AEK_SIMD,
961                          "8-A"),
962         ARMCPUTestParams("cortex-a75", "armv8.2-a", "crypto-neon-fp-armv8",
963                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
964                              AArch64::AEK_FP | AArch64::AEK_SIMD |
965                              AArch64::AEK_RAS | AArch64::AEK_LSE |
966                              AArch64::AEK_RDM | AArch64::AEK_FP16 |
967                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC,
968                          "8.2-A"),
969         ARMCPUTestParams("cortex-a76", "armv8.2-a", "crypto-neon-fp-armv8",
970                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
971                              AArch64::AEK_FP | AArch64::AEK_RDM |
972                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
973                              AArch64::AEK_LSE | AArch64::AEK_FP16 |
974                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
975                              AArch64::AEK_SSBS,
976                          "8.2-A"),
977         ARMCPUTestParams("cortex-a76ae", "armv8.2-a", "crypto-neon-fp-armv8",
978                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
979                              AArch64::AEK_FP | AArch64::AEK_RDM |
980                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
981                              AArch64::AEK_LSE | AArch64::AEK_FP16 |
982                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
983                              AArch64::AEK_SSBS,
984                          "8.2-A"),
985         ARMCPUTestParams("cortex-a77", "armv8.2-a", "crypto-neon-fp-armv8",
986                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
987                              AArch64::AEK_FP | AArch64::AEK_RDM |
988                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
989                              AArch64::AEK_LSE | AArch64::AEK_FP16 |
990                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
991                              AArch64::AEK_SSBS,
992                          "8.2-A"),
993         ARMCPUTestParams("cortex-a78", "armv8.2-a", "crypto-neon-fp-armv8",
994                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
995                              AArch64::AEK_FP | AArch64::AEK_RDM |
996                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
997                              AArch64::AEK_LSE | AArch64::AEK_FP16 |
998                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
999                              AArch64::AEK_SSBS | AArch64::AEK_PROFILE,
1000                          "8.2-A"),
1001         ARMCPUTestParams("cortex-a78c", "armv8.2-a", "crypto-neon-fp-armv8",
1002                          AArch64::AEK_RAS | AArch64::AEK_CRC |
1003                              AArch64::AEK_CRYPTO | AArch64::AEK_FP |
1004                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
1005                              AArch64::AEK_LSE | AArch64::AEK_RDM |
1006                              AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
1007                              AArch64::AEK_RCPC | AArch64::AEK_SSBS |
1008                              AArch64::AEK_PROFILE | AArch64::AEK_FLAGM |
1009                              AArch64::AEK_PAUTH | AArch64::AEK_FP16FML,
1010                          "8.2-A"),
1011         ARMCPUTestParams("cortex-a710", "armv9-a", "neon-fp-armv8",
1012                          AArch64::AEK_CRC | AArch64::AEK_FP |
1013                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
1014                              AArch64::AEK_LSE | AArch64::AEK_RDM |
1015                              AArch64::AEK_RCPC | AArch64::AEK_DOTPROD |
1016                              AArch64::AEK_MTE | AArch64::AEK_FP16FML |
1017                              AArch64::AEK_SVE | AArch64::AEK_SVE2 |
1018                              AArch64::AEK_SVE2BITPERM | AArch64::AEK_PAUTH |
1019                              AArch64::AEK_FLAGM | AArch64::AEK_SB |
1020                              AArch64::AEK_I8MM | AArch64::AEK_BF16,
1021                          "9-A"),
1022         ARMCPUTestParams(
1023             "neoverse-v1", "armv8.4-a", "crypto-neon-fp-armv8",
1024             AArch64::AEK_RAS | AArch64::AEK_SVE | AArch64::AEK_SSBS |
1025                 AArch64::AEK_RCPC | AArch64::AEK_CRC | AArch64::AEK_FP |
1026                 AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
1027                 AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_DOTPROD |
1028                 AArch64::AEK_CRYPTO | AArch64::AEK_FP16 | AArch64::AEK_BF16 |
1029                 AArch64::AEK_PROFILE | AArch64::AEK_RAND |
1030                 AArch64::AEK_FP16FML | AArch64::AEK_I8MM,
1031             "8.4-A"),
1032         ARMCPUTestParams("cortex-r82", "armv8-r", "crypto-neon-fp-armv8",
1033                          AArch64::AEK_CRC | AArch64::AEK_RDM |
1034                              AArch64::AEK_SSBS | AArch64::AEK_DOTPROD |
1035                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1036                              AArch64::AEK_FP16 | AArch64::AEK_FP16FML |
1037                              AArch64::AEK_RAS | AArch64::AEK_RCPC |
1038                              AArch64::AEK_LSE | AArch64::AEK_SB,
1039                          "8-R"),
1040         ARMCPUTestParams("cortex-x1", "armv8.2-a", "crypto-neon-fp-armv8",
1041                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1042                              AArch64::AEK_FP | AArch64::AEK_RDM |
1043                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
1044                              AArch64::AEK_LSE | AArch64::AEK_FP16 |
1045                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
1046                              AArch64::AEK_SSBS | AArch64::AEK_PROFILE,
1047                          "8.2-A"),
1048         ARMCPUTestParams("cortex-x1c", "armv8.2-a", "crypto-neon-fp-armv8",
1049                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1050                              AArch64::AEK_FP | AArch64::AEK_RDM |
1051                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
1052                              AArch64::AEK_LSE | AArch64::AEK_FP16 |
1053                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
1054                              AArch64::AEK_SSBS | AArch64::AEK_PAUTH |
1055                              AArch64::AEK_PROFILE,
1056                          "8.2-A"),
1057         ARMCPUTestParams("cortex-x2", "armv9-a", "neon-fp-armv8",
1058                          AArch64::AEK_CRC | AArch64::AEK_FP |
1059                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
1060                              AArch64::AEK_LSE | AArch64::AEK_RDM |
1061                              AArch64::AEK_RCPC | AArch64::AEK_DOTPROD |
1062                              AArch64::AEK_MTE | AArch64::AEK_PAUTH |
1063                              AArch64::AEK_I8MM | AArch64::AEK_BF16 |
1064                              AArch64::AEK_SVE | AArch64::AEK_SVE2 |
1065                              AArch64::AEK_SVE2BITPERM | AArch64::AEK_SSBS |
1066                              AArch64::AEK_SB | AArch64::AEK_FP16FML,
1067                          "9-A"),
1068         ARMCPUTestParams("cyclone", "armv8-a", "crypto-neon-fp-armv8",
1069                          AArch64::AEK_NONE | AArch64::AEK_CRYPTO |
1070                              AArch64::AEK_FP | AArch64::AEK_SIMD,
1071                          "8-A"),
1072         ARMCPUTestParams("apple-a7", "armv8-a", "crypto-neon-fp-armv8",
1073                          AArch64::AEK_NONE | AArch64::AEK_CRYPTO |
1074                              AArch64::AEK_FP | AArch64::AEK_SIMD,
1075                          "8-A"),
1076         ARMCPUTestParams("apple-a8", "armv8-a", "crypto-neon-fp-armv8",
1077                          AArch64::AEK_NONE | AArch64::AEK_CRYPTO |
1078                              AArch64::AEK_FP | AArch64::AEK_SIMD,
1079                          "8-A"),
1080         ARMCPUTestParams("apple-a9", "armv8-a", "crypto-neon-fp-armv8",
1081                          AArch64::AEK_NONE | AArch64::AEK_CRYPTO |
1082                              AArch64::AEK_FP | AArch64::AEK_SIMD,
1083                          "8-A"),
1084         ARMCPUTestParams("apple-a10", "armv8-a", "crypto-neon-fp-armv8",
1085                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1086                              AArch64::AEK_FP | AArch64::AEK_RDM |
1087                              AArch64::AEK_SIMD,
1088                          "8-A"),
1089         ARMCPUTestParams("apple-a11", "armv8.2-a", "crypto-neon-fp-armv8",
1090                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1091                              AArch64::AEK_FP | AArch64::AEK_LSE |
1092                              AArch64::AEK_RAS | AArch64::AEK_RDM |
1093                              AArch64::AEK_SIMD | AArch64::AEK_FP16,
1094                          "8.2-A"),
1095         ARMCPUTestParams("apple-a12", "armv8.3-a", "crypto-neon-fp-armv8",
1096                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1097                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1098                              AArch64::AEK_LSE | AArch64::AEK_RAS |
1099                              AArch64::AEK_RDM | AArch64::AEK_RCPC |
1100                              AArch64::AEK_FP16,
1101                          "8.3-A"),
1102         ARMCPUTestParams("apple-a13", "armv8.4-a", "crypto-neon-fp-armv8",
1103                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1104                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1105                              AArch64::AEK_LSE | AArch64::AEK_RAS |
1106                              AArch64::AEK_RDM | AArch64::AEK_RCPC |
1107                              AArch64::AEK_DOTPROD | AArch64::AEK_FP16 |
1108                              AArch64::AEK_FP16FML | AArch64::AEK_SHA3,
1109                          "8.4-A"),
1110         ARMCPUTestParams("apple-a14", "armv8.5-a", "crypto-neon-fp-armv8",
1111                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1112                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1113                              AArch64::AEK_LSE | AArch64::AEK_RAS |
1114                              AArch64::AEK_RDM | AArch64::AEK_RCPC |
1115                              AArch64::AEK_DOTPROD | AArch64::AEK_FP16 |
1116                              AArch64::AEK_FP16FML | AArch64::AEK_SHA3,
1117                          "8.5-A"),
1118         ARMCPUTestParams("apple-m1", "armv8.5-a", "crypto-neon-fp-armv8",
1119                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1120                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1121                              AArch64::AEK_LSE | AArch64::AEK_RAS |
1122                              AArch64::AEK_RDM | AArch64::AEK_RCPC |
1123                              AArch64::AEK_DOTPROD | AArch64::AEK_FP16 |
1124                              AArch64::AEK_FP16FML | AArch64::AEK_SHA3,
1125                          "8.5-A"),
1126         ARMCPUTestParams("apple-s4", "armv8.3-a", "crypto-neon-fp-armv8",
1127                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1128                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1129                              AArch64::AEK_LSE | AArch64::AEK_RAS |
1130                              AArch64::AEK_RDM | AArch64::AEK_RCPC |
1131                              AArch64::AEK_FP16,
1132                          "8.3-A"),
1133         ARMCPUTestParams("apple-s5", "armv8.3-a", "crypto-neon-fp-armv8",
1134                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1135                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1136                              AArch64::AEK_LSE | AArch64::AEK_RAS |
1137                              AArch64::AEK_RDM | AArch64::AEK_RCPC |
1138                              AArch64::AEK_FP16,
1139                          "8.3-A"),
1140         ARMCPUTestParams("exynos-m3", "armv8-a", "crypto-neon-fp-armv8",
1141                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1142                              AArch64::AEK_FP | AArch64::AEK_SIMD,
1143                          "8-A"),
1144         ARMCPUTestParams("exynos-m4", "armv8.2-a", "crypto-neon-fp-armv8",
1145                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1146                              AArch64::AEK_DOTPROD | AArch64::AEK_FP |
1147                              AArch64::AEK_FP16 | AArch64::AEK_LSE |
1148                              AArch64::AEK_RAS | AArch64::AEK_RDM |
1149                              AArch64::AEK_SIMD,
1150                          "8.2-A"),
1151         ARMCPUTestParams("exynos-m5", "armv8.2-a", "crypto-neon-fp-armv8",
1152                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1153                              AArch64::AEK_DOTPROD | AArch64::AEK_FP |
1154                              AArch64::AEK_FP16 | AArch64::AEK_LSE |
1155                              AArch64::AEK_RAS | AArch64::AEK_RDM |
1156                              AArch64::AEK_SIMD,
1157                          "8.2-A"),
1158         ARMCPUTestParams("falkor", "armv8-a", "crypto-neon-fp-armv8",
1159                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1160                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1161                              AArch64::AEK_RDM,
1162                          "8-A"),
1163         ARMCPUTestParams("kryo", "armv8-a", "crypto-neon-fp-armv8",
1164                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1165                              AArch64::AEK_FP | AArch64::AEK_SIMD,
1166                          "8-A"),
1167         ARMCPUTestParams("neoverse-e1", "armv8.2-a", "crypto-neon-fp-armv8",
1168                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1169                              AArch64::AEK_DOTPROD | AArch64::AEK_FP |
1170                              AArch64::AEK_FP16 | AArch64::AEK_LSE |
1171                              AArch64::AEK_RAS | AArch64::AEK_RCPC |
1172                              AArch64::AEK_RDM | AArch64::AEK_SIMD |
1173                              AArch64::AEK_SSBS,
1174                          "8.2-A"),
1175         ARMCPUTestParams("neoverse-n1", "armv8.2-a", "crypto-neon-fp-armv8",
1176                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1177                              AArch64::AEK_DOTPROD | AArch64::AEK_FP |
1178                              AArch64::AEK_FP16 | AArch64::AEK_LSE |
1179                              AArch64::AEK_PROFILE | AArch64::AEK_RAS |
1180                              AArch64::AEK_RCPC | AArch64::AEK_RDM |
1181                              AArch64::AEK_SIMD | AArch64::AEK_SSBS,
1182                          "8.2-A"),
1183         ARMCPUTestParams("neoverse-n2", "armv8.5-a", "crypto-neon-fp-armv8",
1184                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1185                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1186                              AArch64::AEK_FP16 | AArch64::AEK_RAS |
1187                              AArch64::AEK_LSE | AArch64::AEK_SVE |
1188                              AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
1189                              AArch64::AEK_RDM | AArch64::AEK_MTE |
1190                              AArch64::AEK_SSBS | AArch64::AEK_SB |
1191                              AArch64::AEK_SVE2 | AArch64::AEK_SVE2BITPERM |
1192                              AArch64::AEK_BF16 | AArch64::AEK_I8MM,
1193                          "8.5-A"),
1194         ARMCPUTestParams(
1195             "neoverse-512tvb", "armv8.4-a", "crypto-neon-fp-armv8",
1196             AArch64::AEK_RAS | AArch64::AEK_SVE | AArch64::AEK_SSBS |
1197                 AArch64::AEK_RCPC | AArch64::AEK_CRC | AArch64::AEK_FP |
1198                 AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
1199                 AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_DOTPROD |
1200                 AArch64::AEK_CRYPTO | AArch64::AEK_FP16 | AArch64::AEK_BF16 |
1201                 AArch64::AEK_PROFILE | AArch64::AEK_RAND |
1202                 AArch64::AEK_FP16FML | AArch64::AEK_I8MM,
1203             "8.4-A"),
1204         ARMCPUTestParams("thunderx2t99", "armv8.1-a", "crypto-neon-fp-armv8",
1205                          AArch64::AEK_NONE | AArch64::AEK_CRC |
1206                              AArch64::AEK_CRYPTO | AArch64::AEK_LSE |
1207                              AArch64::AEK_RDM | AArch64::AEK_FP |
1208                              AArch64::AEK_SIMD,
1209                          "8.1-A"),
1210         ARMCPUTestParams("thunderx3t110", "armv8.3-a", "crypto-neon-fp-armv8",
1211                          AArch64::AEK_NONE | AArch64::AEK_CRC |
1212                              AArch64::AEK_CRYPTO | AArch64::AEK_LSE |
1213                              AArch64::AEK_RDM | AArch64::AEK_FP |
1214                              AArch64::AEK_SIMD | AArch64::AEK_RAS |
1215                              AArch64::AEK_RCPC,
1216                          "8.3-A"),
1217         ARMCPUTestParams("thunderx", "armv8-a", "crypto-neon-fp-armv8",
1218                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1219                              AArch64::AEK_SIMD | AArch64::AEK_FP,
1220                          "8-A"),
1221         ARMCPUTestParams("thunderxt81", "armv8-a", "crypto-neon-fp-armv8",
1222                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1223                              AArch64::AEK_SIMD | AArch64::AEK_FP,
1224                          "8-A"),
1225         ARMCPUTestParams("thunderxt83", "armv8-a", "crypto-neon-fp-armv8",
1226                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1227                              AArch64::AEK_SIMD | AArch64::AEK_FP,
1228                          "8-A"),
1229         ARMCPUTestParams("thunderxt88", "armv8-a", "crypto-neon-fp-armv8",
1230                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1231                              AArch64::AEK_SIMD | AArch64::AEK_FP,
1232                          "8-A"),
1233         ARMCPUTestParams("tsv110", "armv8.2-a", "crypto-neon-fp-armv8",
1234                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1235                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1236                              AArch64::AEK_RAS | AArch64::AEK_LSE |
1237                              AArch64::AEK_RDM | AArch64::AEK_PROFILE |
1238                              AArch64::AEK_FP16 | AArch64::AEK_FP16FML |
1239                              AArch64::AEK_DOTPROD,
1240                          "8.2-A"),
1241         ARMCPUTestParams("a64fx", "armv8.2-a", "crypto-neon-fp-armv8",
1242                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1243                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1244                              AArch64::AEK_FP16 | AArch64::AEK_RAS |
1245                              AArch64::AEK_LSE | AArch64::AEK_SVE |
1246                              AArch64::AEK_RDM,
1247                          "8.2-A"),
1248         ARMCPUTestParams("carmel", "armv8.2-a", "crypto-neon-fp-armv8",
1249                          AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
1250                              AArch64::AEK_FP | AArch64::AEK_SIMD |
1251                              AArch64::AEK_FP16 | AArch64::AEK_RAS |
1252                              AArch64::AEK_LSE | AArch64::AEK_RDM,
1253                          "8.2-A")));
1254 
1255 static constexpr unsigned NumAArch64CPUArchs = 53;
1256 
1257 TEST(TargetParserTest, testAArch64CPUArchList) {
1258   SmallVector<StringRef, NumAArch64CPUArchs> List;
1259   AArch64::fillValidCPUArchList(List);
1260 
1261   // No list exists for these in this test suite, so ensure all are
1262   // valid, and match the expected 'magic' count.
1263   EXPECT_EQ(List.size(), NumAArch64CPUArchs);
1264   for(StringRef CPU : List) {
1265     EXPECT_NE(AArch64::parseCPUArch(CPU), AArch64::ArchKind::INVALID);
1266   }
1267 }
1268 
1269 bool testAArch64Arch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
1270                      unsigned ArchAttr) {
1271   AArch64::ArchKind AK = AArch64::parseArch(Arch);
1272   return (AK != AArch64::ArchKind::INVALID) &&
1273          AArch64::getDefaultCPU(Arch).equals(DefaultCPU) &&
1274          AArch64::getSubArch(AK).equals(SubArch) &&
1275          (AArch64::getArchAttr(AK) == ArchAttr);
1276 }
1277 
1278 TEST(TargetParserTest, testAArch64Arch) {
1279   EXPECT_TRUE(testAArch64Arch("armv8-a", "cortex-a53", "v8",
1280                               ARMBuildAttrs::CPUArch::v8_A));
1281   EXPECT_TRUE(testAArch64Arch("armv8.1-a", "generic", "v8.1a",
1282                               ARMBuildAttrs::CPUArch::v8_A));
1283   EXPECT_TRUE(testAArch64Arch("armv8.2-a", "generic", "v8.2a",
1284                               ARMBuildAttrs::CPUArch::v8_A));
1285   EXPECT_TRUE(testAArch64Arch("armv8.3-a", "generic", "v8.3a",
1286                               ARMBuildAttrs::CPUArch::v8_A));
1287   EXPECT_TRUE(testAArch64Arch("armv8.4-a", "generic", "v8.4a",
1288                               ARMBuildAttrs::CPUArch::v8_A));
1289   EXPECT_TRUE(testAArch64Arch("armv8.5-a", "generic", "v8.5a",
1290                               ARMBuildAttrs::CPUArch::v8_A));
1291   EXPECT_TRUE(testAArch64Arch("armv8.6-a", "generic", "v8.6a",
1292                               ARMBuildAttrs::CPUArch::v8_A));
1293   EXPECT_TRUE(testAArch64Arch("armv8.7-a", "generic", "v8.7a",
1294                               ARMBuildAttrs::CPUArch::v8_A));
1295   EXPECT_TRUE(testAArch64Arch("armv8.8-a", "generic", "v8.8a",
1296                               ARMBuildAttrs::CPUArch::v8_A));
1297   EXPECT_TRUE(testAArch64Arch("armv9-a", "generic", "v9a",
1298                               ARMBuildAttrs::CPUArch::v8_A));
1299   EXPECT_TRUE(testAArch64Arch("armv9.1-a", "generic", "v9.1a",
1300                               ARMBuildAttrs::CPUArch::v8_A));
1301   EXPECT_TRUE(testAArch64Arch("armv9.2-a", "generic", "v9.2a",
1302                               ARMBuildAttrs::CPUArch::v8_A));
1303 }
1304 
1305 bool testAArch64Extension(StringRef CPUName, AArch64::ArchKind AK,
1306                           StringRef ArchExt) {
1307   return AArch64::getDefaultExtensions(CPUName, AK) &
1308          AArch64::parseArchExt(ArchExt);
1309 }
1310 
1311 TEST(TargetParserTest, testAArch64Extension) {
1312   EXPECT_FALSE(testAArch64Extension("cortex-a34",
1313                                     AArch64::ArchKind::INVALID, "ras"));
1314   EXPECT_FALSE(testAArch64Extension("cortex-a35",
1315                                     AArch64::ArchKind::INVALID, "ras"));
1316   EXPECT_FALSE(testAArch64Extension("cortex-a53",
1317                                     AArch64::ArchKind::INVALID, "ras"));
1318   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1319                                     AArch64::ArchKind::INVALID, "ras"));
1320   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1321                                     AArch64::ArchKind::INVALID, "fp16"));
1322   EXPECT_FALSE(testAArch64Extension("cortex-a55",
1323                                     AArch64::ArchKind::INVALID, "fp16fml"));
1324   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1325                                     AArch64::ArchKind::INVALID, "dotprod"));
1326   EXPECT_FALSE(testAArch64Extension("cortex-a57",
1327                                     AArch64::ArchKind::INVALID, "ras"));
1328   EXPECT_FALSE(testAArch64Extension("cortex-a72",
1329                                     AArch64::ArchKind::INVALID, "ras"));
1330   EXPECT_FALSE(testAArch64Extension("cortex-a73",
1331                                     AArch64::ArchKind::INVALID, "ras"));
1332   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1333                                     AArch64::ArchKind::INVALID, "ras"));
1334   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1335                                     AArch64::ArchKind::INVALID, "fp16"));
1336   EXPECT_FALSE(testAArch64Extension("cortex-a75",
1337                                     AArch64::ArchKind::INVALID, "fp16fml"));
1338   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1339                                    AArch64::ArchKind::INVALID, "dotprod"));
1340   EXPECT_TRUE(testAArch64Extension("cortex-r82",
1341                                    AArch64::ArchKind::INVALID, "ras"));
1342   EXPECT_TRUE(testAArch64Extension("cortex-r82",
1343                                    AArch64::ArchKind::INVALID, "fp16"));
1344   EXPECT_TRUE(testAArch64Extension("cortex-r82",
1345                                    AArch64::ArchKind::INVALID, "fp16fml"));
1346   EXPECT_TRUE(testAArch64Extension("cortex-r82",
1347                                    AArch64::ArchKind::INVALID, "dotprod"));
1348   EXPECT_TRUE(testAArch64Extension("cortex-r82",
1349                                    AArch64::ArchKind::INVALID, "lse"));
1350   EXPECT_FALSE(testAArch64Extension("cyclone",
1351                                     AArch64::ArchKind::INVALID, "ras"));
1352   EXPECT_FALSE(testAArch64Extension("exynos-m3",
1353                                     AArch64::ArchKind::INVALID, "ras"));
1354   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1355                                    AArch64::ArchKind::INVALID, "dotprod"));
1356   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1357                                    AArch64::ArchKind::INVALID, "fp16"));
1358   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1359                                    AArch64::ArchKind::INVALID, "lse"));
1360   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1361                                    AArch64::ArchKind::INVALID, "ras"));
1362   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1363                                    AArch64::ArchKind::INVALID, "rdm"));
1364   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1365                                    AArch64::ArchKind::INVALID, "dotprod"));
1366   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1367                                    AArch64::ArchKind::INVALID, "fp16"));
1368   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1369                                    AArch64::ArchKind::INVALID, "lse"));
1370   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1371                                    AArch64::ArchKind::INVALID, "ras"));
1372   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1373                                    AArch64::ArchKind::INVALID, "rdm"));
1374   EXPECT_TRUE(testAArch64Extension("falkor",
1375                                    AArch64::ArchKind::INVALID, "rdm"));
1376   EXPECT_FALSE(testAArch64Extension("kryo",
1377                                     AArch64::ArchKind::INVALID, "ras"));
1378   EXPECT_TRUE(testAArch64Extension("saphira",
1379                                     AArch64::ArchKind::INVALID, "crc"));
1380   EXPECT_TRUE(testAArch64Extension("saphira",
1381                                     AArch64::ArchKind::INVALID, "lse"));
1382   EXPECT_TRUE(testAArch64Extension("saphira",
1383                                    AArch64::ArchKind::INVALID, "rdm"));
1384   EXPECT_TRUE(testAArch64Extension("saphira",
1385                                    AArch64::ArchKind::INVALID, "ras"));
1386   EXPECT_TRUE(testAArch64Extension("saphira",
1387                                    AArch64::ArchKind::INVALID, "rcpc"));
1388   EXPECT_TRUE(testAArch64Extension("saphira",
1389                                    AArch64::ArchKind::INVALID, "profile"));
1390   EXPECT_FALSE(testAArch64Extension("saphira",
1391                                     AArch64::ArchKind::INVALID, "fp16"));
1392   EXPECT_FALSE(testAArch64Extension("thunderx2t99",
1393                                     AArch64::ArchKind::INVALID, "ras"));
1394   EXPECT_FALSE(testAArch64Extension("thunderx",
1395                                     AArch64::ArchKind::INVALID, "lse"));
1396   EXPECT_FALSE(testAArch64Extension("thunderxt81",
1397                                     AArch64::ArchKind::INVALID, "lse"));
1398   EXPECT_FALSE(testAArch64Extension("thunderxt83",
1399                                     AArch64::ArchKind::INVALID, "lse"));
1400   EXPECT_FALSE(testAArch64Extension("thunderxt88",
1401                                     AArch64::ArchKind::INVALID, "lse"));
1402   EXPECT_TRUE(testAArch64Extension("tsv110",
1403                                    AArch64::ArchKind::INVALID, "crypto"));
1404   EXPECT_FALSE(testAArch64Extension("tsv110",
1405                                     AArch64::ArchKind::INVALID, "sha3"));
1406   EXPECT_FALSE(testAArch64Extension("tsv110",
1407                                     AArch64::ArchKind::INVALID, "sm4"));
1408   EXPECT_TRUE(testAArch64Extension("tsv110",
1409                                    AArch64::ArchKind::INVALID, "ras"));
1410   EXPECT_TRUE(testAArch64Extension("tsv110",
1411                                    AArch64::ArchKind::INVALID, "profile"));
1412   EXPECT_TRUE(testAArch64Extension("tsv110",
1413                                    AArch64::ArchKind::INVALID, "fp16"));
1414   EXPECT_TRUE(testAArch64Extension("tsv110",
1415                                    AArch64::ArchKind::INVALID, "fp16fml"));
1416   EXPECT_TRUE(testAArch64Extension("tsv110",
1417                                    AArch64::ArchKind::INVALID, "dotprod"));
1418   EXPECT_TRUE(testAArch64Extension("a64fx",
1419                                    AArch64::ArchKind::INVALID, "fp16"));
1420   EXPECT_TRUE(testAArch64Extension("a64fx",
1421                                    AArch64::ArchKind::INVALID, "sve"));
1422   EXPECT_FALSE(testAArch64Extension("a64fx",
1423                                    AArch64::ArchKind::INVALID, "sve2"));
1424   EXPECT_TRUE(
1425       testAArch64Extension("carmel", AArch64::ArchKind::INVALID, "crypto"));
1426   EXPECT_TRUE(
1427       testAArch64Extension("carmel", AArch64::ArchKind::INVALID, "fp16"));
1428 
1429   EXPECT_FALSE(testAArch64Extension(
1430       "generic", AArch64::ArchKind::ARMV8A, "ras"));
1431   EXPECT_FALSE(testAArch64Extension(
1432       "generic", AArch64::ArchKind::ARMV8_1A, "ras"));
1433   EXPECT_FALSE(testAArch64Extension(
1434       "generic", AArch64::ArchKind::ARMV8_2A, "profile"));
1435   EXPECT_FALSE(testAArch64Extension(
1436       "generic", AArch64::ArchKind::ARMV8_2A, "fp16"));
1437   EXPECT_FALSE(testAArch64Extension(
1438       "generic", AArch64::ArchKind::ARMV8_2A, "fp16fml"));
1439   EXPECT_FALSE(testAArch64Extension(
1440       "generic", AArch64::ArchKind::ARMV8_3A, "fp16"));
1441   EXPECT_FALSE(testAArch64Extension(
1442       "generic", AArch64::ArchKind::ARMV8_3A, "fp16fml"));
1443   EXPECT_FALSE(testAArch64Extension(
1444       "generic", AArch64::ArchKind::ARMV8_4A, "fp16"));
1445   EXPECT_FALSE(testAArch64Extension(
1446       "generic", AArch64::ArchKind::ARMV8_4A, "fp16fml"));
1447 }
1448 
1449 TEST(TargetParserTest, AArch64ExtensionFeatures) {
1450   std::vector<uint64_t> Extensions = {
1451       AArch64::AEK_CRC,         AArch64::AEK_CRYPTO,  AArch64::AEK_FP,
1452       AArch64::AEK_SIMD,        AArch64::AEK_FP16,    AArch64::AEK_PROFILE,
1453       AArch64::AEK_RAS,         AArch64::AEK_LSE,     AArch64::AEK_RDM,
1454       AArch64::AEK_DOTPROD,     AArch64::AEK_SVE,     AArch64::AEK_SVE2,
1455       AArch64::AEK_SVE2AES,     AArch64::AEK_SVE2SM4, AArch64::AEK_SVE2SHA3,
1456       AArch64::AEK_SVE2BITPERM, AArch64::AEK_RCPC,    AArch64::AEK_FP16FML,
1457       AArch64::AEK_SME,         AArch64::AEK_SMEF64,  AArch64::AEK_SMEI64,
1458       AArch64::AEK_PERFMON};
1459 
1460   std::vector<StringRef> Features;
1461 
1462   uint64_t ExtVal = 0;
1463   for (auto Ext : Extensions)
1464     ExtVal |= Ext;
1465 
1466   EXPECT_FALSE(AArch64::getExtensionFeatures(AArch64::AEK_INVALID, Features));
1467   EXPECT_TRUE(!Features.size());
1468 
1469   AArch64::getExtensionFeatures(ExtVal, Features);
1470   EXPECT_EQ(Extensions.size(), Features.size());
1471 
1472   EXPECT_TRUE(llvm::is_contained(Features, "+crc"));
1473   EXPECT_TRUE(llvm::is_contained(Features, "+crypto"));
1474   EXPECT_TRUE(llvm::is_contained(Features, "+fp-armv8"));
1475   EXPECT_TRUE(llvm::is_contained(Features, "+neon"));
1476   EXPECT_TRUE(llvm::is_contained(Features, "+fullfp16"));
1477   EXPECT_TRUE(llvm::is_contained(Features, "+spe"));
1478   EXPECT_TRUE(llvm::is_contained(Features, "+ras"));
1479   EXPECT_TRUE(llvm::is_contained(Features, "+lse"));
1480   EXPECT_TRUE(llvm::is_contained(Features, "+rdm"));
1481   EXPECT_TRUE(llvm::is_contained(Features, "+dotprod"));
1482   EXPECT_TRUE(llvm::is_contained(Features, "+rcpc"));
1483   EXPECT_TRUE(llvm::is_contained(Features, "+fp16fml"));
1484   EXPECT_TRUE(llvm::is_contained(Features, "+sve"));
1485   EXPECT_TRUE(llvm::is_contained(Features, "+sve2"));
1486   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-aes"));
1487   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-sm4"));
1488   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-sha3"));
1489   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-bitperm"));
1490   EXPECT_TRUE(llvm::is_contained(Features, "+sme"));
1491   EXPECT_TRUE(llvm::is_contained(Features, "+sme-f64"));
1492   EXPECT_TRUE(llvm::is_contained(Features, "+sme-i64"));
1493   EXPECT_TRUE(llvm::is_contained(Features, "+perfmon"));
1494 }
1495 
1496 TEST(TargetParserTest, AArch64ArchFeatures) {
1497   std::vector<StringRef> Features;
1498 
1499   for (auto AK : AArch64::ArchKinds) {
1500     if (AK == AArch64::ArchKind::INVALID)
1501       EXPECT_FALSE(AArch64::getArchFeatures(AK, Features));
1502     else
1503       EXPECT_TRUE(AArch64::getArchFeatures(AK, Features));
1504   }
1505 }
1506 
1507 TEST(TargetParserTest, AArch64ArchExtFeature) {
1508   const char *ArchExt[][4] = {
1509       {"crc", "nocrc", "+crc", "-crc"},
1510       {"crypto", "nocrypto", "+crypto", "-crypto"},
1511       {"flagm", "noflagm", "+flagm", "-flagm"},
1512       {"fp", "nofp", "+fp-armv8", "-fp-armv8"},
1513       {"simd", "nosimd", "+neon", "-neon"},
1514       {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
1515       {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
1516       {"profile", "noprofile", "+spe", "-spe"},
1517       {"ras", "noras", "+ras", "-ras"},
1518       {"lse", "nolse", "+lse", "-lse"},
1519       {"rdm", "nordm", "+rdm", "-rdm"},
1520       {"sve", "nosve", "+sve", "-sve"},
1521       {"sve2", "nosve2", "+sve2", "-sve2"},
1522       {"sve2-aes", "nosve2-aes", "+sve2-aes", "-sve2-aes"},
1523       {"sve2-sm4", "nosve2-sm4", "+sve2-sm4", "-sve2-sm4"},
1524       {"sve2-sha3", "nosve2-sha3", "+sve2-sha3", "-sve2-sha3"},
1525       {"sve2-bitperm", "nosve2-bitperm", "+sve2-bitperm", "-sve2-bitperm"},
1526       {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
1527       {"rcpc", "norcpc", "+rcpc", "-rcpc"},
1528       {"rng", "norng", "+rand", "-rand"},
1529       {"memtag", "nomemtag", "+mte", "-mte"},
1530       {"tme", "notme", "+tme", "-tme"},
1531       {"pauth", "nopauth", "+pauth", "-pauth"},
1532       {"ssbs", "nossbs", "+ssbs", "-ssbs"},
1533       {"sb", "nosb", "+sb", "-sb"},
1534       {"predres", "nopredres", "+predres", "-predres"},
1535       {"i8mm", "noi8mm", "+i8mm", "-i8mm"},
1536       {"f32mm", "nof32mm", "+f32mm", "-f32mm"},
1537       {"f64mm", "nof64mm", "+f64mm", "-f64mm"},
1538       {"sme", "nosme", "+sme", "-sme"},
1539       {"sme-f64", "nosme-f64", "+sme-f64", "-sme-f64"},
1540       {"sme-i64", "nosme-i64", "+sme-i64", "-sme-i64"},
1541       {"hbc", "nohbc", "+hbc", "-hbc"},
1542       {"mops", "nomops", "+mops", "-mops"},
1543       {"pmuv3", "nopmuv3", "+perfmon", "-perfmon"},
1544   };
1545 
1546   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
1547     EXPECT_EQ(StringRef(ArchExt[i][2]),
1548               AArch64::getArchExtFeature(ArchExt[i][0]));
1549     EXPECT_EQ(StringRef(ArchExt[i][3]),
1550               AArch64::getArchExtFeature(ArchExt[i][1]));
1551   }
1552 }
1553 
1554 } // namespace
1555