[llvm] 1ae4051 - [ConstraintElimination] Add tests for chained GEPs without inbounds.

Florian Hahn via llvm-commits llvm-commits at lists.llvm.org
Sat Oct 8 11:39:54 PDT 2022


Author: Florian Hahn
Date: 2022-10-08T19:39:37+01:00
New Revision: 1ae405144d62299d87096de4dc2596a04a4fee84

URL: https://github.com/llvm/llvm-project/commit/1ae405144d62299d87096de4dc2596a04a4fee84
DIFF: https://github.com/llvm/llvm-project/commit/1ae405144d62299d87096de4dc2596a04a4fee84.diff

LOG: [ConstraintElimination] Add tests for chained GEPs without inbounds.

Added: 
    

Modified: 
    llvm/test/Transforms/ConstraintElimination/gep-add.ll
    llvm/test/Transforms/ConstraintElimination/gep-sub.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/ConstraintElimination/gep-add.ll b/llvm/test/Transforms/ConstraintElimination/gep-add.ll
index e0fb5c20935ba..0cdc50f03937a 100644
--- a/llvm/test/Transforms/ConstraintElimination/gep-add.ll
+++ b/llvm/test/Transforms/ConstraintElimination/gep-add.ll
@@ -3,8 +3,8 @@
 
 declare void @llvm.assume(i1 noundef) #0
 
