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/Support/ARMBuildAttributes.h"
12 #include "gtest/gtest.h"
13 #include <string>
14 
15 using namespace llvm;
16 
17 namespace {
18 const char *ARMArch[] = {
19     "armv2",       "armv2a",       "armv3",       "armv3m",       "armv4",
20     "armv4t",      "armv5",        "armv5t",      "armv5e",       "armv5te",
21     "armv5tej",    "armv6",        "armv6j",      "armv6k",       "armv6hl",
22     "armv6t2",     "armv6kz",      "armv6z",      "armv6zk",      "armv6-m",
23     "armv6m",      "armv6sm",      "armv6s-m",    "armv7-a",      "armv7",
24     "armv7a",      "armv7ve",      "armv7hl",     "armv7l",       "armv7-r",
25     "armv7r",      "armv7-m",      "armv7m",      "armv7k",       "armv7s",
26     "armv7e-m",    "armv7em",      "armv8-a",     "armv8",        "armv8a",
27     "armv8l",      "armv8.1-a",    "armv8.1a",    "armv8.2-a",    "armv8.2a",
28     "armv8.3-a",   "armv8.3a",     "armv8.4-a",   "armv8.4a",     "armv8.5-a",
29     "armv8.5a",     "armv8-r",     "armv8r",      "armv8-m.base", "armv8m.base",
30     "armv8-m.main", "armv8m.main", "iwmmxt",      "iwmmxt2",      "xscale",
31     "armv8.1-m.main",
32 };
33 
34 bool testARMCPU(StringRef CPUName, StringRef ExpectedArch,
35                 StringRef ExpectedFPU, unsigned ExpectedFlags,
36                 StringRef CPUAttr) {
37   ARM::ArchKind AK = ARM::parseCPUArch(CPUName);
38   bool pass = ARM::getArchName(AK).equals(ExpectedArch);
39   unsigned FPUKind = ARM::getDefaultFPU(CPUName, AK);
40   pass &= ARM::getFPUName(FPUKind).equals(ExpectedFPU);
41 
42   unsigned ExtKind = ARM::getDefaultExtensions(CPUName, AK);
43   if (ExtKind > 1 && (ExtKind & ARM::AEK_NONE))
44     pass &= ((ExtKind ^ ARM::AEK_NONE) == ExpectedFlags);
45   else
46     pass &= (ExtKind == ExpectedFlags);
47   pass &= ARM::getCPUAttr(AK).equals(CPUAttr);
48 
49   return pass;
50 }
51 
52 TEST(TargetParserTest, testARMCPU) {
53   EXPECT_TRUE(testARMCPU("invalid", "invalid", "invalid",
54                          ARM::AEK_NONE, ""));
55   EXPECT_TRUE(testARMCPU("generic", "invalid", "none",
56                          ARM::AEK_NONE, ""));
57 
58   EXPECT_TRUE(testARMCPU("arm2", "armv2", "none",
59                          ARM::AEK_NONE, "2"));
60   EXPECT_TRUE(testARMCPU("arm3", "armv2a", "none",
61                          ARM::AEK_NONE, "2A"));
62   EXPECT_TRUE(testARMCPU("arm6", "armv3", "none",
63                          ARM::AEK_NONE, "3"));
64   EXPECT_TRUE(testARMCPU("arm7m", "armv3m", "none",
65                          ARM::AEK_NONE, "3M"));
66   EXPECT_TRUE(testARMCPU("arm8", "armv4", "none",
67                          ARM::AEK_NONE, "4"));
68   EXPECT_TRUE(testARMCPU("arm810", "armv4", "none",
69                          ARM::AEK_NONE, "4"));
70   EXPECT_TRUE(testARMCPU("strongarm", "armv4", "none",
71                          ARM::AEK_NONE, "4"));
72   EXPECT_TRUE(testARMCPU("strongarm110", "armv4", "none",
73                          ARM::AEK_NONE, "4"));
74   EXPECT_TRUE(testARMCPU("strongarm1100", "armv4", "none",
75                          ARM::AEK_NONE, "4"));
76   EXPECT_TRUE(testARMCPU("strongarm1110", "armv4", "none",
77                          ARM::AEK_NONE, "4"));
78   EXPECT_TRUE(testARMCPU("arm7tdmi", "armv4t", "none",
79                          ARM::AEK_NONE, "4T"));
80   EXPECT_TRUE(testARMCPU("arm7tdmi-s", "armv4t", "none",
81                          ARM::AEK_NONE, "4T"));
82   EXPECT_TRUE(testARMCPU("arm710t", "armv4t", "none",
83                          ARM::AEK_NONE, "4T"));
84   EXPECT_TRUE(testARMCPU("arm720t", "armv4t", "none",
85                          ARM::AEK_NONE, "4T"));
86   EXPECT_TRUE(testARMCPU("arm9", "armv4t", "none",
87                          ARM::AEK_NONE, "4T"));
88   EXPECT_TRUE(testARMCPU("arm9tdmi", "armv4t", "none",
89                          ARM::AEK_NONE, "4T"));
90   EXPECT_TRUE(testARMCPU("arm920", "armv4t", "none",
91                          ARM::AEK_NONE, "4T"));
92   EXPECT_TRUE(testARMCPU("arm920t", "armv4t", "none",
93                          ARM::AEK_NONE, "4T"));
94   EXPECT_TRUE(testARMCPU("arm922t", "armv4t", "none",
95                          ARM::AEK_NONE, "4T"));
96   EXPECT_TRUE(testARMCPU("arm9312", "armv4t", "none",
97                          ARM::AEK_NONE, "4T"));
98   EXPECT_TRUE(testARMCPU("arm940t", "armv4t", "none",
99                          ARM::AEK_NONE, "4T"));
100   EXPECT_TRUE(testARMCPU("ep9312", "armv4t", "none",
101                          ARM::AEK_NONE, "4T"));
102   EXPECT_TRUE(testARMCPU("arm10tdmi", "armv5t", "none",
103                          ARM::AEK_NONE, "5T"));
104   EXPECT_TRUE(testARMCPU("arm1020t", "armv5t", "none",
105                          ARM::AEK_NONE, "5T"));
106   EXPECT_TRUE(testARMCPU("arm9e", "armv5te", "none",
107                          ARM::AEK_DSP, "5TE"));
108   EXPECT_TRUE(testARMCPU("arm946e-s", "armv5te", "none",
109                          ARM::AEK_DSP, "5TE"));
110   EXPECT_TRUE(testARMCPU("arm966e-s", "armv5te", "none",
111                          ARM::AEK_DSP, "5TE"));
112   EXPECT_TRUE(testARMCPU("arm968e-s", "armv5te", "none",
113                          ARM::AEK_DSP, "5TE"));
114   EXPECT_TRUE(testARMCPU("arm10e", "armv5te", "none",
115                          ARM::AEK_DSP, "5TE"));
116   EXPECT_TRUE(testARMCPU("arm1020e", "armv5te", "none",
117                          ARM::AEK_DSP, "5TE"));
118   EXPECT_TRUE(testARMCPU("arm1022e", "armv5te", "none",
119                          ARM::AEK_DSP, "5TE"));
120   EXPECT_TRUE(testARMCPU("arm926ej-s", "armv5tej", "none",
121                          ARM::AEK_DSP, "5TEJ"));
122   EXPECT_TRUE(testARMCPU("arm1136j-s", "armv6", "none",
123                          ARM::AEK_DSP, "6"));
124   EXPECT_TRUE(testARMCPU("arm1136jf-s", "armv6", "vfpv2",
125                          ARM::AEK_DSP, "6"));
126   EXPECT_TRUE(testARMCPU("arm1136jz-s", "armv6", "none",
127                          ARM::AEK_DSP, "6"));
128   EXPECT_TRUE(testARMCPU("arm1176jz-s", "armv6kz", "none",
129                          ARM::AEK_SEC | ARM::AEK_DSP, "6KZ"));
130   EXPECT_TRUE(testARMCPU("mpcore", "armv6k", "vfpv2",
131                          ARM::AEK_DSP, "6K"));
132   EXPECT_TRUE(testARMCPU("mpcorenovfp", "armv6k", "none",
133                          ARM::AEK_DSP, "6K"));
134   EXPECT_TRUE(testARMCPU("arm1176jzf-s", "armv6kz", "vfpv2",
135                          ARM::AEK_SEC | ARM::AEK_DSP, "6KZ"));
136   EXPECT_TRUE(testARMCPU("arm1156t2-s", "armv6t2", "none",
137                          ARM::AEK_DSP, "6T2"));
138   EXPECT_TRUE(testARMCPU("arm1156t2f-s", "armv6t2", "vfpv2",
139                          ARM::AEK_DSP, "6T2"));
140   EXPECT_TRUE(testARMCPU("cortex-m0", "armv6-m", "none",
141                          ARM::AEK_NONE, "6-M"));
142   EXPECT_TRUE(testARMCPU("cortex-m0plus", "armv6-m", "none",
143                          ARM::AEK_NONE, "6-M"));
144   EXPECT_TRUE(testARMCPU("cortex-m1", "armv6-m", "none",
145                          ARM::AEK_NONE, "6-M"));
146   EXPECT_TRUE(testARMCPU("sc000", "armv6-m", "none",
147                          ARM::AEK_NONE, "6-M"));
148   EXPECT_TRUE(testARMCPU("cortex-a5", "armv7-a", "neon-vfpv4",
149                          ARM::AEK_MP | ARM::AEK_SEC | ARM::AEK_DSP, "7-A"));
150   EXPECT_TRUE(testARMCPU("cortex-a7", "armv7-a", "neon-vfpv4",
151                          ARM::AEK_HWDIVTHUMB | ARM::AEK_HWDIVARM | ARM::AEK_MP |
152                              ARM::AEK_SEC | ARM::AEK_VIRT | ARM::AEK_DSP,
153                          "7-A"));
154   EXPECT_TRUE(testARMCPU("cortex-a8", "armv7-a", "neon",
155                          ARM::AEK_SEC | ARM::AEK_DSP, "7-A"));
156   EXPECT_TRUE(testARMCPU("cortex-a9", "armv7-a", "neon-fp16",
157                          ARM::AEK_MP | ARM::AEK_SEC | ARM::AEK_DSP, "7-A"));
158   EXPECT_TRUE(testARMCPU("cortex-a12", "armv7-a", "neon-vfpv4",
159                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
160                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
161                              ARM::AEK_DSP,
162                          "7-A"));
163   EXPECT_TRUE(testARMCPU("cortex-a15", "armv7-a", "neon-vfpv4",
164                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
165                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
166                              ARM::AEK_DSP,
167                          "7-A"));
168   EXPECT_TRUE(testARMCPU("cortex-a17", "armv7-a", "neon-vfpv4",
169                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
170                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
171                              ARM::AEK_DSP,
172                          "7-A"));
173   EXPECT_TRUE(testARMCPU("krait", "armv7-a", "neon-vfpv4",
174                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
175                          "7-A"));
176   EXPECT_TRUE(testARMCPU("cortex-r4", "armv7-r", "none",
177                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7-R"));
178   EXPECT_TRUE(testARMCPU("cortex-r4f", "armv7-r", "vfpv3-d16",
179                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7-R"));
180   EXPECT_TRUE(testARMCPU("cortex-r5", "armv7-r", "vfpv3-d16",
181                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
182                              ARM::AEK_DSP,
183                          "7-R"));
184   EXPECT_TRUE(testARMCPU("cortex-r7", "armv7-r", "vfpv3-d16-fp16",
185                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
186                              ARM::AEK_DSP,
187                          "7-R"));
188   EXPECT_TRUE(testARMCPU("cortex-r8", "armv7-r", "vfpv3-d16-fp16",
189                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
190                              ARM::AEK_DSP,
191                          "7-R"));
192   EXPECT_TRUE(testARMCPU("cortex-r52", "armv8-r", "neon-fp-armv8",
193                          ARM::AEK_CRC | ARM::AEK_MP | ARM::AEK_VIRT |
194                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
195                              ARM::AEK_DSP,
196                          "8-R"));
197   EXPECT_TRUE(
198       testARMCPU("sc300", "armv7-m", "none", ARM::AEK_HWDIVTHUMB, "7-M"));
199   EXPECT_TRUE(
200       testARMCPU("cortex-m3", "armv7-m", "none", ARM::AEK_HWDIVTHUMB, "7-M"));
201   EXPECT_TRUE(testARMCPU("cortex-m4", "armv7e-m", "fpv4-sp-d16",
202                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7E-M"));
203   EXPECT_TRUE(testARMCPU("cortex-m7", "armv7e-m", "fpv5-d16",
204                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7E-M"));
205   EXPECT_TRUE(testARMCPU("cortex-a32", "armv8-a", "crypto-neon-fp-armv8",
206                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
207                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
208                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
209                          "8-A"));
210   EXPECT_TRUE(testARMCPU("cortex-a35", "armv8-a", "crypto-neon-fp-armv8",
211                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
212                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
213                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
214                          "8-A"));
215   EXPECT_TRUE(testARMCPU("cortex-a53", "armv8-a", "crypto-neon-fp-armv8",
216                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
217                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
218                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
219                          "8-A"));
220   EXPECT_TRUE(testARMCPU("cortex-a55", "armv8.2-a", "crypto-neon-fp-armv8",
221                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
222                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
223                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
224                          ARM::AEK_RAS | ARM::AEK_DOTPROD,
225                          "8.2-A"));
226   EXPECT_TRUE(testARMCPU("cortex-a57", "armv8-a", "crypto-neon-fp-armv8",
227                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
228                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
229                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
230                          "8-A"));
231   EXPECT_TRUE(testARMCPU("cortex-a72", "armv8-a", "crypto-neon-fp-armv8",
232                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
233                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
234                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
235                          "8-A"));
236   EXPECT_TRUE(testARMCPU("cortex-a73", "armv8-a", "crypto-neon-fp-armv8",
237                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
238                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
239                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
240                          "8-A"));
241   EXPECT_TRUE(testARMCPU("cortex-a75", "armv8.2-a", "crypto-neon-fp-armv8",
242                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
243                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
244                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
245                          ARM::AEK_RAS | ARM::AEK_DOTPROD,
246                          "8.2-A"));
247   EXPECT_TRUE(testARMCPU("cortex-a76", "armv8.2-a", "crypto-neon-fp-armv8",
248                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
249                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
250                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
251                          ARM::AEK_RAS | ARM::AEK_DOTPROD,
252                          "8.2-A"));
253   EXPECT_TRUE(testARMCPU("cortex-a76ae", "armv8.2-a", "crypto-neon-fp-armv8",
254                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
255                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
256                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
257                          ARM::AEK_RAS | ARM::AEK_DOTPROD,
258                          "8.2-A"));
259   EXPECT_TRUE(testARMCPU("neoverse-n1", "armv8.2-a", "crypto-neon-fp-armv8",
260                         ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
261                         ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
262                         ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
263                         ARM::AEK_RAS | ARM::AEK_DOTPROD,
264                         "8.2-A"));
265   EXPECT_TRUE(testARMCPU("cyclone", "armv8-a", "crypto-neon-fp-armv8",
266                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
267                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
268                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
269                          "8-A"));
270   EXPECT_TRUE(testARMCPU("exynos-m3", "armv8-a", "crypto-neon-fp-armv8",
271                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
272                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
273                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
274                          "8-A"));
275   EXPECT_TRUE(testARMCPU("exynos-m4", "armv8.2-a", "crypto-neon-fp-armv8",
276                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
277                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
278                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_DOTPROD |
279                          ARM::AEK_FP16 | ARM::AEK_RAS,
280                          "8.2-A"));
281   EXPECT_TRUE(testARMCPU("exynos-m5", "armv8.2-a", "crypto-neon-fp-armv8",
282                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
283                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
284                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_DOTPROD |
285                          ARM::AEK_FP16 | ARM::AEK_RAS,
286                          "8.2-A"));
287   EXPECT_TRUE(testARMCPU("cortex-m23", "armv8-m.base", "none",
288                          ARM::AEK_HWDIVTHUMB, "8-M.Baseline"));
289   EXPECT_TRUE(testARMCPU("cortex-m33", "armv8-m.main", "fpv5-sp-d16",
290                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "8-M.Mainline"));
291   EXPECT_TRUE(testARMCPU("cortex-m35p", "armv8-m.main", "fpv5-sp-d16",
292                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "8-M.Mainline"));
293   EXPECT_TRUE(testARMCPU("iwmmxt", "iwmmxt", "none",
294                          ARM::AEK_NONE, "iwmmxt"));
295   EXPECT_TRUE(testARMCPU("xscale", "xscale", "none",
296                          ARM::AEK_NONE, "xscale"));
297   EXPECT_TRUE(testARMCPU("swift", "armv7s", "neon-vfpv4",
298                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
299                          "7-S"));
300 }
301 
302 static constexpr unsigned NumARMCPUArchs = 85;
303 
304 TEST(TargetParserTest, testARMCPUArchList) {
305   SmallVector<StringRef, NumARMCPUArchs> List;
306   ARM::fillValidCPUArchList(List);
307 
308   // No list exists for these in this test suite, so ensure all are
309   // valid, and match the expected 'magic' count.
310   EXPECT_EQ(List.size(), NumARMCPUArchs);
311   for(StringRef CPU : List) {
312     EXPECT_NE(ARM::parseCPUArch(CPU), ARM::ArchKind::INVALID);
313   }
314 }
315 
316 TEST(TargetParserTest, testInvalidARMArch) {
317   auto InvalidArchStrings = {"armv", "armv99", "noarm"};
318   for (const char* InvalidArch : InvalidArchStrings)
319     EXPECT_EQ(ARM::parseArch(InvalidArch), ARM::ArchKind::INVALID);
320 }
321 
322 bool testARMArch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
323                  unsigned ArchAttr) {
324   ARM::ArchKind AK = ARM::parseArch(Arch);
325   bool Result = (AK != ARM::ArchKind::INVALID);
326   Result &= ARM::getDefaultCPU(Arch).equals(DefaultCPU);
327   Result &= ARM::getSubArch(AK).equals(SubArch);
328   Result &= (ARM::getArchAttr(AK) == ArchAttr);
329   return Result;
330 }
331 
332 TEST(TargetParserTest, testARMArch) {
333   EXPECT_TRUE(
334       testARMArch("armv2", "arm2", "v2",
335                           ARMBuildAttrs::CPUArch::Pre_v4));
336   EXPECT_TRUE(
337       testARMArch("armv2a", "arm3", "v2a",
338                           ARMBuildAttrs::CPUArch::Pre_v4));
339   EXPECT_TRUE(
340       testARMArch("armv3", "arm6", "v3",
341                           ARMBuildAttrs::CPUArch::Pre_v4));
342   EXPECT_TRUE(
343       testARMArch("armv3m", "arm7m", "v3m",
344                           ARMBuildAttrs::CPUArch::Pre_v4));
345   EXPECT_TRUE(
346       testARMArch("armv4", "strongarm", "v4",
347                           ARMBuildAttrs::CPUArch::v4));
348   EXPECT_TRUE(
349       testARMArch("armv4t", "arm7tdmi", "v4t",
350                           ARMBuildAttrs::CPUArch::v4T));
351   EXPECT_TRUE(
352       testARMArch("armv5t", "arm10tdmi", "v5",
353                           ARMBuildAttrs::CPUArch::v5T));
354   EXPECT_TRUE(
355       testARMArch("armv5te", "arm1022e", "v5e",
356                           ARMBuildAttrs::CPUArch::v5TE));
357   EXPECT_TRUE(
358       testARMArch("armv5tej", "arm926ej-s", "v5e",
359                           ARMBuildAttrs::CPUArch::v5TEJ));
360   EXPECT_TRUE(
361       testARMArch("armv6", "arm1136jf-s", "v6",
362                           ARMBuildAttrs::CPUArch::v6));
363   EXPECT_TRUE(
364       testARMArch("armv6k", "mpcore", "v6k",
365                           ARMBuildAttrs::CPUArch::v6K));
366   EXPECT_TRUE(
367       testARMArch("armv6t2", "arm1156t2-s", "v6t2",
368                           ARMBuildAttrs::CPUArch::v6T2));
369   EXPECT_TRUE(
370       testARMArch("armv6kz", "arm1176jzf-s", "v6kz",
371                           ARMBuildAttrs::CPUArch::v6KZ));
372   EXPECT_TRUE(
373       testARMArch("armv6-m", "cortex-m0", "v6m",
374                           ARMBuildAttrs::CPUArch::v6_M));
375   EXPECT_TRUE(
376       testARMArch("armv7-a", "generic", "v7",
377                           ARMBuildAttrs::CPUArch::v7));
378   EXPECT_TRUE(
379       testARMArch("armv7ve", "generic", "v7ve",
380                           ARMBuildAttrs::CPUArch::v7));
381   EXPECT_TRUE(
382       testARMArch("armv7-r", "cortex-r4", "v7r",
383                           ARMBuildAttrs::CPUArch::v7));
384   EXPECT_TRUE(
385       testARMArch("armv7-m", "cortex-m3", "v7m",
386                           ARMBuildAttrs::CPUArch::v7));
387   EXPECT_TRUE(
388       testARMArch("armv7e-m", "cortex-m4", "v7em",
389                           ARMBuildAttrs::CPUArch::v7E_M));
390   EXPECT_TRUE(
391       testARMArch("armv8-a", "generic", "v8",
392                           ARMBuildAttrs::CPUArch::v8_A));
393   EXPECT_TRUE(
394       testARMArch("armv8.1-a", "generic", "v8.1a",
395                           ARMBuildAttrs::CPUArch::v8_A));
396   EXPECT_TRUE(
397       testARMArch("armv8.2-a", "generic", "v8.2a",
398                           ARMBuildAttrs::CPUArch::v8_A));
399   EXPECT_TRUE(
400       testARMArch("armv8.3-a", "generic", "v8.3a",
401                           ARMBuildAttrs::CPUArch::v8_A));
402   EXPECT_TRUE(
403       testARMArch("armv8.4-a", "generic", "v8.4a",
404                           ARMBuildAttrs::CPUArch::v8_A));
405   EXPECT_TRUE(
406       testARMArch("armv8.5-a", "generic", "v8.5a",
407                           ARMBuildAttrs::CPUArch::v8_A));
408   EXPECT_TRUE(
409       testARMArch("armv8-r", "cortex-r52", "v8r",
410                           ARMBuildAttrs::CPUArch::v8_R));
411   EXPECT_TRUE(
412       testARMArch("armv8-m.base", "generic", "v8m.base",
413                           ARMBuildAttrs::CPUArch::v8_M_Base));
414   EXPECT_TRUE(
415       testARMArch("armv8-m.main", "generic", "v8m.main",
416                           ARMBuildAttrs::CPUArch::v8_M_Main));
417   EXPECT_TRUE(
418       testARMArch("armv8.1-m.main", "generic", "v8.1m.main",
419                           ARMBuildAttrs::CPUArch::v8_1_M_Main));
420   EXPECT_TRUE(
421       testARMArch("iwmmxt", "iwmmxt", "",
422                           ARMBuildAttrs::CPUArch::v5TE));
423   EXPECT_TRUE(
424       testARMArch("iwmmxt2", "generic", "",
425                           ARMBuildAttrs::CPUArch::v5TE));
426   EXPECT_TRUE(
427       testARMArch("xscale", "xscale", "v5e",
428                           ARMBuildAttrs::CPUArch::v5TE));
429   EXPECT_TRUE(
430       testARMArch("armv7s", "swift", "v7s",
431                           ARMBuildAttrs::CPUArch::v7));
432   EXPECT_TRUE(
433       testARMArch("armv7k", "generic", "v7k",
434                           ARMBuildAttrs::CPUArch::v7));
435 }
436 
437 bool testARMExtension(StringRef CPUName,ARM::ArchKind ArchKind, StringRef ArchExt) {
438   return ARM::getDefaultExtensions(CPUName, ArchKind) &
439          ARM::parseArchExt(ArchExt);
440 }
441 
442 TEST(TargetParserTest, testARMExtension) {
443   EXPECT_FALSE(testARMExtension("arm2", ARM::ArchKind::INVALID, "thumb"));
444   EXPECT_FALSE(testARMExtension("arm3", ARM::ArchKind::INVALID, "thumb"));
445   EXPECT_FALSE(testARMExtension("arm6", ARM::ArchKind::INVALID, "thumb"));
446   EXPECT_FALSE(testARMExtension("arm7m", ARM::ArchKind::INVALID, "thumb"));
447   EXPECT_FALSE(testARMExtension("strongarm", ARM::ArchKind::INVALID, "dsp"));
448   EXPECT_FALSE(testARMExtension("arm7tdmi", ARM::ArchKind::INVALID, "dsp"));
449   EXPECT_FALSE(testARMExtension("arm10tdmi",
450                                 ARM::ArchKind::INVALID, "simd"));
451   EXPECT_FALSE(testARMExtension("arm1022e", ARM::ArchKind::INVALID, "simd"));
452   EXPECT_FALSE(testARMExtension("arm926ej-s",
453                                 ARM::ArchKind::INVALID, "simd"));
454   EXPECT_FALSE(testARMExtension("arm1136jf-s",
455                                 ARM::ArchKind::INVALID, "crypto"));
456   EXPECT_FALSE(testARMExtension("arm1176j-s",
457                                 ARM::ArchKind::INVALID, "crypto"));
458   EXPECT_FALSE(testARMExtension("arm1156t2-s",
459                                 ARM::ArchKind::INVALID, "crypto"));
460   EXPECT_FALSE(testARMExtension("arm1176jzf-s",
461                                 ARM::ArchKind::INVALID, "crypto"));
462   EXPECT_FALSE(testARMExtension("cortex-m0",
463                                 ARM::ArchKind::INVALID, "crypto"));
464   EXPECT_FALSE(testARMExtension("cortex-a8",
465                                 ARM::ArchKind::INVALID, "crypto"));
466   EXPECT_FALSE(testARMExtension("cortex-r4",
467                                 ARM::ArchKind::INVALID, "crypto"));
468   EXPECT_FALSE(testARMExtension("cortex-m3",
469                                 ARM::ArchKind::INVALID, "crypto"));
470   EXPECT_FALSE(testARMExtension("cortex-a53",
471                                 ARM::ArchKind::INVALID, "ras"));
472   EXPECT_FALSE(testARMExtension("cortex-a53",
473                                 ARM::ArchKind::INVALID, "fp16"));
474   EXPECT_TRUE(testARMExtension("cortex-a55",
475                                 ARM::ArchKind::INVALID, "fp16"));
476   EXPECT_FALSE(testARMExtension("cortex-a55",
477                                 ARM::ArchKind::INVALID, "fp16fml"));
478   EXPECT_TRUE(testARMExtension("cortex-a75",
479                                 ARM::ArchKind::INVALID, "fp16"));
480   EXPECT_FALSE(testARMExtension("cortex-a75",
481                                 ARM::ArchKind::INVALID, "fp16fml"));
482   EXPECT_FALSE(testARMExtension("cortex-r52",
483                                 ARM::ArchKind::INVALID, "ras"));
484   EXPECT_FALSE(testARMExtension("iwmmxt", ARM::ArchKind::INVALID, "crc"));
485   EXPECT_FALSE(testARMExtension("xscale", ARM::ArchKind::INVALID, "crc"));
486   EXPECT_FALSE(testARMExtension("swift", ARM::ArchKind::INVALID, "crc"));
487 
488   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2, "thumb"));
489   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2A, "thumb"));
490   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3, "thumb"));
491   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3M, "thumb"));
492   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4, "dsp"));
493   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4T, "dsp"));
494   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5T, "simd"));
495   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TE, "simd"));
496   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TEJ, "simd"));
497   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6, "crypto"));
498   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6K, "crypto"));
499   EXPECT_FALSE(testARMExtension("generic",
500                                 ARM::ArchKind::ARMV6T2, "crypto"));
501   EXPECT_FALSE(testARMExtension("generic",
502                                 ARM::ArchKind::ARMV6KZ, "crypto"));
503   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6M, "crypto"));
504   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7A, "crypto"));
505   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7R, "crypto"));
506   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7M, "crypto"));
507   EXPECT_FALSE(testARMExtension("generic",
508                                 ARM::ArchKind::ARMV7EM, "crypto"));
509   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8A, "ras"));
510   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_1A, "ras"));
511   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "profile"));
512   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16"));
513   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16fml"));
514   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16"));
515   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16fml"));
516   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16"));
517   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16fml"));
518   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8R, "ras"));
519   EXPECT_FALSE(testARMExtension("generic",
520                                 ARM::ArchKind::ARMV8MBaseline, "crc"));
521   EXPECT_FALSE(testARMExtension("generic",
522                                 ARM::ArchKind::ARMV8MMainline, "crc"));
523   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT, "crc"));
524   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT2, "crc"));
525   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::XSCALE, "crc"));
526   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7S, "crypto"));
527   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7K, "crypto"));
528 }
529 
530 TEST(TargetParserTest, ARMFPUVersion) {
531   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
532        FK <= ARM::FPUKind::FK_LAST;
533        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
534     if (FK == ARM::FK_LAST || ARM::getFPUName(FK) == "invalid" ||
535         ARM::getFPUName(FK) == "none" || ARM::getFPUName(FK) == "softvfp")
536       EXPECT_EQ(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
537     else
538       EXPECT_NE(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
539 }
540 
541 TEST(TargetParserTest, ARMFPUNeonSupportLevel) {
542   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
543        FK <= ARM::FPUKind::FK_LAST;
544        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
545     if (FK == ARM::FK_LAST ||
546         ARM::getFPUName(FK).find("neon") == std::string::npos)
547       EXPECT_EQ(ARM::NeonSupportLevel::None,
548                  ARM::getFPUNeonSupportLevel(FK));
549     else
550       EXPECT_NE(ARM::NeonSupportLevel::None,
551                 ARM::getFPUNeonSupportLevel(FK));
552 }
553 
554 TEST(TargetParserTest, ARMFPURestriction) {
555   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
556        FK <= ARM::FPUKind::FK_LAST;
557        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1)) {
558     if (FK == ARM::FK_LAST ||
559         (ARM::getFPUName(FK).find("d16") == std::string::npos &&
560          ARM::getFPUName(FK).find("vfpv3xd") == std::string::npos))
561       EXPECT_EQ(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
562     else
563       EXPECT_NE(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
564   }
565 }
566 
567 TEST(TargetParserTest, ARMExtensionFeatures) {
568   std::map<unsigned, std::vector<StringRef>> Extensions;
569 
570   for (auto &Ext : ARM::ARCHExtNames) {
571     if (Ext.Feature && Ext.NegFeature)
572       Extensions[Ext.ID] = { StringRef(Ext.Feature),
573                              StringRef(Ext.NegFeature) };
574   }
575 
576   Extensions[ARM::AEK_HWDIVARM]   = { "+hwdiv-arm", "-hwdiv-arm" };
577   Extensions[ARM::AEK_HWDIVTHUMB] = { "+hwdiv",     "-hwdiv" };
578 
579   std::vector<StringRef> Features;
580 
581   EXPECT_FALSE(ARM::getExtensionFeatures(ARM::AEK_INVALID, Features));
582 
583   for (auto &E : Extensions) {
584     // test +extension
585     Features.clear();
586     ARM::getExtensionFeatures(E.first, Features);
587     auto Found =
588         std::find(std::begin(Features), std::end(Features), E.second.at(0));
589     EXPECT_TRUE(Found != std::end(Features));
590     EXPECT_TRUE(Extensions.size() == Features.size());
591 
592     // test -extension
593     Features.clear();
594     ARM::getExtensionFeatures(~E.first, Features);
595     Found = std::find(std::begin(Features), std::end(Features), E.second.at(1));
596     EXPECT_TRUE(Found != std::end(Features));
597     EXPECT_TRUE(Extensions.size() == Features.size());
598   }
599 }
600 
601 TEST(TargetParserTest, ARMFPUFeatures) {
602   std::vector<StringRef> Features;
603   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
604        FK <= ARM::FPUKind::FK_LAST;
605        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
606     EXPECT_TRUE((FK == ARM::FK_INVALID || FK >= ARM::FK_LAST)
607                     ? !ARM::getFPUFeatures(FK, Features)
608                     : ARM::getFPUFeatures(FK, Features));
609 }
610 
611 TEST(TargetParserTest, ARMArchExtFeature) {
612   const char *ArchExt[][4] = {{"crc", "nocrc", "+crc", "-crc"},
613                               {"crypto", "nocrypto", "+crypto", "-crypto"},
614                               {"dsp", "nodsp", "+dsp", "-dsp"},
615                               {"fp", "nofp", nullptr, nullptr},
616                               {"idiv", "noidiv", nullptr, nullptr},
617                               {"mp", "nomp", nullptr, nullptr},
618                               {"simd", "nosimd", nullptr, nullptr},
619                               {"sec", "nosec", nullptr, nullptr},
620                               {"virt", "novirt", nullptr, nullptr},
621                               {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
622                               {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
623                               {"ras", "noras", "+ras", "-ras"},
624                               {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
625                               {"os", "noos", nullptr, nullptr},
626                               {"iwmmxt", "noiwmmxt", nullptr, nullptr},
627                               {"iwmmxt2", "noiwmmxt2", nullptr, nullptr},
628                               {"maverick", "maverick", nullptr, nullptr},
629                               {"xscale", "noxscale", nullptr, nullptr},
630                               {"sb", "nosb", "+sb", "-sb"},
631                               {"mve", "nomve", "+mve", "-mve"},
632                               {"mve.fp", "nomve.fp", "+mve.fp", "-mve.fp"}};
633 
634   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
635     EXPECT_EQ(StringRef(ArchExt[i][2]), ARM::getArchExtFeature(ArchExt[i][0]));
636     EXPECT_EQ(StringRef(ArchExt[i][3]), ARM::getArchExtFeature(ArchExt[i][1]));
637   }
638 }
639 
640 TEST(TargetParserTest, ARMparseHWDiv) {
641   const char *hwdiv[] = {"thumb", "arm", "arm,thumb", "thumb,arm"};
642 
643   for (unsigned i = 0; i < array_lengthof(hwdiv); i++)
644     EXPECT_NE(ARM::AEK_INVALID, ARM::parseHWDiv((StringRef)hwdiv[i]));
645 }
646 
647 TEST(TargetParserTest, ARMparseArchEndianAndISA) {
648   const char *Arch[] = {
649       "v2",   "v2a",    "v3",    "v3m",    "v4",    "v4t",    "v5",    "v5t",
650       "v5e",  "v5te",   "v5tej", "v6",     "v6j",   "v6k",    "v6hl",  "v6t2",
651       "v6kz", "v6z",    "v6zk",  "v6-m",   "v6m",   "v6sm",   "v6s-m", "v7-a",
652       "v7",   "v7a",    "v7ve",  "v7hl",   "v7l",   "v7-r",   "v7r",   "v7-m",
653       "v7m",  "v7k",    "v7s",   "v7e-m",  "v7em",  "v8-a",   "v8",    "v8a",
654       "v8l",  "v8.1-a", "v8.1a", "v8.2-a", "v8.2a", "v8.3-a", "v8.3a", "v8.4-a",
655       "v8.4a", "v8.5-a","v8.5a", "v8-r",   "v8m.base", "v8m.main", "v8.1m.main"
656   };
657 
658   for (unsigned i = 0; i < array_lengthof(Arch); i++) {
659     std::string arm_1 = "armeb" + (std::string)(Arch[i]);
660     std::string arm_2 = "arm" + (std::string)(Arch[i]) + "eb";
661     std::string arm_3 = "arm" + (std::string)(Arch[i]);
662     std::string thumb_1 = "thumbeb" + (std::string)(Arch[i]);
663     std::string thumb_2 = "thumb" + (std::string)(Arch[i]) + "eb";
664     std::string thumb_3 = "thumb" + (std::string)(Arch[i]);
665 
666     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_1));
667     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_2));
668     EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(arm_3));
669 
670     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_1));
671     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_2));
672     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_3));
673     if (i >= 4) {
674       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_1));
675       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_2));
676       EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(thumb_3));
677 
678       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_1));
679       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_2));
680       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_3));
681     }
682   }
683 
684   EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian("aarch64"));
685   EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian("arm64_32"));
686   EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian("aarch64_be"));
687 
688   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64"));
689   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64_be"));
690   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64"));
691   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64_be"));
692   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64_32"));
693   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64_32"));
694 }
695 
696 TEST(TargetParserTest, ARMparseArchProfile) {
697   for (unsigned i = 0; i < array_lengthof(ARMArch); i++) {
698     switch (ARM::parseArch(ARMArch[i])) {
699     case ARM::ArchKind::ARMV6M:
700     case ARM::ArchKind::ARMV7M:
701     case ARM::ArchKind::ARMV7EM:
702     case ARM::ArchKind::ARMV8MMainline:
703     case ARM::ArchKind::ARMV8MBaseline:
704     case ARM::ArchKind::ARMV8_1MMainline:
705       EXPECT_EQ(ARM::ProfileKind::M, ARM::parseArchProfile(ARMArch[i]));
706       break;
707     case ARM::ArchKind::ARMV7R:
708     case ARM::ArchKind::ARMV8R:
709       EXPECT_EQ(ARM::ProfileKind::R, ARM::parseArchProfile(ARMArch[i]));
710       break;
711     case ARM::ArchKind::ARMV7A:
712     case ARM::ArchKind::ARMV7VE:
713     case ARM::ArchKind::ARMV7K:
714     case ARM::ArchKind::ARMV8A:
715     case ARM::ArchKind::ARMV8_1A:
716     case ARM::ArchKind::ARMV8_2A:
717     case ARM::ArchKind::ARMV8_3A:
718     case ARM::ArchKind::ARMV8_4A:
719     case ARM::ArchKind::ARMV8_5A:
720       EXPECT_EQ(ARM::ProfileKind::A, ARM::parseArchProfile(ARMArch[i]));
721       break;
722     default:
723       EXPECT_EQ(ARM::ProfileKind::INVALID, ARM::parseArchProfile(ARMArch[i]));
724       break;
725     }
726   }
727 }
728 
729 TEST(TargetParserTest, ARMparseArchVersion) {
730   for (unsigned i = 0; i < array_lengthof(ARMArch); i++)
731     if (((std::string)ARMArch[i]).substr(0, 4) == "armv")
732       EXPECT_EQ((ARMArch[i][4] - 48u), ARM::parseArchVersion(ARMArch[i]));
733     else
734       EXPECT_EQ(5u, ARM::parseArchVersion(ARMArch[i]));
735 }
736 
737 bool testAArch64CPU(StringRef CPUName, StringRef ExpectedArch,
738                     StringRef ExpectedFPU, unsigned ExpectedFlags,
739                     StringRef CPUAttr) {
740   AArch64::ArchKind AK = AArch64::parseCPUArch(CPUName);
741   bool pass = AArch64::getArchName(AK).equals(ExpectedArch);
742 
743   unsigned ExtKind = AArch64::getDefaultExtensions(CPUName, AK);
744   if (ExtKind > 1 && (ExtKind & AArch64::AEK_NONE))
745     pass &= ((ExtKind ^ AArch64::AEK_NONE) == ExpectedFlags);
746   else
747     pass &= (ExtKind == ExpectedFlags);
748 
749   pass &= AArch64::getCPUAttr(AK).equals(CPUAttr);
750 
751   return pass;
752 }
753 
754 TEST(TargetParserTest, testAArch64CPU) {
755   EXPECT_TRUE(testAArch64CPU(
756       "invalid", "invalid", "invalid",
757       AArch64::AEK_NONE, ""));
758   EXPECT_TRUE(testAArch64CPU(
759       "generic", "invalid", "none",
760       AArch64::AEK_NONE, ""));
761 
762   EXPECT_TRUE(testAArch64CPU(
763       "cortex-a35", "armv8-a", "crypto-neon-fp-armv8",
764       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
765       AArch64::AEK_SIMD, "8-A"));
766   EXPECT_TRUE(testAArch64CPU(
767       "cortex-a53", "armv8-a", "crypto-neon-fp-armv8",
768       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
769       AArch64::AEK_SIMD, "8-A"));
770   EXPECT_TRUE(testAArch64CPU(
771       "cortex-a55", "armv8.2-a", "crypto-neon-fp-armv8",
772       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
773       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
774       AArch64::AEK_RDM | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
775       AArch64::AEK_RCPC, "8.2-A"));
776   EXPECT_TRUE(testAArch64CPU(
777       "cortex-a57", "armv8-a", "crypto-neon-fp-armv8",
778       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
779       AArch64::AEK_SIMD, "8-A"));
780   EXPECT_TRUE(testAArch64CPU(
781       "cortex-a65", "armv8.2-a", "crypto-neon-fp-armv8",
782       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
783       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
784       AArch64::AEK_RAS | AArch64::AEK_RCPC | AArch64::AEK_RDM |
785       AArch64::AEK_SIMD | AArch64::AEK_SSBS,
786       "8.2-A"));
787   EXPECT_TRUE(testAArch64CPU(
788       "cortex-a65ae", "armv8.2-a", "crypto-neon-fp-armv8",
789       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
790       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
791       AArch64::AEK_RAS | AArch64::AEK_RCPC | AArch64::AEK_RDM |
792       AArch64::AEK_SIMD | AArch64::AEK_SSBS,
793       "8.2-A"));
794   EXPECT_TRUE(testAArch64CPU(
795       "cortex-a72", "armv8-a", "crypto-neon-fp-armv8",
796       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
797       AArch64::AEK_SIMD, "8-A"));
798   EXPECT_TRUE(testAArch64CPU(
799       "cortex-a73", "armv8-a", "crypto-neon-fp-armv8",
800       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
801       AArch64::AEK_SIMD, "8-A"));
802   EXPECT_TRUE(testAArch64CPU(
803       "cortex-a75", "armv8.2-a", "crypto-neon-fp-armv8",
804       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
805       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
806       AArch64::AEK_RDM | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
807       AArch64::AEK_RCPC, "8.2-A"));
808   EXPECT_TRUE(testAArch64CPU(
809       "cortex-a76", "armv8.2-a", "crypto-neon-fp-armv8",
810       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
811       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_RAS |
812       AArch64::AEK_LSE | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
813       AArch64::AEK_RCPC| AArch64::AEK_SSBS, "8.2-A"));
814   EXPECT_TRUE(testAArch64CPU(
815       "cortex-a76ae", "armv8.2-a", "crypto-neon-fp-armv8",
816       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
817       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_RAS |
818       AArch64::AEK_LSE | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
819       AArch64::AEK_RCPC| AArch64::AEK_SSBS, "8.2-A"));
820   EXPECT_TRUE(testAArch64CPU(
821       "cyclone", "armv8-a", "crypto-neon-fp-armv8",
822       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
823   EXPECT_TRUE(testAArch64CPU(
824       "apple-a7", "armv8-a", "crypto-neon-fp-armv8",
825       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
826   EXPECT_TRUE(testAArch64CPU(
827       "apple-a8", "armv8-a", "crypto-neon-fp-armv8",
828       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
829   EXPECT_TRUE(testAArch64CPU(
830       "apple-a9", "armv8-a", "crypto-neon-fp-armv8",
831       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
832   EXPECT_TRUE(testAArch64CPU("apple-a10", "armv8-a", "crypto-neon-fp-armv8",
833                              AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
834                                  AArch64::AEK_FP | AArch64::AEK_RDM |
835                                  AArch64::AEK_SIMD,
836                              "8-A"));
837   EXPECT_TRUE(testAArch64CPU("apple-a11", "armv8.2-a", "crypto-neon-fp-armv8",
838                              AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
839                                  AArch64::AEK_FP | AArch64::AEK_LSE |
840                                  AArch64::AEK_RAS | AArch64::AEK_RDM |
841                                  AArch64::AEK_SIMD,
842                              "8.2-A"));
843   EXPECT_TRUE(testAArch64CPU(
844       "apple-a12", "armv8.3-a", "crypto-neon-fp-armv8",
845       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
846           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
847           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_FP16,
848       "8.3-A"));
849   EXPECT_TRUE(testAArch64CPU(
850       "apple-a13", "armv8.4-a", "crypto-neon-fp-armv8",
851       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
852           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
853           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_DOTPROD |
854           AArch64::AEK_FP16 | AArch64::AEK_FP16FML,
855       "8.4-A"));
856   EXPECT_TRUE(testAArch64CPU(
857       "apple-s4", "armv8.3-a", "crypto-neon-fp-armv8",
858       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
859           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
860           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_FP16,
861       "8.3-A"));
862   EXPECT_TRUE(testAArch64CPU(
863       "apple-s5", "armv8.3-a", "crypto-neon-fp-armv8",
864       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
865           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
866           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_FP16,
867       "8.3-A"));
868   EXPECT_TRUE(testAArch64CPU(
869       "exynos-m3", "armv8-a", "crypto-neon-fp-armv8",
870       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
871       AArch64::AEK_SIMD, "8-A"));
872   EXPECT_TRUE(testAArch64CPU(
873       "exynos-m4", "armv8.2-a", "crypto-neon-fp-armv8",
874       AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
875       AArch64::AEK_DOTPROD | AArch64::AEK_FP | AArch64::AEK_FP16 |
876       AArch64::AEK_LSE | AArch64::AEK_RAS | AArch64::AEK_RDM |
877       AArch64::AEK_SIMD, "8.2-A"));
878   EXPECT_TRUE(testAArch64CPU(
879       "exynos-m5", "armv8.2-a", "crypto-neon-fp-armv8",
880       AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
881       AArch64::AEK_DOTPROD | AArch64::AEK_FP | AArch64::AEK_FP16 |
882       AArch64::AEK_LSE | AArch64::AEK_RAS | AArch64::AEK_RDM |
883       AArch64::AEK_SIMD, "8.2-A"));
884   EXPECT_TRUE(testAArch64CPU(
885       "falkor", "armv8-a", "crypto-neon-fp-armv8",
886       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
887       AArch64::AEK_SIMD | AArch64::AEK_RDM, "8-A"));
888   EXPECT_TRUE(testAArch64CPU(
889       "kryo", "armv8-a", "crypto-neon-fp-armv8",
890       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
891       AArch64::AEK_SIMD, "8-A"));
892   EXPECT_TRUE(testAArch64CPU(
893      "neoverse-e1", "armv8.2-a", "crypto-neon-fp-armv8",
894       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
895       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
896       AArch64::AEK_RAS | AArch64::AEK_RCPC | AArch64::AEK_RDM |
897       AArch64::AEK_SIMD | AArch64::AEK_SSBS,
898      "8.2-A"));
899   EXPECT_TRUE(testAArch64CPU(
900      "neoverse-n1", "armv8.2-a", "crypto-neon-fp-armv8",
901       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
902       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
903       AArch64::AEK_PROFILE | AArch64::AEK_RAS | AArch64::AEK_RCPC |
904       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_SSBS,
905      "8.2-A"));
906   EXPECT_TRUE(testAArch64CPU(
907       "thunderx2t99", "armv8.1-a", "crypto-neon-fp-armv8",
908       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_LSE |
909       AArch64::AEK_RDM | AArch64::AEK_FP | AArch64::AEK_SIMD, "8.1-A"));
910   EXPECT_TRUE(testAArch64CPU(
911       "thunderx", "armv8-a", "crypto-neon-fp-armv8",
912       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
913       AArch64::AEK_FP | AArch64::AEK_PROFILE,
914       "8-A"));
915   EXPECT_TRUE(testAArch64CPU(
916       "thunderxt81", "armv8-a", "crypto-neon-fp-armv8",
917       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
918       AArch64::AEK_FP | AArch64::AEK_PROFILE,
919       "8-A"));
920   EXPECT_TRUE(testAArch64CPU(
921       "thunderxt83", "armv8-a", "crypto-neon-fp-armv8",
922       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
923       AArch64::AEK_FP | AArch64::AEK_PROFILE,
924       "8-A"));
925   EXPECT_TRUE(testAArch64CPU(
926       "thunderxt88", "armv8-a", "crypto-neon-fp-armv8",
927       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
928       AArch64::AEK_FP | AArch64::AEK_PROFILE,
929       "8-A"));
930   EXPECT_TRUE(testAArch64CPU(
931       "tsv110", "armv8.2-a", "crypto-neon-fp-armv8",
932       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
933       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
934       AArch64::AEK_RDM | AArch64::AEK_PROFILE | AArch64::AEK_FP16 |
935       AArch64::AEK_FP16FML | AArch64::AEK_DOTPROD,
936       "8.2-A"));
937 }
938 
939 static constexpr unsigned NumAArch64CPUArchs = 35;
940 
941 TEST(TargetParserTest, testAArch64CPUArchList) {
942   SmallVector<StringRef, NumAArch64CPUArchs> List;
943   AArch64::fillValidCPUArchList(List);
944 
945   // No list exists for these in this test suite, so ensure all are
946   // valid, and match the expected 'magic' count.
947   EXPECT_EQ(List.size(), NumAArch64CPUArchs);
948   for(StringRef CPU : List) {
949     EXPECT_NE(AArch64::parseCPUArch(CPU), AArch64::ArchKind::INVALID);
950   }
951 }
952 
953 bool testAArch64Arch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
954                      unsigned ArchAttr) {
955   AArch64::ArchKind AK = AArch64::parseArch(Arch);
956   return (AK != AArch64::ArchKind::INVALID) &
957          AArch64::getDefaultCPU(Arch).equals(DefaultCPU) &
958          AArch64::getSubArch(AK).equals(SubArch) &
959          (AArch64::getArchAttr(AK) == ArchAttr);
960 }
961 
962 TEST(TargetParserTest, testAArch64Arch) {
963   EXPECT_TRUE(testAArch64Arch("armv8-a", "cortex-a53", "v8",
964                               ARMBuildAttrs::CPUArch::v8_A));
965   EXPECT_TRUE(testAArch64Arch("armv8.1-a", "generic", "v8.1a",
966                               ARMBuildAttrs::CPUArch::v8_A));
967   EXPECT_TRUE(testAArch64Arch("armv8.2-a", "generic", "v8.2a",
968                               ARMBuildAttrs::CPUArch::v8_A));
969   EXPECT_TRUE(testAArch64Arch("armv8.3-a", "generic", "v8.3a",
970                               ARMBuildAttrs::CPUArch::v8_A));
971   EXPECT_TRUE(testAArch64Arch("armv8.4-a", "generic", "v8.4a",
972                               ARMBuildAttrs::CPUArch::v8_A));
973   EXPECT_TRUE(testAArch64Arch("armv8.5-a", "generic", "v8.5a",
974                               ARMBuildAttrs::CPUArch::v8_A));
975 }
976 
977 bool testAArch64Extension(StringRef CPUName, AArch64::ArchKind AK,
978                           StringRef ArchExt) {
979   return AArch64::getDefaultExtensions(CPUName, AK) &
980          AArch64::parseArchExt(ArchExt);
981 }
982 
983 TEST(TargetParserTest, testAArch64Extension) {
984   EXPECT_FALSE(testAArch64Extension("cortex-a35",
985                                     AArch64::ArchKind::INVALID, "ras"));
986   EXPECT_FALSE(testAArch64Extension("cortex-a53",
987                                     AArch64::ArchKind::INVALID, "ras"));
988   EXPECT_TRUE(testAArch64Extension("cortex-a55",
989                                     AArch64::ArchKind::INVALID, "ras"));
990   EXPECT_FALSE(testAArch64Extension("cortex-a57",
991                                     AArch64::ArchKind::INVALID, "ras"));
992   EXPECT_FALSE(testAArch64Extension("cortex-a72",
993                                     AArch64::ArchKind::INVALID, "ras"));
994   EXPECT_FALSE(testAArch64Extension("cortex-a73",
995                                     AArch64::ArchKind::INVALID, "ras"));
996   EXPECT_TRUE(testAArch64Extension("cortex-a75",
997                                     AArch64::ArchKind::INVALID, "ras"));
998   EXPECT_FALSE(testAArch64Extension("cyclone",
999                                     AArch64::ArchKind::INVALID, "ras"));
1000   EXPECT_FALSE(testAArch64Extension("exynos-m3",
1001                                     AArch64::ArchKind::INVALID, "ras"));
1002   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1003                                    AArch64::ArchKind::INVALID, "dotprod"));
1004   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1005                                    AArch64::ArchKind::INVALID, "fp16"));
1006   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1007                                    AArch64::ArchKind::INVALID, "lse"));
1008   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1009                                    AArch64::ArchKind::INVALID, "ras"));
1010   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1011                                    AArch64::ArchKind::INVALID, "rdm"));
1012   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1013                                    AArch64::ArchKind::INVALID, "dotprod"));
1014   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1015                                    AArch64::ArchKind::INVALID, "fp16"));
1016   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1017                                    AArch64::ArchKind::INVALID, "lse"));
1018   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1019                                    AArch64::ArchKind::INVALID, "ras"));
1020   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1021                                    AArch64::ArchKind::INVALID, "rdm"));
1022   EXPECT_TRUE(testAArch64Extension("falkor",
1023                                    AArch64::ArchKind::INVALID, "rdm"));
1024   EXPECT_FALSE(testAArch64Extension("kryo",
1025                                     AArch64::ArchKind::INVALID, "ras"));
1026   EXPECT_TRUE(testAArch64Extension("saphira",
1027                                     AArch64::ArchKind::INVALID, "crc"));
1028   EXPECT_TRUE(testAArch64Extension("saphira",
1029                                     AArch64::ArchKind::INVALID, "lse"));
1030   EXPECT_TRUE(testAArch64Extension("saphira",
1031                                    AArch64::ArchKind::INVALID, "rdm"));
1032   EXPECT_TRUE(testAArch64Extension("saphira",
1033                                    AArch64::ArchKind::INVALID, "ras"));
1034   EXPECT_TRUE(testAArch64Extension("saphira",
1035                                    AArch64::ArchKind::INVALID, "rcpc"));
1036   EXPECT_TRUE(testAArch64Extension("saphira",
1037                                    AArch64::ArchKind::INVALID, "profile"));
1038   EXPECT_FALSE(testAArch64Extension("saphira",
1039                                     AArch64::ArchKind::INVALID, "fp16"));
1040   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1041                                     AArch64::ArchKind::INVALID, "fp16"));
1042   EXPECT_FALSE(testAArch64Extension("cortex-a55",
1043                                     AArch64::ArchKind::INVALID, "fp16fml"));
1044   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1045                                     AArch64::ArchKind::INVALID, "dotprod"));
1046   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1047                                     AArch64::ArchKind::INVALID, "fp16"));
1048   EXPECT_FALSE(testAArch64Extension("cortex-a75",
1049                                     AArch64::ArchKind::INVALID, "fp16fml"));
1050   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1051                                     AArch64::ArchKind::INVALID, "dotprod"));
1052   EXPECT_FALSE(testAArch64Extension("thunderx2t99",
1053                                     AArch64::ArchKind::INVALID, "ras"));
1054   EXPECT_FALSE(testAArch64Extension("thunderx",
1055                                     AArch64::ArchKind::INVALID, "lse"));
1056   EXPECT_FALSE(testAArch64Extension("thunderxt81",
1057                                     AArch64::ArchKind::INVALID, "lse"));
1058   EXPECT_FALSE(testAArch64Extension("thunderxt83",
1059                                     AArch64::ArchKind::INVALID, "lse"));
1060   EXPECT_FALSE(testAArch64Extension("thunderxt88",
1061                                     AArch64::ArchKind::INVALID, "lse"));
1062   EXPECT_TRUE(testAArch64Extension("tsv110",
1063                                    AArch64::ArchKind::INVALID, "crypto"));
1064   EXPECT_FALSE(testAArch64Extension("tsv110",
1065                                     AArch64::ArchKind::INVALID, "sha3"));
1066   EXPECT_FALSE(testAArch64Extension("tsv110",
1067                                     AArch64::ArchKind::INVALID, "sm4"));
1068   EXPECT_TRUE(testAArch64Extension("tsv110",
1069                                    AArch64::ArchKind::INVALID, "ras"));
1070   EXPECT_TRUE(testAArch64Extension("tsv110",
1071                                    AArch64::ArchKind::INVALID, "profile"));
1072   EXPECT_TRUE(testAArch64Extension("tsv110",
1073                                    AArch64::ArchKind::INVALID, "fp16"));
1074   EXPECT_TRUE(testAArch64Extension("tsv110",
1075                                    AArch64::ArchKind::INVALID, "fp16fml"));
1076   EXPECT_TRUE(testAArch64Extension("tsv110",
1077                                    AArch64::ArchKind::INVALID, "dotprod"));
1078 
1079   EXPECT_FALSE(testAArch64Extension(
1080       "generic", AArch64::ArchKind::ARMV8A, "ras"));
1081   EXPECT_FALSE(testAArch64Extension(
1082       "generic", AArch64::ArchKind::ARMV8_1A, "ras"));
1083   EXPECT_FALSE(testAArch64Extension(
1084       "generic", AArch64::ArchKind::ARMV8_2A, "profile"));
1085   EXPECT_FALSE(testAArch64Extension(
1086       "generic", AArch64::ArchKind::ARMV8_2A, "fp16"));
1087   EXPECT_FALSE(testAArch64Extension(
1088       "generic", AArch64::ArchKind::ARMV8_2A, "fp16fml"));
1089   EXPECT_FALSE(testAArch64Extension(
1090       "generic", AArch64::ArchKind::ARMV8_3A, "fp16"));
1091   EXPECT_FALSE(testAArch64Extension(
1092       "generic", AArch64::ArchKind::ARMV8_3A, "fp16fml"));
1093   EXPECT_FALSE(testAArch64Extension(
1094       "generic", AArch64::ArchKind::ARMV8_4A, "fp16"));
1095   EXPECT_FALSE(testAArch64Extension(
1096       "generic", AArch64::ArchKind::ARMV8_4A, "fp16fml"));
1097 }
1098 
1099 TEST(TargetParserTest, AArch64ExtensionFeatures) {
1100   std::vector<unsigned> Extensions = {
1101     AArch64::AEK_CRC,      AArch64::AEK_CRYPTO,
1102     AArch64::AEK_FP,       AArch64::AEK_SIMD,
1103     AArch64::AEK_FP16,     AArch64::AEK_PROFILE,
1104     AArch64::AEK_RAS,      AArch64::AEK_LSE,
1105     AArch64::AEK_RDM,      AArch64::AEK_DOTPROD,
1106     AArch64::AEK_SVE,      AArch64::AEK_SVE2,
1107     AArch64::AEK_SVE2AES,  AArch64::AEK_SVE2SM4,
1108     AArch64::AEK_SVE2SHA3, AArch64::AEK_SVE2BITPERM,
1109     AArch64::AEK_RCPC,     AArch64::AEK_FP16FML };
1110 
1111   std::vector<StringRef> Features;
1112 
1113   unsigned ExtVal = 0;
1114   for (auto Ext : Extensions)
1115     ExtVal |= Ext;
1116 
1117   EXPECT_FALSE(AArch64::getExtensionFeatures(AArch64::AEK_INVALID, Features));
1118   EXPECT_TRUE(!Features.size());
1119 
1120   AArch64::getExtensionFeatures(ExtVal, Features);
1121   EXPECT_TRUE(Extensions.size() == Features.size());
1122 
1123   auto B = std::begin(Features);
1124   auto E = std::end(Features);
1125 
1126   EXPECT_TRUE(std::find(B, E, "+crc") != E);
1127   EXPECT_TRUE(std::find(B, E, "+crypto") != E);
1128   EXPECT_TRUE(std::find(B, E, "+fp-armv8") != E);
1129   EXPECT_TRUE(std::find(B, E, "+neon") != E);
1130   EXPECT_TRUE(std::find(B, E, "+fullfp16") != E);
1131   EXPECT_TRUE(std::find(B, E, "+spe") != E);
1132   EXPECT_TRUE(std::find(B, E, "+ras") != E);
1133   EXPECT_TRUE(std::find(B, E, "+lse") != E);
1134   EXPECT_TRUE(std::find(B, E, "+rdm") != E);
1135   EXPECT_TRUE(std::find(B, E, "+dotprod") != E);
1136   EXPECT_TRUE(std::find(B, E, "+rcpc") != E);
1137   EXPECT_TRUE(std::find(B, E, "+fp16fml") != E);
1138   EXPECT_TRUE(std::find(B, E, "+sve") != E);
1139   EXPECT_TRUE(std::find(B, E, "+sve2") != E);
1140   EXPECT_TRUE(std::find(B, E, "+sve2-aes") != E);
1141   EXPECT_TRUE(std::find(B, E, "+sve2-sm4") != E);
1142   EXPECT_TRUE(std::find(B, E, "+sve2-sha3") != E);
1143   EXPECT_TRUE(std::find(B, E, "+sve2-bitperm") != E);
1144 }
1145 
1146 TEST(TargetParserTest, AArch64ArchFeatures) {
1147   std::vector<StringRef> Features;
1148 
1149   for (auto AK : AArch64::ArchKinds)
1150     EXPECT_TRUE((AK == AArch64::ArchKind::INVALID)
1151                     ? !AArch64::getArchFeatures(AK, Features)
1152                     : AArch64::getArchFeatures(AK, Features));
1153 }
1154 
1155 TEST(TargetParserTest, AArch64ArchExtFeature) {
1156   const char *ArchExt[][4] = {{"crc", "nocrc", "+crc", "-crc"},
1157                               {"crypto", "nocrypto", "+crypto", "-crypto"},
1158                               {"fp", "nofp", "+fp-armv8", "-fp-armv8"},
1159                               {"simd", "nosimd", "+neon", "-neon"},
1160                               {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
1161                               {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
1162                               {"profile", "noprofile", "+spe", "-spe"},
1163                               {"ras", "noras", "+ras", "-ras"},
1164                               {"lse", "nolse", "+lse", "-lse"},
1165                               {"rdm", "nordm", "+rdm", "-rdm"},
1166                               {"sve", "nosve", "+sve", "-sve"},
1167                               {"sve2", "nosve2", "+sve2", "-sve2"},
1168                               {"sve2-aes", "nosve2-aes", "+sve2-aes",
1169                                "-sve2-aes"},
1170                               {"sve2-sm4", "nosve2-sm4", "+sve2-sm4",
1171                                "-sve2-sm4"},
1172                               {"sve2-sha3", "nosve2-sha3", "+sve2-sha3",
1173                                "-sve2-sha3"},
1174                               {"sve2-bitperm", "nosve2-bitperm",
1175                                "+sve2-bitperm", "-sve2-bitperm"},
1176                               {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
1177                               {"rcpc", "norcpc", "+rcpc", "-rcpc" },
1178                               {"rng", "norng", "+rand", "-rand"},
1179                               {"memtag", "nomemtag", "+mte", "-mte"},
1180                               {"tme", "notme", "+tme", "-tme"},
1181                               {"ssbs", "nossbs", "+ssbs", "-ssbs"},
1182                               {"sb", "nosb", "+sb", "-sb"},
1183                               {"predres", "nopredres", "+predres", "-predres"}
1184 };
1185 
1186   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
1187     EXPECT_EQ(StringRef(ArchExt[i][2]),
1188               AArch64::getArchExtFeature(ArchExt[i][0]));
1189     EXPECT_EQ(StringRef(ArchExt[i][3]),
1190               AArch64::getArchExtFeature(ArchExt[i][1]));
1191   }
1192 }
1193 }
1194