1 //===----------- TargetParser.cpp - Target Parser -------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/Support/TargetParser.h"
11 #include "llvm/ADT/STLExtras.h"
12 #include "llvm/Support/ARMBuildAttributes.h"
13 #include "gtest/gtest.h"
14 #include <string>
15 
16 using namespace llvm;
17 
18 namespace {
19 const char *ARMArch[] = {
20     "armv2",       "armv2a",       "armv3",       "armv3m",    "armv4",
21     "armv4t",      "armv5",        "armv5t",      "armv5e",    "armv5te",
22     "armv5tej",    "armv6",        "armv6j",      "armv6k",    "armv6hl",
23     "armv6t2",     "armv6kz",      "armv6z",      "armv6zk",   "armv6-m",
24     "armv6m",      "armv6sm",      "armv6s-m",    "armv7-a",   "armv7",
25     "armv7a",      "armv7ve",      "armv7hl",     "armv7l",    "armv7-r",
26     "armv7r",      "armv7-m",      "armv7m",      "armv7k",    "armv7s",
27     "armv7e-m",    "armv7em",      "armv8-a",     "armv8",     "armv8a",
28     "armv8l",      "armv8.1-a",    "armv8.1a",    "armv8.2-a", "armv8.2a",
29     "armv8.3-a",   "armv8.3a",     "armv8-r",     "armv8r",    "armv8-m.base",
30     "armv8m.base", "armv8-m.main", "armv8m.main", "iwmmxt",    "iwmmxt2",
31     "xscale"};
32 
33 bool testARMCPU(StringRef CPUName, StringRef ExpectedArch,
34                 StringRef ExpectedFPU, unsigned ExpectedFlags,
35                 StringRef CPUAttr) {
36   ARM::ArchKind AK = ARM::parseCPUArch(CPUName);
37   bool pass = ARM::getArchName(AK).equals(ExpectedArch);
38   unsigned FPUKind = ARM::getDefaultFPU(CPUName, AK);
39   pass &= ARM::getFPUName(FPUKind).equals(ExpectedFPU);
40 
41   unsigned ExtKind = ARM::getDefaultExtensions(CPUName, AK);
42   if (ExtKind > 1 && (ExtKind & ARM::AEK_NONE))
43     pass &= ((ExtKind ^ ARM::AEK_NONE) == ExpectedFlags);
44   else
45     pass &= (ExtKind == ExpectedFlags);
46 
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("arm1176j-s", "armv6k", "none",
129                          ARM::AEK_DSP, "6K"));
130   EXPECT_TRUE(testARMCPU("arm1176jz-s", "armv6kz", "none",
131                          ARM::AEK_SEC | ARM::AEK_DSP, "6KZ"));
132   EXPECT_TRUE(testARMCPU("mpcore", "armv6k", "vfpv2",
133                          ARM::AEK_DSP, "6K"));
134   EXPECT_TRUE(testARMCPU("mpcorenovfp", "armv6k", "none",
135                          ARM::AEK_DSP, "6K"));
136   EXPECT_TRUE(testARMCPU("arm1176jzf-s", "armv6kz", "vfpv2",
137                          ARM::AEK_SEC | ARM::AEK_DSP, "6KZ"));
138   EXPECT_TRUE(testARMCPU("arm1156t2-s", "armv6t2", "none",
139                          ARM::AEK_DSP, "6T2"));
140   EXPECT_TRUE(testARMCPU("arm1156t2f-s", "armv6t2", "vfpv2",
141                          ARM::AEK_DSP, "6T2"));
142   EXPECT_TRUE(testARMCPU("cortex-m0", "armv6-m", "none",
143                          ARM::AEK_NONE, "6-M"));
144   EXPECT_TRUE(testARMCPU("cortex-m0plus", "armv6-m", "none",
145                          ARM::AEK_NONE, "6-M"));
146   EXPECT_TRUE(testARMCPU("cortex-m1", "armv6-m", "none",
147                          ARM::AEK_NONE, "6-M"));
148   EXPECT_TRUE(testARMCPU("sc000", "armv6-m", "none",
149                          ARM::AEK_NONE, "6-M"));
150   EXPECT_TRUE(testARMCPU("cortex-a5", "armv7-a", "neon-vfpv4",
151                          ARM::AEK_MP | ARM::AEK_SEC | ARM::AEK_DSP, "7-A"));
152   EXPECT_TRUE(testARMCPU("cortex-a7", "armv7-a", "neon-vfpv4",
153                          ARM::AEK_HWDIVTHUMB | ARM::AEK_HWDIVARM | ARM::AEK_MP |
154                              ARM::AEK_SEC | ARM::AEK_VIRT | ARM::AEK_DSP,
155                          "7-A"));
156   EXPECT_TRUE(testARMCPU("cortex-a8", "armv7-a", "neon",
157                          ARM::AEK_SEC | ARM::AEK_DSP, "7-A"));
158   EXPECT_TRUE(testARMCPU("cortex-a9", "armv7-a", "neon-fp16",
159                          ARM::AEK_MP | ARM::AEK_SEC | ARM::AEK_DSP, "7-A"));
160   EXPECT_TRUE(testARMCPU("cortex-a12", "armv7-a", "neon-vfpv4",
161                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
162                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
163                              ARM::AEK_DSP,
164                          "7-A"));
165   EXPECT_TRUE(testARMCPU("cortex-a15", "armv7-a", "neon-vfpv4",
166                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
167                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
168                              ARM::AEK_DSP,
169                          "7-A"));
170   EXPECT_TRUE(testARMCPU("cortex-a17", "armv7-a", "neon-vfpv4",
171                          ARM::AEK_SEC | ARM::AEK_MP | ARM::AEK_VIRT |
172                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
173                              ARM::AEK_DSP,
174                          "7-A"));
175   EXPECT_TRUE(testARMCPU("krait", "armv7-a", "neon-vfpv4",
176                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
177                          "7-A"));
178   EXPECT_TRUE(testARMCPU("cortex-r4", "armv7-r", "none",
179                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7-R"));
180   EXPECT_TRUE(testARMCPU("cortex-r4f", "armv7-r", "vfpv3-d16",
181                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7-R"));
182   EXPECT_TRUE(testARMCPU("cortex-r5", "armv7-r", "vfpv3-d16",
183                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
184                              ARM::AEK_DSP,
185                          "7-R"));
186   EXPECT_TRUE(testARMCPU("cortex-r7", "armv7-r", "vfpv3-d16-fp16",
187                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
188                              ARM::AEK_DSP,
189                          "7-R"));
190   EXPECT_TRUE(testARMCPU("cortex-r8", "armv7-r", "vfpv3-d16-fp16",
191                          ARM::AEK_MP | ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
192                              ARM::AEK_DSP,
193                          "7-R"));
194   EXPECT_TRUE(testARMCPU("cortex-r52", "armv8-r", "neon-fp-armv8",
195                          ARM::AEK_CRC | ARM::AEK_MP | ARM::AEK_VIRT |
196                              ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB |
197                              ARM::AEK_DSP,
198                          "8-R"));
199   EXPECT_TRUE(
200       testARMCPU("sc300", "armv7-m", "none", ARM::AEK_HWDIVTHUMB, "7-M"));
201   EXPECT_TRUE(
202       testARMCPU("cortex-m3", "armv7-m", "none", ARM::AEK_HWDIVTHUMB, "7-M"));
203   EXPECT_TRUE(testARMCPU("cortex-m4", "armv7e-m", "fpv4-sp-d16",
204                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7E-M"));
205   EXPECT_TRUE(testARMCPU("cortex-m7", "armv7e-m", "fpv5-d16",
206                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "7E-M"));
207   EXPECT_TRUE(testARMCPU("cortex-a32", "armv8-a", "crypto-neon-fp-armv8",
208                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
209                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
210                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
211                          "8-A"));
212   EXPECT_TRUE(testARMCPU("cortex-a35", "armv8-a", "crypto-neon-fp-armv8",
213                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
214                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
215                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
216                          "8-A"));
217   EXPECT_TRUE(testARMCPU("cortex-a53", "armv8-a", "crypto-neon-fp-armv8",
218                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
219                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
220                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
221                          "8-A"));
222   EXPECT_TRUE(testARMCPU("cortex-a55", "armv8.2-a", "crypto-neon-fp-armv8",
223                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
224                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
225                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
226                          ARM::AEK_RAS | ARM::AEK_DOTPROD,
227                          "8.2-A"));
228   EXPECT_TRUE(testARMCPU("cortex-a57", "armv8-a", "crypto-neon-fp-armv8",
229                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
230                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
231                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
232                          "8-A"));
233   EXPECT_TRUE(testARMCPU("cortex-a72", "armv8-a", "crypto-neon-fp-armv8",
234                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
235                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
236                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
237                          "8-A"));
238   EXPECT_TRUE(testARMCPU("cortex-a73", "armv8-a", "crypto-neon-fp-armv8",
239                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
240                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
241                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
242                          "8-A"));
243   EXPECT_TRUE(testARMCPU("cortex-a75", "armv8.2-a", "crypto-neon-fp-armv8",
244                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
245                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
246                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP | ARM::AEK_FP16 |
247                          ARM::AEK_RAS | ARM::AEK_DOTPROD,
248                          "8.2-A"));
249   EXPECT_TRUE(testARMCPU("cyclone", "armv8-a", "crypto-neon-fp-armv8",
250                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
251                              ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
252                              ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
253                          "8-A"));
254   EXPECT_TRUE(testARMCPU("exynos-m1", "armv8-a", "crypto-neon-fp-armv8",
255                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
256                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
257                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
258                          "8-A"));
259   EXPECT_TRUE(testARMCPU("exynos-m2", "armv8-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,
263                          "8-A"));
264   EXPECT_TRUE(testARMCPU("exynos-m3", "armv8-a", "crypto-neon-fp-armv8",
265                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
266                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
267                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
268                          "8-A"));
269   EXPECT_TRUE(testARMCPU("exynos-m4", "armv8-a", "crypto-neon-fp-armv8",
270                          ARM::AEK_CRC | ARM::AEK_SEC | ARM::AEK_MP |
271                          ARM::AEK_VIRT | ARM::AEK_HWDIVARM |
272                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
273                          "8-A"));
274   EXPECT_TRUE(testARMCPU("cortex-m23", "armv8-m.base", "none",
275                          ARM::AEK_HWDIVTHUMB, "8-M.Baseline"));
276   EXPECT_TRUE(testARMCPU("cortex-m33", "armv8-m.main", "fpv5-sp-d16",
277                          ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP, "8-M.Mainline"));
278   EXPECT_TRUE(testARMCPU("iwmmxt", "iwmmxt", "none",
279                          ARM::AEK_NONE, "iwmmxt"));
280   EXPECT_TRUE(testARMCPU("xscale", "xscale", "none",
281                          ARM::AEK_NONE, "xscale"));
282   EXPECT_TRUE(testARMCPU("swift", "armv7s", "neon-vfpv4",
283                          ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_DSP,
284                          "7-S"));
285 }
286 
287 static constexpr unsigned NumARMCPUArchs = 83;
288 
289 TEST(TargetParserTest, testARMCPUArchList) {
290   SmallVector<StringRef, NumARMCPUArchs> List;
291   ARM::fillValidCPUArchList(List);
292 
293   // No list exists for these in this test suite, so ensure all are
294   // valid, and match the expected 'magic' count.
295   EXPECT_EQ(List.size(), NumARMCPUArchs);
296   for(StringRef CPU : List) {
297     EXPECT_NE(ARM::parseCPUArch(CPU), ARM::ArchKind::INVALID);
298   }
299 }
300 
301 TEST(TargetParserTest, testInvalidARMArch) {
302   auto InvalidArchStrings = {"armv", "armv99", "noarm"};
303   for (const char* InvalidArch : InvalidArchStrings)
304     EXPECT_EQ(ARM::parseArch(InvalidArch), ARM::ArchKind::INVALID);
305 }
306 
307 bool testARMArch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
308                  unsigned ArchAttr) {
309   ARM::ArchKind AK = ARM::parseArch(Arch);
310   return (AK!= ARM::ArchKind::INVALID) &
311          ARM::getDefaultCPU(Arch).equals(DefaultCPU) &
312          ARM::getSubArch(AK).equals(SubArch) &
313          (ARM::getArchAttr(AK) == ArchAttr);
314 }
315 
316 TEST(TargetParserTest, testARMArch) {
317   EXPECT_TRUE(
318       testARMArch("armv2", "arm2", "v2",
319                           ARMBuildAttrs::CPUArch::Pre_v4));
320   EXPECT_TRUE(
321       testARMArch("armv2a", "arm3", "v2a",
322                           ARMBuildAttrs::CPUArch::Pre_v4));
323   EXPECT_TRUE(
324       testARMArch("armv3", "arm6", "v3",
325                           ARMBuildAttrs::CPUArch::Pre_v4));
326   EXPECT_TRUE(
327       testARMArch("armv3m", "arm7m", "v3m",
328                           ARMBuildAttrs::CPUArch::Pre_v4));
329   EXPECT_TRUE(
330       testARMArch("armv4", "strongarm", "v4",
331                           ARMBuildAttrs::CPUArch::v4));
332   EXPECT_TRUE(
333       testARMArch("armv4t", "arm7tdmi", "v4t",
334                           ARMBuildAttrs::CPUArch::v4T));
335   EXPECT_TRUE(
336       testARMArch("armv5t", "arm10tdmi", "v5",
337                           ARMBuildAttrs::CPUArch::v5T));
338   EXPECT_TRUE(
339       testARMArch("armv5te", "arm1022e", "v5e",
340                           ARMBuildAttrs::CPUArch::v5TE));
341   EXPECT_TRUE(
342       testARMArch("armv5tej", "arm926ej-s", "v5e",
343                           ARMBuildAttrs::CPUArch::v5TEJ));
344   EXPECT_TRUE(
345       testARMArch("armv6", "arm1136jf-s", "v6",
346                           ARMBuildAttrs::CPUArch::v6));
347   EXPECT_TRUE(
348       testARMArch("armv6k", "arm1176j-s", "v6k",
349                           ARMBuildAttrs::CPUArch::v6K));
350   EXPECT_TRUE(
351       testARMArch("armv6t2", "arm1156t2-s", "v6t2",
352                           ARMBuildAttrs::CPUArch::v6T2));
353   EXPECT_TRUE(
354       testARMArch("armv6kz", "arm1176jzf-s", "v6kz",
355                           ARMBuildAttrs::CPUArch::v6KZ));
356   EXPECT_TRUE(
357       testARMArch("armv6-m", "cortex-m0", "v6m",
358                           ARMBuildAttrs::CPUArch::v6_M));
359   EXPECT_TRUE(
360       testARMArch("armv7-a", "generic", "v7",
361                           ARMBuildAttrs::CPUArch::v7));
362   EXPECT_TRUE(
363       testARMArch("armv7ve", "generic", "v7ve",
364                           ARMBuildAttrs::CPUArch::v7));
365   EXPECT_TRUE(
366       testARMArch("armv7-r", "cortex-r4", "v7r",
367                           ARMBuildAttrs::CPUArch::v7));
368   EXPECT_TRUE(
369       testARMArch("armv7-m", "cortex-m3", "v7m",
370                           ARMBuildAttrs::CPUArch::v7));
371   EXPECT_TRUE(
372       testARMArch("armv7e-m", "cortex-m4", "v7em",
373                           ARMBuildAttrs::CPUArch::v7E_M));
374   EXPECT_TRUE(
375       testARMArch("armv8-a", "generic", "v8",
376                           ARMBuildAttrs::CPUArch::v8_A));
377   EXPECT_TRUE(
378       testARMArch("armv8.1-a", "generic", "v8.1a",
379                           ARMBuildAttrs::CPUArch::v8_A));
380   EXPECT_TRUE(
381       testARMArch("armv8.2-a", "generic", "v8.2a",
382                           ARMBuildAttrs::CPUArch::v8_A));
383   EXPECT_TRUE(
384       testARMArch("armv8.3-a", "generic", "v8.3a",
385                           ARMBuildAttrs::CPUArch::v8_A));
386   EXPECT_TRUE(
387       testARMArch("armv8-r", "cortex-r52", "v8r",
388                           ARMBuildAttrs::CPUArch::v8_R));
389   EXPECT_TRUE(
390       testARMArch("armv8-m.base", "generic", "v8m.base",
391                           ARMBuildAttrs::CPUArch::v8_M_Base));
392   EXPECT_TRUE(
393       testARMArch("armv8-m.main", "generic", "v8m.main",
394                           ARMBuildAttrs::CPUArch::v8_M_Main));
395   EXPECT_TRUE(
396       testARMArch("iwmmxt", "iwmmxt", "",
397                           ARMBuildAttrs::CPUArch::v5TE));
398   EXPECT_TRUE(
399       testARMArch("iwmmxt2", "generic", "",
400                           ARMBuildAttrs::CPUArch::v5TE));
401   EXPECT_TRUE(
402       testARMArch("xscale", "xscale", "v5e",
403                           ARMBuildAttrs::CPUArch::v5TE));
404   EXPECT_TRUE(
405       testARMArch("armv7s", "swift", "v7s",
406                           ARMBuildAttrs::CPUArch::v7));
407   EXPECT_TRUE(
408       testARMArch("armv7k", "generic", "v7k",
409                           ARMBuildAttrs::CPUArch::v7));
410 }
411 
412 bool testARMExtension(StringRef CPUName,ARM::ArchKind ArchKind, StringRef ArchExt) {
413   return ARM::getDefaultExtensions(CPUName, ArchKind) &
414          ARM::parseArchExt(ArchExt);
415 }
416 
417 TEST(TargetParserTest, testARMExtension) {
418   EXPECT_FALSE(testARMExtension("arm2", ARM::ArchKind::INVALID, "thumb"));
419   EXPECT_FALSE(testARMExtension("arm3", ARM::ArchKind::INVALID, "thumb"));
420   EXPECT_FALSE(testARMExtension("arm6", ARM::ArchKind::INVALID, "thumb"));
421   EXPECT_FALSE(testARMExtension("arm7m", ARM::ArchKind::INVALID, "thumb"));
422   EXPECT_FALSE(testARMExtension("strongarm", ARM::ArchKind::INVALID, "dsp"));
423   EXPECT_FALSE(testARMExtension("arm7tdmi", ARM::ArchKind::INVALID, "dsp"));
424   EXPECT_FALSE(testARMExtension("arm10tdmi",
425                                 ARM::ArchKind::INVALID, "simd"));
426   EXPECT_FALSE(testARMExtension("arm1022e", ARM::ArchKind::INVALID, "simd"));
427   EXPECT_FALSE(testARMExtension("arm926ej-s",
428                                 ARM::ArchKind::INVALID, "simd"));
429   EXPECT_FALSE(testARMExtension("arm1136jf-s",
430                                 ARM::ArchKind::INVALID, "crypto"));
431   EXPECT_FALSE(testARMExtension("arm1176j-s",
432                                 ARM::ArchKind::INVALID, "crypto"));
433   EXPECT_FALSE(testARMExtension("arm1156t2-s",
434                                 ARM::ArchKind::INVALID, "crypto"));
435   EXPECT_FALSE(testARMExtension("arm1176jzf-s",
436                                 ARM::ArchKind::INVALID, "crypto"));
437   EXPECT_FALSE(testARMExtension("cortex-m0",
438                                 ARM::ArchKind::INVALID, "crypto"));
439   EXPECT_FALSE(testARMExtension("cortex-a8",
440                                 ARM::ArchKind::INVALID, "crypto"));
441   EXPECT_FALSE(testARMExtension("cortex-r4",
442                                 ARM::ArchKind::INVALID, "crypto"));
443   EXPECT_FALSE(testARMExtension("cortex-m3",
444                                 ARM::ArchKind::INVALID, "crypto"));
445   EXPECT_FALSE(testARMExtension("cortex-a53",
446                                 ARM::ArchKind::INVALID, "ras"));
447   EXPECT_FALSE(testARMExtension("cortex-a53",
448                                 ARM::ArchKind::INVALID, "fp16"));
449   EXPECT_TRUE(testARMExtension("cortex-a55",
450                                 ARM::ArchKind::INVALID, "fp16"));
451   EXPECT_FALSE(testARMExtension("cortex-a55",
452                                 ARM::ArchKind::INVALID, "fp16fml"));
453   EXPECT_TRUE(testARMExtension("cortex-a75",
454                                 ARM::ArchKind::INVALID, "fp16"));
455   EXPECT_FALSE(testARMExtension("cortex-a75",
456                                 ARM::ArchKind::INVALID, "fp16fml"));
457   EXPECT_FALSE(testARMExtension("cortex-r52",
458                                 ARM::ArchKind::INVALID, "ras"));
459   EXPECT_FALSE(testARMExtension("iwmmxt", ARM::ArchKind::INVALID, "crc"));
460   EXPECT_FALSE(testARMExtension("xscale", ARM::ArchKind::INVALID, "crc"));
461   EXPECT_FALSE(testARMExtension("swift", ARM::ArchKind::INVALID, "crc"));
462 
463   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2, "thumb"));
464   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV2A, "thumb"));
465   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3, "thumb"));
466   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV3M, "thumb"));
467   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4, "dsp"));
468   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV4T, "dsp"));
469   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5T, "simd"));
470   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TE, "simd"));
471   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV5TEJ, "simd"));
472   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6, "crypto"));
473   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6K, "crypto"));
474   EXPECT_FALSE(testARMExtension("generic",
475                                 ARM::ArchKind::ARMV6T2, "crypto"));
476   EXPECT_FALSE(testARMExtension("generic",
477                                 ARM::ArchKind::ARMV6KZ, "crypto"));
478   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV6M, "crypto"));
479   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7A, "crypto"));
480   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7R, "crypto"));
481   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7M, "crypto"));
482   EXPECT_FALSE(testARMExtension("generic",
483                                 ARM::ArchKind::ARMV7EM, "crypto"));
484   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8A, "ras"));
485   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_1A, "ras"));
486   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "spe"));
487   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16"));
488   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_2A, "fp16fml"));
489   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16"));
490   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_3A, "fp16fml"));
491   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16"));
492   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8_4A, "fp16fml"));
493   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV8R, "ras"));
494   EXPECT_FALSE(testARMExtension("generic",
495                                 ARM::ArchKind::ARMV8MBaseline, "crc"));
496   EXPECT_FALSE(testARMExtension("generic",
497                                 ARM::ArchKind::ARMV8MMainline, "crc"));
498   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT, "crc"));
499   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::IWMMXT2, "crc"));
500   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::XSCALE, "crc"));
501   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7S, "crypto"));
502   EXPECT_FALSE(testARMExtension("generic", ARM::ArchKind::ARMV7K, "crypto"));
503 }
504 
505 TEST(TargetParserTest, ARMFPUVersion) {
506   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
507        FK <= ARM::FPUKind::FK_LAST;
508        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
509     if (FK == ARM::FK_LAST || ARM::getFPUName(FK) == "invalid" ||
510         ARM::getFPUName(FK) == "none" || ARM::getFPUName(FK) == "softvfp")
511       EXPECT_EQ(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
512     else
513       EXPECT_NE(ARM::FPUVersion::NONE, ARM::getFPUVersion(FK));
514 }
515 
516 TEST(TargetParserTest, ARMFPUNeonSupportLevel) {
517   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
518        FK <= ARM::FPUKind::FK_LAST;
519        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
520     if (FK == ARM::FK_LAST ||
521         ARM::getFPUName(FK).find("neon") == std::string::npos)
522       EXPECT_EQ(ARM::NeonSupportLevel::None,
523                  ARM::getFPUNeonSupportLevel(FK));
524     else
525       EXPECT_NE(ARM::NeonSupportLevel::None,
526                 ARM::getFPUNeonSupportLevel(FK));
527 }
528 
529 TEST(TargetParserTest, ARMFPURestriction) {
530   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
531        FK <= ARM::FPUKind::FK_LAST;
532        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1)) {
533     if (FK == ARM::FK_LAST ||
534         (ARM::getFPUName(FK).find("d16") == std::string::npos &&
535          ARM::getFPUName(FK).find("vfpv3xd") == std::string::npos))
536       EXPECT_EQ(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
537     else
538       EXPECT_NE(ARM::FPURestriction::None, ARM::getFPURestriction(FK));
539   }
540 }
541 
542 TEST(TargetParserTest, ARMExtensionFeatures) {
543   std::vector<StringRef> Features;
544   unsigned Extensions = ARM::AEK_CRC | ARM::AEK_CRYPTO | ARM::AEK_DSP |
545                         ARM::AEK_HWDIVARM | ARM::AEK_HWDIVTHUMB | ARM::AEK_MP |
546                         ARM::AEK_SEC | ARM::AEK_VIRT | ARM::AEK_RAS | ARM::AEK_FP16 |
547                         ARM::AEK_FP16FML;
548 
549   for (unsigned i = 0; i <= Extensions; i++)
550     EXPECT_TRUE(i == 0 ? !ARM::getExtensionFeatures(i, Features)
551                        : ARM::getExtensionFeatures(i, Features));
552 }
553 
554 TEST(TargetParserTest, ARMFPUFeatures) {
555   std::vector<StringRef> Features;
556   for (ARM::FPUKind FK = static_cast<ARM::FPUKind>(0);
557        FK <= ARM::FPUKind::FK_LAST;
558        FK = static_cast<ARM::FPUKind>(static_cast<unsigned>(FK) + 1))
559     EXPECT_TRUE((FK == ARM::FK_INVALID || FK >= ARM::FK_LAST)
560                     ? !ARM::getFPUFeatures(FK, Features)
561                     : ARM::getFPUFeatures(FK, Features));
562 }
563 
564 TEST(TargetParserTest, ARMArchExtFeature) {
565   const char *ArchExt[][4] = {{"crc", "nocrc", "+crc", "-crc"},
566                               {"crypto", "nocrypto", "+crypto", "-crypto"},
567                               {"dsp", "nodsp", "+dsp", "-dsp"},
568                               {"fp", "nofp", nullptr, nullptr},
569                               {"idiv", "noidiv", nullptr, nullptr},
570                               {"mp", "nomp", nullptr, nullptr},
571                               {"simd", "nosimd", nullptr, nullptr},
572                               {"sec", "nosec", nullptr, nullptr},
573                               {"virt", "novirt", nullptr, nullptr},
574                               {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
575                               {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
576                               {"ras", "noras", "+ras", "-ras"},
577                               {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
578                               {"os", "noos", nullptr, nullptr},
579                               {"iwmmxt", "noiwmmxt", nullptr, nullptr},
580                               {"iwmmxt2", "noiwmmxt2", nullptr, nullptr},
581                               {"maverick", "maverick", nullptr, nullptr},
582                               {"xscale", "noxscale", nullptr, nullptr}};
583 
584   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
585     EXPECT_EQ(StringRef(ArchExt[i][2]), ARM::getArchExtFeature(ArchExt[i][0]));
586     EXPECT_EQ(StringRef(ArchExt[i][3]), ARM::getArchExtFeature(ArchExt[i][1]));
587   }
588 }
589 
590 TEST(TargetParserTest, ARMparseHWDiv) {
591   const char *hwdiv[] = {"thumb", "arm", "arm,thumb", "thumb,arm"};
592 
593   for (unsigned i = 0; i < array_lengthof(hwdiv); i++)
594     EXPECT_NE(ARM::AEK_INVALID, ARM::parseHWDiv((StringRef)hwdiv[i]));
595 }
596 
597 TEST(TargetParserTest, ARMparseArchEndianAndISA) {
598   const char *Arch[] = {
599       "v2",   "v2a",    "v3",    "v3m",    "v4",    "v4t",    "v5",    "v5t",
600       "v5e",  "v5te",   "v5tej", "v6",     "v6j",   "v6k",    "v6hl",  "v6t2",
601       "v6kz", "v6z",    "v6zk",  "v6-m",   "v6m",   "v6sm",   "v6s-m", "v7-a",
602       "v7",   "v7a",    "v7ve",  "v7hl",   "v7l",   "v7-r",   "v7r",   "v7-m",
603       "v7m",  "v7k",    "v7s",   "v7e-m",  "v7em",  "v8-a",   "v8",    "v8a",
604       "v8l",  "v8.1-a", "v8.1a", "v8.2-a", "v8.2a", "v8.3-a", "v8.3a", "v8-r"};
605 
606   for (unsigned i = 0; i < array_lengthof(Arch); i++) {
607     std::string arm_1 = "armeb" + (std::string)(Arch[i]);
608     std::string arm_2 = "arm" + (std::string)(Arch[i]) + "eb";
609     std::string arm_3 = "arm" + (std::string)(Arch[i]);
610     std::string thumb_1 = "thumbeb" + (std::string)(Arch[i]);
611     std::string thumb_2 = "thumb" + (std::string)(Arch[i]) + "eb";
612     std::string thumb_3 = "thumb" + (std::string)(Arch[i]);
613 
614     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_1));
615     EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(arm_2));
616     EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(arm_3));
617 
618     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_1));
619     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_2));
620     EXPECT_EQ(ARM::ISAKind::ARM, ARM::parseArchISA(arm_3));
621     if (i >= 4) {
622       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_1));
623       EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian(thumb_2));
624       EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian(thumb_3));
625 
626       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_1));
627       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_2));
628       EXPECT_EQ(ARM::ISAKind::THUMB, ARM::parseArchISA(thumb_3));
629     }
630   }
631 
632   EXPECT_EQ(ARM::EndianKind::LITTLE, ARM::parseArchEndian("aarch64"));
633   EXPECT_EQ(ARM::EndianKind::BIG, ARM::parseArchEndian("aarch64_be"));
634 
635   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64"));
636   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("aarch64_be"));
637   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64"));
638   EXPECT_EQ(ARM::ISAKind::AARCH64, ARM::parseArchISA("arm64_be"));
639 }
640 
641 TEST(TargetParserTest, ARMparseArchProfile) {
642   for (unsigned i = 0; i < array_lengthof(ARMArch); i++) {
643     switch (ARM::parseArch(ARMArch[i])) {
644     case ARM::ArchKind::ARMV6M:
645     case ARM::ArchKind::ARMV7M:
646     case ARM::ArchKind::ARMV7EM:
647     case ARM::ArchKind::ARMV8MMainline:
648     case ARM::ArchKind::ARMV8MBaseline:
649       EXPECT_EQ(ARM::ProfileKind::M, ARM::parseArchProfile(ARMArch[i]));
650       break;
651     case ARM::ArchKind::ARMV7R:
652     case ARM::ArchKind::ARMV8R:
653       EXPECT_EQ(ARM::ProfileKind::R, ARM::parseArchProfile(ARMArch[i]));
654       break;
655     case ARM::ArchKind::ARMV7A:
656     case ARM::ArchKind::ARMV7VE:
657     case ARM::ArchKind::ARMV7K:
658     case ARM::ArchKind::ARMV8A:
659     case ARM::ArchKind::ARMV8_1A:
660     case ARM::ArchKind::ARMV8_2A:
661     case ARM::ArchKind::ARMV8_3A:
662       EXPECT_EQ(ARM::ProfileKind::A, ARM::parseArchProfile(ARMArch[i]));
663       break;
664     default:
665       EXPECT_EQ(ARM::ProfileKind::INVALID, ARM::parseArchProfile(ARMArch[i]));
666       break;
667     }
668   }
669 }
670 
671 TEST(TargetParserTest, ARMparseArchVersion) {
672   for (unsigned i = 0; i < array_lengthof(ARMArch); i++)
673     if (((std::string)ARMArch[i]).substr(0, 4) == "armv")
674       EXPECT_EQ((ARMArch[i][4] - 48u), ARM::parseArchVersion(ARMArch[i]));
675     else
676       EXPECT_EQ(5u, ARM::parseArchVersion(ARMArch[i]));
677 }
678 
679 bool testAArch64CPU(StringRef CPUName, StringRef ExpectedArch,
680                     StringRef ExpectedFPU, unsigned ExpectedFlags,
681                     StringRef CPUAttr) {
682   AArch64::ArchKind AK = AArch64::parseCPUArch(CPUName);
683   bool pass = AArch64::getArchName(AK).equals(ExpectedArch);
684   unsigned FPUKind = AArch64::getDefaultFPU(CPUName, AK);
685   pass &= AArch64::getFPUName(FPUKind).equals(ExpectedFPU);
686 
687   unsigned ExtKind = AArch64::getDefaultExtensions(CPUName, AK);
688   if (ExtKind > 1 && (ExtKind & AArch64::AEK_NONE))
689     pass &= ((ExtKind ^ AArch64::AEK_NONE) == ExpectedFlags);
690   else
691     pass &= (ExtKind == ExpectedFlags);
692 
693   pass &= AArch64::getCPUAttr(AK).equals(CPUAttr);
694 
695   return pass;
696 }
697 
698 TEST(TargetParserTest, testAArch64CPU) {
699   EXPECT_TRUE(testAArch64CPU(
700       "invalid", "invalid", "invalid",
701       AArch64::AEK_NONE, ""));
702   EXPECT_TRUE(testAArch64CPU(
703       "generic", "invalid", "none",
704       AArch64::AEK_NONE, ""));
705 
706   EXPECT_TRUE(testAArch64CPU(
707       "cortex-a35", "armv8-a", "crypto-neon-fp-armv8",
708       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
709       AArch64::AEK_SIMD, "8-A"));
710   EXPECT_TRUE(testAArch64CPU(
711       "cortex-a53", "armv8-a", "crypto-neon-fp-armv8",
712       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
713       AArch64::AEK_SIMD, "8-A"));
714   EXPECT_TRUE(testAArch64CPU(
715       "cortex-a55", "armv8.2-a", "crypto-neon-fp-armv8",
716       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
717       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
718       AArch64::AEK_RDM | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
719       AArch64::AEK_RCPC, "8.2-A"));
720   EXPECT_TRUE(testAArch64CPU(
721       "cortex-a57", "armv8-a", "crypto-neon-fp-armv8",
722       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
723       AArch64::AEK_SIMD, "8-A"));
724   EXPECT_TRUE(testAArch64CPU(
725       "cortex-a72", "armv8-a", "crypto-neon-fp-armv8",
726       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
727       AArch64::AEK_SIMD, "8-A"));
728   EXPECT_TRUE(testAArch64CPU(
729       "cortex-a73", "armv8-a", "crypto-neon-fp-armv8",
730       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
731       AArch64::AEK_SIMD, "8-A"));
732   EXPECT_TRUE(testAArch64CPU(
733       "cortex-a75", "armv8.2-a", "crypto-neon-fp-armv8",
734       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
735       AArch64::AEK_SIMD | AArch64::AEK_RAS | AArch64::AEK_LSE |
736       AArch64::AEK_RDM | AArch64::AEK_FP16 | AArch64::AEK_DOTPROD |
737       AArch64::AEK_RCPC, "8.2-A"));
738   EXPECT_TRUE(testAArch64CPU(
739       "cyclone", "armv8-a", "crypto-neon-fp-armv8",
740       AArch64::AEK_CRYPTO | AArch64::AEK_FP | AArch64::AEK_SIMD, "8-A"));
741   EXPECT_TRUE(testAArch64CPU(
742       "exynos-m1", "armv8-a", "crypto-neon-fp-armv8",
743       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
744       AArch64::AEK_SIMD, "8-A"));
745   EXPECT_TRUE(testAArch64CPU(
746       "exynos-m2", "armv8-a", "crypto-neon-fp-armv8",
747       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
748       AArch64::AEK_SIMD, "8-A"));
749   EXPECT_TRUE(testAArch64CPU(
750       "exynos-m3", "armv8-a", "crypto-neon-fp-armv8",
751       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
752       AArch64::AEK_SIMD, "8-A"));
753   EXPECT_TRUE(testAArch64CPU(
754       "exynos-m4", "armv8-a", "crypto-neon-fp-armv8",
755       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
756       AArch64::AEK_SIMD, "8-A"));
757   EXPECT_TRUE(testAArch64CPU(
758       "falkor", "armv8-a", "crypto-neon-fp-armv8",
759       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
760       AArch64::AEK_SIMD | AArch64::AEK_RDM, "8-A"));
761   EXPECT_TRUE(testAArch64CPU(
762       "kryo", "armv8-a", "crypto-neon-fp-armv8",
763       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_FP |
764       AArch64::AEK_SIMD, "8-A"));
765   EXPECT_TRUE(testAArch64CPU(
766       "thunderx2t99", "armv8.1-a", "crypto-neon-fp-armv8",
767       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_LSE |
768       AArch64::AEK_RDM | AArch64::AEK_FP | AArch64::AEK_SIMD, "8.1-A"));
769   EXPECT_TRUE(testAArch64CPU(
770       "thunderx", "armv8-a", "crypto-neon-fp-armv8",
771       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
772       AArch64::AEK_FP | AArch64::AEK_PROFILE,
773       "8-A"));
774   EXPECT_TRUE(testAArch64CPU(
775       "thunderxt81", "armv8-a", "crypto-neon-fp-armv8",
776       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
777       AArch64::AEK_FP | AArch64::AEK_PROFILE,
778       "8-A"));
779   EXPECT_TRUE(testAArch64CPU(
780       "thunderxt83", "armv8-a", "crypto-neon-fp-armv8",
781       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
782       AArch64::AEK_FP | AArch64::AEK_PROFILE,
783       "8-A"));
784   EXPECT_TRUE(testAArch64CPU(
785       "thunderxt88", "armv8-a", "crypto-neon-fp-armv8",
786       AArch64::AEK_CRC | AArch64::AEK_CRYPTO | AArch64::AEK_SIMD |
787       AArch64::AEK_FP | AArch64::AEK_PROFILE,
788       "8-A"));
789 }
790 
791 static constexpr unsigned NumAArch64CPUArchs = 20;
792 
793 TEST(TargetParserTest, testAArch64CPUArchList) {
794   SmallVector<StringRef, NumAArch64CPUArchs> List;
795   AArch64::fillValidCPUArchList(List);
796 
797   // No list exists for these in this test suite, so ensure all are
798   // valid, and match the expected 'magic' count.
799   EXPECT_EQ(List.size(), NumAArch64CPUArchs);
800   for(StringRef CPU : List) {
801     EXPECT_NE(AArch64::parseCPUArch(CPU), AArch64::ArchKind::INVALID);
802   }
803 }
804 
805 bool testAArch64Arch(StringRef Arch, StringRef DefaultCPU, StringRef SubArch,
806                      unsigned ArchAttr) {
807   AArch64::ArchKind AK = AArch64::parseArch(Arch);
808   return (AK != AArch64::ArchKind::INVALID) &
809          AArch64::getDefaultCPU(Arch).equals(DefaultCPU) &
810          AArch64::getSubArch(AK).equals(SubArch) &
811          (AArch64::getArchAttr(AK) == ArchAttr);
812 }
813 
814 TEST(TargetParserTest, testAArch64Arch) {
815   EXPECT_TRUE(testAArch64Arch("armv8-a", "cortex-a53", "v8",
816                               ARMBuildAttrs::CPUArch::v8_A));
817   EXPECT_TRUE(testAArch64Arch("armv8.1-a", "generic", "v8.1a",
818                               ARMBuildAttrs::CPUArch::v8_A));
819   EXPECT_TRUE(testAArch64Arch("armv8.2-a", "generic", "v8.2a",
820                               ARMBuildAttrs::CPUArch::v8_A));
821   EXPECT_TRUE(testAArch64Arch("armv8.3-a", "generic", "v8.3a",
822                               ARMBuildAttrs::CPUArch::v8_A));
823 }
824 
825 bool testAArch64Extension(StringRef CPUName, AArch64::ArchKind AK,
826                           StringRef ArchExt) {
827   return AArch64::getDefaultExtensions(CPUName, AK) &
828          AArch64::parseArchExt(ArchExt);
829 }
830 
831 TEST(TargetParserTest, testAArch64Extension) {
832   EXPECT_FALSE(testAArch64Extension("cortex-a35",
833                                     AArch64::ArchKind::INVALID, "ras"));
834   EXPECT_FALSE(testAArch64Extension("cortex-a53",
835                                     AArch64::ArchKind::INVALID, "ras"));
836   EXPECT_TRUE(testAArch64Extension("cortex-a55",
837                                     AArch64::ArchKind::INVALID, "ras"));
838   EXPECT_FALSE(testAArch64Extension("cortex-a57",
839                                     AArch64::ArchKind::INVALID, "ras"));
840   EXPECT_FALSE(testAArch64Extension("cortex-a72",
841                                     AArch64::ArchKind::INVALID, "ras"));
842   EXPECT_FALSE(testAArch64Extension("cortex-a73",
843                                     AArch64::ArchKind::INVALID, "ras"));
844   EXPECT_TRUE(testAArch64Extension("cortex-a75",
845                                     AArch64::ArchKind::INVALID, "ras"));
846   EXPECT_FALSE(testAArch64Extension("cyclone",
847                                     AArch64::ArchKind::INVALID, "ras"));
848   EXPECT_FALSE(testAArch64Extension("exynos-m1",
849                                     AArch64::ArchKind::INVALID, "ras"));
850   EXPECT_FALSE(testAArch64Extension("exynos-m2",
851                                     AArch64::ArchKind::INVALID, "ras"));
852   EXPECT_FALSE(testAArch64Extension("exynos-m3",
853                                     AArch64::ArchKind::INVALID, "ras"));
854   EXPECT_FALSE(testAArch64Extension("exynos-m4",
855                                     AArch64::ArchKind::INVALID, "ras"));
856   EXPECT_TRUE(testAArch64Extension("falkor",
857                                    AArch64::ArchKind::INVALID, "rdm"));
858   EXPECT_FALSE(testAArch64Extension("kryo",
859                                     AArch64::ArchKind::INVALID, "ras"));
860   EXPECT_TRUE(testAArch64Extension("saphira",
861                                     AArch64::ArchKind::INVALID, "crc"));
862   EXPECT_TRUE(testAArch64Extension("saphira",
863                                     AArch64::ArchKind::INVALID, "lse"));
864   EXPECT_TRUE(testAArch64Extension("saphira",
865                                    AArch64::ArchKind::INVALID, "rdm"));
866   EXPECT_TRUE(testAArch64Extension("saphira",
867                                    AArch64::ArchKind::INVALID, "ras"));
868   EXPECT_TRUE(testAArch64Extension("saphira",
869                                    AArch64::ArchKind::INVALID, "rcpc"));
870   EXPECT_TRUE(testAArch64Extension("saphira",
871                                    AArch64::ArchKind::INVALID, "profile"));
872   EXPECT_FALSE(testAArch64Extension("saphira",
873                                     AArch64::ArchKind::INVALID, "fp16"));
874   EXPECT_TRUE(testAArch64Extension("cortex-a55",
875                                     AArch64::ArchKind::INVALID, "fp16"));
876   EXPECT_FALSE(testAArch64Extension("cortex-a55",
877                                     AArch64::ArchKind::INVALID, "fp16fml"));
878   EXPECT_TRUE(testAArch64Extension("cortex-a75",
879                                     AArch64::ArchKind::INVALID, "fp16"));
880   EXPECT_FALSE(testAArch64Extension("cortex-a75",
881                                     AArch64::ArchKind::INVALID, "fp16fml"));
882   EXPECT_FALSE(testAArch64Extension("thunderx2t99",
883                                     AArch64::ArchKind::INVALID, "ras"));
884   EXPECT_FALSE(testAArch64Extension("thunderx",
885                                     AArch64::ArchKind::INVALID, "lse"));
886   EXPECT_FALSE(testAArch64Extension("thunderxt81",
887                                     AArch64::ArchKind::INVALID, "lse"));
888   EXPECT_FALSE(testAArch64Extension("thunderxt83",
889                                     AArch64::ArchKind::INVALID, "lse"));
890   EXPECT_FALSE(testAArch64Extension("thunderxt88",
891                                     AArch64::ArchKind::INVALID, "lse"));
892 
893   EXPECT_FALSE(testAArch64Extension(
894       "generic", AArch64::ArchKind::ARMV8A, "ras"));
895   EXPECT_FALSE(testAArch64Extension(
896       "generic", AArch64::ArchKind::ARMV8_1A, "ras"));
897   EXPECT_FALSE(testAArch64Extension(
898       "generic", AArch64::ArchKind::ARMV8_2A, "spe"));
899   EXPECT_FALSE(testAArch64Extension(
900       "generic", AArch64::ArchKind::ARMV8_2A, "fp16"));
901   EXPECT_FALSE(testAArch64Extension(
902       "generic", AArch64::ArchKind::ARMV8_2A, "fp16fml"));
903   EXPECT_FALSE(testAArch64Extension(
904       "generic", AArch64::ArchKind::ARMV8_3A, "fp16"));
905   EXPECT_FALSE(testAArch64Extension(
906       "generic", AArch64::ArchKind::ARMV8_3A, "fp16fml"));
907   EXPECT_FALSE(testAArch64Extension(
908       "generic", AArch64::ArchKind::ARMV8_4A, "fp16"));
909   EXPECT_FALSE(testAArch64Extension(
910       "generic", AArch64::ArchKind::ARMV8_4A, "fp16fml"));
911 }
912 
913 TEST(TargetParserTest, AArch64ExtensionFeatures) {
914   std::vector<StringRef> Features;
915   unsigned Extensions = AArch64::AEK_CRC | AArch64::AEK_CRYPTO |
916                         AArch64::AEK_FP | AArch64::AEK_SIMD |
917                         AArch64::AEK_FP16 | AArch64::AEK_PROFILE |
918                         AArch64::AEK_RAS | AArch64::AEK_LSE |
919                         AArch64::AEK_RDM | AArch64::AEK_SVE |
920                         AArch64::AEK_DOTPROD | AArch64::AEK_RCPC |
921                         AArch64::AEK_FP16FML;
922 
923   for (unsigned i = 0; i <= Extensions; i++)
924     EXPECT_TRUE(i == 0 ? !AArch64::getExtensionFeatures(i, Features)
925                        : AArch64::getExtensionFeatures(i, Features));
926 }
927 
928 TEST(TargetParserTest, AArch64ArchFeatures) {
929   std::vector<StringRef> Features;
930   AArch64::ArchKind ArchKinds[] = {
931 #define AARCH64_ARCH(NAME, ID, CPU_ATTR, SUB_ARCH, ARCH_ATTR, ARCH_FPU, ARCH_BASE_EXT)       \
932      AArch64::ArchKind::ID,
933 #include "llvm/Support/AArch64TargetParser.def"
934   };
935 
936   for (auto AK : ArchKinds)
937     EXPECT_TRUE((AK == AArch64::ArchKind::INVALID)
938                     ? !AArch64::getArchFeatures(AK, Features)
939                     : AArch64::getArchFeatures(AK, Features));
940 }
941 
942 TEST(TargetParserTest, AArch64ArchExtFeature) {
943   const char *ArchExt[][4] = {{"crc", "nocrc", "+crc", "-crc"},
944                               {"crypto", "nocrypto", "+crypto", "-crypto"},
945                               {"fp", "nofp", "+fp-armv8", "-fp-armv8"},
946                               {"simd", "nosimd", "+neon", "-neon"},
947                               {"fp16", "nofp16", "+fullfp16", "-fullfp16"},
948                               {"fp16fml", "nofp16fml", "+fp16fml", "-fp16fml"},
949                               {"profile", "noprofile", "+spe", "-spe"},
950                               {"ras", "noras", "+ras", "-ras"},
951                               {"lse", "nolse", "+lse", "-lse"},
952                               {"rdm", "nordm", "+rdm", "-rdm"},
953                               {"sve", "nosve", "+sve", "-sve"},
954                               {"dotprod", "nodotprod", "+dotprod", "-dotprod"},
955                               {"rcpc", "norcpc", "+rcpc", "-rcpc" }};
956 
957   for (unsigned i = 0; i < array_lengthof(ArchExt); i++) {
958     EXPECT_EQ(StringRef(ArchExt[i][2]),
959               AArch64::getArchExtFeature(ArchExt[i][0]));
960     EXPECT_EQ(StringRef(ArchExt[i][3]),
961               AArch64::getArchExtFeature(ArchExt[i][1]));
962   }
963 }
964 }
965