[llvm] 22fe400 - [ConstraintElimination] Add tests for chained GEPs with const offsets.
Florian Hahn via llvm-commits
llvm-commits at lists.llvm.org
Mon Oct 3 07:56:05 PDT 2022
Author: Florian Hahn
Date: 2022-10-03T15:55:36+01:00
New Revision: 22fe40068ca35aeb7cfdbc752f07ad9e2400e2a5
URL: https://github.com/llvm/llvm-project/commit/22fe40068ca35aeb7cfdbc752f07ad9e2400e2a5
DIFF: https://github.com/llvm/llvm-project/commit/22fe40068ca35aeb7cfdbc752f07ad9e2400e2a5.diff
LOG: [ConstraintElimination] Add tests for chained GEPs with const offsets.
Add extra tests for chained GEPs where the second GEP has a constant
offset. Inspired by missed optimizations from #51358.
Added:
llvm/test/Transforms/ConstraintElimination/gep-add.ll
llvm/test/Transforms/ConstraintElimination/gep-sub.ll
Modified:
Removed:
################################################################################
diff --git a/llvm/test/Transforms/ConstraintElimination/gep-add.ll b/llvm/test/Transforms/ConstraintElimination/gep-add.ll
new file mode 100644
index 0000000000000..f1f532109945a
--- /dev/null
+++ b/llvm/test/Transforms/ConstraintElimination/gep-add.ll
@@ -0,0 +1,167 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -passes=constraint-elimination -S %s | FileCheck %s
+
+declare void @llvm.assume(i1 noundef) #0
+
+define i1 @gep_add_1_uge(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_add_1_uge(
+; 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 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 true, true
+; 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 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 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_ult(ptr %dst, ptr %lower, ptr %upper) {
+; CHECK-LABEL: @gep_add_1_ult(
+; CHECK-NEXT: [[END:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i64 4
+; CHECK-NEXT: [[PRE:%.*]] = icmp ult ptr [[END]], [[UPPER:%.*]]
+; CHECK-NEXT: call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT: [[DST_ADD_1:%.*]] = getelementptr inbounds i8, ptr [[DST]], i64 1
+; CHECK-NEXT: [[DST_ADD_3:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_1]], i64 2
+; CHECK-NEXT: [[CMP_ADD_3:%.*]] = icmp ult ptr [[DST_ADD_3]], [[UPPER]]
+; CHECK-NEXT: [[DST_ADD_4:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_1]], i64 3
+; CHECK-NEXT: [[CMP_ADD_4:%.*]] = icmp ult ptr [[DST_ADD_4]], [[UPPER]]
+; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[CMP_ADD_3]], [[CMP_ADD_4]]
+; CHECK-NEXT: [[DST_ADD_5:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_1]], i64 4
+; CHECK-NEXT: [[CMP_ADD_5:%.*]] = icmp ult ptr [[DST_ADD_5]], [[UPPER]]
+; CHECK-NEXT: [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_ADD_5]]
+; CHECK-NEXT: ret i1 [[RES_2]]
+;
+ %end = getelementptr inbounds i8, ptr %dst, i64 4
+ %pre = icmp ult ptr %end, %upper
+ call void @llvm.assume(i1 %pre)
+ %dst.add.1 = getelementptr inbounds i8, ptr %dst, i64 1
+ %dst.add.3 = getelementptr inbounds i8, ptr %dst.add.1, i64 2
+ %cmp.add.3 = icmp ult ptr %dst.add.3, %upper
+ %dst.add.4 = getelementptr inbounds i8, ptr %dst.add.1, i64 3
+ %cmp.add.4 = icmp ult ptr %dst.add.4, %upper
+ %res.1 = xor i1 %cmp.add.3, %cmp.add.4
+ %dst.add.5 = getelementptr inbounds i8, ptr %dst.add.1, i64 4
+ %cmp.add.5 = icmp ult ptr %dst.add.5, %upper
+ %res.2 = xor i1 %res.1, %cmp.add.5
+ ret i1 %res.2
+}
+
+define i1 @gep_add_ult_var_idx(ptr %dst, ptr %upper, i8 %idx) {
+; CHECK-LABEL: @gep_add_ult_var_idx(
+; CHECK-NEXT: [[NOT_ZERO:%.*]] = icmp ne i8 [[IDX:%.*]], 0
+; CHECK-NEXT: call void @llvm.assume(i1 [[NOT_ZERO]])
+; CHECK-NEXT: [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT: [[DST_ADD_IDX:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i16 [[IDX_EXT]]
+; CHECK-NEXT: [[PRE:%.*]] = icmp ult ptr [[DST_ADD_IDX]], [[UPPER:%.*]]
+; CHECK-NEXT: call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT: [[DST_ADD_1:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 1
+; CHECK-NEXT: [[CMP_ADD_1:%.*]] = icmp ule ptr [[DST_ADD_1]], [[UPPER]]
+; CHECK-NEXT: [[DST_ADD_2:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 2
+; CHECK-NEXT: [[CMP_ADD_2:%.*]] = icmp ule ptr [[DST_ADD_2]], [[UPPER]]
+; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[CMP_ADD_1]], [[CMP_ADD_2]]
+; CHECK-NEXT: ret i1 [[RES_1]]
+;
+ %not.zero = icmp ne i8 %idx, 0
+ call void @llvm.assume(i1 %not.zero)
+ %idx.ext = zext i8 %idx to i16
+ %dst.add.idx = getelementptr inbounds i8, ptr %dst, i16 %idx.ext
+ %pre = icmp ult ptr %dst.add.idx, %upper
+ call void @llvm.assume(i1 %pre)
+ %dst.add.1 = getelementptr inbounds i8, ptr %dst.add.idx, i64 1
+ %cmp.add.1 = icmp ule ptr %dst.add.1, %upper
+ %dst.add.2 = getelementptr inbounds i8, ptr %dst.add.idx, i64 2
+ %cmp.add.2 = icmp ule ptr %dst.add.2, %upper
+ %res.1 = xor i1 %cmp.add.1, %cmp.add.2
+ ret i1 %res.1
+}
+
+define i1 @gep_add_ult_var_idx_sgt_1(ptr %dst, ptr %upper, i8 %idx) {
+; CHECK-LABEL: @gep_add_ult_var_idx_sgt_1(
+; CHECK-NEXT: [[SGT_1:%.*]] = icmp sgt i8 [[IDX:%.*]], 1
+; CHECK-NEXT: call void @llvm.assume(i1 [[SGT_1]])
+; CHECK-NEXT: [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT: [[DST_ADD_IDX:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i16 [[IDX_EXT]]
+; CHECK-NEXT: [[PRE:%.*]] = icmp ult ptr [[DST_ADD_IDX]], [[UPPER:%.*]]
+; CHECK-NEXT: call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT: [[DST_ADD_1:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 1
+; CHECK-NEXT: [[CMP_ADD_1:%.*]] = icmp ule ptr [[DST_ADD_1]], [[UPPER]]
+; CHECK-NEXT: [[DST_ADD_2:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 2
+; CHECK-NEXT: [[CMP_ADD_2:%.*]] = icmp ule ptr [[DST_ADD_2]], [[UPPER]]
+; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[CMP_ADD_1]], [[CMP_ADD_2]]
+; CHECK-NEXT: ret i1 [[RES_1]]
+;
+ %sgt.1 = icmp sgt i8 %idx, 1
+ call void @llvm.assume(i1 %sgt.1)
+ %idx.ext = zext i8 %idx to i16
+ %dst.add.idx = getelementptr inbounds i8, ptr %dst, i16 %idx.ext
+ %pre = icmp ult ptr %dst.add.idx, %upper
+ call void @llvm.assume(i1 %pre)
+ %dst.add.1 = getelementptr inbounds i8, ptr %dst.add.idx, i64 1
+ %cmp.add.1 = icmp ule ptr %dst.add.1, %upper
+ %dst.add.2 = getelementptr inbounds i8, ptr %dst.add.idx, i64 2
+ %cmp.add.2 = icmp ule ptr %dst.add.2, %upper
+ %res.1 = xor i1 %cmp.add.1, %cmp.add.2
+ 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(
+; 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 inbounds 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 inbounds 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 inbounds 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 inbounds i8, ptr %dst.add.idx, i64 1
+ %cmp.idx.1 = icmp ult ptr %dst.add.idx.1, %upper
+ %dst.add.idx.2 = getelementptr inbounds 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 inbounds 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
new file mode 100644
index 0000000000000..7825d03502c35
--- /dev/null
+++ b/llvm/test/Transforms/ConstraintElimination/gep-sub.ll
@@ -0,0 +1,225 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -passes=constraint-elimination -S %s | FileCheck %s
+
+declare void @llvm.assume(i1 noundef)
+
+define i1 @gep_sub_1_uge(ptr %dst, ptr %lower) {
+; CHECK-LABEL: @gep_sub_1_uge(
+; 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 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 inbounds 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_ult(ptr %dst, ptr %upper) {
+; CHECK-LABEL: @gep_sub_1_ult(
+; CHECK-NEXT: [[DST_ADD_4:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i64 4
+; CHECK-NEXT: [[PRE:%.*]] = icmp ult ptr [[DST_ADD_4]], [[UPPER:%.*]]
+; 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 inbounds i8, ptr [[DST_ADD_3]], i64 -1
+; CHECK-NEXT: [[CMP_SUB_1:%.*]] = icmp ult ptr [[DST_SUB_1]], [[UPPER]]
+; CHECK-NEXT: [[DST_SUB_3:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_3]], i64 -3
+; CHECK-NEXT: [[CMP_SUB_3:%.*]] = icmp ult ptr [[DST_SUB_3]], [[UPPER]]
+; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[CMP_SUB_1]], [[CMP_SUB_3]]
+; CHECK-NEXT: ret i1 [[RES_1]]
+;
+ %dst.add.4 = getelementptr inbounds i8, ptr %dst, i64 4
+ %pre = icmp ult ptr %dst.add.4, %upper
+ call void @llvm.assume(i1 %pre)
+ %dst.add.3 = getelementptr inbounds i8, ptr %dst, i64 3
+ %dst.sub.1 = getelementptr inbounds i8, ptr %dst.add.3, i64 -1
+ %cmp.sub.1 = icmp ult ptr %dst.sub.1, %upper
+ %dst.sub.3 = getelementptr inbounds i8, ptr %dst.add.3, i64 -3
+ %cmp.sub.3 = icmp ult ptr %dst.sub.3, %upper
+ %res.1 = xor i1 %cmp.sub.1, %cmp.sub.3
+ ret i1 %res.1
+}
+
+define i1 @gep_sub_ult_var_idx(ptr %dst, ptr %upper, i8 %idx) {
+; CHECK-LABEL: @gep_sub_ult_var_idx(
+; CHECK-NEXT: [[NOT_ZERO:%.*]] = icmp ne i8 [[IDX:%.*]], 0
+; CHECK-NEXT: call void @llvm.assume(i1 [[NOT_ZERO]])
+; CHECK-NEXT: [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT: [[DST_ADD_IDX:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i16 [[IDX_EXT]]
+; CHECK-NEXT: [[PRE:%.*]] = icmp ult ptr [[DST_ADD_IDX]], [[UPPER:%.*]]
+; CHECK-NEXT: call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT: [[DST_SUB_1:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 -1
+; CHECK-NEXT: [[CMP_SUB_1:%.*]] = icmp ult ptr [[DST_SUB_1]], [[UPPER]]
+; CHECK-NEXT: [[DST_SUB_2:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 -2
+; CHECK-NEXT: [[CMP_SUB_2:%.*]] = icmp ult ptr [[DST_SUB_2]], [[UPPER]]
+; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[CMP_SUB_1]], [[CMP_SUB_2]]
+; CHECK-NEXT: [[DST_SUB_1_SUB_1:%.*]] = getelementptr inbounds i8, ptr [[DST_SUB_1]], i64 -1
+; CHECK-NEXT: [[CMP_SUB_1_SUB_1:%.*]] = icmp ult ptr [[DST_SUB_1_SUB_1]], [[UPPER]]
+; CHECK-NEXT: [[CMP_SUB_1_SUB_1_EQ:%.*]] = icmp eq ptr [[DST_SUB_1_SUB_1]], [[DST_SUB_2]]
+; CHECK-NEXT: [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_SUB_1_SUB_1]]
+; CHECK-NEXT: [[RES_3:%.*]] = xor i1 [[RES_2]], [[CMP_SUB_1_SUB_1_EQ]]
+; CHECK-NEXT: ret i1 [[RES_3]]
+;
+ %not.zero = icmp ne i8 %idx, 0
+ call void @llvm.assume(i1 %not.zero)
+ %idx.ext = zext i8 %idx to i16
+ %dst.add.idx = getelementptr inbounds i8, ptr %dst, i16 %idx.ext
+ %pre = icmp ult ptr %dst.add.idx, %upper
+ call void @llvm.assume(i1 %pre)
+ %dst.sub.1 = getelementptr inbounds i8, ptr %dst.add.idx, i64 -1
+ %cmp.sub.1 = icmp ult ptr %dst.sub.1, %upper
+ %dst.sub.2 = getelementptr inbounds i8, ptr %dst.add.idx, i64 -2
+ %cmp.sub.2 = icmp ult ptr %dst.sub.2, %upper
+ %res.1 = xor i1 %cmp.sub.1, %cmp.sub.2
+ %dst.sub.1.sub.1 = getelementptr inbounds i8, ptr %dst.sub.1, i64 -1
+ %cmp.sub.1.sub.1 = icmp ult ptr %dst.sub.1.sub.1, %upper
+ %cmp.sub.1.sub.1.eq = icmp eq ptr %dst.sub.1.sub.1, %dst.sub.2
+ %res.2 = xor i1 %res.1, %cmp.sub.1.sub.1
+ %res.3 = xor i1 %res.2, %cmp.sub.1.sub.1.eq
+ ret i1 %res.3
+}
+
+define i1 @gep_sub_ult_var_idx_sgt_1(ptr %dst, ptr %upper, i8 %idx) {
+; CHECK-LABEL: @gep_sub_ult_var_idx_sgt_1(
+; CHECK-NEXT: [[SGT_1:%.*]] = icmp sgt i8 [[IDX:%.*]], 1
+; CHECK-NEXT: call void @llvm.assume(i1 [[SGT_1]])
+; CHECK-NEXT: [[IDX_EXT:%.*]] = zext i8 [[IDX]] to i16
+; CHECK-NEXT: [[DST_ADD_IDX:%.*]] = getelementptr inbounds i8, ptr [[DST:%.*]], i16 [[IDX_EXT]]
+; CHECK-NEXT: [[PRE:%.*]] = icmp ult ptr [[DST_ADD_IDX]], [[UPPER:%.*]]
+; CHECK-NEXT: call void @llvm.assume(i1 [[PRE]])
+; CHECK-NEXT: [[DST_SUB_1:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 -1
+; CHECK-NEXT: [[CMP_SUB_1:%.*]] = icmp ult ptr [[DST_SUB_1]], [[UPPER]]
+; CHECK-NEXT: [[DST_SUB_2:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 -2
+; CHECK-NEXT: [[CMP_SUB_2:%.*]] = icmp ult ptr [[DST_SUB_2]], [[UPPER]]
+; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[CMP_SUB_1]], [[CMP_SUB_2]]
+; CHECK-NEXT: [[DST_SUB_3:%.*]] = getelementptr inbounds i8, ptr [[DST_ADD_IDX]], i64 -3
+; CHECK-NEXT: [[CMP_SUB_3:%.*]] = icmp ult ptr [[DST_SUB_3]], [[UPPER]]
+; CHECK-NEXT: [[RES_2:%.*]] = xor i1 [[RES_1]], [[CMP_SUB_3]]
+; CHECK-NEXT: ret i1 [[RES_2]]
+;
+ %sgt.1 = icmp sgt i8 %idx, 1
+ call void @llvm.assume(i1 %sgt.1)
+ %idx.ext = zext i8 %idx to i16
+ %dst.add.idx = getelementptr inbounds i8, ptr %dst, i16 %idx.ext
+ %pre = icmp ult ptr %dst.add.idx, %upper
+ call void @llvm.assume(i1 %pre)
+ %dst.sub.1 = getelementptr inbounds i8, ptr %dst.add.idx, i64 -1
+ %cmp.sub.1 = icmp ult ptr %dst.sub.1, %upper
+ %dst.sub.2 = getelementptr inbounds i8, ptr %dst.add.idx, i64 -2
+ %cmp.sub.2 = icmp ult ptr %dst.sub.2, %upper
+ %res.1 = xor i1 %cmp.sub.1, %cmp.sub.2
+ %dst.sub.3 = getelementptr inbounds i8, ptr %dst.add.idx, i64 -3
+ %cmp.sub.3 = icmp ult ptr %dst.sub.3, %upper
+ %res.2 = xor i1 %res.1, %cmp.sub.3
+ 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(
+; 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 inbounds 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 inbounds 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_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
+; 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 inbounds i8, ptr [[DST_ADD_LEN]], i64 -2
+; 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 inbounds i8, ptr %dst.add.len, i64 -2
+ %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_ult_var_idx_len_sgt_1(ptr %dst, ptr %upper, i8 %len, i8 %idx) {
+; CHECK-LABEL: @gep_sub_ult_var_idx_len_sgt_1(
+; CHECK-NEXT: [[SGT_1:%.*]] = icmp sgt i8 [[LEN:%.*]], 1
+; CHECK-NEXT: call void @llvm.assume(i1 [[SGT_1]])
+; 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 inbounds i8, ptr [[DST_ADD_LEN]], i64 -2
+; 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]]
+;
+ %sgt.1 = icmp sgt i8 %len, 1
+ call void @llvm.assume(i1 %sgt.1)
+ %len.ext = zext i8 %len to i16
+ %dst.add.len = getelementptr inbounds i8, ptr %dst, i16 %len.ext
+ %dst.sub.1 = getelementptr inbounds i8, ptr %dst.add.len, i64 -2
+ %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
+}
More information about the llvm-commits
mailing list