-define i1 @gep_add_1_uge(ptr %dst, ptr %lower) {
-; CHECK-LABEL: @gep_add_1_uge(
+define i1 @gep_add_1_uge_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_add_1_uge_inbounds(
 ; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
 ; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
 ; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr inbounds i8, ptr [[DST]], i64 3
@@ -32,6 +32,93 @@ define i1 @gep_add_1_uge(ptr %dst, ptr %lower) {
   ret i1 %res.2
 }
 
+define i1 @gep_add_1_uge_only_inner_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_add_1_uge_only_inner_inbounds(
+; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr inbounds i8, ptr [[DST]], i64 3
+; CHECK-NEXT:    [[DST_ADD_1:%.*]] = getelementptr inbounds i8, ptr [[DST]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_1:%.*]] = icmp uge ptr [[DST_ADD_1]], [[LOWER]]
+; CHECK-NEXT:    [[DST_ADD_2:%.*]] = getelementptr i8, ptr [[DST_ADD_1]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_3:%.*]] = icmp uge ptr [[DST_ADD_3]], [[LOWER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 true, true
+; CHECK-NEXT:    [[DST_ADD_4:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 3
+; CHECK-NEXT:    [[CMP_ADD_4:%.*]] = icmp uge ptr [[DST_ADD_4]], [[LOWER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_ADD_4]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %pre = icmp uge ptr %dst, %lower
+  call void @llvm.assume(i1 %pre)
+  %dst.add.3 = getelementptr inbounds i8, ptr %dst, i64 3
+  %dst.add.1 = getelementptr inbounds i8, ptr %dst, i64 1
+  %cmp.add.1 = icmp uge ptr %dst.add.1, %lower
+  %dst.add.2 = getelementptr i8, ptr %dst.add.1, i64 1
+  %cmp.add.3 = icmp uge ptr %dst.add.3, %lower
+  %res.1 = xor i1 %cmp.add.1, %cmp.add.3
+  %dst.add.4 = getelementptr i8, ptr %dst.add.3, i64 3
+  %cmp.add.4 = icmp uge ptr %dst.add.4, %lower
+  %res.2 = xor i1 %res.1, %cmp.add.4
+  ret i1 %res.2
+}
+
+define i1 @gep_add_1_uge_only_outer_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_add_1_uge_only_outer_inbounds(
+; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr i8, ptr [[DST]], i64 3
+; CHECK-NEXT:    [[DST_ADD_1:%.*]] = getelementptr i8, ptr [[DST]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_1:%.*]] = icmp uge ptr [[DST_ADD_1]], [[LOWER]]
+; CHECK-NEXT:    [[DST_ADD_2:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_1]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_3:%.*]] = icmp uge ptr [[DST_ADD_3]], [[LOWER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 [[CMP_ADD_1]], [[CMP_ADD_3]]
+; CHECK-NEXT:    [[DST_ADD_4:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_3]], i64 3
+; CHECK-NEXT:    [[CMP_ADD_4:%.*]] = icmp uge ptr [[DST_ADD_4]], [[LOWER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_ADD_4]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %pre = icmp uge ptr %dst, %lower
+  call void @llvm.assume(i1 %pre)
+  %dst.add.3 = getelementptr i8, ptr %dst, i64 3
+  %dst.add.1 = getelementptr i8, ptr %dst, i64 1
+  %cmp.add.1 = icmp uge ptr %dst.add.1, %lower
+  %dst.add.2 = getelementptr inbounds i8, ptr %dst.add.1, i64 1
+  %cmp.add.3 = icmp uge ptr %dst.add.3, %lower
+  %res.1 = xor i1 %cmp.add.1, %cmp.add.3
+  %dst.add.4 = getelementptr inbounds i8, ptr %dst.add.3, i64 3
+  %cmp.add.4 = icmp uge ptr %dst.add.4, %lower
+  %res.2 = xor i1 %res.1, %cmp.add.4
+  ret i1 %res.2
+}
+
+define i1 @gep_add_1_uge_no_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_add_1_uge_no_inbounds(
+; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr i8, ptr [[DST]], i64 3
+; CHECK-NEXT:    [[DST_ADD_1:%.*]] = getelementptr i8, ptr [[DST]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_1:%.*]] = icmp uge ptr [[DST_ADD_1]], [[LOWER]]
+; CHECK-NEXT:    [[DST_ADD_2:%.*]] = getelementptr i8, ptr [[DST_ADD_1]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_3:%.*]] = icmp uge ptr [[DST_ADD_3]], [[LOWER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 [[CMP_ADD_1]], [[CMP_ADD_3]]
+; CHECK-NEXT:    [[DST_ADD_4:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 3
+; CHECK-NEXT:    [[CMP_ADD_4:%.*]] = icmp uge ptr [[DST_ADD_4]], [[LOWER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_ADD_4]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %pre = icmp uge ptr %dst, %lower
+  call void @llvm.assume(i1 %pre)
+  %dst.add.3 = getelementptr i8, ptr %dst, i64 3
+  %dst.add.1 = getelementptr i8, ptr %dst, i64 1
+  %cmp.add.1 = icmp uge ptr %dst.add.1, %lower
+  %dst.add.2 = getelementptr i8, ptr %dst.add.1, i64 1
+  %cmp.add.3 = icmp uge ptr %dst.add.3, %lower
+  %res.1 = xor i1 %cmp.add.1, %cmp.add.3
+  %dst.add.4 = getelementptr i8, ptr %dst.add.3, i64 3
+  %cmp.add.4 = icmp uge ptr %dst.add.4, %lower
+  %res.2 = xor i1 %res.1, %cmp.add.4
+  ret i1 %res.2
+}
+
 define i1 @gep_add_1_ult(ptr %dst, ptr %lower, ptr %upper) {
 ; CHECK-LABEL: @gep_add_1_ult(
 ; CHECK-NEXT:    [[END:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i64 4
@@ -121,8 +208,8 @@ define i1 @gep_add_ult_var_idx_sgt_1(ptr %dst, ptr %upper, i8 %idx) {
   ret i1 %res.1
 }
 
-define i1 @gep_add_1_ult_var_idx(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
-; CHECK-LABEL: @gep_add_1_ult_var_idx(
+define i1 @gep_add_1_ult_var_idx_inbounds(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
+; CHECK-LABEL: @gep_add_1_ult_var_idx_inbounds(
 ; CHECK-NEXT:    [[NOT_ZERO:%.*]] = icmp ne i8 [[LEN:%.*]], 0
 ; CHECK-NEXT:    call void @llvm.assume(i1 [[NOT_ZERO]])
 ; CHECK-NEXT:    [[LEN_EXT:%.*]] = zext i8 [[LEN]] to i16
@@ -165,3 +252,93 @@ define i1 @gep_add_1_ult_var_idx(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
   %res.2 = xor i1 %res.1, %cmp.idx.3
   ret i1 %res.2
 }
+
+define i1 @gep_add_1_ult_var_idx_only_inner_inbounds(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
+; CHECK-LABEL: @gep_add_1_ult_var_idx_only_inner_inbounds(
+; CHECK-NEXT:    [[NOT_ZERO:%.*]] = icmp ne i8 [[LEN:%.*]], 0
+; CHECK-NEXT:    call void @llvm.assume(i1 [[NOT_ZERO]])
+; CHECK-NEXT:    [[LEN_EXT:%.*]] = zext i8 [[LEN]] to i16
+; CHECK-NEXT:    [[DST_ADD_LEN:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i16 [[LEN_EXT]]
+; CHECK-NEXT:    [[DST_ADD_1:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_LEN]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_1:%.*]] = icmp ult ptr [[DST_ADD_1]], [[UPPER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_ADD_1]])
+; CHECK-NEXT:    [[CMP_IDX_ULT_LEN:%.*]] = icmp ult i8 [[IDX:%.*]], [[LEN]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_IDX_ULT_LEN]])
+; CHECK-NEXT:    [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT:    [[DST_ADD_IDX:%.*]] = getelementptr inbounds i8, ptr [[DST]], i16 [[IDX_EXT]]
+; CHECK-NEXT:    [[DST_ADD_IDX_1:%.*]] = getelementptr i8, ptr [[DST_ADD_IDX]], i64 1
+; CHECK-NEXT:    [[CMP_IDX_1:%.*]] = icmp ult ptr [[DST_ADD_IDX_1]], [[UPPER]]
+; CHECK-NEXT:    [[DST_ADD_IDX_2:%.*]] = getelementptr i8, ptr [[DST_ADD_IDX]], i64 2
+; CHECK-NEXT:    [[CMP_IDX_2:%.*]] = icmp ult ptr [[DST_ADD_IDX_2]], [[UPPER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 [[CMP_IDX_1]], [[CMP_IDX_2]]
+; CHECK-NEXT:    [[DST_ADD_IDX_3:%.*]] = getelementptr i8, ptr [[DST_ADD_IDX]], i64 3
+; CHECK-NEXT:    [[CMP_IDX_3:%.*]] = icmp ult ptr [[DST_ADD_IDX_3]], [[UPPER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_IDX_3]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %not.zero = icmp ne i8 %len, 0
+  call void @llvm.assume(i1 %not.zero)
+  %len.ext = zext i8 %len to i16
+  %dst.add.len = getelementptr inbounds i8, ptr %dst, i16 %len.ext
+  %dst.add.1 = getelementptr inbounds i8, ptr %dst.add.len, i64 1
+  %cmp.add.1 = icmp ult ptr %dst.add.1, %upper
+  call void @llvm.assume(i1 %cmp.add.1)
+  %cmp.idx.ult.len = icmp ult i8 %idx, %len
+  call void @llvm.assume(i1 %cmp.idx.ult.len)
+  %idx.ext = zext i8 %idx to i16
+  %dst.add.idx = getelementptr inbounds i8, ptr %dst, i16 %idx.ext
+  %dst.add.idx.1 = getelementptr i8, ptr %dst.add.idx, i64 1
+  %cmp.idx.1 = icmp ult ptr %dst.add.idx.1, %upper
+  %dst.add.idx.2 = getelementptr i8, ptr %dst.add.idx, i64 2
+  %cmp.idx.2 = icmp ult ptr %dst.add.idx.2, %upper
+  %res.1 = xor i1 %cmp.idx.1, %cmp.idx.2
+  %dst.add.idx.3 = getelementptr i8, ptr %dst.add.idx, i64 3
+  %cmp.idx.3 = icmp ult ptr %dst.add.idx.3, %upper
+  %res.2 = xor i1 %res.1, %cmp.idx.3
+  ret i1 %res.2
+}
+
+define i1 @gep_add_1_ult_var_idx_no_inbounds(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
+; CHECK-LABEL: @gep_add_1_ult_var_idx_no_inbounds(
+; CHECK-NEXT:    [[NOT_ZERO:%.*]] = icmp ne i8 [[LEN:%.*]], 0
+; CHECK-NEXT:    call void @llvm.assume(i1 [[NOT_ZERO]])
+; CHECK-NEXT:    [[LEN_EXT:%.*]] = zext i8 [[LEN]] to i16
+; CHECK-NEXT:    [[DST_ADD_LEN:%.*]] = getelementptr i8, ptr [[DST:%.*]], i16 [[LEN_EXT]]
+; CHECK-NEXT:    [[DST_ADD_1:%.*]] = getelementptr i8, ptr [[DST_ADD_LEN]], i64 1
+; CHECK-NEXT:    [[CMP_ADD_1:%.*]] = icmp ult ptr [[DST_ADD_1]], [[UPPER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_ADD_1]])
+; CHECK-NEXT:    [[CMP_IDX_ULT_LEN:%.*]] = icmp ult i8 [[IDX:%.*]], [[LEN]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_IDX_ULT_LEN]])
+; CHECK-NEXT:    [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT:    [[DST_ADD_IDX:%.*]] = getelementptr i8, ptr [[DST]], i16 [[IDX_EXT]]
+; CHECK-NEXT:    [[DST_ADD_IDX_1:%.*]] = getelementptr i8, ptr [[DST_ADD_IDX]], i64 1
+; CHECK-NEXT:    [[CMP_IDX_1:%.*]] = icmp ult ptr [[DST_ADD_IDX_1]], [[UPPER]]
+; CHECK-NEXT:    [[DST_ADD_IDX_2:%.*]] = getelementptr i8, ptr [[DST_ADD_IDX]], i64 2
+; CHECK-NEXT:    [[CMP_IDX_2:%.*]] = icmp ult ptr [[DST_ADD_IDX_2]], [[UPPER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 [[CMP_IDX_1]], [[CMP_IDX_2]]
+; CHECK-NEXT:    [[DST_ADD_IDX_3:%.*]] = getelementptr i8, ptr [[DST_ADD_IDX]], i64 3
+; CHECK-NEXT:    [[CMP_IDX_3:%.*]] = icmp ult ptr [[DST_ADD_IDX_3]], [[UPPER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_IDX_3]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %not.zero = icmp ne i8 %len, 0
+  call void @llvm.assume(i1 %not.zero)
+  %len.ext = zext i8 %len to i16
+  %dst.add.len = getelementptr i8, ptr %dst, i16 %len.ext
+  %dst.add.1 = getelementptr i8, ptr %dst.add.len, i64 1
+  %cmp.add.1 = icmp ult ptr %dst.add.1, %upper
+  call void @llvm.assume(i1 %cmp.add.1)
+  %cmp.idx.ult.len = icmp ult i8 %idx, %len
+  call void @llvm.assume(i1 %cmp.idx.ult.len)
+  %idx.ext = zext i8 %idx to i16
+  %dst.add.idx = getelementptr i8, ptr %dst, i16 %idx.ext
+  %dst.add.idx.1 = getelementptr i8, ptr %dst.add.idx, i64 1
+  %cmp.idx.1 = icmp ult ptr %dst.add.idx.1, %upper
+  %dst.add.idx.2 = getelementptr i8, ptr %dst.add.idx, i64 2
+  %cmp.idx.2 = icmp ult ptr %dst.add.idx.2, %upper
+  %res.1 = xor i1 %cmp.idx.1, %cmp.idx.2
+  %dst.add.idx.3 = getelementptr i8, ptr %dst.add.idx, i64 3
+  %cmp.idx.3 = icmp ult ptr %dst.add.idx.3, %upper
+  %res.2 = xor i1 %res.1, %cmp.idx.3
+  ret i1 %res.2
+}

diff  --git a/llvm/test/Transforms/ConstraintElimination/gep-sub.ll b/llvm/test/Transforms/ConstraintElimination/gep-sub.ll
index ab65d250fe19a..ec317156c3a75 100644
--- a/llvm/test/Transforms/ConstraintElimination/gep-sub.ll
+++ b/llvm/test/Transforms/ConstraintElimination/gep-sub.ll
@@ -3,8 +3,8 @@
 
 declare void @llvm.assume(i1 noundef)
 
-define i1 @gep_sub_1_uge(ptr %dst, ptr %lower) {
-; CHECK-LABEL: @gep_sub_1_uge(
+define i1 @gep_sub_1_uge_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_sub_1_uge_inbounds(
 ; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
 ; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
 ; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr inbounds i8, ptr [[DST]], i64 3
@@ -32,6 +32,93 @@ define i1 @gep_sub_1_uge(ptr %dst, ptr %lower) {
   ret i1 %res.2
 }
 
+define i1 @gep_sub_1_uge_only_inner_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_sub_1_uge_only_inner_inbounds(
+; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr inbounds i8, ptr [[DST]], i64 3
+; CHECK-NEXT:    [[DST_SUB_1:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 -1
+; CHECK-NEXT:    [[CMP_SUB_1:%.*]] = icmp uge ptr [[DST_SUB_1]], [[LOWER]]
+; CHECK-NEXT:    [[DST_SUB_3:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 -3
+; CHECK-NEXT:    [[CMP_SUB_3:%.*]] = icmp uge ptr [[DST_SUB_3]], [[LOWER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 [[CMP_SUB_1]], [[CMP_SUB_3]]
+; CHECK-NEXT:    [[DST_SUB_4:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 -4
+; CHECK-NEXT:    [[CMP_SUB_4:%.*]] = icmp uge ptr [[DST_SUB_4]], [[LOWER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_SUB_4]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %pre = icmp uge ptr %dst, %lower
+  call void @llvm.assume(i1 %pre)
+  %dst.add.3 = getelementptr inbounds i8, ptr %dst, i64 3
+  %dst.sub.1 = getelementptr i8, ptr %dst.add.3, i64 -1
+  %cmp.sub.1 = icmp uge ptr %dst.sub.1, %lower
+  %dst.sub.3 = getelementptr i8, ptr %dst.add.3, i64 -3
+  %cmp.sub.3 = icmp uge ptr %dst.sub.3, %lower
+  %res.1 = xor i1 %cmp.sub.1, %cmp.sub.3
+  %dst.sub.4 = getelementptr i8, ptr %dst.add.3, i64 -4
+  %cmp.sub.4 = icmp uge ptr %dst.sub.4, %lower
+  %res.2 = xor i1 %res.1, %cmp.sub.4
+  ret i1 %res.2
+}
+
+define i1 @gep_sub_1_uge_only_outer_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_sub_1_uge_only_outer_inbounds(
+; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr i8, ptr [[DST]], i64 3
+; CHECK-NEXT:    [[DST_SUB_1:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_3]], i64 -1
+; CHECK-NEXT:    [[CMP_SUB_1:%.*]] = icmp uge ptr [[DST_SUB_1]], [[LOWER]]
+; CHECK-NEXT:    [[DST_SUB_3:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_3]], i64 -3
+; CHECK-NEXT:    [[CMP_SUB_3:%.*]] = icmp uge ptr [[DST_SUB_3]], [[LOWER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 [[CMP_SUB_1]], [[CMP_SUB_3]]
+; CHECK-NEXT:    [[DST_SUB_4:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_3]], i64 -4
+; CHECK-NEXT:    [[CMP_SUB_4:%.*]] = icmp uge ptr [[DST_SUB_4]], [[LOWER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_SUB_4]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %pre = icmp uge ptr %dst, %lower
+  call void @llvm.assume(i1 %pre)
+  %dst.add.3 = getelementptr i8, ptr %dst, i64 3
+  %dst.sub.1 = getelementptr inbounds i8, ptr %dst.add.3, i64 -1
+  %cmp.sub.1 = icmp uge ptr %dst.sub.1, %lower
+  %dst.sub.3 = getelementptr inbounds i8, ptr %dst.add.3, i64 -3
+  %cmp.sub.3 = icmp uge ptr %dst.sub.3, %lower
+  %res.1 = xor i1 %cmp.sub.1, %cmp.sub.3
+  %dst.sub.4 = getelementptr inbounds i8, ptr %dst.add.3, i64 -4
+  %cmp.sub.4 = icmp uge ptr %dst.sub.4, %lower
+  %res.2 = xor i1 %res.1, %cmp.sub.4
+  ret i1 %res.2
+}
+
+define i1 @gep_sub_1_uge_no_inbounds(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_sub_1_uge_no_inbounds(
+; CHECK-NEXT:    [[PRE:%.*]] = icmp uge ptr [[DST:%.*]], [[LOWER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT:    [[DST_ADD_3:%.*]] = getelementptr i8, ptr [[DST]], i64 3
+; CHECK-NEXT:    [[DST_SUB_1:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 -1
+; CHECK-NEXT:    [[CMP_SUB_1:%.*]] = icmp uge ptr [[DST_SUB_1]], [[LOWER]]
+; CHECK-NEXT:    [[DST_SUB_3:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 -3
+; CHECK-NEXT:    [[CMP_SUB_3:%.*]] = icmp uge ptr [[DST_SUB_3]], [[LOWER]]
+; CHECK-NEXT:    [[RES_1:%.*]] = xor i1 [[CMP_SUB_1]], [[CMP_SUB_3]]
+; CHECK-NEXT:    [[DST_SUB_4:%.*]] = getelementptr i8, ptr [[DST_ADD_3]], i64 -4
+; CHECK-NEXT:    [[CMP_SUB_4:%.*]] = icmp uge ptr [[DST_SUB_4]], [[LOWER]]
+; CHECK-NEXT:    [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_SUB_4]]
+; CHECK-NEXT:    ret i1 [[RES_2]]
+;
+  %pre = icmp uge ptr %dst, %lower
+  call void @llvm.assume(i1 %pre)
+  %dst.add.3 = getelementptr i8, ptr %dst, i64 3
+  %dst.sub.1 = getelementptr i8, ptr %dst.add.3, i64 -1
+  %cmp.sub.1 = icmp uge ptr %dst.sub.1, %lower
+  %dst.sub.3 = getelementptr i8, ptr %dst.add.3, i64 -3
+  %cmp.sub.3 = icmp uge ptr %dst.sub.3, %lower
+  %res.1 = xor i1 %cmp.sub.1, %cmp.sub.3
+  %dst.sub.4 = getelementptr i8, ptr %dst.add.3, i64 -4
+  %cmp.sub.4 = icmp uge ptr %dst.sub.4, %lower
+  %res.2 = xor i1 %res.1, %cmp.sub.4
+  ret i1 %res.2
+}
+
 define i1 @gep_sub_1_ult(ptr %dst, ptr %upper) {
 ; CHECK-LABEL: @gep_sub_1_ult(
 ; CHECK-NEXT:    [[DST_ADD_4:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i64 4
@@ -131,8 +218,8 @@ define i1 @gep_sub_ult_var_idx_sgt_1(ptr %dst, ptr %upper, i8 %idx) {
   ret i1 %res.2
 }
 
-define i1 @gep_sub_1_ult_var_idx(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
-; CHECK-LABEL: @gep_sub_1_ult_var_idx(
+define i1 @gep_sub_1_ult_var_idx_inbounds(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
+; CHECK-LABEL: @gep_sub_1_ult_var_idx_inbounds(
 ; CHECK-NEXT:    [[NOT_ZERO:%.*]] = icmp ne i8 [[LEN:%.*]], 0
 ; CHECK-NEXT:    call void @llvm.assume(i1 [[NOT_ZERO]])
 ; CHECK-NEXT:    [[LEN_EXT:%.*]] = zext i8 [[LEN]] to i16
@@ -162,6 +249,68 @@ define i1 @gep_sub_1_ult_var_idx(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
   ret i1 %cmp.idx
 }
 
+define i1 @gep_sub_1_ult_var_idx_only_inner_inbounds(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
+; CHECK-LABEL: @gep_sub_1_ult_var_idx_only_inner_inbounds(
+; CHECK-NEXT:    [[NOT_ZERO:%.*]] = icmp ne i8 [[LEN:%.*]], 0
+; CHECK-NEXT:    call void @llvm.assume(i1 [[NOT_ZERO]])
+; CHECK-NEXT:    [[LEN_EXT:%.*]] = zext i8 [[LEN]] to i16
+; CHECK-NEXT:    [[DST_ADD_LEN:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i16 [[LEN_EXT]]
+; CHECK-NEXT:    [[DST_SUB_1:%.*]] = getelementptr i8, ptr [[DST_ADD_LEN]], i64 -1
+; CHECK-NEXT:    [[CMP_SUB_1:%.*]] = icmp ult ptr [[DST_SUB_1]], [[UPPER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_SUB_1]])
+; CHECK-NEXT:    [[CMP_IDX_ULT_LEN:%.*]] = icmp ult i8 [[IDX:%.*]], [[LEN]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_IDX_ULT_LEN]])
+; CHECK-NEXT:    [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT:    [[DST_ADD_IDX:%.*]] = getelementptr inbounds i8, ptr [[DST]], i16 [[IDX_EXT]]
+; CHECK-NEXT:    [[CMP_IDX:%.*]] = icmp ult ptr [[DST_ADD_IDX]], [[UPPER]]
+; CHECK-NEXT:    ret i1 [[CMP_IDX]]
+;
+  %not.zero = icmp ne i8 %len, 0
+  call void @llvm.assume(i1 %not.zero)
+  %len.ext = zext i8 %len to i16
+  %dst.add.len = getelementptr inbounds i8, ptr %dst, i16 %len.ext
+  %dst.sub.1 = getelementptr i8, ptr %dst.add.len, i64 -1
+  %cmp.sub.1 = icmp ult ptr %dst.sub.1, %upper
+  call void @llvm.assume(i1 %cmp.sub.1)
+  %cmp.idx.ult.len = icmp ult i8 %idx, %len
+  call void @llvm.assume(i1 %cmp.idx.ult.len)
+  %idx.ext = zext i8 %idx to i16
+  %dst.add.idx = getelementptr inbounds i8, ptr %dst, i16 %idx.ext
+  %cmp.idx = icmp ult ptr %dst.add.idx, %upper
+  ret i1 %cmp.idx
+}
+
+define i1 @gep_sub_1_ult_var_idx_no_inbounds(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
+; CHECK-LABEL: @gep_sub_1_ult_var_idx_no_inbounds(
+; CHECK-NEXT:    [[NOT_ZERO:%.*]] = icmp ne i8 [[LEN:%.*]], 0
+; CHECK-NEXT:    call void @llvm.assume(i1 [[NOT_ZERO]])
+; CHECK-NEXT:    [[LEN_EXT:%.*]] = zext i8 [[LEN]] to i16
+; CHECK-NEXT:    [[DST_ADD_LEN:%.*]] = getelementptr i8, ptr [[DST:%.*]], i16 [[LEN_EXT]]
+; CHECK-NEXT:    [[DST_SUB_1:%.*]] = getelementptr i8, ptr [[DST_ADD_LEN]], i64 -1
+; CHECK-NEXT:    [[CMP_SUB_1:%.*]] = icmp ult ptr [[DST_SUB_1]], [[UPPER:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_SUB_1]])
+; CHECK-NEXT:    [[CMP_IDX_ULT_LEN:%.*]] = icmp ult i8 [[IDX:%.*]], [[LEN]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP_IDX_ULT_LEN]])
+; CHECK-NEXT:    [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT:    [[DST_ADD_IDX:%.*]] = getelementptr i8, ptr [[DST]], i16 [[IDX_EXT]]
+; CHECK-NEXT:    [[CMP_IDX:%.*]] = icmp ult ptr [[DST_ADD_IDX]], [[UPPER]]
+; CHECK-NEXT:    ret i1 [[CMP_IDX]]
+;
+  %not.zero = icmp ne i8 %len, 0
+  call void @llvm.assume(i1 %not.zero)
+  %len.ext = zext i8 %len to i16
+  %dst.add.len = getelementptr i8, ptr %dst, i16 %len.ext
+  %dst.sub.1 = getelementptr i8, ptr %dst.add.len, i64 -1
+  %cmp.sub.1 = icmp ult ptr %dst.sub.1, %upper
+  call void @llvm.assume(i1 %cmp.sub.1)
+  %cmp.idx.ult.len = icmp ult i8 %idx, %len
+  call void @llvm.assume(i1 %cmp.idx.ult.len)
+  %idx.ext = zext i8 %idx to i16
+  %dst.add.idx = getelementptr i8, ptr %dst, i16 %idx.ext
+  %cmp.idx = icmp ult ptr %dst.add.idx, %upper
+  ret i1 %cmp.idx
+}
+
 define i1 @gep_sub_2_ult_var_idx(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
 ; CHECK-LABEL: @gep_sub_2_ult_var_idx(
 ; CHECK-NEXT:    [[NOT_ZERO:%.*]] = icmp ne i8 [[LEN:%.*]], 0


        


More information about the llvm-commits mailing list