[BPI] Look through bitcasts in calcZeroHeuristic
authorWei Wang <apollo.mobility@gmail.com>
Fri, 13 Nov 2020 19:17:47 +0000 (11:17 -0800)
committerWei Wang <apollo.mobility@gmail.com>
Tue, 17 Nov 2020 17:33:05 +0000 (09:33 -0800)
Constant hoisting may hide the constant value behind bitcast for And's
operand. Track down the constant to make the BFI result consistent
regardless of hoisting.

Differential Revision: https://reviews.llvm.org/D91450

llvm/lib/Analysis/BranchProbabilityInfo.cpp
llvm/test/Analysis/BranchProbabilityInfo/hoist.ll [new file with mode: 0644]

index ca4b580..ac5dbe1 100644 (file)
@@ -881,7 +881,7 @@ bool BranchProbabilityInfo::calcZeroHeuristics(const BasicBlock *BB,
   // we don't have information about probabilities.
   if (Instruction *LHS = dyn_cast<Instruction>(CI->getOperand(0)))
     if (LHS->getOpcode() == Instruction::And)
-      if (ConstantInt *AndRHS = dyn_cast<ConstantInt>(LHS->getOperand(1)))
+      if (ConstantInt *AndRHS = GetConstantInt(LHS->getOperand(1)))
         if (AndRHS->getValue().isPowerOf2())
           return false;
 
diff --git a/llvm/test/Analysis/BranchProbabilityInfo/hoist.ll b/llvm/test/Analysis/BranchProbabilityInfo/hoist.ll
new file mode 100644 (file)
index 0000000..1daa5d1
--- /dev/null
@@ -0,0 +1,63 @@
+; RUN: opt < %s -analyze -branch-prob -enable-new-pm=0 | FileCheck %s
+; RUN: opt < %s -passes='print<branch-prob>' -disable-output 2>&1 | FileCheck %s
+
+; CHECK-LABEL: no_hoist
+; CHECK: edge entry -> if.end probability is 0x40000000 / 0x80000000 = 50.00%
+; CHECK: edge entry -> if.then probability is 0x40000000 / 0x80000000 = 50.00%
+; CHECK: edge if.end -> if.end4 probability is 0x40000000 / 0x80000000 = 50.00%
+; CHECK: edge if.end -> if.then3 probability is 0x40000000 / 0x80000000 = 50.00%
+define dso_local void @no_hoist(i64 %arg1, i64 %arg2) local_unnamed_addr #0 {
+entry:
+  %and = and i64 %arg1, 1152921504606846976
+  %tobool.not = icmp eq i64 %and, 0
+  br i1 %tobool.not, label %if.end, label %if.then
+
+if.then:                                          ; preds = %entry
+  tail call void @bar()
+  br label %if.end
+
+if.end:                                           ; preds = %if.then, %entry
+  %and1 = and i64 %arg2, 1152921504606846976
+  %tobool2.not = icmp eq i64 %and1, 0
+  br i1 %tobool2.not, label %if.end4, label %if.then3
+
+if.then3:                                         ; preds = %if.end
+  tail call void @baz()
+  br label %if.end4
+
+if.end4:                                          ; preds = %if.then3, %if.end
+  ret void
+}
+
+; CHECK-LABEL: hoist
+; CHECK: edge entry -> if.end probability is 0x40000000 / 0x80000000 = 50.00%
+; CHECK: edge entry -> if.then probability is 0x40000000 / 0x80000000 = 50.00%
+; CHECK: edge if.end -> if.end4 probability is 0x40000000 / 0x80000000 = 50.00%
+; CHECK: edge if.end -> if.then3 probability is 0x40000000 / 0x80000000 = 50.00%
+define dso_local void @hoist(i64 %arg1, i64 %arg2) local_unnamed_addr #0 {
+entry:
+  %const = bitcast i64 1152921504606846976 to i64
+  %and = and i64 %arg1, %const
+  %tobool.not = icmp eq i64 %and, 0
+  br i1 %tobool.not, label %if.end, label %if.then
+
+if.then:                                          ; preds = %entry
+  tail call void @bar()
+  br label %if.end
+
+if.end:                                           ; preds = %if.then, %entry
+  %and1 = and i64 %arg2, %const
+  %tobool2.not = icmp eq i64 %and1, 0
+  br i1 %tobool2.not, label %if.end4, label %if.then3
+
+if.then3:                                         ; preds = %if.end
+  tail call void @baz()
+  br label %if.end4
+
+if.end4:                                          ; preds = %if.then3, %if.end
+  ret void
+}
+
+declare dso_local void @bar() local_unnamed_addr #1
+
+declare dso_local void @baz() local_unnamed_addr #1