[PATCH] D74747: [JumpThreading] Handle unconditional PredBB when threading jumps through two basic blocks

[Fangrui Song via Phabricator via llvm-commits]

MaskRay updated this revision to Diff 245193.
MaskRay added a comment.

Improve bb names


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D74747/new/

https://reviews.llvm.org/D74747

Files:
  llvm/lib/Transforms/Scalar/JumpThreading.cpp
  llvm/test/Transforms/JumpThreading/thread-two-bbs6.ll


Index: llvm/test/Transforms/JumpThreading/thread-two-bbs6.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/JumpThreading/thread-two-bbs6.ll
@@ -0,0 +1,48 @@
+;; Test that we can handle unconditional PredBB when threading jumps through two
+;; successive basic blocks.
+; RUN: opt -S -passes='function(jump-threading)' < %s | FileCheck %s
+
+target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+define i32 @f(i32* %0) {
+; CHECK-LABEL: @f
+; CHECK: br i1 %good, label %pred.bb, label %bb.thread
+entry:
+  %size = call i64 @get_size(i32* %0)
+  %good = icmp ugt i64 %size, 3
+  br i1 %good, label %pred.bb, label %pred.pred.bb
+
+; CHECK:      pred.bb:
+; CHECK-NEXT:  %v = load i32, i32* %0
+; CHECK-NEXT:  br label %bb
+; CHECK-EMPTY:
+; CHECK-NEXT: bb.thread:
+; CHECK-NEXT:  call void @effect()
+; CHECK-NEXT:  %v1 = load i32, i32* %0
+; CHECK-NEXT:  call void @effect1(i8* blockaddress(@f, %bb))
+; CHECK:      bb:
+; CHECK-NEXT:  %v2 = phi i32 [ %v, %pred.bb ]
+; CHECK-NEXT:  call void @effect1(i8* blockaddress(@f, %bb))
+pred.pred.bb:                                        ; preds = %entry
+  call void @effect()
+  br label %pred.bb
+pred.bb:                                             ; preds = %pred.pred.bb, %entry
+  %v = load i32, i32* %0
+  br label %bb
+
+bb:                                                  ; preds = %pred.bb
+  call void @effect1(i8* blockaddress(@f, %bb))
+  br i1 %good, label %cont2, label %cont1
+
+cont1:                                               ; preds = %bb
+  br i1 %good, label %exit, label %cont2
+cont2:                                               ; preds = %bb
+  br label %exit
+exit:                                                ; preds = %cont1, %cont2
+  ret i32 %v
+}
+
+declare i64 @get_size(i32*)
+declare void @effect()
+declare void @effect1(i8*)
Index: llvm/lib/Transforms/Scalar/JumpThreading.cpp
===================================================================
--- llvm/lib/Transforms/Scalar/JumpThreading.cpp
+++ llvm/lib/Transforms/Scalar/JumpThreading.cpp
@@ -2250,8 +2250,9 @@
 
   AddPHINodeEntriesForMappedBlock(PredBBBranch->getSuccessor(0), PredBB, NewBB,
                                   ValueMapping);
-  AddPHINodeEntriesForMappedBlock(PredBBBranch->getSuccessor(1), PredBB, NewBB,
-                                  ValueMapping);
+  if (PredBBBranch->isConditional())
+    AddPHINodeEntriesForMappedBlock(PredBBBranch->getSuccessor(1), PredBB,
+                                    NewBB, ValueMapping);
 
   DTU->applyUpdatesPermissive(
       {{DominatorTree::Insert, NewBB, CondBr->getSuccessor(0)},


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D74747.245193.patch
Type: text/x-patch
Size: 2739 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200218/7dd7f25a/attachment.bin>