Skip to content

Commit 3a3ad9f

Browse files
committed
[SimplifyCFG][LICM] Add metadata speculation tests (NFC)
1 parent 256c672 commit 3a3ad9f

File tree

2 files changed

+131
-0
lines changed

2 files changed

+131
-0
lines changed
Lines changed: 73 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,73 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals --version 2
2+
; RUN: opt -S -passes=licm < %s| FileCheck %s
3+
4+
declare void @foo(...) memory(none)
5+
6+
define void @test_unconditional(i1 %c, ptr dereferenceable(8) align 8 %p) {
7+
; CHECK-LABEL: define void @test_unconditional
8+
; CHECK-SAME: (i1 [[C:%.*]], ptr align 8 dereferenceable(8) [[P:%.*]]) {
9+
; CHECK-NEXT: [[V1:%.*]] = load i32, ptr [[P]], align 4, !range [[RNG0:![0-9]+]]
10+
; CHECK-NEXT: [[V2:%.*]] = load ptr, ptr [[P]], align 8, !nonnull !1, !noundef !1
11+
; CHECK-NEXT: [[V3:%.*]] = load ptr, ptr [[P]], align 8, !dereferenceable !2, !align !2
12+
; CHECK-NEXT: br label [[LOOP:%.*]]
13+
; CHECK: loop:
14+
; CHECK-NEXT: call void @foo(i32 [[V1]], ptr [[V2]], ptr [[V3]])
15+
; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
16+
; CHECK: exit:
17+
; CHECK-NEXT: ret void
18+
;
19+
br label %loop
20+
21+
loop:
22+
%v1 = load i32, ptr %p, !range !{i32 0, i32 10}
23+
%v2 = load ptr, ptr %p, !nonnull !{}, !noundef !{}
24+
%v3 = load ptr, ptr %p, !align !{i64 4}, !dereferenceable !{i64 4}
25+
call void @foo(i32 %v1, ptr %v2, ptr %v3)
26+
br i1 %c, label %loop, label %exit
27+
28+
exit:
29+
ret void
30+
}
31+
32+
define void @test_conditional(i1 %c, i1 %c2, ptr dereferenceable(8) align 8 %p) {
33+
; CHECK-LABEL: define void @test_conditional
34+
; CHECK-SAME: (i1 [[C:%.*]], i1 [[C2:%.*]], ptr align 8 dereferenceable(8) [[P:%.*]]) {
35+
; CHECK-NEXT: [[V1:%.*]] = load i32, ptr [[P]], align 4
36+
; CHECK-NEXT: [[V2:%.*]] = load ptr, ptr [[P]], align 8
37+
; CHECK-NEXT: [[V3:%.*]] = load ptr, ptr [[P]], align 8
38+
; CHECK-NEXT: br label [[LOOP:%.*]]
39+
; CHECK: loop:
40+
; CHECK-NEXT: br i1 [[C]], label [[IF:%.*]], label [[LATCH:%.*]]
41+
; CHECK: if:
42+
; CHECK-NEXT: call void @foo(i32 [[V1]], ptr [[V2]], ptr [[V3]])
43+
; CHECK-NEXT: br label [[LATCH]]
44+
; CHECK: latch:
45+
; CHECK-NEXT: br i1 [[C2]], label [[LOOP]], label [[EXIT:%.*]]
46+
; CHECK: exit:
47+
; CHECK-NEXT: ret void
48+
;
49+
br label %loop
50+
51+
loop:
52+
br i1 %c, label %if, label %latch
53+
54+
if:
55+
%v1 = load i32, ptr %p, !range !{i32 0, i32 10}
56+
%v2 = load ptr, ptr %p, !nonnull !{}, !noundef !{}
57+
%v3 = load ptr, ptr %p, !align !{i64 4}, !dereferenceable !{i64 4}
58+
call void @foo(i32 %v1, ptr %v2, ptr %v3)
59+
br label %latch
60+
61+
latch:
62+
br i1 %c2, label %loop, label %exit
63+
64+
exit:
65+
ret void
66+
}
67+
;.
68+
; CHECK: attributes #[[ATTR0:[0-9]+]] = { memory(none) }
69+
;.
70+
; CHECK: [[RNG0]] = !{i32 0, i32 10}
71+
; CHECK: [[META1:![0-9]+]] = !{}
72+
; CHECK: [[META2:![0-9]+]] = !{i64 4}
73+
;.

llvm/test/Transforms/SimplifyCFG/hoist-with-metadata.ll

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,64 @@ out:
6161
ret void
6262
}
6363

64+
define i32 @speculate_range(i1 %c, ptr dereferenceable(8) align 8 %p) {
65+
; CHECK-LABEL: @speculate_range(
66+
; CHECK-NEXT: entry:
67+
; CHECK-NEXT: [[V:%.*]] = load i32, ptr [[P:%.*]], align 4
68+
; CHECK-NEXT: [[SPEC_SELECT:%.*]] = select i1 [[C:%.*]], i32 [[V]], i32 0
69+
; CHECK-NEXT: ret i32 [[SPEC_SELECT]]
70+
;
71+
entry:
72+
br i1 %c, label %if, label %join
73+
74+
if:
75+
%v = load i32, ptr %p, !range !{i32 0, i32 10}
76+
br label %join
77+
78+
join:
79+
%phi = phi i32 [ %v, %if ], [ 0, %entry ]
80+
ret i32 %phi
81+
}
82+
83+
define ptr @speculate_nonnull(i1 %c, ptr dereferenceable(8) align 8 %p) {
84+
; CHECK-LABEL: @speculate_nonnull(
85+
; CHECK-NEXT: entry:
86+
; CHECK-NEXT: [[V:%.*]] = load ptr, ptr [[P:%.*]], align 8
87+
; CHECK-NEXT: [[SPEC_SELECT:%.*]] = select i1 [[C:%.*]], ptr [[V]], ptr null
88+
; CHECK-NEXT: ret ptr [[SPEC_SELECT]]
89+
;
90+
entry:
91+
br i1 %c, label %if, label %join
92+
93+
if:
94+
%v = load ptr, ptr %p, !nonnull !{}, !noundef !{}
95+
br label %join
96+
97+
join:
98+
%phi = phi ptr [ %v, %if ], [ null, %entry ]
99+
ret ptr %phi
100+
}
101+
102+
103+
define ptr @speculate_align(i1 %c, ptr dereferenceable(8) align 8 %p) {
104+
; CHECK-LABEL: @speculate_align(
105+
; CHECK-NEXT: entry:
106+
; CHECK-NEXT: [[V:%.*]] = load ptr, ptr [[P:%.*]], align 8
107+
; CHECK-NEXT: [[SPEC_SELECT:%.*]] = select i1 [[C:%.*]], ptr [[V]], ptr null
108+
; CHECK-NEXT: ret ptr [[SPEC_SELECT]]
109+
;
110+
entry:
111+
br i1 %c, label %if, label %join
112+
113+
if:
114+
%v = load ptr, ptr %p, !align !{i64 4}, !dereferenceable !{i64 4}
115+
br label %join
116+
117+
join:
118+
%phi = phi ptr [ %v, %if ], [ null, %entry ]
119+
ret ptr %phi
120+
}
121+
64122
!0 = !{ i8 0, i8 1 }
65123
!1 = !{ i8 3, i8 5 }
66124
!2 = !{}

0 commit comments

Comments
 (0)