[llvm] 798754f - [ConstraintElim] Add multi-exit tests for #94610.
Florian Hahn via llvm-commits
llvm-commits at lists.llvm.org
Fri Jun 28 14:29:51 PDT 2024
Author: Florian Hahn
Date: 2024-06-28T22:29:17+01:00
New Revision: 798754f6c6fe19d19984ca8370806cbcdf5f8294
URL: https://github.com/llvm/llvm-project/commit/798754f6c6fe19d19984ca8370806cbcdf5f8294
DIFF: https://github.com/llvm/llvm-project/commit/798754f6c6fe19d19984ca8370806cbcdf5f8294.diff
LOG: [ConstraintElim] Add multi-exit tests for #94610.
Additional test coverage with multi-exit loops for
https://github.com/llvm/llvm-project/pull/94610.
Added:
Modified:
llvm/test/Transforms/ConstraintElimination/induction-condition-in-loop-exit.ll
Removed:
################################################################################
diff --git a/llvm/test/Transforms/ConstraintElimination/induction-condition-in-loop-exit.ll b/llvm/test/Transforms/ConstraintElimination/induction-condition-in-loop-exit.ll
index 969c4c3c71a92..05c7c9e5b90d8 100644
--- a/llvm/test/Transforms/ConstraintElimination/induction-condition-in-loop-exit.ll
+++ b/llvm/test/Transforms/ConstraintElimination/induction-condition-in-loop-exit.ll
@@ -481,4 +481,290 @@ exit:
ret i1 %u
}
+define i1 @multi_exiting_loop_eq_
diff erent_exits_const_compare_known(ptr %s) {
+; CHECK-LABEL: define i1 @multi_exiting_loop_eq_
diff erent_exits_const_compare_known(
+; CHECK-SAME: ptr [[S:%.*]]) {
+; CHECK-NEXT: [[ENTRY:.*]]:
+; CHECK-NEXT: br label %[[LOOP_HEADER:.*]]
+; CHECK: [[LOOP_HEADER]]:
+; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
+; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp eq i32 [[IV]], 1234
+; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label %[[EXIT_1:.*]], label %[[LOOP_LATCH]]
+; CHECK: [[LOOP_LATCH]]:
+; CHECK-NEXT: [[ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[S]], i32 [[IV]]
+; CHECK-NEXT: [[TMP0:%.*]] = load i8, ptr [[ARRAYIDX]], align 1
+; CHECK-NEXT: [[LATCH_C:%.*]] = icmp ult i8 [[TMP0]], 10
+; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i32 [[IV]], 1
+; CHECK-NEXT: br i1 [[LATCH_C]], label %[[LOOP_HEADER]], label %[[EXIT_2:.*]]
+; CHECK: [[EXIT_1]]:
+; CHECK-NEXT: ret i1 true
+; CHECK: [[EXIT_2]]:
+; CHECK-NEXT: ret i1 true
+;
+entry:
+ br label %loop.header
+
+loop.header:
+ %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
+ %exitcond.not = icmp eq i32 %iv, 1234
+ br i1 %exitcond.not, label %exit.1, label %loop.latch
+
+loop.latch:
+ %arrayidx = getelementptr inbounds i8, ptr %s, i32 %iv
+ %0 = load i8, ptr %arrayidx, align 1
+ %latch.c = icmp ult i8 %0, 10
+ %iv.next = add nuw nsw i32 %iv, 1
+ br i1 %latch.c, label %loop.header, label %exit.2
+
+exit.1:
+ %t.1 = icmp ult i32 %iv, 1235
+ ret i1 %t.1
+
+exit.2:
+ %t.2 = icmp ult i32 %iv, 1235
+ ret i1 %t.2
+}
+
+define i1 @multi_exiting_loop_eq_
diff erent_exits_2_const_compare_known(ptr %s, i1 %c.1, i1 %c.2) {
+; CHECK-LABEL: define i1 @multi_exiting_loop_eq_
diff erent_exits_2_const_compare_known(
+; CHECK-SAME: ptr [[S:%.*]], i1 [[C_1:%.*]], i1 [[C_2:%.*]]) {
+; CHECK-NEXT: [[ENTRY:.*]]:
+; CHECK-NEXT: br label %[[LOOP_HEADER:.*]]
+; CHECK: [[LOOP_HEADER]]:
+; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
+; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp eq i32 [[IV]], 1234
+; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label %[[EXIT_1:.*]], label %[[ELSE_1:.*]]
+; CHECK: [[ELSE_1]]:
+; CHECK-NEXT: br i1 [[C_1]], label %[[EXIT_1]], label %[[ELSE_2:.*]]
+; CHECK: [[ELSE_2]]:
+; CHECK-NEXT: br i1 [[C_2]], label %[[EXIT_2:.*]], label %[[LOOP_LATCH]]
+; CHECK: [[LOOP_LATCH]]:
+; CHECK-NEXT: [[ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[S]], i32 [[IV]]
+; CHECK-NEXT: [[TMP0:%.*]] = load i8, ptr [[ARRAYIDX]], align 1
+; CHECK-NEXT: [[LATCH_C:%.*]] = icmp ult i8 [[TMP0]], 10
+; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i32 [[IV]], 1
+; CHECK-NEXT: br i1 [[LATCH_C]], label %[[LOOP_HEADER]], label %[[EXIT_2]]
+; CHECK: [[EXIT_1]]:
+; CHECK-NEXT: [[T_1:%.*]] = icmp ult i32 [[IV]], 1235
+; CHECK-NEXT: ret i1 [[T_1]]
+; CHECK: [[EXIT_2]]:
+; CHECK-NEXT: ret i1 true
+;
+entry:
+ br label %loop.header
+
+loop.header:
+ %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
+ %exitcond.not = icmp eq i32 %iv, 1234
+ br i1 %exitcond.not, label %exit.1, label %else.1
+
+else.1:
+ br i1 %c.1, label %exit.1, label %else.2
+
+else.2:
+ br i1 %c.2, label %exit.2, label %loop.latch
+
+loop.latch:
+ %arrayidx = getelementptr inbounds i8, ptr %s, i32 %iv
+ %0 = load i8, ptr %arrayidx, align 1
+ %latch.c = icmp ult i8 %0, 10
+ %iv.next = add nuw nsw i32 %iv, 1
+ br i1 %latch.c, label %loop.header, label %exit.2
+
+exit.1:
+ %t.1 = icmp ult i32 %iv, 1235
+ ret i1 %t.1
+
+exit.2:
+ %t.2 = icmp ult i32 %iv, 1235
+ ret i1 %t.2
+}
+
+
+define i1 @multi_exiting_loop_eq_
diff erent_exits_compare_not_known(ptr %s, i32 %N) {
+; CHECK-LABEL: define i1 @multi_exiting_loop_eq_
diff erent_exits_compare_not_known(
+; CHECK-SAME: ptr [[S:%.*]], i32 [[N:%.*]]) {
+; CHECK-NEXT: [[ENTRY:.*]]:
+; CHECK-NEXT: br label %[[LOOP_HEADER:.*]]
+; CHECK: [[LOOP_HEADER]]:
+; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
+; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp eq i32 [[IV]], [[N]]
+; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label %[[EXIT_1:.*]], label %[[LOOP_LATCH]]
+; CHECK: [[LOOP_LATCH]]:
+; CHECK-NEXT: [[ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[S]], i32 [[IV]]
+; CHECK-NEXT: [[TMP0:%.*]] = load i8, ptr [[ARRAYIDX]], align 1
+; CHECK-NEXT: [[LATCH_C:%.*]] = icmp ult i8 [[TMP0]], 10
+; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i32 [[IV]], 1
+; CHECK-NEXT: br i1 [[LATCH_C]], label %[[LOOP_HEADER]], label %[[EXIT_2:.*]]
+; CHECK: [[EXIT_1]]:
+; CHECK-NEXT: ret i1 false
+; CHECK: [[EXIT_2]]:
+; CHECK-NEXT: ret i1 true
+;
+entry:
+ br label %loop.header
+loop.header:
+ %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
+ %exitcond.not = icmp eq i32 %iv, %N
+ br i1 %exitcond.not, label %exit.1, label %loop.latch
+
+loop.latch:
+ %arrayidx = getelementptr inbounds i8, ptr %s, i32 %iv
+ %0 = load i8, ptr %arrayidx, align 1
+ %latch.c = icmp ult i8 %0, 10
+ %iv.next = add nuw nsw i32 %iv, 1
+ br i1 %latch.c, label %loop.header, label %exit.2
+
+exit.1:
+ %t.1 = icmp ult i32 %iv, %N
+ ret i1 %t.1
+
+exit.2:
+ %t.2 = icmp ult i32 %iv, %N
+ ret i1 %t.2
+}
+
+define i1 @multi_exiting_loop_eq_
diff erent_exits_2_compare_not_known(ptr %s, i32 %N, i1 %c.1, i1 %c.2) {
+; CHECK-LABEL: define i1 @multi_exiting_loop_eq_
diff erent_exits_2_compare_not_known(
+; CHECK-SAME: ptr [[S:%.*]], i32 [[N:%.*]], i1 [[C_1:%.*]], i1 [[C_2:%.*]]) {
+; CHECK-NEXT: [[ENTRY:.*]]:
+; CHECK-NEXT: br label %[[LOOP_HEADER:.*]]
+; CHECK: [[LOOP_HEADER]]:
+; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
+; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp eq i32 [[IV]], [[N]]
+; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label %[[EXIT_1:.*]], label %[[ELSE_1:.*]]
+; CHECK: [[ELSE_1]]:
+; CHECK-NEXT: br i1 [[C_1]], label %[[EXIT_1]], label %[[ELSE_2:.*]]
+; CHECK: [[ELSE_2]]:
+; CHECK-NEXT: br i1 [[C_2]], label %[[EXIT_2:.*]], label %[[LOOP_LATCH]]
+; CHECK: [[LOOP_LATCH]]:
+; CHECK-NEXT: [[ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[S]], i32 [[IV]]
+; CHECK-NEXT: [[TMP0:%.*]] = load i8, ptr [[ARRAYIDX]], align 1
+; CHECK-NEXT: [[LATCH_C:%.*]] = icmp ult i8 [[TMP0]], 10
+; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i32 [[IV]], 1
+; CHECK-NEXT: br i1 [[LATCH_C]], label %[[LOOP_HEADER]], label %[[EXIT_2]]
+; CHECK: [[EXIT_1]]:
+; CHECK-NEXT: [[T_1:%.*]] = icmp ult i32 [[IV]], [[N]]
+; CHECK-NEXT: ret i1 [[T_1]]
+; CHECK: [[EXIT_2]]:
+; CHECK-NEXT: ret i1 true
+;
+entry:
+ br label %loop.header
+
+loop.header:
+ %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
+ %exitcond.not = icmp eq i32 %iv, %N
+ br i1 %exitcond.not, label %exit.1, label %else.1
+
+else.1:
+ br i1 %c.1, label %exit.1, label %else.2
+
+else.2:
+ br i1 %c.2, label %exit.2, label %loop.latch
+
+loop.latch:
+ %arrayidx = getelementptr inbounds i8, ptr %s, i32 %iv
+ %0 = load i8, ptr %arrayidx, align 1
+ %latch.c = icmp ult i8 %0, 10
+ %iv.next = add nuw nsw i32 %iv, 1
+ br i1 %latch.c, label %loop.header, label %exit.2
+
+exit.1:
+ %t.1 = icmp ult i32 %iv, %N
+ ret i1 %t.1
+
+exit.2:
+ %t.2 = icmp ult i32 %iv, %N
+ ret i1 %t.2
+}
+
+define i1 @multi_exiting_loop_ne_
diff erent_exits_const_compare_known(ptr %s) {
+; CHECK-LABEL: define i1 @multi_exiting_loop_ne_
diff erent_exits_const_compare_known(
+; CHECK-SAME: ptr [[S:%.*]]) {
+; CHECK-NEXT: [[ENTRY:.*]]:
+; CHECK-NEXT: br label %[[LOOP_HEADER:.*]]
+; CHECK: [[LOOP_HEADER]]:
+; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 1, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
+; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp ne i32 [[IV]], 1234
+; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label %[[LOOP_LATCH]], label %[[EXIT_1:.*]]
+; CHECK: [[LOOP_LATCH]]:
+; CHECK-NEXT: [[ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[S]], i32 [[IV]]
+; CHECK-NEXT: [[TMP0:%.*]] = load i8, ptr [[ARRAYIDX]], align 1
+; CHECK-NEXT: [[LATCH_C:%.*]] = icmp ult i8 [[TMP0]], 10
+; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i32 [[IV]], 1
+; CHECK-NEXT: br i1 [[LATCH_C]], label %[[LOOP_HEADER]], label %[[EXIT_2:.*]]
+; CHECK: [[EXIT_1]]:
+; CHECK-NEXT: ret i1 true
+; CHECK: [[EXIT_2]]:
+; CHECK-NEXT: ret i1 true
+;
+entry:
+ br label %loop.header
+
+loop.header:
+ %iv = phi i32 [ 1, %entry ], [ %iv.next, %loop.latch ]
+ %exitcond.not = icmp ne i32 %iv, 1234
+ br i1 %exitcond.not, label %loop.latch, label %exit.1
+
+loop.latch:
+ %arrayidx = getelementptr inbounds i8, ptr %s, i32 %iv
+ %0 = load i8, ptr %arrayidx, align 1
+ %latch.c = icmp ult i8 %0, 10
+ %iv.next = add nuw nsw i32 %iv, 1
+ br i1 %latch.c, label %loop.header, label %exit.2
+
+exit.1:
+ %t.1 = icmp ult i32 %iv, 1235
+ ret i1 %t.1
+
+exit.2:
+ %t.2 = icmp ult i32 %iv, 1235
+ ret i1 %t.2
+}
+
+define i1 @multi_exiting_loop_ne_
diff erent_exits_compare_not_known(ptr %s, i32 %N) {
+; CHECK-LABEL: define i1 @multi_exiting_loop_ne_
diff erent_exits_compare_not_known(
+; CHECK-SAME: ptr [[S:%.*]], i32 [[N:%.*]]) {
+; CHECK-NEXT: [[ENTRY:.*]]:
+; CHECK-NEXT: br label %[[LOOP_HEADER:.*]]
+; CHECK: [[LOOP_HEADER]]:
+; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 1, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
+; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp ne i32 [[IV]], [[N]]
+; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label %[[LOOP_LATCH]], label %[[EXIT_1:.*]]
+; CHECK: [[LOOP_LATCH]]:
+; CHECK-NEXT: [[ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[S]], i32 [[IV]]
+; CHECK-NEXT: [[TMP0:%.*]] = load i8, ptr [[ARRAYIDX]], align 1
+; CHECK-NEXT: [[LATCH_C:%.*]] = icmp ult i8 [[TMP0]], 10
+; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i32 [[IV]], 1
+; CHECK-NEXT: br i1 [[LATCH_C]], label %[[LOOP_HEADER]], label %[[EXIT_2:.*]]
+; CHECK: [[EXIT_1]]:
+; CHECK-NEXT: ret i1 false
+; CHECK: [[EXIT_2]]:
+; CHECK-NEXT: [[T_2:%.*]] = icmp ult i32 [[IV]], [[N]]
+; CHECK-NEXT: ret i1 [[T_2]]
+;
+entry:
+ br label %loop.header
+
+loop.header:
+ %iv = phi i32 [ 1, %entry ], [ %iv.next, %loop.latch ]
+ %exitcond.not = icmp ne i32 %iv, %N
+ br i1 %exitcond.not, label %loop.latch, label %exit.1
+
+loop.latch:
+ %arrayidx = getelementptr inbounds i8, ptr %s, i32 %iv
+ %0 = load i8, ptr %arrayidx, align 1
+ %latch.c = icmp ult i8 %0, 10
+ %iv.next = add nuw nsw i32 %iv, 1
+ br i1 %latch.c, label %loop.header, label %exit.2
+
+exit.1:
+ %t.1 = icmp ult i32 %iv, %N
+ ret i1 %t.1
+
+exit.2:
+ %t.2 = icmp ult i32 %iv, %N
+ ret i1 %t.2
+}
More information about the llvm-commits
mailing list