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