[llvm] 41f905d - [CVP] Add tests for using block values in conditions (NFC)

[Nikita Popov via llvm-commits]

Author: Nikita Popov
Date: 2023-12-13T10:52:42+01:00
New Revision: 41f905ddf007b08afa41e3e22326ecea446f2cfe

URL: https://github.com/llvm/llvm-project/commit/41f905ddf007b08afa41e3e22326ecea446f2cfe
DIFF: https://github.com/llvm/llvm-project/commit/41f905ddf007b08afa41e3e22326ecea446f2cfe.diff

LOG: [CVP] Add tests for using block values in conditions (NFC)

Added: 
    llvm/test/Transforms/CorrelatedValuePropagation/cond-using-block-value.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/CorrelatedValuePropagation/cond-using-block-value.ll b/llvm/test/Transforms/CorrelatedValuePropagation/cond-using-block-value.ll
new file mode 100644
index 00000000000000..d30b31d317a6de
--- /dev/null
+++ b/llvm/test/Transforms/CorrelatedValuePropagation/cond-using-block-value.ll
@@ -0,0 +1,98 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4
+; RUN: opt -S -passes=correlated-propagation < %s | FileCheck %s
+
+declare void @use(i1)
+
+define void @test_icmp_from_implied_cond(i32 %a, i32 %b) {
+; CHECK-LABEL: define void @test_icmp_from_implied_cond(
+; CHECK-SAME: i32 [[A:%.*]], i32 [[B:%.*]]) {
+; CHECK-NEXT:    [[A_CMP:%.*]] = icmp ugt i32 [[A]], 32
+; CHECK-NEXT:    br i1 [[A_CMP]], label [[END:%.*]], label [[L1:%.*]]
+; CHECK:       l1:
+; CHECK-NEXT:    [[COND:%.*]] = icmp ult i32 [[B]], [[A]]
+; CHECK-NEXT:    br i1 [[COND]], label [[L2:%.*]], label [[END]]
+; CHECK:       l2:
+; CHECK-NEXT:    [[B_CMP1:%.*]] = icmp ult i32 [[B]], 32
+; CHECK-NEXT:    call void @use(i1 [[B_CMP1]])
+; CHECK-NEXT:    [[B_CMP2:%.*]] = icmp ult i32 [[B]], 31
+; CHECK-NEXT:    call void @use(i1 [[B_CMP2]])
+; CHECK-NEXT:    ret void
+; CHECK:       end:
+; CHECK-NEXT:    ret void
+;
+  %a.cmp = icmp ugt i32 %a, 32
+  br i1 %a.cmp, label %end, label %l1
+
+l1:
+  %cond = icmp ult i32 %b, %a
+  br i1 %cond, label %l2, label %end
+
+l2:
+  %b.cmp1 = icmp ult i32 %b, 32
+  call void @use(i1 %b.cmp1)
+  %b.cmp2 = icmp ult i32 %b, 31
+  call void @use(i1 %b.cmp2)
+  ret void
+
+end:
+  ret void
+}
+
+define i64 @test_sext_from_implied_cond(i32 %a, i32 %b) {
+; CHECK-LABEL: define i64 @test_sext_from_implied_cond(
+; CHECK-SAME: i32 [[A:%.*]], i32 [[B:%.*]]) {
+; CHECK-NEXT:    [[A_CMP:%.*]] = icmp slt i32 [[A]], 0
+; CHECK-NEXT:    br i1 [[A_CMP]], label [[END:%.*]], label [[L1:%.*]]
+; CHECK:       l1:
+; CHECK-NEXT:    [[COND:%.*]] = icmp ult i32 [[B]], [[A]]
+; CHECK-NEXT:    br i1 [[COND]], label [[L2:%.*]], label [[END]]
+; CHECK:       l2:
+; CHECK-NEXT:    [[SEXT:%.*]] = sext i32 [[B]] to i64
+; CHECK-NEXT:    ret i64 [[SEXT]]
+; CHECK:       end:
+; CHECK-NEXT:    ret i64 0
+;
+  %a.cmp = icmp slt i32 %a, 0
+  br i1 %a.cmp, label %end, label %l1
+
+l1:
+  %cond = icmp ult i32 %b, %a
+  br i1 %cond, label %l2, label %end
+
+l2:
+  %sext = sext i32 %b to i64
+  ret i64 %sext
+
+end:
+  ret i64 0
+}
+
+define void @test_icmp_from_implied_range(i16 %x, i32 %b) {
+; CHECK-LABEL: define void @test_icmp_from_implied_range(
+; CHECK-SAME: i16 [[X:%.*]], i32 [[B:%.*]]) {
+; CHECK-NEXT:    [[A:%.*]] = zext i16 [[X]] to i32
+; CHECK-NEXT:    [[COND:%.*]] = icmp ult i32 [[B]], [[A]]
+; CHECK-NEXT:    br i1 [[COND]], label [[L1:%.*]], label [[END:%.*]]
+; CHECK:       l1:
+; CHECK-NEXT:    [[B_CMP1:%.*]] = icmp ult i32 [[B]], 65535
+; CHECK-NEXT:    call void @use(i1 [[B_CMP1]])
+; CHECK-NEXT:    [[B_CMP2:%.*]] = icmp ult i32 [[B]], 65534
+; CHECK-NEXT:    call void @use(i1 [[B_CMP2]])
+; CHECK-NEXT:    ret void
+; CHECK:       end:
+; CHECK-NEXT:    ret void
+;
+  %a = zext i16 %x to i32
+  %cond = icmp ult i32 %b, %a
+  br i1 %cond, label %l1, label %end
+
+l1:
+  %b.cmp1 = icmp ult i32 %b, 65535
+  call void @use(i1 %b.cmp1)
+  %b.cmp2 = icmp ult i32 %b, 65534
+  call void @use(i1 %b.cmp2)
+  ret void
+
+end:
+  ret void
+}