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