1; RUN: opt -passes=argpromotion -S %s | FileCheck %s
2
3define internal i32 @callee2(i32* noundef %0) {
4; CHECK-LABEL: define {{[^@]+}}@callee2
5; CHECK-SAME: (i32 [[P_0:%.*]], i32 [[P_1:%.*]]) {
6; CHECK-NEXT:    [[SUM:%.*]] = add nsw i32 [[P_0]], [[P_1]]
7; CHECK-NEXT:    ret i32 [[SUM]]
8;
9  %2 = getelementptr inbounds i32, i32* %0, i64 0
10  %3 = load i32, i32* %2, align 4
11  %4 = getelementptr inbounds i32, i32* %0, i64 1
12  %5 = load i32, i32* %4, align 4
13  %6 = add nsw i32 %3, %5
14  ret i32 %6
15}
16
17define i32 @caller2(i32 %0, i32 %1) {
18; CHECK-LABEL: define {{[^@]+}}@caller2
19; CHECK-SAME: (i32 [[P_0:%.*]], i32 [[P_1:%.*]]) {
20; CHECK-NEXT:    [[TMP1:%.*]] = alloca [2 x i32], align 4
21; CHECK-NEXT:    [[PL_0:%.*]] = getelementptr inbounds [2 x i32], [2 x i32]* [[TMP1]], i64 0, i64 0
22; CHECK-NEXT:    store i32 [[P_0]], i32* [[PL_0]], align 4
23; CHECK-NEXT:    [[PL_1:%.*]] = getelementptr inbounds i32, i32* [[PL_0]], i64 1
24; CHECK-NEXT:    store i32 [[P_1]], i32* [[PL_1]], align 4
25; CHECK-NEXT:    [[PL_2:%.*]] = getelementptr inbounds [2 x i32], [2 x i32]* [[TMP1]], i64 0, i64 0
26; CHECK-NEXT:    [[VAL_0:%.*]] = load i32, i32* [[PL_2]], align 4
27; CHECK-NEXT:    [[PL_3:%.*]] = getelementptr i32, i32* [[PL_2]], i64 1
28; CHECK-NEXT:    [[VAL_1:%.*]] = load i32, i32* [[PL_3]], align 4
29; CHECK-NEXT:    [[RES:%.*]] = call i32 @callee2(i32 [[VAL_0]], i32 [[VAL_1]])
30; CHECK-NEXT:    ret i32 [[RES]]
31;
32  %3 = alloca [2 x i32], align 4
33  %4 = getelementptr inbounds [2 x i32], [2 x i32]* %3, i64 0, i64 0
34  store i32 %0, i32* %4, align 4
35  %5 = getelementptr inbounds i32, i32* %4, i64 1
36  store i32 %1, i32* %5, align 4
37  %6 = getelementptr inbounds [2 x i32], [2 x i32]* %3, i64 0, i64 0
38  %7 = call i32 @callee2(i32* noundef %6)
39  ret i32 %7
40}
41
42define internal i32 @callee3(i32* noundef %0) {
43; CHECK-LABEL: define {{[^@]+}}@callee3
44; CHECK-SAME: (i32* noundef [[P_0:%.*]]) {
45; CHECK-NEXT:    [[PL_0:%.*]] = getelementptr inbounds i32, i32* [[P_0]], i64 0
46; CHECK-NEXT:    [[VAL_0:%.*]] = load i32, i32* [[PL_0]], align 4
47; CHECK-NEXT:    [[PL_1:%.*]] = getelementptr inbounds i32, i32* [[P_0]], i64 1
48; CHECK-NEXT:    [[VAL_1:%.*]] = load i32, i32* [[PL_1]], align 4
49; CHECK-NEXT:    [[SUM_0:%.*]] = add nsw i32 [[VAL_0]], [[VAL_1]]
50; CHECK-NEXT:    [[PL_2:%.*]] = getelementptr inbounds i32, i32* [[P_0]], i64 2
51; CHECK-NEXT:    [[VAL_2:%.*]] = load i32, i32* [[PL_2]], align 4
52; CHECK-NEXT:    [[RES:%.*]] = add nsw i32 [[SUM_0]], [[VAL_2]]
53; CHECK-NEXT:    ret i32 [[RES]]
54;
55  %2 = getelementptr inbounds i32, i32* %0, i64 0
56  %3 = load i32, i32* %2, align 4
57  %4 = getelementptr inbounds i32, i32* %0, i64 1
58  %5 = load i32, i32* %4, align 4
59  %6 = add nsw i32 %3, %5
60  %7 = getelementptr inbounds i32, i32* %0, i64 2
61  %8 = load i32, i32* %7, align 4
62  %9 = add nsw i32 %6, %8
63  ret i32 %9
64}
65
66define i32 @caller3(i32 %0, i32 %1, i32 %2) {
67; CHECK-LABEL: define {{[^@]+}}@caller3
68; CHECK-SAME: (i32 [[P_0:%.*]], i32 [[P_1:%.*]], i32 [[P_2:%.*]]) {
69; CHECK-NEXT:    [[TMP1:%.*]] = alloca [3 x i32], align 4
70; CHECK-NEXT:    [[PL_0:%.*]] = getelementptr inbounds [3 x i32], [3 x i32]* [[TMP1]], i64 0, i64 0
71; CHECK-NEXT:    store i32 [[P_0]], i32* [[PL_0]], align 4
72; CHECK-NEXT:    [[PL_1:%.*]] = getelementptr inbounds i32, i32* [[PL_0]], i64 1
73; CHECK-NEXT:    store i32 [[P_1]], i32* [[PL_1]], align 4
74; CHECK-NEXT:    [[PL_2:%.*]] = getelementptr inbounds i32, i32* [[PL_1]], i64 1
75; CHECK-NEXT:    store i32 [[P_2]], i32* [[PL_2]], align 4
76; CHECK-NEXT:    [[PL_3:%.*]] = getelementptr inbounds [3 x i32], [3 x i32]* [[TMP1]], i64 0, i64 0
77; CHECK-NEXT:    [[RES:%.*]] = call i32 @callee3(i32* noundef [[PL_3]])
78; CHECK-NEXT:    ret i32 [[RES]]
79;
80  %4 = alloca [3 x i32], align 4
81  %5 = getelementptr inbounds [3 x i32], [3 x i32]* %4, i64 0, i64 0
82  store i32 %0, i32* %5, align 4
83  %6 = getelementptr inbounds i32, i32* %5, i64 1
84  store i32 %1, i32* %6, align 4
85  %7 = getelementptr inbounds i32, i32* %6, i64 1
86  store i32 %2, i32* %7, align 4
87  %8 = getelementptr inbounds [3 x i32], [3 x i32]* %4, i64 0, i64 0
88  %9 = call i32 @callee3(i32* noundef %8)
89  ret i32 %9
90}
91