[llvm] 1718e64 - [SimplifyCFG] Add additional tests with assume (NFC)

[Nikita Popov via llvm-commits]

Author: Nikita Popov
Date: 2023-06-28T14:22:19+02:00
New Revision: 1718e640dbda341f5583159348ad380637ba6de5

URL: https://github.com/llvm/llvm-project/commit/1718e640dbda341f5583159348ad380637ba6de5
DIFF: https://github.com/llvm/llvm-project/commit/1718e640dbda341f5583159348ad380637ba6de5.diff

LOG: [SimplifyCFG] Add additional tests with assume (NFC)

Added: 
    

Modified: 
    llvm/test/Transforms/SimplifyCFG/assume.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/SimplifyCFG/assume.ll b/llvm/test/Transforms/SimplifyCFG/assume.ll
index f8e179354cc13..eaebf376a9d72 100644
--- a/llvm/test/Transforms/SimplifyCFG/assume.ll
+++ b/llvm/test/Transforms/SimplifyCFG/assume.ll
@@ -130,5 +130,69 @@ join:
   ret i32 %phi
 }
 
+define void @empty_block_with_assume(i1 %c, i32 %x) {
+; CHECK-LABEL: @empty_block_with_assume(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    br i1 [[C:%.*]], label [[IF:%.*]], label [[ELSE:%.*]]
+; CHECK:       if:
+; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i32 [[X:%.*]], 0
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP]])
+; CHECK-NEXT:    br label [[JOIN:%.*]]
+; CHECK:       else:
+; CHECK-NEXT:    call void @dummy()
+; CHECK-NEXT:    br label [[JOIN]]
+; CHECK:       join:
+; CHECK-NEXT:    ret void
+;
+entry:
+  br i1 %c, label %if, label %else
+
+if:
+  %cmp = icmp ne i32 %x, 0
+  call void @llvm.assume(i1 %cmp)
+  br label %join
+
+else:
+  call void @dummy()
+  br label %join
+
+join:
+  ret void
+}
+
+define void @not_empty_block_with_assume(i1 %c) {
+; CHECK-LABEL: @not_empty_block_with_assume(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    br i1 [[C:%.*]], label [[IF:%.*]], label [[ELSE:%.*]]
+; CHECK:       if:
+; CHECK-NEXT:    [[X:%.*]] = call i32 @may_have_side_effect()
+; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i32 [[X]], 0
+; CHECK-NEXT:    call void @llvm.assume(i1 [[CMP]])
+; CHECK-NEXT:    br label [[JOIN:%.*]]
+; CHECK:       else:
+; CHECK-NEXT:    call void @dummy()
+; CHECK-NEXT:    br label [[JOIN]]
+; CHECK:       join:
+; CHECK-NEXT:    ret void
+;
+entry:
+  br i1 %c, label %if, label %else
+
+if:
+  %x = call i32 @may_have_side_effect()
+  %cmp = icmp ne i32 %x, 0
+  call void @llvm.assume(i1 %cmp)
+  br label %join
+
+else:
+  call void @dummy()
+  br label %join
+
+join:
+  ret void
+}
+
+declare void @dummy()
+declare i32 @may_have_side_effect()
 declare void @llvm.assume(i1) nounwind