aboutsummaryrefslogtreecommitdiff
path: root/llvm/test/Transforms/InstCombine/sink-dereferenceable-assume.ll
blob: 8ceb310fc8f71ded083bc7cdeb45f4d1b740dce9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 6
; RUN: opt -p instcombine -S %s | FileCheck %s

define i64 @test_dereferenceable_assume(ptr %p, ptr %q, i1 %c.0) {
; CHECK-LABEL: define i64 @test_dereferenceable_assume(
; CHECK-SAME: ptr [[P:%.*]], ptr [[Q:%.*]], i1 [[C_0:%.*]]) {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    [[P_INT:%.*]] = ptrtoint ptr [[P]] to i64
; CHECK-NEXT:    [[Q_INT:%.*]] = ptrtoint ptr [[Q]] to i64
; CHECK-NEXT:    [[DIFF:%.*]] = sub i64 [[Q_INT]], [[P_INT]]
; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "dereferenceable"(ptr [[P]], i64 [[DIFF]]) ]
; CHECK-NEXT:    br i1 [[C_0]], label %[[THEN:.*]], label %[[ELSE:.*]]
; CHECK:       [[THEN]]:
; CHECK-NEXT:    ret i64 [[DIFF]]
; CHECK:       [[ELSE]]:
; CHECK-NEXT:    ret i64 0
;
entry:
  %p_int = ptrtoint ptr %p to i64
  %q_int = ptrtoint ptr %q to i64
  %diff = sub i64 %q_int, %p_int
  call void @llvm.assume(i1 true) [ "dereferenceable"(ptr %p, i64 %diff) ]
  br i1 %c.0, label %then, label %else

then:
  ret i64 %diff

else:
  ret i64 0
}

define i64 @test_sink_with_dereferenceable_assume_same_block_as_user(ptr %p, ptr %q, i1 %c.0) {
; CHECK-LABEL: define i64 @test_sink_with_dereferenceable_assume_same_block_as_user(
; CHECK-SAME: ptr [[P:%.*]], ptr [[Q:%.*]], i1 [[C_0:%.*]]) {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    br i1 [[C_0]], label %[[THEN:.*]], label %[[ELSE:.*]]
; CHECK:       [[THEN]]:
; CHECK-NEXT:    [[Q_INT:%.*]] = ptrtoint ptr [[Q]] to i64
; CHECK-NEXT:    [[P_INT:%.*]] = ptrtoint ptr [[P]] to i64
; CHECK-NEXT:    [[DIFF:%.*]] = sub i64 [[Q_INT]], [[P_INT]]
; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "dereferenceable"(ptr [[P]], i64 [[DIFF]]) ]
; CHECK-NEXT:    ret i64 [[DIFF]]
; CHECK:       [[ELSE]]:
; CHECK-NEXT:    ret i64 0
;
entry:
  %p_int = ptrtoint ptr %p to i64
  %q_int = ptrtoint ptr %q to i64
  %diff = sub i64 %q_int, %p_int
  br i1 %c.0, label %then, label %else

then:
  call void @llvm.assume(i1 true) [ "dereferenceable"(ptr %p, i64 %diff) ]
  ret i64 %diff

else:
  ret i64 0
}

define i64 @test_sink_with_multiple_users_dominated_by_deref(ptr %p, ptr %q, i1 %c.0, i1 %c.1) {
; CHECK-LABEL: define i64 @test_sink_with_multiple_users_dominated_by_deref(
; CHECK-SAME: ptr [[P:%.*]], ptr [[Q:%.*]], i1 [[C_0:%.*]], i1 [[C_1:%.*]]) {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    [[P_INT:%.*]] = ptrtoint ptr [[P]] to i64
; CHECK-NEXT:    [[Q_INT:%.*]] = ptrtoint ptr [[Q]] to i64
; CHECK-NEXT:    [[DIFF:%.*]] = sub i64 [[Q_INT]], [[P_INT]]
; CHECK-NEXT:    br i1 [[C_0]], label %[[THEN:.*]], label %[[ELSE:.*]]
; CHECK:       [[THEN]]:
; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "dereferenceable"(ptr [[P]], i64 [[DIFF]]) ]
; CHECK-NEXT:    br i1 [[C_1]], label %[[THEN_2:.*]], label %[[ELSE]]
; CHECK:       [[THEN_2]]:
; CHECK-NEXT:    [[DOUBLED:%.*]] = shl i64 [[DIFF]], 1
; CHECK-NEXT:    ret i64 [[DOUBLED]]
; CHECK:       [[ELSE]]:
; CHECK-NEXT:    ret i64 0
;
entry:
  %p_int = ptrtoint ptr %p to i64
  %q_int = ptrtoint ptr %q to i64
  %diff = sub i64 %q_int, %p_int
  br i1 %c.0, label %then, label %else

then:
  call void @llvm.assume(i1 true) [ "dereferenceable"(ptr %p, i64 %diff) ]
  br i1 %c.1, label %then.2, label %else

then.2:
  %doubled = mul i64 %diff, 2
  ret i64 %doubled

else:
  ret i64 0
}

define i64 @test_deref_user_does_not_dominate_other_user(ptr %p, ptr %q, i1 %c.0) {
; CHECK-LABEL: define i64 @test_deref_user_does_not_dominate_other_user(
; CHECK-SAME: ptr [[P:%.*]], ptr [[Q:%.*]], i1 [[C_0:%.*]]) {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    [[P_INT:%.*]] = ptrtoint ptr [[P]] to i64
; CHECK-NEXT:    [[Q_INT:%.*]] = ptrtoint ptr [[Q]] to i64
; CHECK-NEXT:    [[DIFF:%.*]] = sub i64 [[Q_INT]], [[P_INT]]
; CHECK-NEXT:    br i1 [[C_0]], label %[[MIDDLE:.*]], label %[[EXIT:.*]]
; CHECK:       [[MIDDLE]]:
; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "dereferenceable"(ptr [[P]], i64 [[DIFF]]) ]
; CHECK-NEXT:    br label %[[EXIT]]
; CHECK:       [[EXIT]]:
; CHECK-NEXT:    ret i64 [[DIFF]]
;
entry:
  %p_int = ptrtoint ptr %p to i64
  %q_int = ptrtoint ptr %q to i64
  %diff = sub i64 %q_int, %p_int
  br i1 %c.0, label %middle, label %exit

middle:
  call void @llvm.assume(i1 true) [ "dereferenceable"(ptr %p, i64 %diff) ]
  br label %exit

exit:
  ret i64 %diff
}

declare void @llvm.assume(i1 noundef)