Skip to content

Commit 5ab5810

Browse files
committed
[PhaseOrdering] Add additional test for switch with GEPs (NFC)
1 parent 3974d89 commit 5ab5810

File tree

2 files changed

+148
-75
lines changed

2 files changed

+148
-75
lines changed

llvm/test/Transforms/PhaseOrdering/switch_different_gep_types.ll

Lines changed: 0 additions & 75 deletions
This file was deleted.
Lines changed: 148 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,148 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 3
2+
; RUN: opt -S -passes='default<O1>' < %s | FileCheck %s
3+
; RUN: opt -S -passes='default<O2>' < %s | FileCheck %s
4+
; RUN: opt -S -passes='default<O3>' < %s | FileCheck %s
5+
6+
%"OpKind::Zero" = type { [1 x i32], i32 }
7+
%"OpKind::One" = type { [1 x i32], i32, i16, [1 x i16] }
8+
%"OpKind::Two" = type { [1 x i32], i32, i16, i16 }
9+
%"OpKind::Three" = type { [1 x i32], i32, i16, i16, i16, [1 x i16] }
10+
11+
; FIXME: The switch should be optimized away.
12+
define i32 @test(ptr %ptr) {
13+
; CHECK-LABEL: define i32 @test(
14+
; CHECK-SAME: ptr nocapture readonly [[PTR:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {
15+
; CHECK-NEXT: start:
16+
; CHECK-NEXT: [[T:%.*]] = load i32, ptr [[PTR]], align 4
17+
; CHECK-NEXT: switch i32 [[T]], label [[DEFAULT:%.*]] [
18+
; CHECK-NEXT: i32 0, label [[BB4:%.*]]
19+
; CHECK-NEXT: i32 1, label [[BB5:%.*]]
20+
; CHECK-NEXT: i32 2, label [[BB6:%.*]]
21+
; CHECK-NEXT: i32 3, label [[BB7:%.*]]
22+
; CHECK-NEXT: ]
23+
; CHECK: default:
24+
; CHECK-NEXT: unreachable
25+
; CHECK: bb4:
26+
; CHECK-NEXT: [[GEP0:%.*]] = getelementptr inbounds %"OpKind::Zero", ptr [[PTR]], i64 0, i32 1
27+
; CHECK-NEXT: br label [[EXIT:%.*]]
28+
; CHECK: bb5:
29+
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds %"OpKind::One", ptr [[PTR]], i64 0, i32 1
30+
; CHECK-NEXT: br label [[EXIT]]
31+
; CHECK: bb6:
32+
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds %"OpKind::Two", ptr [[PTR]], i64 0, i32 1
33+
; CHECK-NEXT: br label [[EXIT]]
34+
; CHECK: bb7:
35+
; CHECK-NEXT: [[GEP3:%.*]] = getelementptr inbounds %"OpKind::Three", ptr [[PTR]], i64 0, i32 1
36+
; CHECK-NEXT: br label [[EXIT]]
37+
; CHECK: exit:
38+
; CHECK-NEXT: [[PHI:%.*]] = phi ptr [ [[GEP3]], [[BB7]] ], [ [[GEP2]], [[BB6]] ], [ [[GEP1]], [[BB5]] ], [ [[GEP0]], [[BB4]] ]
39+
; CHECK-NEXT: [[RET:%.*]] = load i32, ptr [[PHI]], align 4
40+
; CHECK-NEXT: ret i32 [[RET]]
41+
;
42+
start:
43+
%t = load i32, ptr %ptr, align 4
44+
switch i32 %t, label %default [
45+
i32 0, label %bb4
46+
i32 1, label %bb5
47+
i32 2, label %bb6
48+
i32 3, label %bb7
49+
]
50+
51+
default:
52+
unreachable
53+
54+
bb4:
55+
%gep0 = getelementptr inbounds %"OpKind::Zero", ptr %ptr, i64 0, i32 1
56+
br label %exit
57+
58+
bb5:
59+
%gep1 = getelementptr inbounds %"OpKind::One", ptr %ptr, i64 0, i32 1
60+
br label %exit
61+
62+
bb6:
63+
%gep2 = getelementptr inbounds %"OpKind::Two", ptr %ptr, i64 0, i32 1
64+
br label %exit
65+
66+
bb7:
67+
%gep3 = getelementptr inbounds %"OpKind::Three", ptr %ptr, i64 0, i32 1
68+
br label %exit
69+
70+
exit:
71+
%phi = phi ptr [ %gep3, %bb7 ], [ %gep2, %bb6 ], [ %gep1, %bb5 ], [ %gep0, %bb4 ]
72+
%ret = load i32, ptr %phi, align 4
73+
ret i32 %ret
74+
}
75+
76+
%X = type { i64, i64, i64, i64, i64, i64 }
77+
78+
; FIXME: The switch should be optimized away.
79+
define void @test2(ptr %self, i64 %v, i64 %ix) {
80+
; CHECK-LABEL: define void @test2(
81+
; CHECK-SAME: ptr nocapture writeonly [[SELF:%.*]], i64 [[V:%.*]], i64 [[IX:%.*]]) local_unnamed_addr #[[ATTR1:[0-9]+]] {
82+
; CHECK-NEXT: start:
83+
; CHECK-NEXT: switch i64 [[IX]], label [[DEFAULT:%.*]] [
84+
; CHECK-NEXT: i64 1, label [[BB3:%.*]]
85+
; CHECK-NEXT: i64 2, label [[BB4:%.*]]
86+
; CHECK-NEXT: i64 3, label [[BB5:%.*]]
87+
; CHECK-NEXT: i64 4, label [[BB6:%.*]]
88+
; CHECK-NEXT: i64 5, label [[BB7:%.*]]
89+
; CHECK-NEXT: ]
90+
; CHECK: default:
91+
; CHECK-NEXT: unreachable
92+
; CHECK: bb3:
93+
; CHECK-NEXT: [[GEP1:%.*]] = getelementptr inbounds [[X:%.*]], ptr [[SELF]], i64 0, i32 1
94+
; CHECK-NEXT: br label [[BB8:%.*]]
95+
; CHECK: bb4:
96+
; CHECK-NEXT: [[GEP2:%.*]] = getelementptr inbounds [[X]], ptr [[SELF]], i64 0, i32 2
97+
; CHECK-NEXT: br label [[BB8]]
98+
; CHECK: bb5:
99+
; CHECK-NEXT: [[GEP3:%.*]] = getelementptr inbounds [[X]], ptr [[SELF]], i64 0, i32 3
100+
; CHECK-NEXT: br label [[BB8]]
101+
; CHECK: bb6:
102+
; CHECK-NEXT: [[GEP4:%.*]] = getelementptr inbounds [[X]], ptr [[SELF]], i64 0, i32 4
103+
; CHECK-NEXT: br label [[BB8]]
104+
; CHECK: bb7:
105+
; CHECK-NEXT: [[GEP5:%.*]] = getelementptr inbounds [[X]], ptr [[SELF]], i64 0, i32 5
106+
; CHECK-NEXT: br label [[BB8]]
107+
; CHECK: bb8:
108+
; CHECK-NEXT: [[PTR:%.*]] = phi ptr [ [[GEP5]], [[BB7]] ], [ [[GEP4]], [[BB6]] ], [ [[GEP3]], [[BB5]] ], [ [[GEP2]], [[BB4]] ], [ [[GEP1]], [[BB3]] ]
109+
; CHECK-NEXT: store i64 [[V]], ptr [[PTR]], align 8
110+
; CHECK-NEXT: ret void
111+
;
112+
start:
113+
switch i64 %ix, label %default [
114+
i64 1, label %bb3
115+
i64 2, label %bb4
116+
i64 3, label %bb5
117+
i64 4, label %bb6
118+
i64 5, label %bb7
119+
]
120+
121+
default:
122+
unreachable
123+
124+
bb3:
125+
%gep1 = getelementptr inbounds %X, ptr %self, i64 0, i32 1
126+
br label %bb8
127+
128+
bb4:
129+
%gep2 = getelementptr inbounds %X, ptr %self, i64 0, i32 2
130+
br label %bb8
131+
132+
bb5:
133+
%gep3 = getelementptr inbounds %X, ptr %self, i64 0, i32 3
134+
br label %bb8
135+
136+
bb6:
137+
%gep4 = getelementptr inbounds %X, ptr %self, i64 0, i32 4
138+
br label %bb8
139+
140+
bb7:
141+
%gep5 = getelementptr inbounds %X, ptr %self, i64 0, i32 5
142+
br label %bb8
143+
144+
bb8:
145+
%ptr = phi ptr [ %gep5, %bb7 ], [ %gep4, %bb6 ], [ %gep3, %bb5 ], [ %gep2, %bb4 ], [ %gep1, %bb3 ]
146+
store i64 %v, ptr %ptr, align 8
147+
ret void
148+
}

0 commit comments

Comments
 (0)