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