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.6-a",   "armv8.6a",     "armv8-r",     "armv8r",
30     "armv8-m.base", "armv8m.base", "armv8-m.main", "armv8m.main", "iwmmxt",
31     "iwmmxt2",      "xscale",      "armv8.1-m.main",
32 };
33 
34 bool testARMCPU(StringRef CPUName, StringRef ExpectedArch,
35                 StringRef ExpectedFPU, uint64_t 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   uint64_t 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("cortex-a77", "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("cortex-a78", "armv8.2-a", "crypto-neon-fp-armv8",
266                          ARM::AEK_DOTPROD | ARM::AEK_FP16 |
267                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
268                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
269                          ARM::AEK_DSP | ARM::AEK_CRC | ARM::AEK_RAS,
270                          "8.2-A"));
271   EXPECT_TRUE(testARMCPU("cortex-x1", "armv8.2-a", "crypto-neon-fp-armv8",
272                          ARM::AEK_RAS | ARM::AEK_FP16 | ARM::AEK_DOTPROD |
273                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
274                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
275                          ARM::AEK_DSP | ARM::AEK_CRC | ARM::AEK_RAS,
276                          "8.2-A"));
277   EXPECT_TRUE(testARMCPU("neoverse-n1", "armv8.2-a", "crypto-neon-fp-armv8",
278                         ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
279                         ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
280                         ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
281                         ARM::AEK_RAS | ARM::AEK_DOTPROD,
282                         "8.2-A"));
283   EXPECT_TRUE(testARMCPU("cyclone", "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   EXPECT_TRUE(testARMCPU("exynos-m3", "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   EXPECT_TRUE(testARMCPU("exynos-m4", "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 | ARM::AEK_DOTPROD |
297                          ARM::AEK_FP16 | ARM::AEK_RAS,
298                          "8.2-A"));
299   EXPECT_TRUE(testARMCPU("exynos-m5", "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 | ARM::AEK_DOTPROD |
303                          ARM::AEK_FP16 | ARM::AEK_RAS,
304                          "8.2-A"));
305   EXPECT_TRUE(testARMCPU("cortex-m23", "armv8-m.base", "none",
306                          ARM::AEK_HWDIVTHUMB, "8-M.Baseline"));
307   EXPECT_TRUE(testARMCPU("cortex-m33", "armv8-m.main", "fpv5-sp-d16",
308                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "8-M.Mainline"));
309   EXPECT_TRUE(testARMCPU("cortex-m35p", "armv8-m.main", "fpv5-sp-d16",
310                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "8-M.Mainline"));
311   EXPECT_TRUE(testARMCPU("cortex-m55", "armv8.1-m.main", "fp-armv8-fullfp16-d16",
312                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_SIMD |
313                          ARM::AEK_FP         | ARM::AEK_RAS | ARM::AEK_LOB |
314                          ARM::AEK_FP16,
315                         "8.1-M.Mainline"));
316   EXPECT_TRUE(testARMCPU("iwmmxt", "iwmmxt", "none",
317                          ARM::AEK_NONE, "iwmmxt"));
318   EXPECT_TRUE(testARMCPU("xscale", "xscale", "none",
319                          ARM::AEK_NONE, "xscale"));
320   EXPECT_TRUE(testARMCPU("swift", "armv7s", "neon-vfpv4",
321                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
322                          "7-S"));
323 }
324 
325 static constexpr unsigned NumARMCPUArchs = 89;
326 
327 TEST(TargetParserTest, testARMCPUArchList) {
328   SmallVector<StringRef, NumARMCPUArchs> List;
329   ARM::fillValidCPUArchList(List);
330 
331   // No list exists for these in this test suite, so ensure all are
332   // valid, and match the expected 'magic' count.
333   EXPECT_EQ(List.size(), NumARMCPUArchs);
334   for(StringRef CPU : List) {
335     EXPECT_NE(ARM::parseCPUArch(CPU), ARM::ArchKind::INVALID);
336   }
337 }
338 
339 TEST(TargetParserTest, testInvalidARMArch) {
340   auto InvalidArchStrings = {"armv", "armv99", "noarm"};
341   for (const char* InvalidArch : InvalidArchStrings)
342     EXPECT_EQ(ARM::parseArch(InvalidArch), ARM::ArchKind::INVALID);
343 }
344 
345 bool testARMArch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
346                  unsigned ArchAttr) {
347   ARM::ArchKind AK = ARM::parseArch(Arch);
348   bool Result = (AK != ARM::ArchKind::INVALID);
349   Result &= ARM::getDefaultCPU(Arch).equals(DefaultCPU);
350   Result &= ARM::getSubArch(AK).equals(SubArch);
351   Result &= (ARM::getArchAttr(AK) == ArchAttr);
352   return Result;
353 }
354 
355 TEST(TargetParserTest, testARMArch) {
356   EXPECT_TRUE(
357       testARMArch("armv2", "arm2", "v2",
358                           ARMBuildAttrs::CPUArch::Pre_v4));
359   EXPECT_TRUE(
360       testARMArch("armv2a", "arm3", "v2a",
361                           ARMBuildAttrs::CPUArch::Pre_v4));
362   EXPECT_TRUE(
363       testARMArch("armv3", "arm6", "v3",
364                           ARMBuildAttrs::CPUArch::Pre_v4));
365   EXPECT_TRUE(
366       testARMArch("armv3m", "arm7m", "v3m",
367                           ARMBuildAttrs::CPUArch::Pre_v4));
368   EXPECT_TRUE(
369       testARMArch("armv4", "strongarm", "v4",
370                           ARMBuildAttrs::CPUArch::v4));
371   EXPECT_TRUE(
372       testARMArch("armv4t", "arm7tdmi", "v4t",
373                           ARMBuildAttrs::CPUArch::v4T));
374   EXPECT_TRUE(
375       testARMArch("armv5t", "arm10tdmi", "v5",
376                           ARMBuildAttrs::CPUArch::v5T));
377   EXPECT_TRUE(
378       testARMArch("armv5te", "arm1022e", "v5e",
379                           ARMBuildAttrs::CPUArch::v5TE));
380   EXPECT_TRUE(
381       testARMArch("armv5tej", "arm926ej-s", "v5e",
382                           ARMBuildAttrs::CPUArch::v5TEJ));
383   EXPECT_TRUE(
384       testARMArch("armv6", "arm1136jf-s", "v6",
385                           ARMBuildAttrs::CPUArch::v6));
386   EXPECT_TRUE(
387       testARMArch("armv6k", "mpcore", "v6k",
388                           ARMBuildAttrs::CPUArch::v6K));
389   EXPECT_TRUE(
390       testARMArch("armv6t2", "arm1156t2-s", "v6t2",
391                           ARMBuildAttrs::CPUArch::v6T2));
392   EXPECT_TRUE(
393       testARMArch("armv6kz", "arm1176jzf-s", "v6kz",
394                           ARMBuildAttrs::CPUArch::v6KZ));
395   EXPECT_TRUE(
396       testARMArch("armv6-m", "cortex-m0", "v6m",
397                           ARMBuildAttrs::CPUArch::v6_M));
398   EXPECT_TRUE(
399       testARMArch("armv7-a", "generic", "v7",
400                           ARMBuildAttrs::CPUArch::v7));
401   EXPECT_TRUE(
402       testARMArch("armv7ve", "generic", "v7ve",
403                           ARMBuildAttrs::CPUArch::v7));
404   EXPECT_TRUE(
405       testARMArch("armv7-r", "cortex-r4", "v7r",
406                           ARMBuildAttrs::CPUArch::v7));
407   EXPECT_TRUE(
408       testARMArch("armv7-m", "cortex-m3", "v7m",
409                           ARMBuildAttrs::CPUArch::v7));
410   EXPECT_TRUE(
411       testARMArch("armv7e-m", "cortex-m4", "v7em",
412                           ARMBuildAttrs::CPUArch::v7E_M));
413   EXPECT_TRUE(
414       testARMArch("armv8-a", "generic", "v8",
415                           ARMBuildAttrs::CPUArch::v8_A));
416   EXPECT_TRUE(
417       testARMArch("armv8.1-a", "generic", "v8.1a",
418                           ARMBuildAttrs::CPUArch::v8_A));
419   EXPECT_TRUE(
420       testARMArch("armv8.2-a", "generic", "v8.2a",
421                           ARMBuildAttrs::CPUArch::v8_A));
422   EXPECT_TRUE(
423       testARMArch("armv8.3-a", "generic", "v8.3a",
424                           ARMBuildAttrs::CPUArch::v8_A));
425   EXPECT_TRUE(
426       testARMArch("armv8.4-a", "generic", "v8.4a",
427                           ARMBuildAttrs::CPUArch::v8_A));
428   EXPECT_TRUE(
429       testARMArch("armv8.5-a", "generic", "v8.5a",
430                           ARMBuildAttrs::CPUArch::v8_A));
431   EXPECT_TRUE(
432       testARMArch("armv8.6-a", "generic", "v8.6a",
433                           ARMBuildAttrs::CPUArch::v8_A));
434   EXPECT_TRUE(
435       testARMArch("armv8-r", "cortex-r52", "v8r",
436                           ARMBuildAttrs::CPUArch::v8_R));
437   EXPECT_TRUE(
438       testARMArch("armv8-m.base", "generic", "v8m.base",
439                           ARMBuildAttrs::CPUArch::v8_M_Base));
440   EXPECT_TRUE(
441       testARMArch("armv8-m.main", "generic", "v8m.main",
442                           ARMBuildAttrs::CPUArch::v8_M_Main));
443   EXPECT_TRUE(
444       testARMArch("armv8.1-m.main", "generic", "v8.1m.main",
445                           ARMBuildAttrs::CPUArch::v8_1_M_Main));
446   EXPECT_TRUE(
447       testARMArch("iwmmxt", "iwmmxt", "",
448                           ARMBuildAttrs::CPUArch::v5TE));
449   EXPECT_TRUE(
450       testARMArch("iwmmxt2", "generic", "",
451                           ARMBuildAttrs::CPUArch::v5TE));
452   EXPECT_TRUE(
453       testARMArch("xscale", "xscale", "v5e",
454                           ARMBuildAttrs::CPUArch::v5TE));
455   EXPECT_TRUE(
456       testARMArch("armv7s", "swift", "v7s",
457                           ARMBuildAttrs::CPUArch::v7));
458   EXPECT_TRUE(
459       testARMArch("armv7k", "generic", "v7k",
460                           ARMBuildAttrs::CPUArch::v7));
461 }
462 
463 bool testARMExtension(StringRef CPUName,ARM::ArchKind ArchKind, StringRef ArchExt) {
464   return ARM::getDefaultExtensions(CPUName, ArchKind) &
465          ARM::parseArchExt(ArchExt);
466 }
467 
468 TEST(TargetParserTest, testARMExtension) {
469   EXPECT_FALSE(testARMExtension("arm2", ARM::ArchKind::INVALID, "thumb"));
470   EXPECT_FALSE(testARMExtension("arm3", ARM::ArchKind::INVALID, "thumb"));
471   EXPECT_FALSE(testARMExtension("arm6", ARM::ArchKind::INVALID, "thumb"));
472   EXPECT_FALSE(testARMExtension("arm7m", ARM::ArchKind::INVALID, "thumb"));
473   EXPECT_FALSE(testARMExtension("strongarm", ARM::ArchKind::INVALID, "dsp"));
474   EXPECT_FALSE(testARMExtension("arm7tdmi", ARM::ArchKind::INVALID, "dsp"));
475   EXPECT_FALSE(testARMExtension("arm10tdmi",
476                                 ARM::ArchKind::INVALID, "simd"));
477   EXPECT_FALSE(testARMExtension("arm1022e", ARM::ArchKind::INVALID, "simd"));
478   EXPECT_FALSE(testARMExtension("arm926ej-s",
479                                 ARM::ArchKind::INVALID, "simd"));
480   EXPECT_FALSE(testARMExtension("arm1136jf-s",
481                                 ARM::ArchKind::INVALID, "crypto"));
482   EXPECT_FALSE(testARMExtension("arm1176j-s",
483                                 ARM::ArchKind::INVALID, "crypto"));
484   EXPECT_FALSE(testARMExtension("arm1156t2-s",
485                                 ARM::ArchKind::INVALID, "crypto"));
486   EXPECT_FALSE(testARMExtension("arm1176jzf-s",
487                                 ARM::ArchKind::INVALID, "crypto"));
488   EXPECT_FALSE(testARMExtension("cortex-m0",
489                                 ARM::ArchKind::INVALID, "crypto"));
490   EXPECT_FALSE(testARMExtension("cortex-a8",
491                                 ARM::ArchKind::INVALID, "crypto"));
492   EXPECT_FALSE(testARMExtension("cortex-r4",
493                                 ARM::ArchKind::INVALID, "crypto"));
494   EXPECT_FALSE(testARMExtension("cortex-m3",
495                                 ARM::ArchKind::INVALID, "crypto"));
496   EXPECT_FALSE(testARMExtension("cortex-a53",
497                                 ARM::ArchKind::INVALID, "ras"));
498   EXPECT_FALSE(testARMExtension("cortex-a53",
499                                 ARM::ArchKind::INVALID, "fp16"));
500   EXPECT_TRUE(testARMExtension("cortex-a55",
501                                 ARM::ArchKind::INVALID, "fp16"));
502   EXPECT_FALSE(testARMExtension("cortex-a55",
503                                 ARM::ArchKind::INVALID, "fp16fml"));
504   EXPECT_TRUE(testARMExtension("cortex-a75",
505                                 ARM::ArchKind::INVALID, "fp16"));
506   EXPECT_FALSE(testARMExtension("cortex-a75",
507                                 ARM::ArchKind::INVALID, "fp16fml"));
508   EXPECT_FALSE(testARMExtension("cortex-r52",
509                                 ARM::ArchKind::INVALID, "ras"));
510   EXPECT_FALSE(testARMExtension("iwmmxt", ARM::ArchKind::INVALID, "crc"));
511   EXPECT_FALSE(testARMExtension("xscale", ARM::ArchKind::INVALID, "crc"));
512   EXPECT_FALSE(testARMExtension("swift", ARM::ArchKind::INVALID, "crc"));
513 
514   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2, "thumb"));
515   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2A, "thumb"));
516   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3, "thumb"));
517   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3M, "thumb"));
518   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4, "dsp"));
519   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4T, "dsp"));
520   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5T, "simd"));
521   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TE, "simd"));
522   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TEJ, "simd"));
523   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6, "crypto"));
524   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6K, "crypto"));
525   EXPECT_FALSE(testARMExtension("generic",
526                                 ARM::ArchKind::ARMV6T2, "crypto"));
527   EXPECT_FALSE(testARMExtension("generic",
528                                 ARM::ArchKind::ARMV6KZ, "crypto"));
529   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6M, "crypto"));
530   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7A, "crypto"));
531   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7R, "crypto"));
532   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7M, "crypto"));
533   EXPECT_FALSE(testARMExtension("generic",
534                                 ARM::ArchKind::ARMV7EM, "crypto"));
535   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8A, "ras"));
536   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_1A, "ras"));
537   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "profile"));
538   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16"));
539   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16fml"));
540   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16"));
541   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16fml"));
542   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16"));
543   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16fml"));
544   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8R, "ras"));
545   EXPECT_FALSE(testARMExtension("generic",
546                                 ARM::ArchKind::ARMV8MBaseline, "crc"));
547   EXPECT_FALSE(testARMExtension("generic",
548                                 ARM::ArchKind::ARMV8MMainline, "crc"));
549   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT, "crc"));
550   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT2, "crc"));
551   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::XSCALE, "crc"));
552   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7S, "crypto"));
553   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7K, "crypto"));
554 }
555 
556 TEST(TargetParserTest, ARMFPUVersion) {
557   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
558        FK <= ARM::FPUKind::FK_LAST;
559        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
560     if (FK == ARM::FK_LAST || ARM::getFPUName(FK) == "invalid" ||
561         ARM::getFPUName(FK) == "none" || ARM::getFPUName(FK) == "softvfp")
562       EXPECT_EQ(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
563     else
564       EXPECT_NE(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
565 }
566 
567 TEST(TargetParserTest, ARMFPUNeonSupportLevel) {
568   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
569        FK <= ARM::FPUKind::FK_LAST;
570        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
571     if (FK == ARM::FK_LAST ||
572         ARM::getFPUName(FK).find("neon") == std::string::npos)
573       EXPECT_EQ(ARM::NeonSupportLevel::None,
574                  ARM::getFPUNeonSupportLevel(FK));
575     else
576       EXPECT_NE(ARM::NeonSupportLevel::None,
577                 ARM::getFPUNeonSupportLevel(FK));
578 }
579 
580 TEST(TargetParserTest, ARMFPURestriction) {
581   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
582        FK <= ARM::FPUKind::FK_LAST;
583        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1)) {
584     if (FK == ARM::FK_LAST ||
585         (ARM::getFPUName(FK).find("d16") == std::string::npos &&
586          ARM::getFPUName(FK).find("vfpv3xd") == std::string::npos))
587       EXPECT_EQ(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
588     else
589       EXPECT_NE(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
590   }
591 }
592 
593 TEST(TargetParserTest, ARMExtensionFeatures) {
594   std::map<uint64_t, std::vector<StringRef>> Extensions;
595 
596   for (auto &Ext : ARM::ARCHExtNames) {
597     if (Ext.Feature && Ext.NegFeature)
598       Extensions[Ext.ID] = { StringRef(Ext.Feature),
599                              StringRef(Ext.NegFeature) };
600   }
601 
602   Extensions[ARM::AEK_HWDIVARM]   = { "+hwdiv-arm", "-hwdiv-arm" };
603   Extensions[ARM::AEK_HWDIVTHUMB] = { "+hwdiv",     "-hwdiv" };
604 
605   std::vector<StringRef> Features;
606 
607   EXPECT_FALSE(ARM::getExtensionFeatures(ARM::AEK_INVALID, Features));
608 
609   for (auto &E : Extensions) {
610     // test +extension
611     Features.clear();
612     ARM::getExtensionFeatures(E.first, Features);
613     EXPECT_TRUE(llvm::is_contained(Features, E.second.at(0)));
614     EXPECT_TRUE(Extensions.size() == Features.size());
615 
616     // test -extension
617     Features.clear();
618     ARM::getExtensionFeatures(~E.first, Features);
619     EXPECT_TRUE(llvm::is_contained(Features, E.second.at(1)));
620     EXPECT_TRUE(Extensions.size() == Features.size());
621   }
622 }
623 
624 TEST(TargetParserTest, ARMFPUFeatures) {
625   std::vector<StringRef> Features;
626   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
627        FK <= ARM::FPUKind::FK_LAST;
628        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
629     EXPECT_TRUE((FK == ARM::FK_INVALID || FK >= ARM::FK_LAST)
630                     ? !ARM::getFPUFeatures(FK, Features)
631                     : ARM::getFPUFeatures(FK, Features));
632 }
633 
634 TEST(TargetParserTest, ARMArchExtFeature) {
635   const char *ArchExt[][4] = {{"crc", "nocrc", "+crc", "-crc"},
636                               {"crypto", "nocrypto", "+crypto", "-crypto"},
637                               {"dsp", "nodsp", "+dsp", "-dsp"},
638                               {"fp", "nofp", nullptr, nullptr},
639                               {"idiv", "noidiv", nullptr, nullptr},
640                               {"mp", "nomp", nullptr, nullptr},
641                               {"simd", "nosimd", nullptr, nullptr},
642                               {"sec", "nosec", nullptr, nullptr},
643                               {"virt", "novirt", nullptr, nullptr},
644                               {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
645                               {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
646                               {"ras", "noras", "+ras", "-ras"},
647                               {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
648                               {"os", "noos", nullptr, nullptr},
649                               {"iwmmxt", "noiwmmxt", nullptr, nullptr},
650                               {"iwmmxt2", "noiwmmxt2", nullptr, nullptr},
651                               {"maverick", "maverick", nullptr, nullptr},
652                               {"xscale", "noxscale", nullptr, nullptr},
653                               {"sb", "nosb", "+sb", "-sb"},
654                               {"i8mm", "noi8mm", "+i8mm", "-i8mm"},
655                               {"mve", "nomve", "+mve", "-mve"},
656                               {"mve.fp", "nomve.fp", "+mve.fp", "-mve.fp"}};
657 
658   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
659     EXPECT_EQ(StringRef(ArchExt[i][2]), ARM::getArchExtFeature(ArchExt[i][0]));
660     EXPECT_EQ(StringRef(ArchExt[i][3]), ARM::getArchExtFeature(ArchExt[i][1]));
661   }
662 }
663 
664 static bool
665 testArchExtDependency(const char *ArchExt,
666                       const std::initializer_list<const char *> &Expected) {
667   std::vector<StringRef> Features;
668   unsigned FPUID;
669 
670   if (!ARM::appendArchExtFeatures("", ARM::ArchKind::ARMV8_1MMainline, ArchExt,
671                                   Features, FPUID))
672     return false;
673 
674   return llvm::all_of(Expected, [&](StringRef Ext) {
675     return llvm::is_contained(Features, Ext);
676   });
677 }
678 
679 TEST(TargetParserTest, ARMArchExtDependencies) {
680   EXPECT_TRUE(testArchExtDependency("mve", {"+mve", "+dsp"}));
681   EXPECT_TRUE(testArchExtDependency("mve.fp", {"+mve.fp", "+mve", "+dsp"}));
682   EXPECT_TRUE(testArchExtDependency("nodsp", {"-dsp", "-mve", "-mve.fp"}));
683   EXPECT_TRUE(testArchExtDependency("nomve", {"-mve", "-mve.fp"}));
684 }
685 
686 TEST(TargetParserTest, ARMparseHWDiv) {
687   const char *hwdiv[] = {"thumb", "arm", "arm,thumb", "thumb,arm"};
688 
689   for (unsigned i = 0; i < array_lengthof(hwdiv); i++)
690     EXPECT_NE(ARM::AEK_INVALID, ARM::parseHWDiv((StringRef)hwdiv[i]));
691 }
692 
693 TEST(TargetParserTest, ARMparseArchEndianAndISA) {
694   const char *Arch[] = {
695       "v2",   "v2a",    "v3",    "v3m",    "v4",    "v4t",    "v5",    "v5t",
696       "v5e",  "v5te",   "v5tej", "v6",     "v6j",   "v6k",    "v6hl",  "v6t2",
697       "v6kz", "v6z",    "v6zk",  "v6-m",   "v6m",   "v6sm",   "v6s-m", "v7-a",
698       "v7",   "v7a",    "v7ve",  "v7hl",   "v7l",   "v7-r",   "v7r",   "v7-m",
699       "v7m",  "v7k",    "v7s",   "v7e-m",  "v7em",  "v8-a",   "v8",    "v8a",
700       "v8l",  "v8.1-a", "v8.1a", "v8.2-a", "v8.2a", "v8.3-a", "v8.3a", "v8.4-a",
701       "v8.4a", "v8.5-a","v8.5a", "v8.6-a", "v8.6a", "v8-r",   "v8m.base", "v8m.main", "v8.1m.main"
702   };
703 
704   for (unsigned i = 0; i < array_lengthof(Arch); i++) {
705     std::string arm_1 = "armeb" + (std::string)(Arch[i]);
706     std::string arm_2 = "arm" + (std::string)(Arch[i]) + "eb";
707     std::string arm_3 = "arm" + (std::string)(Arch[i]);
708     std::string thumb_1 = "thumbeb" + (std::string)(Arch[i]);
709     std::string thumb_2 = "thumb" + (std::string)(Arch[i]) + "eb";
710     std::string thumb_3 = "thumb" + (std::string)(Arch[i]);
711 
712     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_1));
713     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_2));
714     EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(arm_3));
715 
716     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_1));
717     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_2));
718     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_3));
719     if (i >= 4) {
720       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_1));
721       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_2));
722       EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(thumb_3));
723 
724       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_1));
725       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_2));
726       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_3));
727     }
728   }
729 
730   EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian("aarch64"));
731   EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian("arm64_32"));
732   EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian("aarch64_be"));
733 
734   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64"));
735   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64_be"));
736   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64"));
737   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64_be"));
738   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64_32"));
739   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64_32"));
740 }
741 
742 TEST(TargetParserTest, ARMparseArchProfile) {
743   for (unsigned i = 0; i < array_lengthof(ARMArch); i++) {
744     switch (ARM::parseArch(ARMArch[i])) {
745     case ARM::ArchKind::ARMV6M:
746     case ARM::ArchKind::ARMV7M:
747     case ARM::ArchKind::ARMV7EM:
748     case ARM::ArchKind::ARMV8MMainline:
749     case ARM::ArchKind::ARMV8MBaseline:
750     case ARM::ArchKind::ARMV8_1MMainline:
751       EXPECT_EQ(ARM::ProfileKind::M, ARM::parseArchProfile(ARMArch[i]));
752       break;
753     case ARM::ArchKind::ARMV7R:
754     case ARM::ArchKind::ARMV8R:
755       EXPECT_EQ(ARM::ProfileKind::R, ARM::parseArchProfile(ARMArch[i]));
756       break;
757     case ARM::ArchKind::ARMV7A:
758     case ARM::ArchKind::ARMV7VE:
759     case ARM::ArchKind::ARMV7K:
760     case ARM::ArchKind::ARMV8A:
761     case ARM::ArchKind::ARMV8_1A:
762     case ARM::ArchKind::ARMV8_2A:
763     case ARM::ArchKind::ARMV8_3A:
764     case ARM::ArchKind::ARMV8_4A:
765     case ARM::ArchKind::ARMV8_5A:
766     case ARM::ArchKind::ARMV8_6A:
767       EXPECT_EQ(ARM::ProfileKind::A, ARM::parseArchProfile(ARMArch[i]));
768       break;
769     default:
770       EXPECT_EQ(ARM::ProfileKind::INVALID, ARM::parseArchProfile(ARMArch[i]));
771       break;
772     }
773   }
774 }
775 
776 TEST(TargetParserTest, ARMparseArchVersion) {
777   for (unsigned i = 0; i < array_lengthof(ARMArch); i++)
778     if (((std::string)ARMArch[i]).substr(0, 4) == "armv")
779       EXPECT_EQ((ARMArch[i][4] - 48u), ARM::parseArchVersion(ARMArch[i]));
780     else
781       EXPECT_EQ(5u, ARM::parseArchVersion(ARMArch[i]));
782 }
783 
784 bool testAArch64CPU(StringRef CPUName, StringRef ExpectedArch,
785                     StringRef ExpectedFPU, uint64_t ExpectedFlags,
786                     StringRef CPUAttr) {
787   AArch64::ArchKind AK = AArch64::parseCPUArch(CPUName);
788   bool pass = AArch64::getArchName(AK).equals(ExpectedArch);
789 
790   uint64_t ExtKind = AArch64::getDefaultExtensions(CPUName, AK);
791   if (ExtKind > 1 && (ExtKind & AArch64::AEK_NONE))
792     pass &= ((ExtKind ^ AArch64::AEK_NONE) == ExpectedFlags);
793   else
794     pass &= (ExtKind == ExpectedFlags);
795 
796   pass &= AArch64::getCPUAttr(AK).equals(CPUAttr);
797 
798   return pass;
799 }
800 
801 TEST(TargetParserTest, testAArch64CPU) {
802   EXPECT_TRUE(testAArch64CPU(
803       "invalid", "invalid", "invalid",
804       AArch64::AEK_NONE, ""));
805   EXPECT_TRUE(testAArch64CPU(
806       "generic", "invalid", "none",
807       AArch64::AEK_NONE, ""));
808 
809   EXPECT_TRUE(testAArch64CPU(
810       "cortex-a34", "armv8-a", "crypto-neon-fp-armv8",
811       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
812       AArch64::AEK_SIMD, "8-A"));
813   EXPECT_TRUE(testAArch64CPU(
814       "cortex-a35", "armv8-a", "crypto-neon-fp-armv8",
815       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
816       AArch64::AEK_SIMD, "8-A"));
817   EXPECT_TRUE(testAArch64CPU(
818       "cortex-a53", "armv8-a", "crypto-neon-fp-armv8",
819       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
820       AArch64::AEK_SIMD, "8-A"));
821   EXPECT_TRUE(testAArch64CPU(
822       "cortex-a55", "armv8.2-a", "crypto-neon-fp-armv8",
823       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
824       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
825       AArch64::AEK_RDM | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
826       AArch64::AEK_RCPC, "8.2-A"));
827   EXPECT_TRUE(testAArch64CPU(
828       "cortex-a57", "armv8-a", "crypto-neon-fp-armv8",
829       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
830       AArch64::AEK_SIMD, "8-A"));
831   EXPECT_TRUE(testAArch64CPU(
832       "cortex-a65", "armv8.2-a", "crypto-neon-fp-armv8",
833       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
834       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
835       AArch64::AEK_RAS | AArch64::AEK_RCPC | AArch64::AEK_RDM |
836       AArch64::AEK_SIMD | AArch64::AEK_SSBS,
837       "8.2-A"));
838   EXPECT_TRUE(testAArch64CPU(
839       "cortex-a65ae", "armv8.2-a", "crypto-neon-fp-armv8",
840       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
841       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
842       AArch64::AEK_RAS | AArch64::AEK_RCPC | AArch64::AEK_RDM |
843       AArch64::AEK_SIMD | AArch64::AEK_SSBS,
844       "8.2-A"));
845   EXPECT_TRUE(testAArch64CPU(
846       "cortex-a72", "armv8-a", "crypto-neon-fp-armv8",
847       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
848       AArch64::AEK_SIMD, "8-A"));
849   EXPECT_TRUE(testAArch64CPU(
850       "cortex-a73", "armv8-a", "crypto-neon-fp-armv8",
851       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
852       AArch64::AEK_SIMD, "8-A"));
853   EXPECT_TRUE(testAArch64CPU(
854       "cortex-a75", "armv8.2-a", "crypto-neon-fp-armv8",
855       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
856       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
857       AArch64::AEK_RDM | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
858       AArch64::AEK_RCPC, "8.2-A"));
859   EXPECT_TRUE(testAArch64CPU(
860       "cortex-a76", "armv8.2-a", "crypto-neon-fp-armv8",
861       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
862       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_RAS |
863       AArch64::AEK_LSE | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
864       AArch64::AEK_RCPC| AArch64::AEK_SSBS, "8.2-A"));
865   EXPECT_TRUE(testAArch64CPU(
866       "cortex-a76ae", "armv8.2-a", "crypto-neon-fp-armv8",
867       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
868       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_RAS |
869       AArch64::AEK_LSE | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
870       AArch64::AEK_RCPC| AArch64::AEK_SSBS, "8.2-A"));
871   EXPECT_TRUE(testAArch64CPU(
872       "cortex-a77", "armv8.2-a", "crypto-neon-fp-armv8",
873       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
874       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_RAS |
875       AArch64::AEK_LSE | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
876       AArch64::AEK_RCPC | AArch64::AEK_SSBS, "8.2-A"));
877   EXPECT_TRUE(testAArch64CPU(
878       "cortex-a78", "armv8.2-a", "crypto-neon-fp-armv8",
879       AArch64::AEK_CRC | AArch64::AEK_CRYPTO  | AArch64::AEK_FP |
880       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_RAS |
881       AArch64::AEK_LSE | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
882       AArch64::AEK_RCPC | AArch64::AEK_SSBS,
883       "8.2-A"));
884   EXPECT_TRUE(testAArch64CPU(
885       "cortex-x1", "armv8.2-a", "crypto-neon-fp-armv8",
886       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
887       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_RAS |
888       AArch64::AEK_LSE | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
889       AArch64::AEK_RCPC | AArch64::AEK_SSBS,
890       "8.2-A"));
891   EXPECT_TRUE(testAArch64CPU(
892       "cyclone", "armv8-a", "crypto-neon-fp-armv8",
893       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
894   EXPECT_TRUE(testAArch64CPU(
895       "apple-a7", "armv8-a", "crypto-neon-fp-armv8",
896       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
897   EXPECT_TRUE(testAArch64CPU(
898       "apple-a8", "armv8-a", "crypto-neon-fp-armv8",
899       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
900   EXPECT_TRUE(testAArch64CPU(
901       "apple-a9", "armv8-a", "crypto-neon-fp-armv8",
902       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
903   EXPECT_TRUE(testAArch64CPU("apple-a10", "armv8-a", "crypto-neon-fp-armv8",
904                              AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
905                                  AArch64::AEK_FP | AArch64::AEK_RDM |
906                                  AArch64::AEK_SIMD,
907                              "8-A"));
908   EXPECT_TRUE(testAArch64CPU("apple-a11", "armv8.2-a", "crypto-neon-fp-armv8",
909                              AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
910                                  AArch64::AEK_FP | AArch64::AEK_LSE |
911                                  AArch64::AEK_RAS | AArch64::AEK_RDM |
912                                  AArch64::AEK_SIMD,
913                              "8.2-A"));
914   EXPECT_TRUE(testAArch64CPU(
915       "apple-a12", "armv8.3-a", "crypto-neon-fp-armv8",
916       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
917           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
918           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_FP16,
919       "8.3-A"));
920   EXPECT_TRUE(testAArch64CPU(
921       "apple-a13", "armv8.4-a", "crypto-neon-fp-armv8",
922       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
923           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
924           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_DOTPROD |
925           AArch64::AEK_FP16 | AArch64::AEK_FP16FML,
926       "8.4-A"));
927   EXPECT_TRUE(testAArch64CPU(
928       "apple-s4", "armv8.3-a", "crypto-neon-fp-armv8",
929       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
930           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
931           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_FP16,
932       "8.3-A"));
933   EXPECT_TRUE(testAArch64CPU(
934       "apple-s5", "armv8.3-a", "crypto-neon-fp-armv8",
935       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
936           AArch64::AEK_SIMD | AArch64::AEK_LSE | AArch64::AEK_RAS |
937           AArch64::AEK_RDM | AArch64::AEK_RCPC | AArch64::AEK_FP16,
938       "8.3-A"));
939   EXPECT_TRUE(testAArch64CPU(
940       "exynos-m3", "armv8-a", "crypto-neon-fp-armv8",
941       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
942       AArch64::AEK_SIMD, "8-A"));
943   EXPECT_TRUE(testAArch64CPU(
944       "exynos-m4", "armv8.2-a", "crypto-neon-fp-armv8",
945       AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
946       AArch64::AEK_DOTPROD | AArch64::AEK_FP | AArch64::AEK_FP16 |
947       AArch64::AEK_LSE | AArch64::AEK_RAS | AArch64::AEK_RDM |
948       AArch64::AEK_SIMD, "8.2-A"));
949   EXPECT_TRUE(testAArch64CPU(
950       "exynos-m5", "armv8.2-a", "crypto-neon-fp-armv8",
951       AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
952       AArch64::AEK_DOTPROD | AArch64::AEK_FP | AArch64::AEK_FP16 |
953       AArch64::AEK_LSE | AArch64::AEK_RAS | AArch64::AEK_RDM |
954       AArch64::AEK_SIMD, "8.2-A"));
955   EXPECT_TRUE(testAArch64CPU(
956       "falkor", "armv8-a", "crypto-neon-fp-armv8",
957       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
958       AArch64::AEK_SIMD | AArch64::AEK_RDM, "8-A"));
959   EXPECT_TRUE(testAArch64CPU(
960       "kryo", "armv8-a", "crypto-neon-fp-armv8",
961       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
962       AArch64::AEK_SIMD, "8-A"));
963   EXPECT_TRUE(testAArch64CPU(
964      "neoverse-e1", "armv8.2-a", "crypto-neon-fp-armv8",
965       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
966       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
967       AArch64::AEK_RAS | AArch64::AEK_RCPC | AArch64::AEK_RDM |
968       AArch64::AEK_SIMD | AArch64::AEK_SSBS,
969      "8.2-A"));
970   EXPECT_TRUE(testAArch64CPU(
971      "neoverse-n1", "armv8.2-a", "crypto-neon-fp-armv8",
972       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_DOTPROD |
973       AArch64::AEK_FP | AArch64::AEK_FP16 | AArch64::AEK_LSE |
974       AArch64::AEK_PROFILE | AArch64::AEK_RAS | AArch64::AEK_RCPC |
975       AArch64::AEK_RDM | AArch64::AEK_SIMD | AArch64::AEK_SSBS,
976      "8.2-A"));
977   EXPECT_TRUE(testAArch64CPU(
978       "thunderx2t99", "armv8.1-a", "crypto-neon-fp-armv8",
979       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_LSE |
980       AArch64::AEK_RDM | AArch64::AEK_FP | AArch64::AEK_SIMD, "8.1-A"));
981   EXPECT_TRUE(testAArch64CPU(
982       "thunderx3t110", "armv8.3-a", "crypto-neon-fp-armv8",
983       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_LSE |
984           AArch64::AEK_RDM | AArch64::AEK_FP | AArch64::AEK_SIMD |
985           AArch64::AEK_PROFILE | AArch64::AEK_RAS | AArch64::AEK_RAND |
986           AArch64::AEK_RCPC,
987       "8.3-A"));
988   EXPECT_TRUE(testAArch64CPU(
989       "thunderx", "armv8-a", "crypto-neon-fp-armv8",
990       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
991       AArch64::AEK_FP | AArch64::AEK_PROFILE,
992       "8-A"));
993   EXPECT_TRUE(testAArch64CPU(
994       "thunderxt81", "armv8-a", "crypto-neon-fp-armv8",
995       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
996       AArch64::AEK_FP | AArch64::AEK_PROFILE,
997       "8-A"));
998   EXPECT_TRUE(testAArch64CPU(
999       "thunderxt83", "armv8-a", "crypto-neon-fp-armv8",
1000       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
1001       AArch64::AEK_FP | AArch64::AEK_PROFILE,
1002       "8-A"));
1003   EXPECT_TRUE(testAArch64CPU(
1004       "thunderxt88", "armv8-a", "crypto-neon-fp-armv8",
1005       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
1006       AArch64::AEK_FP | AArch64::AEK_PROFILE,
1007       "8-A"));
1008   EXPECT_TRUE(testAArch64CPU(
1009       "tsv110", "armv8.2-a", "crypto-neon-fp-armv8",
1010       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
1011       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
1012       AArch64::AEK_RDM | AArch64::AEK_PROFILE | AArch64::AEK_FP16 |
1013       AArch64::AEK_FP16FML | AArch64::AEK_DOTPROD,
1014       "8.2-A"));
1015   EXPECT_TRUE(testAArch64CPU(
1016       "a64fx", "armv8.2-a", "crypto-neon-fp-armv8",
1017       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
1018       AArch64::AEK_SIMD | AArch64::AEK_FP16 | AArch64::AEK_RAS |
1019       AArch64::AEK_LSE | AArch64::AEK_SVE | AArch64::AEK_RDM,
1020       "8.2-A"));
1021   EXPECT_TRUE(testAArch64CPU(
1022       "carmel", "armv8.2-a", "crypto-neon-fp-armv8",
1023       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
1024       AArch64::AEK_SIMD | AArch64::AEK_FP16 | AArch64::AEK_RAS |
1025       AArch64::AEK_LSE | AArch64::AEK_RDM,
1026       "8.2-A"));
1027 }
1028 
1029 static constexpr unsigned NumAArch64CPUArchs = 42;
1030 
1031 TEST(TargetParserTest, testAArch64CPUArchList) {
1032   SmallVector<StringRef, NumAArch64CPUArchs> List;
1033   AArch64::fillValidCPUArchList(List);
1034 
1035   // No list exists for these in this test suite, so ensure all are
1036   // valid, and match the expected 'magic' count.
1037   EXPECT_EQ(List.size(), NumAArch64CPUArchs);
1038   for(StringRef CPU : List) {
1039     EXPECT_NE(AArch64::parseCPUArch(CPU), AArch64::ArchKind::INVALID);
1040   }
1041 }
1042 
1043 bool testAArch64Arch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
1044                      unsigned ArchAttr) {
1045   AArch64::ArchKind AK = AArch64::parseArch(Arch);
1046   return (AK != AArch64::ArchKind::INVALID) &
1047          AArch64::getDefaultCPU(Arch).equals(DefaultCPU) &
1048          AArch64::getSubArch(AK).equals(SubArch) &
1049          (AArch64::getArchAttr(AK) == ArchAttr);
1050 }
1051 
1052 TEST(TargetParserTest, testAArch64Arch) {
1053   EXPECT_TRUE(testAArch64Arch("armv8-a", "cortex-a53", "v8",
1054                               ARMBuildAttrs::CPUArch::v8_A));
1055   EXPECT_TRUE(testAArch64Arch("armv8.1-a", "generic", "v8.1a",
1056                               ARMBuildAttrs::CPUArch::v8_A));
1057   EXPECT_TRUE(testAArch64Arch("armv8.2-a", "generic", "v8.2a",
1058                               ARMBuildAttrs::CPUArch::v8_A));
1059   EXPECT_TRUE(testAArch64Arch("armv8.3-a", "generic", "v8.3a",
1060                               ARMBuildAttrs::CPUArch::v8_A));
1061   EXPECT_TRUE(testAArch64Arch("armv8.4-a", "generic", "v8.4a",
1062                               ARMBuildAttrs::CPUArch::v8_A));
1063   EXPECT_TRUE(testAArch64Arch("armv8.5-a", "generic", "v8.5a",
1064                               ARMBuildAttrs::CPUArch::v8_A));
1065   EXPECT_TRUE(testAArch64Arch("armv8.6-a", "generic", "v8.6a",
1066                               ARMBuildAttrs::CPUArch::v8_A));
1067 }
1068 
1069 bool testAArch64Extension(StringRef CPUName, AArch64::ArchKind AK,
1070                           StringRef ArchExt) {
1071   return AArch64::getDefaultExtensions(CPUName, AK) &
1072          AArch64::parseArchExt(ArchExt);
1073 }
1074 
1075 TEST(TargetParserTest, testAArch64Extension) {
1076   EXPECT_FALSE(testAArch64Extension("cortex-a34",
1077                                     AArch64::ArchKind::INVALID, "ras"));
1078   EXPECT_FALSE(testAArch64Extension("cortex-a35",
1079                                     AArch64::ArchKind::INVALID, "ras"));
1080   EXPECT_FALSE(testAArch64Extension("cortex-a53",
1081                                     AArch64::ArchKind::INVALID, "ras"));
1082   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1083                                     AArch64::ArchKind::INVALID, "ras"));
1084   EXPECT_FALSE(testAArch64Extension("cortex-a57",
1085                                     AArch64::ArchKind::INVALID, "ras"));
1086   EXPECT_FALSE(testAArch64Extension("cortex-a72",
1087                                     AArch64::ArchKind::INVALID, "ras"));
1088   EXPECT_FALSE(testAArch64Extension("cortex-a73",
1089                                     AArch64::ArchKind::INVALID, "ras"));
1090   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1091                                     AArch64::ArchKind::INVALID, "ras"));
1092   EXPECT_FALSE(testAArch64Extension("cyclone",
1093                                     AArch64::ArchKind::INVALID, "ras"));
1094   EXPECT_FALSE(testAArch64Extension("exynos-m3",
1095                                     AArch64::ArchKind::INVALID, "ras"));
1096   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1097                                    AArch64::ArchKind::INVALID, "dotprod"));
1098   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1099                                    AArch64::ArchKind::INVALID, "fp16"));
1100   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1101                                    AArch64::ArchKind::INVALID, "lse"));
1102   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1103                                    AArch64::ArchKind::INVALID, "ras"));
1104   EXPECT_TRUE(testAArch64Extension("exynos-m4",
1105                                    AArch64::ArchKind::INVALID, "rdm"));
1106   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1107                                    AArch64::ArchKind::INVALID, "dotprod"));
1108   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1109                                    AArch64::ArchKind::INVALID, "fp16"));
1110   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1111                                    AArch64::ArchKind::INVALID, "lse"));
1112   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1113                                    AArch64::ArchKind::INVALID, "ras"));
1114   EXPECT_TRUE(testAArch64Extension("exynos-m5",
1115                                    AArch64::ArchKind::INVALID, "rdm"));
1116   EXPECT_TRUE(testAArch64Extension("falkor",
1117                                    AArch64::ArchKind::INVALID, "rdm"));
1118   EXPECT_FALSE(testAArch64Extension("kryo",
1119                                     AArch64::ArchKind::INVALID, "ras"));
1120   EXPECT_TRUE(testAArch64Extension("saphira",
1121                                     AArch64::ArchKind::INVALID, "crc"));
1122   EXPECT_TRUE(testAArch64Extension("saphira",
1123                                     AArch64::ArchKind::INVALID, "lse"));
1124   EXPECT_TRUE(testAArch64Extension("saphira",
1125                                    AArch64::ArchKind::INVALID, "rdm"));
1126   EXPECT_TRUE(testAArch64Extension("saphira",
1127                                    AArch64::ArchKind::INVALID, "ras"));
1128   EXPECT_TRUE(testAArch64Extension("saphira",
1129                                    AArch64::ArchKind::INVALID, "rcpc"));
1130   EXPECT_TRUE(testAArch64Extension("saphira",
1131                                    AArch64::ArchKind::INVALID, "profile"));
1132   EXPECT_FALSE(testAArch64Extension("saphira",
1133                                     AArch64::ArchKind::INVALID, "fp16"));
1134   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1135                                     AArch64::ArchKind::INVALID, "fp16"));
1136   EXPECT_FALSE(testAArch64Extension("cortex-a55",
1137                                     AArch64::ArchKind::INVALID, "fp16fml"));
1138   EXPECT_TRUE(testAArch64Extension("cortex-a55",
1139                                     AArch64::ArchKind::INVALID, "dotprod"));
1140   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1141                                     AArch64::ArchKind::INVALID, "fp16"));
1142   EXPECT_FALSE(testAArch64Extension("cortex-a75",
1143                                     AArch64::ArchKind::INVALID, "fp16fml"));
1144   EXPECT_TRUE(testAArch64Extension("cortex-a75",
1145                                     AArch64::ArchKind::INVALID, "dotprod"));
1146   EXPECT_FALSE(testAArch64Extension("thunderx2t99",
1147                                     AArch64::ArchKind::INVALID, "ras"));
1148   EXPECT_FALSE(testAArch64Extension("thunderx",
1149                                     AArch64::ArchKind::INVALID, "lse"));
1150   EXPECT_FALSE(testAArch64Extension("thunderxt81",
1151                                     AArch64::ArchKind::INVALID, "lse"));
1152   EXPECT_FALSE(testAArch64Extension("thunderxt83",
1153                                     AArch64::ArchKind::INVALID, "lse"));
1154   EXPECT_FALSE(testAArch64Extension("thunderxt88",
1155                                     AArch64::ArchKind::INVALID, "lse"));
1156   EXPECT_TRUE(testAArch64Extension("tsv110",
1157                                    AArch64::ArchKind::INVALID, "crypto"));
1158   EXPECT_FALSE(testAArch64Extension("tsv110",
1159                                     AArch64::ArchKind::INVALID, "sha3"));
1160   EXPECT_FALSE(testAArch64Extension("tsv110",
1161                                     AArch64::ArchKind::INVALID, "sm4"));
1162   EXPECT_TRUE(testAArch64Extension("tsv110",
1163                                    AArch64::ArchKind::INVALID, "ras"));
1164   EXPECT_TRUE(testAArch64Extension("tsv110",
1165                                    AArch64::ArchKind::INVALID, "profile"));
1166   EXPECT_TRUE(testAArch64Extension("tsv110",
1167                                    AArch64::ArchKind::INVALID, "fp16"));
1168   EXPECT_TRUE(testAArch64Extension("tsv110",
1169                                    AArch64::ArchKind::INVALID, "fp16fml"));
1170   EXPECT_TRUE(testAArch64Extension("tsv110",
1171                                    AArch64::ArchKind::INVALID, "dotprod"));
1172   EXPECT_TRUE(testAArch64Extension("a64fx",
1173                                    AArch64::ArchKind::INVALID, "fp16"));
1174   EXPECT_TRUE(testAArch64Extension("a64fx",
1175                                    AArch64::ArchKind::INVALID, "sve"));
1176   EXPECT_FALSE(testAArch64Extension("a64fx",
1177                                    AArch64::ArchKind::INVALID, "sve2"));
1178   EXPECT_TRUE(
1179       testAArch64Extension("carmel", AArch64::ArchKind::INVALID, "crypto"));
1180   EXPECT_TRUE(
1181       testAArch64Extension("carmel", AArch64::ArchKind::INVALID, "fp16"));
1182 
1183   EXPECT_FALSE(testAArch64Extension(
1184       "generic", AArch64::ArchKind::ARMV8A, "ras"));
1185   EXPECT_FALSE(testAArch64Extension(
1186       "generic", AArch64::ArchKind::ARMV8_1A, "ras"));
1187   EXPECT_FALSE(testAArch64Extension(
1188       "generic", AArch64::ArchKind::ARMV8_2A, "profile"));
1189   EXPECT_FALSE(testAArch64Extension(
1190       "generic", AArch64::ArchKind::ARMV8_2A, "fp16"));
1191   EXPECT_FALSE(testAArch64Extension(
1192       "generic", AArch64::ArchKind::ARMV8_2A, "fp16fml"));
1193   EXPECT_FALSE(testAArch64Extension(
1194       "generic", AArch64::ArchKind::ARMV8_3A, "fp16"));
1195   EXPECT_FALSE(testAArch64Extension(
1196       "generic", AArch64::ArchKind::ARMV8_3A, "fp16fml"));
1197   EXPECT_FALSE(testAArch64Extension(
1198       "generic", AArch64::ArchKind::ARMV8_4A, "fp16"));
1199   EXPECT_FALSE(testAArch64Extension(
1200       "generic", AArch64::ArchKind::ARMV8_4A, "fp16fml"));
1201 }
1202 
1203 TEST(TargetParserTest, AArch64ExtensionFeatures) {
1204   std::vector<uint64_t> Extensions = {
1205     AArch64::AEK_CRC,      AArch64::AEK_CRYPTO,
1206     AArch64::AEK_FP,       AArch64::AEK_SIMD,
1207     AArch64::AEK_FP16,     AArch64::AEK_PROFILE,
1208     AArch64::AEK_RAS,      AArch64::AEK_LSE,
1209     AArch64::AEK_RDM,      AArch64::AEK_DOTPROD,
1210     AArch64::AEK_SVE,      AArch64::AEK_SVE2,
1211     AArch64::AEK_SVE2AES,  AArch64::AEK_SVE2SM4,
1212     AArch64::AEK_SVE2SHA3, AArch64::AEK_SVE2BITPERM,
1213     AArch64::AEK_RCPC,     AArch64::AEK_FP16FML };
1214 
1215   std::vector<StringRef> Features;
1216 
1217   uint64_t ExtVal = 0;
1218   for (auto Ext : Extensions)
1219     ExtVal |= Ext;
1220 
1221   EXPECT_FALSE(AArch64::getExtensionFeatures(AArch64::AEK_INVALID, Features));
1222   EXPECT_TRUE(!Features.size());
1223 
1224   AArch64::getExtensionFeatures(ExtVal, Features);
1225   EXPECT_TRUE(Extensions.size() == Features.size());
1226 
1227   EXPECT_TRUE(llvm::is_contained(Features, "+crc"));
1228   EXPECT_TRUE(llvm::is_contained(Features, "+crypto"));
1229   EXPECT_TRUE(llvm::is_contained(Features, "+fp-armv8"));
1230   EXPECT_TRUE(llvm::is_contained(Features, "+neon"));
1231   EXPECT_TRUE(llvm::is_contained(Features, "+fullfp16"));
1232   EXPECT_TRUE(llvm::is_contained(Features, "+spe"));
1233   EXPECT_TRUE(llvm::is_contained(Features, "+ras"));
1234   EXPECT_TRUE(llvm::is_contained(Features, "+lse"));
1235   EXPECT_TRUE(llvm::is_contained(Features, "+rdm"));
1236   EXPECT_TRUE(llvm::is_contained(Features, "+dotprod"));
1237   EXPECT_TRUE(llvm::is_contained(Features, "+rcpc"));
1238   EXPECT_TRUE(llvm::is_contained(Features, "+fp16fml"));
1239   EXPECT_TRUE(llvm::is_contained(Features, "+sve"));
1240   EXPECT_TRUE(llvm::is_contained(Features, "+sve2"));
1241   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-aes"));
1242   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-sm4"));
1243   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-sha3"));
1244   EXPECT_TRUE(llvm::is_contained(Features, "+sve2-bitperm"));
1245 }
1246 
1247 TEST(TargetParserTest, AArch64ArchFeatures) {
1248   std::vector<StringRef> Features;
1249 
1250   for (auto AK : AArch64::ArchKinds)
1251     EXPECT_TRUE((AK == AArch64::ArchKind::INVALID)
1252                     ? !AArch64::getArchFeatures(AK, Features)
1253                     : AArch64::getArchFeatures(AK, Features));
1254 }
1255 
1256 TEST(TargetParserTest, AArch64ArchExtFeature) {
1257   const char *ArchExt[][4] = {{"crc", "nocrc", "+crc", "-crc"},
1258                               {"crypto", "nocrypto", "+crypto", "-crypto"},
1259                               {"fp", "nofp", "+fp-armv8", "-fp-armv8"},
1260                               {"simd", "nosimd", "+neon", "-neon"},
1261                               {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
1262                               {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
1263                               {"profile", "noprofile", "+spe", "-spe"},
1264                               {"ras", "noras", "+ras", "-ras"},
1265                               {"lse", "nolse", "+lse", "-lse"},
1266                               {"rdm", "nordm", "+rdm", "-rdm"},
1267                               {"sve", "nosve", "+sve", "-sve"},
1268                               {"sve2", "nosve2", "+sve2", "-sve2"},
1269                               {"sve2-aes", "nosve2-aes", "+sve2-aes",
1270                                "-sve2-aes"},
1271                               {"sve2-sm4", "nosve2-sm4", "+sve2-sm4",
1272                                "-sve2-sm4"},
1273                               {"sve2-sha3", "nosve2-sha3", "+sve2-sha3",
1274                                "-sve2-sha3"},
1275                               {"sve2-bitperm", "nosve2-bitperm",
1276                                "+sve2-bitperm", "-sve2-bitperm"},
1277                               {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
1278                               {"rcpc", "norcpc", "+rcpc", "-rcpc" },
1279                               {"rng", "norng", "+rand", "-rand"},
1280                               {"memtag", "nomemtag", "+mte", "-mte"},
1281                               {"tme", "notme", "+tme", "-tme"},
1282                               {"ssbs", "nossbs", "+ssbs", "-ssbs"},
1283                               {"sb", "nosb", "+sb", "-sb"},
1284                               {"predres", "nopredres", "+predres", "-predres"},
1285                               {"i8mm", "noi8mm", "+i8mm", "-i8mm"},
1286                               {"f32mm", "nof32mm", "+f32mm", "-f32mm"},
1287                               {"f64mm", "nof64mm", "+f64mm", "-f64mm"},
1288 };
1289 
1290   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
1291     EXPECT_EQ(StringRef(ArchExt[i][2]),
1292               AArch64::getArchExtFeature(ArchExt[i][0]));
1293     EXPECT_EQ(StringRef(ArchExt[i][3]),
1294               AArch64::getArchExtFeature(ArchExt[i][1]));
1295   }
1296 }
1297 }
1298