xref: /llvm-project/llvm/unittests/Bitcode/DataLayoutUpgradeTest.cpp (revision f0415f2a456d54daaa231c228d2c9f4ef2ce9b89)
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("e-m:o-i64:64-i128:128-n32:64-S128",
23                                             "x86_64-apple-macosx");
24   EXPECT_EQ(DL1, "e-m:e-p:32:32-p270:32:32-p271:32:32-p272:64:64-i64:64"
25                  "-f80:128-n8:16:32:64-S128");
26   EXPECT_EQ(DL2, "e-m:w-p:32:32-p270:32:32-p271:32:32-p272:64:64-i64:64"
27                  "-f80:128-n8:16:32-S32");
28   EXPECT_EQ(DL3, "e-m:o-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128"
29                  "-n32:64-S128");
30 
31   // Check that AMDGPU targets add -G1 if it's not present.
32   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32", "r600"), "e-p:32:32-G1");
33   // and that ANDGCN adds p7 and p8 as well.
34   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64", "amdgcn"),
35             "e-p:64:64-G1-ni:7:8-p7:160:256:256:32-p8:128:128");
36   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-G1", "amdgcn"),
37             "e-p:64:64-G1-ni:7:8-p7:160:256:256:32-p8:128:128");
38   // but that r600 does not.
39   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G1", "r600"), "e-p:32:32-G1");
40 
41   // Ensure that the non-integral direction for address space 8 doesn't get
42   // added in to pointer declarations.
43   EXPECT_EQ(
44       UpgradeDataLayoutString(
45           "e-p:64:64-p1:64:64-p2:32:32-p3:32:32-p4:64:64-p5:32:32-p6:32:32-i64:"
46           "64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-"
47           "v1024:1024-v2048:2048-n32:64-S32-A5-G1-ni:7",
48           "amdgcn"),
49       "e-p:64:64-p1:64:64-p2:32:32-p3:32:32-p4:64:64-p5:32:32-p6:32:32-i64:64-"
50       "v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:"
51       "1024-v2048:2048-n32:64-S32-A5-G1-ni:7:8-p7:160:256:256:32-p8:128:128");
52 
53   // Check that RISCV64 upgrades -n64 to -n32:64.
54   EXPECT_EQ(UpgradeDataLayoutString("e-m:e-p:64:64-i64:64-i128:128-n64-S128",
55                                     "riscv64"),
56             "e-m:e-p:64:64-i64:64-i128:128-n32:64-S128");
57 }
58 
59 TEST(DataLayoutUpgradeTest, NoDataLayoutUpgrade) {
60   std::string DL1 = UpgradeDataLayoutString(
61       "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32"
62       "-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128"
63       "-n8:16:32:64-S128",
64       "x86_64-unknown-linux-gnu");
65   std::string DL2 = UpgradeDataLayoutString("e-p:32:32", "i686-apple-darwin9");
66   std::string DL3 = UpgradeDataLayoutString("e-m:e-i64:64-n32:64",
67                                             "powerpc64le-unknown-linux-gnu");
68   std::string DL4 =
69       UpgradeDataLayoutString("e-m:o-i64:64-i128:128-n32:64-S128", "aarch64--");
70   EXPECT_EQ(DL1, "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64"
71                  "-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64"
72                  "-f80:128:128-n8:16:32:64-S128");
73   EXPECT_EQ(DL2, "e-p:32:32");
74   EXPECT_EQ(DL3, "e-m:e-i64:64-n32:64");
75   EXPECT_EQ(DL4, "e-m:o-i64:64-i128:128-n32:64-S128");
76 
77   // Check that AMDGPU targets don't add -G1 if there is already a -G flag.
78   EXPECT_EQ(UpgradeDataLayoutString("e-p:32:32-G2", "r600"), "e-p:32:32-G2");
79   EXPECT_EQ(UpgradeDataLayoutString("G2", "r600"), "G2");
80   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-G2", "amdgcn"),
81             "e-p:64:64-G2-ni:7:8-p7:160:256:256:32-p8:128:128");
82   EXPECT_EQ(UpgradeDataLayoutString("G2-e-p:64:64", "amdgcn"),
83             "G2-e-p:64:64-ni:7:8-p7:160:256:256:32-p8:128:128");
84   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-G0", "amdgcn"),
85             "e-p:64:64-G0-ni:7:8-p7:160:256:256:32-p8:128:128");
86 
87   // Check that AMDGCN targets don't add already declared address space 7.
88   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-p7:64:64", "amdgcn"),
89             "e-p:64:64-p7:64:64-G1-ni:7:8-p8:128:128");
90   EXPECT_EQ(UpgradeDataLayoutString("p7:64:64-G2-e-p:64:64", "amdgcn"),
91             "p7:64:64-G2-e-p:64:64-ni:7:8-p8:128:128");
92   EXPECT_EQ(UpgradeDataLayoutString("e-p:64:64-p7:64:64-G1", "amdgcn"),
93             "e-p:64:64-p7:64:64-G1-ni:7:8-p8:128:128");
94 }
95 
96 TEST(DataLayoutUpgradeTest, EmptyDataLayout) {
97   std::string DL1 = UpgradeDataLayoutString("", "x86_64-unknown-linux-gnu");
98   std::string DL2 = UpgradeDataLayoutString(
99       "e-m:e-p:32:32-i64:64-f80:128-n8:16:32:64-S128", "");
100   EXPECT_EQ(DL1, "");
101   EXPECT_EQ(DL2, "e-m:e-p:32:32-i64:64-f80:128-n8:16:32:64-S128");
102 
103   // Check that AMDGPU targets add G1 if it's not present.
104   EXPECT_EQ(UpgradeDataLayoutString("", "r600"), "G1");
105   EXPECT_EQ(UpgradeDataLayoutString("", "amdgcn"),
106             "G1-ni:7:8-p7:160:256:256:32-p8:128:128");
107 }
108 
109 } // end namespace
110