[InstCombine] Add tests for assume implication (NFC)

diff --git a/llvm/test/Transforms/InstCombine/assume_icmp.ll b/llvm/test/Transforms/InstCombine/assume_icmp.ll
new file mode 100644 (file)
index 0000000..5da2567
--- /dev/null
+++ b/llvm/test/Transforms/InstCombine/assume_icmp.ll
@@ -0,0 +1,130 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -instcombine -S | FileCheck %s
+
+declare void @llvm.assume(i1)
+declare void @use(i1)
+
+define void @basic_ugt(i32 %x, i32 %y) {
+; CHECK-LABEL: @basic_ugt(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp ugt i32 [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP1]])
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp ugt i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP2]])
+; CHECK-NEXT:    [[CMP3:%.*]] = icmp uge i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP3]])
+; CHECK-NEXT:    [[CMP4:%.*]] = icmp ult i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP4]])
+; CHECK-NEXT:    [[CMP5:%.*]] = icmp ule i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP5]])
+; CHECK-NEXT:    [[CMP6:%.*]] = icmp ugt i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP6]])
+; CHECK-NEXT:    [[CMP7:%.*]] = icmp uge i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP7]])
+; CHECK-NEXT:    [[CMP8:%.*]] = icmp ult i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP8]])
+; CHECK-NEXT:    [[CMP9:%.*]] = icmp ule i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP9]])
+; CHECK-NEXT:    ret void
+;
+  %cmp1 = icmp ugt i32 %x, %y
+  call void @llvm.assume(i1 %cmp1)
+
+  %cmp2 = icmp ugt i32 %x, %y
+  call void @use(i1 %cmp2)
+  %cmp3 = icmp uge i32 %x, %y
+  call void @use(i1 %cmp3)
+  %cmp4 = icmp ult i32 %x, %y
+  call void @use(i1 %cmp4)
+  %cmp5 = icmp ule i32 %x, %y
+  call void @use(i1 %cmp5)
+
+  %cmp6 = icmp ugt i32 %y, %x
+  call void @use(i1 %cmp6)
+  %cmp7 = icmp uge i32 %y, %x
+  call void @use(i1 %cmp7)
+  %cmp8 = icmp ult i32 %y, %x
+  call void @use(i1 %cmp8)
+  %cmp9 = icmp ule i32 %y, %x
+  call void @use(i1 %cmp9)
+
+  ret void
+}
+
+define void @basic_uge(i32 %x, i32 %y) {
+; CHECK-LABEL: @basic_uge(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp uge i32 [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP1]])
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp ugt i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP2]])
+; CHECK-NEXT:    [[CMP3:%.*]] = icmp uge i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP3]])
+; CHECK-NEXT:    [[CMP4:%.*]] = icmp ult i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP4]])
+; CHECK-NEXT:    [[CMP5:%.*]] = icmp ule i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP5]])
+; CHECK-NEXT:    [[CMP6:%.*]] = icmp ugt i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP6]])
+; CHECK-NEXT:    [[CMP7:%.*]] = icmp uge i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP7]])
+; CHECK-NEXT:    [[CMP8:%.*]] = icmp ult i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP8]])
+; CHECK-NEXT:    [[CMP9:%.*]] = icmp ule i32 [[Y]], [[X]]
+; CHECK-NEXT:    call void @use(i1 [[CMP9]])
+; CHECK-NEXT:    ret void
+;
+  %cmp1 = icmp uge i32 %x, %y
+  call void @llvm.assume(i1 %cmp1)
+
+  %cmp2 = icmp ugt i32 %x, %y
+  call void @use(i1 %cmp2)
+  %cmp3 = icmp uge i32 %x, %y
+  call void @use(i1 %cmp3)
+  %cmp4 = icmp ult i32 %x, %y
+  call void @use(i1 %cmp4)
+  %cmp5 = icmp ule i32 %x, %y
+  call void @use(i1 %cmp5)
+
+  %cmp6 = icmp ugt i32 %y, %x
+  call void @use(i1 %cmp6)
+  %cmp7 = icmp uge i32 %y, %x
+  call void @use(i1 %cmp7)
+  %cmp8 = icmp ult i32 %y, %x
+  call void @use(i1 %cmp8)
+  %cmp9 = icmp ule i32 %y, %x
+  call void @use(i1 %cmp9)
+
+  ret void
+}
+
+define void @and(i32 %x, i32 %y, i32 %z) {
+; CHECK-LABEL: @and(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp ugt i32 [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp ugt i32 [[Z:%.*]], [[Y]]
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP1]])
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP2]])
+; CHECK-NEXT:    [[CMP3:%.*]] = icmp ugt i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP3]])
+; CHECK-NEXT:    [[CMP4:%.*]] = icmp uge i32 [[X]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP4]])
+; CHECK-NEXT:    [[CMP5:%.*]] = icmp ugt i32 [[Z]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP5]])
+; CHECK-NEXT:    [[CMP6:%.*]] = icmp uge i32 [[Z]], [[Y]]
+; CHECK-NEXT:    call void @use(i1 [[CMP6]])
+; CHECK-NEXT:    ret void
+;
+  %cmp1 = icmp ugt i32 %x, %y
+  %cmp2 = icmp ugt i32 %z, %y
+  %and = and i1 %cmp1, %cmp2
+  call void @llvm.assume(i1 %and)
+
+  %cmp3 = icmp ugt i32 %x, %y
+  call void @use(i1 %cmp3)
+  %cmp4 = icmp uge i32 %x, %y
+  call void @use(i1 %cmp4)
+  %cmp5 = icmp ugt i32 %z, %y
+  call void @use(i1 %cmp5)
+  %cmp6 = icmp uge i32 %z, %y
+  call void @use(i1 %cmp6)
+
+  ret void
+}