[PowerPC] Do not rely on MachineSink threshold to optimize CR spilling

spill_p0_setb.ll and knowCRBitSpill.ll rely on a probability threshold
in MachineSink to pull down CRSET/CRUNSET. It's worth to do so anyway
and let register allocation decide otherwise.
Additionally, make the interface accept a const MI.

Author: guy-david

llvm/lib/Target/PowerPC/PPCInstrInfo.cpp

@@ -1072,6 +1072,13 @@ Register PPCInstrInfo::isLoadFromStackSlot(const MachineInstr &MI,
   return 0;
 }
 
+/// Sink down CodeGen-only, cheap instructions to allow further
+/// optimizations which are only applied intra-block.
+bool PPCInstrInfo::shouldBreakCriticalEdgeToSink(MachineInstr &MI) const {
+  // These can turn into immediates, see PPCRegisterInfo::lowerCRBitRestore.
+  return MI.getOpcode() == PPC::CRSET || MI.getOpcode() == PPC::CRUNSET;
+}
+
 // For opcodes with the ReMaterializable flag set, this function is called to
 // verify the instruction is really rematable.
 bool PPCInstrInfo::isReallyTriviallyReMaterializable(

llvm/lib/Target/PowerPC/PPCInstrInfo.h

@@ -424,6 +424,7 @@ class PPCInstrInfo : public PPCGenInstrInfo {
                              unsigned &SubIdx) const override;
   Register isLoadFromStackSlot(const MachineInstr &MI,
                                int &FrameIndex) const override;
+  bool shouldBreakCriticalEdgeToSink(MachineInstr &MI) const override;
   bool isReallyTriviallyReMaterializable(const MachineInstr &MI) const override;
   Register isStoreToStackSlot(const MachineInstr &MI,
                               int &FrameIndex) const override;

llvm/test/CodeGen/PowerPC/sink-crset-crunset.mir

@@ -0,0 +1,27 @@
+# RUN: llc -run-pass=machine-sink -mtriple=powerpc64-unknown-linux-gnu -o - %s | FileCheck %s
+
+---
+name:            sink-crset-crunset
+body:             |
+  ; CHECK-LABEL: bb.0.entry
+  bb.0.entry:
+    successors: %bb.1(0x40000000), %bb.2(0x40000000)
+    liveins: $r3
+    ; CHECK-NOT: CRSET
+    ; CHECK-NOT: CRUNSET
+    %cond:crrc = CMPLWI killed renamable $r3, 0
+    %a:crbitrc = CRSET
+    %b:crbitrc = CRUNSET
+    BCC 76, killed %cond, %bb.2
+  ; CHECK-LABEL: bb.1
+  bb.1:
+    ; CHECK-DAG: %a:crbitrc = CRSET
+    ; CHECK-DAG: %b:crbitrc = CRUNSET
+    ; CHECK-DAG: %and:crbitrc = CRAND killed %a, %b
+    %and:crbitrc = CRAND killed %a, %b
+  ; CHECK-LABEL: bb.2
+  bb.2:
+    ; CHECK: BLR
+    BLR implicit $lr, implicit $rm
+
+...

llvm/test/CodeGen/PowerPC/spill_p9_setb.ll

@@ -16,21 +16,21 @@
 
 define void @p9_setb_spill() {
 ; CHECK-P9-LABEL: p9_setb_spill:
-; CHECK-P9:       # %bb.1: # %if.then
+; CHECK-P9:       .LBB0_2: # %if.then
 ; CHECK-P9-DAG:    crnot 4*cr[[CREG:.*]]+lt, eq
 ; CHECK-P9-DAG:    setb [[REG1:.*]], cr[[CREG]]
 ; CHECK-P9-DAG:    stw [[REG1]]
 ; CHECK-P9:        blr
-; CHECK-P9:        .LBB0_4: # %if.then1
+; CHECK-P9:       .LBB0_5: # %if.then1
 ;
 ; CHECK-P8-LABEL: p9_setb_spill:
-; CHECK-P8:       # %bb.1: # %if.then
+; CHECK-P8:       .LBB0_2: # %if.then
 ; CHECK-P8-DAG:    crnot 4*cr[[CREG2:.*]]+lt, eq
 ; CHECK-P8-DAG:    mfocrf [[REG2:.*]],
 ; CHECK-P8-DAG:    rlwinm [[REG2]], [[REG2]]
 ; CHECK-P8-DAG:    stw [[REG2]]
 ; CHECK-P8:        blr
-; CHECK-P8:        .LBB0_4: # %if.then1
+; CHECK-P8:       .LBB0_5: # %if.then1
 entry:
   br i1 undef, label %if.end, label %if.then