xref: /llvm-project/llvm/unittests/Bitcode/DataLayoutUpgradeTest.cpp (revision 46edc02eaac81bfdace3e1f906751cad114790da)
1 //===- DataLayoutUpgradeTest.cpp - Tests for DataLayout upgrades ----------===//
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/IR/AutoUpgrade.h"
10 #include "gtest/gtest.h"
11 
12 using namespace llvm;
13 
14 namespace {
15 
16 TEST(DataLayoutUpgradeTest, ValidDataLayoutUpgrade) {
17   std::string DL1 =
18       UpgradeDataLayoutString("e-m:e-p:32:32-i64:64-f80:128-n8:16:32:64-S128",
19                               "x86_64-unknown-linux-gnu");
20   std::string DL2 = UpgradeDataLayoutString(
21       "e-m:w-p:32:32-i64:64-f80:32-n8:16:32-S32", "i686-pc-windows-msvc");
22   std::string DL3 = UpgradeDataLayoutString(
23       "e-m:o-i64:64-f80:128-n8:16:32:64-S128", "x86_64-apple-macosx");
24   std::string DL4 =
25       UpgradeDataLayoutString("e-m:o-i64:64-i128:128-n32:64-S128", "aarch64--");
26   EXPECT_EQ(DL1,
27             "e-m:e-p:32:32-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128"
28             "-f80:128-n8:16:32:64-S128");
29   EXPECT_EQ(DL2,
30             "e-m:w-p:32:32-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128"
31             "-f80:128-n8:16:32-S32");
32   EXPECT_EQ(DL3, "e-m:o-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128-f80:"
33                  "128-n8:16:32:64-S128");
34   EXPECT_EQ(DL4, "e-m:o-i64:64-i128:128-n32:64-S128-Fn32");
35 
36   // Check that AMDGPU targets add -G1 if it's not present.
37   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "r600"), "e-p:32:32-G1");
38   // and that ANDGCN adds p7 and p8 as well.
39   EXPECT_EQ(
40       UpgradeDataLayoutString("e-p:64:64", "amdgcn"),
41       "e-p:64:64-G1-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
42   EXPECT_EQ(
43       UpgradeDataLayoutString("e-p:64:64-G1", "amdgcn"),
44       "e-p:64:64-G1-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
45   // but that r600 does not.
46   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G1", "r600"), "e-p:32:32-G1");
47 
48   // Ensure that the non-integral direction for address space 8 doesn't get
49   // added in to pointer declarations.
50   EXPECT_EQ(
51       UpgradeDataLayoutString(
52           "e-p:64:64-p1:64:64-p2:32:32-p3:32:32-p4:64:64-p5:32:32-p6:32:32-i64:"
53           "64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-"
54           "v1024:1024-v2048:2048-n32:64-S32-A5-G1-ni:7",
55           "amdgcn"),
56       "e-p:64:64-p1:64:64-p2:32:32-p3:32:32-p4:64:64-p5:32:32-p6:32:32-i64:64-"
57       "v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:"
58       "1024-v2048:2048-n32:64-S32-A5-G1-ni:7:8:9-p7:160:256:256:32-p8:128:128-"
59       "p9:192:256:256:32");
60 
61   // Check that RISCV64 upgrades -n64 to -n32:64.
62   EXPECT_EQ(UpgradeDataLayoutString("e-m:e-p:64:64-i64:64-i128:128-n64-S128",
63                                     "riscv64"),
64             "e-m:e-p:64:64-i64:64-i128:128-n32:64-S128");
65 
66   // Check that LoongArch64 upgrades -n64 to -n32:64.
67   EXPECT_EQ(UpgradeDataLayoutString("e-m:e-p:64:64-i64:64-i128:128-n64-S128",
68                                     "loongarch64"),
69             "e-m:e-p:64:64-i64:64-i128:128-n32:64-S128");
70 
71   // Check that SPIR && SPIRV targets add -G1 if it's not present.
72   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spir"), "e-p:32:32-G1");
73   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spir64"), "e-p:32:32-G1");
74   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spirv32"), "e-p:32:32-G1");
75   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spirv64"), "e-p:32:32-G1");
76   // but that SPIRV Logical does not.
77   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spirv"), "e-p:32:32");
78 }
79 
80 TEST(DataLayoutUpgradeTest, NoDataLayoutUpgrade) {
81   std::string DL1 = UpgradeDataLayoutString(
82       "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128-"
83       "f32:32:32"
84       "-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128"
85       "-n8:16:32:64-S128",
86       "x86_64-unknown-linux-gnu");
87   std::string DL2 = UpgradeDataLayoutString("e-m:e-i64:64-n32:64",
88                                             "powerpc64le-unknown-linux-gnu");
89   std::string DL3 = UpgradeDataLayoutString(
90       "e-m:o-i64:64-i128:128-n32:64-S128-Fn32", "aarch64--");
91   EXPECT_EQ(
92       DL1,
93       "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128"
94       "-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64"
95       "-f80:128:128-n8:16:32:64-S128");
96   EXPECT_EQ(DL2, "e-m:e-i64:64-n32:64");
97   EXPECT_EQ(DL3, "e-m:o-i64:64-i128:128-n32:64-S128-Fn32");
98 
99   // Check that AMDGPU targets don't add -G1 if there is already a -G flag.
100   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "r600"), "e-p:32:32-G2");
101   EXPECT_EQ(UpgradeDataLayoutString("G2", "r600"), "G2");
102   EXPECT_EQ(
103       UpgradeDataLayoutString("e-p:64:64-G2", "amdgcn"),
104       "e-p:64:64-G2-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
105   EXPECT_EQ(
106       UpgradeDataLayoutString("G2-e-p:64:64", "amdgcn"),
107       "G2-e-p:64:64-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
108   EXPECT_EQ(
109       UpgradeDataLayoutString("e-p:64:64-G0", "amdgcn"),
110       "e-p:64:64-G0-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
111 
112   // Check that AMDGCN targets don't add already declared address space 7.
113   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-p7:64:64", "amdgcn"),
114             "e-p:64:64-p7:64:64-G1-ni:7:8:9-p8:128:128-p9:192:256:256:32");
115   EXPECT_EQ(UpgradeDataLayoutString("p7:64:64-G2-e-p:64:64", "amdgcn"),
116             "p7:64:64-G2-e-p:64:64-ni:7:8:9-p8:128:128-p9:192:256:256:32");
117   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-p7:64:64-G1", "amdgcn"),
118             "e-p:64:64-p7:64:64-G1-ni:7:8:9-p8:128:128-p9:192:256:256:32");
119 
120   // Check that SPIR & SPIRV targets don't add -G1 if there is already a -G
121   // flag.
122   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spir"), "e-p:32:32-G2");
123   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spir64"), "e-p:32:32-G2");
124   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spirv32"), "e-p:32:32-G2");
125   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spirv64"), "e-p:32:32-G2");
126   EXPECT_EQ(UpgradeDataLayoutString("G2", "spir"), "G2");
127   EXPECT_EQ(UpgradeDataLayoutString("G2", "spir64"), "G2");
128   EXPECT_EQ(UpgradeDataLayoutString("G2", "spirv32"), "G2");
129   EXPECT_EQ(UpgradeDataLayoutString("G2", "spirv64"), "G2");
130 }
131 
132 TEST(DataLayoutUpgradeTest, EmptyDataLayout) {
133   std::string DL1 = UpgradeDataLayoutString("", "x86_64-unknown-linux-gnu");
134   std::string DL2 = UpgradeDataLayoutString(
135       "e-m:e-p:32:32-i64:64-f80:128-n8:16:32:64-S128", "");
136   EXPECT_EQ(DL1, "");
137   EXPECT_EQ(DL2, "e-m:e-p:32:32-i64:64-f80:128-n8:16:32:64-S128");
138 
139   // Check that AMDGPU targets add G1 if it's not present.
140   EXPECT_EQ(UpgradeDataLayoutString("", "r600"), "G1");
141   EXPECT_EQ(UpgradeDataLayoutString("", "amdgcn"),
142             "G1-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
143 
144   // Check that SPIR & SPIRV targets add G1 if it's not present.
145   EXPECT_EQ(UpgradeDataLayoutString("", "spir"), "G1");
146   EXPECT_EQ(UpgradeDataLayoutString("", "spir64"), "G1");
147   EXPECT_EQ(UpgradeDataLayoutString("", "spirv32"), "G1");
148   EXPECT_EQ(UpgradeDataLayoutString("", "spirv64"), "G1");
149   // but SPIRV Logical does not.
150   EXPECT_EQ(UpgradeDataLayoutString("", "spirv"), "");
151 }
152 
153 } // end namespace
154