xref: /llvm-project/llvm/unittests/Bitcode/DataLayoutUpgradeTest.cpp (revision dbad963a69fd7b16c6838f81b61167fbf00a413c)
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 SPARC targets add -i128:128.
72   EXPECT_EQ(
73       UpgradeDataLayoutString("E-m:e-p:32:32-i64:64-f128:64-n32-S64", "sparc"),
74       "E-m:e-p:32:32-i64:64-i128:128-f128:64-n32-S64");
75   EXPECT_EQ(UpgradeDataLayoutString("E-m:e-i64:64-n32:64-S128", "sparcv9"),
76             "E-m:e-i64:64-i128:128-n32:64-S128");
77 
78   // Check that SPIR && SPIRV targets add -G1 if it's not present.
79   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spir"), "e-p:32:32-G1");
80   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spir64"), "e-p:32:32-G1");
81   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spirv32"), "e-p:32:32-G1");
82   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spirv64"), "e-p:32:32-G1");
83   // but that SPIRV Logical does not.
84   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "spirv"), "e-p:32:32");
85 }
86 
87 TEST(DataLayoutUpgradeTest, NoDataLayoutUpgrade) {
88   std::string DL1 = UpgradeDataLayoutString(
89       "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128-"
90       "f32:32:32"
91       "-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128"
92       "-n8:16:32:64-S128",
93       "x86_64-unknown-linux-gnu");
94   std::string DL2 = UpgradeDataLayoutString("e-m:e-i64:64-n32:64",
95                                             "powerpc64le-unknown-linux-gnu");
96   std::string DL3 = UpgradeDataLayoutString(
97       "e-m:o-i64:64-i128:128-n32:64-S128-Fn32", "aarch64--");
98   EXPECT_EQ(
99       DL1,
100       "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128"
101       "-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64"
102       "-f80:128:128-n8:16:32:64-S128");
103   EXPECT_EQ(DL2, "e-m:e-i64:64-n32:64");
104   EXPECT_EQ(DL3, "e-m:o-i64:64-i128:128-n32:64-S128-Fn32");
105 
106   // Check that AMDGPU targets don't add -G1 if there is already a -G flag.
107   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "r600"), "e-p:32:32-G2");
108   EXPECT_EQ(UpgradeDataLayoutString("G2", "r600"), "G2");
109   EXPECT_EQ(
110       UpgradeDataLayoutString("e-p:64:64-G2", "amdgcn"),
111       "e-p:64:64-G2-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
112   EXPECT_EQ(
113       UpgradeDataLayoutString("G2-e-p:64:64", "amdgcn"),
114       "G2-e-p:64:64-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
115   EXPECT_EQ(
116       UpgradeDataLayoutString("e-p:64:64-G0", "amdgcn"),
117       "e-p:64:64-G0-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
118 
119   // Check that AMDGCN targets don't add already declared address space 7.
120   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-p7:64:64", "amdgcn"),
121             "e-p:64:64-p7:64:64-G1-ni:7:8:9-p8:128:128-p9:192:256:256:32");
122   EXPECT_EQ(UpgradeDataLayoutString("p7:64:64-G2-e-p:64:64", "amdgcn"),
123             "p7:64:64-G2-e-p:64:64-ni:7:8:9-p8:128:128-p9:192:256:256:32");
124   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-p7:64:64-G1", "amdgcn"),
125             "e-p:64:64-p7:64:64-G1-ni:7:8:9-p8:128:128-p9:192:256:256:32");
126 
127   // Check that SPIR & SPIRV targets don't add -G1 if there is already a -G
128   // flag.
129   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spir"), "e-p:32:32-G2");
130   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spir64"), "e-p:32:32-G2");
131   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spirv32"), "e-p:32:32-G2");
132   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "spirv64"), "e-p:32:32-G2");
133   EXPECT_EQ(UpgradeDataLayoutString("G2", "spir"), "G2");
134   EXPECT_EQ(UpgradeDataLayoutString("G2", "spir64"), "G2");
135   EXPECT_EQ(UpgradeDataLayoutString("G2", "spirv32"), "G2");
136   EXPECT_EQ(UpgradeDataLayoutString("G2", "spirv64"), "G2");
137 }
138 
139 TEST(DataLayoutUpgradeTest, EmptyDataLayout) {
140   std::string DL1 = UpgradeDataLayoutString("", "x86_64-unknown-linux-gnu");
141   std::string DL2 = UpgradeDataLayoutString(
142       "e-m:e-p:32:32-i64:64-f80:128-n8:16:32:64-S128", "");
143   EXPECT_EQ(DL1, "");
144   EXPECT_EQ(DL2, "e-m:e-p:32:32-i64:64-f80:128-n8:16:32:64-S128");
145 
146   // Check that AMDGPU targets add G1 if it's not present.
147   EXPECT_EQ(UpgradeDataLayoutString("", "r600"), "G1");
148   EXPECT_EQ(UpgradeDataLayoutString("", "amdgcn"),
149             "G1-ni:7:8:9-p7:160:256:256:32-p8:128:128-p9:192:256:256:32");
150 
151   // Check that SPIR & SPIRV targets add G1 if it's not present.
152   EXPECT_EQ(UpgradeDataLayoutString("", "spir"), "G1");
153   EXPECT_EQ(UpgradeDataLayoutString("", "spir64"), "G1");
154   EXPECT_EQ(UpgradeDataLayoutString("", "spirv32"), "G1");
155   EXPECT_EQ(UpgradeDataLayoutString("", "spirv64"), "G1");
156   // but SPIRV Logical does not.
157   EXPECT_EQ(UpgradeDataLayoutString("", "spirv"), "");
158 }
159 
160 } // end namespace
161