1 //===- HexagonNewValueJump.cpp - Hexagon Backend New Value Jump -----------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This implements NewValueJump pass in Hexagon.
10 // Ideally, we should merge this as a Peephole pass prior to register
11 // allocation, but because we have a spill in between the feeder and new value
12 // jump instructions, we are forced to write after register allocation.
13 // Having said that, we should re-attempt to pull this earlier at some point
14 // in future.
15 
16 // The basic approach looks for sequence of predicated jump, compare instruciton
17 // that genereates the predicate and, the feeder to the predicate. Once it finds
18 // all, it collapses compare and jump instruction into a new value jump
19 // intstructions.
20 //
21 //===----------------------------------------------------------------------===//
22 
23 #include "llvm/InitializePasses.h"
24 #include "Hexagon.h"
25 #include "HexagonInstrInfo.h"
26 #include "HexagonRegisterInfo.h"
27 #include "HexagonSubtarget.h"
28 #include "llvm/ADT/Statistic.h"
29 #include "llvm/CodeGen/MachineBasicBlock.h"
30 #include "llvm/CodeGen/MachineBranchProbabilityInfo.h"
31 #include "llvm/CodeGen/MachineFunction.h"
32 #include "llvm/CodeGen/MachineFunctionPass.h"
33 #include "llvm/CodeGen/MachineInstr.h"
34 #include "llvm/CodeGen/MachineInstrBuilder.h"
35 #include "llvm/CodeGen/MachineOperand.h"
36 #include "llvm/CodeGen/MachineRegisterInfo.h"
37 #include "llvm/CodeGen/TargetOpcodes.h"
38 #include "llvm/CodeGen/TargetRegisterInfo.h"
39 #include "llvm/CodeGen/TargetSubtargetInfo.h"
40 #include "llvm/IR/DebugLoc.h"
41 #include "llvm/MC/MCInstrDesc.h"
42 #include "llvm/Pass.h"
43 #include "llvm/Support/BranchProbability.h"
44 #include "llvm/Support/CommandLine.h"
45 #include "llvm/Support/Debug.h"
46 #include "llvm/Support/ErrorHandling.h"
47 #include "llvm/Support/MathExtras.h"
48 #include "llvm/Support/raw_ostream.h"
49 #include <cassert>
50 #include <cstdint>
51 #include <iterator>
52 
53 using namespace llvm;
54 
55 #define DEBUG_TYPE "hexagon-nvj"
56 
57 STATISTIC(NumNVJGenerated, "Number of New Value Jump Instructions created");
58 
59 static cl::opt<int> DbgNVJCount("nvj-count", cl::init(-1), cl::Hidden,
60     cl::desc("Maximum number of predicated jumps to be converted to "
61     "New Value Jump"));
62 
63 static cl::opt<bool> DisableNewValueJumps("disable-nvjump", cl::Hidden,
64     cl::ZeroOrMore, cl::init(false),
65     cl::desc("Disable New Value Jumps"));
66 
67 namespace llvm {
68 
69 FunctionPass *createHexagonNewValueJump();
70 void initializeHexagonNewValueJumpPass(PassRegistry&);
71 
72 } // end namespace llvm
73 
74 namespace {
75 
76   struct HexagonNewValueJump : public MachineFunctionPass {
77     static char ID;
78 
HexagonNewValueJump__anonee0125660111::HexagonNewValueJump79     HexagonNewValueJump() : MachineFunctionPass(ID) {}
80 
getAnalysisUsage__anonee0125660111::HexagonNewValueJump81     void getAnalysisUsage(AnalysisUsage &AU) const override {
82       AU.addRequired<MachineBranchProbabilityInfo>();
83       MachineFunctionPass::getAnalysisUsage(AU);
84     }
85 
getPassName__anonee0125660111::HexagonNewValueJump86     StringRef getPassName() const override { return "Hexagon NewValueJump"; }
87 
88     bool runOnMachineFunction(MachineFunction &Fn) override;
89 
getRequiredProperties__anonee0125660111::HexagonNewValueJump90     MachineFunctionProperties getRequiredProperties() const override {
91       return MachineFunctionProperties().set(
92           MachineFunctionProperties::Property::NoVRegs);
93     }
94 
95   private:
96     const HexagonInstrInfo *QII;
97     const HexagonRegisterInfo *QRI;
98 
99     /// A handle to the branch probability pass.
100     const MachineBranchProbabilityInfo *MBPI;
101 
102     bool isNewValueJumpCandidate(const MachineInstr &MI) const;
103   };
104 
105 } // end anonymous namespace
106 
107 char HexagonNewValueJump::ID = 0;
108 
109 INITIALIZE_PASS_BEGIN(HexagonNewValueJump, "hexagon-nvj",
110                       "Hexagon NewValueJump", false, false)
INITIALIZE_PASS_DEPENDENCY(MachineBranchProbabilityInfo)111 INITIALIZE_PASS_DEPENDENCY(MachineBranchProbabilityInfo)
112 INITIALIZE_PASS_END(HexagonNewValueJump, "hexagon-nvj",
113                     "Hexagon NewValueJump", false, false)
114 
115 // We have identified this II could be feeder to NVJ,
116 // verify that it can be.
117 static bool canBeFeederToNewValueJump(const HexagonInstrInfo *QII,
118                                       const TargetRegisterInfo *TRI,
119                                       MachineBasicBlock::iterator II,
120                                       MachineBasicBlock::iterator end,
121                                       MachineBasicBlock::iterator skip,
122                                       MachineFunction &MF) {
123   // Predicated instruction can not be feeder to NVJ.
124   if (QII->isPredicated(*II))
125     return false;
126 
127   // Bail out if feederReg is a paired register (double regs in
128   // our case). One would think that we can check to see if a given
129   // register cmpReg1 or cmpReg2 is a sub register of feederReg
130   // using -- if (QRI->isSubRegister(feederReg, cmpReg1) logic
131   // before the callsite of this function
132   // But we can not as it comes in the following fashion.
133   //    %d0 = Hexagon_S2_lsr_r_p killed %d0, killed %r2
134   //    %r0 = KILL %r0, implicit killed %d0
135   //    %p0 = CMPEQri killed %r0, 0
136   // Hence, we need to check if it's a KILL instruction.
137   if (II->getOpcode() == TargetOpcode::KILL)
138     return false;
139 
140   if (II->isImplicitDef())
141     return false;
142 
143   if (QII->isSolo(*II))
144     return false;
145 
146   if (QII->isFloat(*II))
147     return false;
148 
149   // Make sure that the (unique) def operand is a register from IntRegs.
150   bool HadDef = false;
151   for (const MachineOperand &Op : II->operands()) {
152     if (!Op.isReg() || !Op.isDef())
153       continue;
154     if (HadDef)
155       return false;
156     HadDef = true;
157     if (!Hexagon::IntRegsRegClass.contains(Op.getReg()))
158       return false;
159   }
160   assert(HadDef);
161 
162   // Make sure there is no 'def' or 'use' of any of the uses of
163   // feeder insn between its definition, this MI and jump, jmpInst
164   // skipping compare, cmpInst.
165   // Here's the example.
166   //    r21=memub(r22+r24<<#0)
167   //    p0 = cmp.eq(r21, #0)
168   //    r4=memub(r3+r21<<#0)
169   //    if (p0.new) jump:t .LBB29_45
170   // Without this check, it will be converted into
171   //    r4=memub(r3+r21<<#0)
172   //    r21=memub(r22+r24<<#0)
173   //    p0 = cmp.eq(r21, #0)
174   //    if (p0.new) jump:t .LBB29_45
175   // and result WAR hazards if converted to New Value Jump.
176   for (unsigned i = 0; i < II->getNumOperands(); ++i) {
177     if (II->getOperand(i).isReg() &&
178         (II->getOperand(i).isUse() || II->getOperand(i).isDef())) {
179       MachineBasicBlock::iterator localII = II;
180       ++localII;
181       Register Reg = II->getOperand(i).getReg();
182       for (MachineBasicBlock::iterator localBegin = localII; localBegin != end;
183            ++localBegin) {
184         if (localBegin == skip)
185           continue;
186         // Check for Subregisters too.
187         if (localBegin->modifiesRegister(Reg, TRI) ||
188             localBegin->readsRegister(Reg, TRI))
189           return false;
190       }
191     }
192   }
193   return true;
194 }
195 
196 // These are the common checks that need to performed
197 // to determine if
198 // 1. compare instruction can be moved before jump.
199 // 2. feeder to the compare instruction can be moved before jump.
commonChecksToProhibitNewValueJump(bool afterRA,MachineBasicBlock::iterator MII)200 static bool commonChecksToProhibitNewValueJump(bool afterRA,
201                           MachineBasicBlock::iterator MII) {
202   // If store in path, bail out.
203   if (MII->mayStore())
204     return false;
205 
206   // if call in path, bail out.
207   if (MII->isCall())
208     return false;
209 
210   // if NVJ is running prior to RA, do the following checks.
211   if (!afterRA) {
212     // The following Target Opcode instructions are spurious
213     // to new value jump. If they are in the path, bail out.
214     // KILL sets kill flag on the opcode. It also sets up a
215     // single register, out of pair.
216     //    %d0 = S2_lsr_r_p killed %d0, killed %r2
217     //    %r0 = KILL %r0, implicit killed %d0
218     //    %p0 = C2_cmpeqi killed %r0, 0
219     // PHI can be anything after RA.
220     // COPY can remateriaze things in between feeder, compare and nvj.
221     if (MII->getOpcode() == TargetOpcode::KILL ||
222         MII->getOpcode() == TargetOpcode::PHI ||
223         MII->getOpcode() == TargetOpcode::COPY)
224       return false;
225 
226     // The following pseudo Hexagon instructions sets "use" and "def"
227     // of registers by individual passes in the backend. At this time,
228     // we don't know the scope of usage and definitions of these
229     // instructions.
230     if (MII->getOpcode() == Hexagon::LDriw_pred ||
231         MII->getOpcode() == Hexagon::STriw_pred)
232       return false;
233   }
234 
235   return true;
236 }
237 
canCompareBeNewValueJump(const HexagonInstrInfo * QII,const TargetRegisterInfo * TRI,MachineBasicBlock::iterator II,unsigned pReg,bool secondReg,bool optLocation,MachineBasicBlock::iterator end,MachineFunction & MF)238 static bool canCompareBeNewValueJump(const HexagonInstrInfo *QII,
239                                      const TargetRegisterInfo *TRI,
240                                      MachineBasicBlock::iterator II,
241                                      unsigned pReg,
242                                      bool secondReg,
243                                      bool optLocation,
244                                      MachineBasicBlock::iterator end,
245                                      MachineFunction &MF) {
246   MachineInstr &MI = *II;
247 
248   // If the second operand of the compare is an imm, make sure it's in the
249   // range specified by the arch.
250   if (!secondReg) {
251     const MachineOperand &Op2 = MI.getOperand(2);
252     if (!Op2.isImm())
253       return false;
254 
255     int64_t v = Op2.getImm();
256     bool Valid = false;
257 
258     switch (MI.getOpcode()) {
259       case Hexagon::C2_cmpeqi:
260       case Hexagon::C4_cmpneqi:
261       case Hexagon::C2_cmpgti:
262       case Hexagon::C4_cmpltei:
263         Valid = (isUInt<5>(v) || v == -1);
264         break;
265       case Hexagon::C2_cmpgtui:
266       case Hexagon::C4_cmplteui:
267         Valid = isUInt<5>(v);
268         break;
269       case Hexagon::S2_tstbit_i:
270       case Hexagon::S4_ntstbit_i:
271         Valid = (v == 0);
272         break;
273     }
274 
275     if (!Valid)
276       return false;
277   }
278 
279   unsigned cmpReg1, cmpOp2 = 0; // cmpOp2 assignment silences compiler warning.
280   cmpReg1 = MI.getOperand(1).getReg();
281 
282   if (secondReg) {
283     cmpOp2 = MI.getOperand(2).getReg();
284 
285     // If the same register appears as both operands, we cannot generate a new
286     // value compare. Only one operand may use the .new suffix.
287     if (cmpReg1 == cmpOp2)
288       return false;
289 
290     // Make sure that the second register is not from COPY
291     // at machine code level, we don't need this, but if we decide
292     // to move new value jump prior to RA, we would be needing this.
293     MachineRegisterInfo &MRI = MF.getRegInfo();
294     if (!Register::isPhysicalRegister(cmpOp2)) {
295       MachineInstr *def = MRI.getVRegDef(cmpOp2);
296       if (def->getOpcode() == TargetOpcode::COPY)
297         return false;
298     }
299   }
300 
301   // Walk the instructions after the compare (predicate def) to the jump,
302   // and satisfy the following conditions.
303   ++II;
304   for (MachineBasicBlock::iterator localII = II; localII != end; ++localII) {
305     if (localII->isDebugInstr())
306       continue;
307 
308     // Check 1.
309     // If "common" checks fail, bail out.
310     if (!commonChecksToProhibitNewValueJump(optLocation, localII))
311       return false;
312 
313     // Check 2.
314     // If there is a def or use of predicate (result of compare), bail out.
315     if (localII->modifiesRegister(pReg, TRI) ||
316         localII->readsRegister(pReg, TRI))
317       return false;
318 
319     // Check 3.
320     // If there is a def of any of the use of the compare (operands of compare),
321     // bail out.
322     // Eg.
323     //    p0 = cmp.eq(r2, r0)
324     //    r2 = r4
325     //    if (p0.new) jump:t .LBB28_3
326     if (localII->modifiesRegister(cmpReg1, TRI) ||
327         (secondReg && localII->modifiesRegister(cmpOp2, TRI)))
328       return false;
329   }
330   return true;
331 }
332 
333 // Given a compare operator, return a matching New Value Jump compare operator.
334 // Make sure that MI here is included in isNewValueJumpCandidate.
getNewValueJumpOpcode(MachineInstr * MI,int reg,bool secondRegNewified,MachineBasicBlock * jmpTarget,const MachineBranchProbabilityInfo * MBPI)335 static unsigned getNewValueJumpOpcode(MachineInstr *MI, int reg,
336                                       bool secondRegNewified,
337                                       MachineBasicBlock *jmpTarget,
338                                       const MachineBranchProbabilityInfo
339                                       *MBPI) {
340   bool taken = false;
341   MachineBasicBlock *Src = MI->getParent();
342   const BranchProbability Prediction =
343     MBPI->getEdgeProbability(Src, jmpTarget);
344 
345   if (Prediction >= BranchProbability(1,2))
346     taken = true;
347 
348   switch (MI->getOpcode()) {
349     case Hexagon::C2_cmpeq:
350       return taken ? Hexagon::J4_cmpeq_t_jumpnv_t
351                    : Hexagon::J4_cmpeq_t_jumpnv_nt;
352 
353     case Hexagon::C2_cmpeqi:
354       if (reg >= 0)
355         return taken ? Hexagon::J4_cmpeqi_t_jumpnv_t
356                      : Hexagon::J4_cmpeqi_t_jumpnv_nt;
357       return taken ? Hexagon::J4_cmpeqn1_t_jumpnv_t
358                    : Hexagon::J4_cmpeqn1_t_jumpnv_nt;
359 
360     case Hexagon::C4_cmpneqi:
361       if (reg >= 0)
362         return taken ? Hexagon::J4_cmpeqi_f_jumpnv_t
363                      : Hexagon::J4_cmpeqi_f_jumpnv_nt;
364       return taken ? Hexagon::J4_cmpeqn1_f_jumpnv_t :
365                      Hexagon::J4_cmpeqn1_f_jumpnv_nt;
366 
367     case Hexagon::C2_cmpgt:
368       if (secondRegNewified)
369         return taken ? Hexagon::J4_cmplt_t_jumpnv_t
370                      : Hexagon::J4_cmplt_t_jumpnv_nt;
371       return taken ? Hexagon::J4_cmpgt_t_jumpnv_t
372                    : Hexagon::J4_cmpgt_t_jumpnv_nt;
373 
374     case Hexagon::C2_cmpgti:
375       if (reg >= 0)
376         return taken ? Hexagon::J4_cmpgti_t_jumpnv_t
377                      : Hexagon::J4_cmpgti_t_jumpnv_nt;
378       return taken ? Hexagon::J4_cmpgtn1_t_jumpnv_t
379                    : Hexagon::J4_cmpgtn1_t_jumpnv_nt;
380 
381     case Hexagon::C2_cmpgtu:
382       if (secondRegNewified)
383         return taken ? Hexagon::J4_cmpltu_t_jumpnv_t
384                      : Hexagon::J4_cmpltu_t_jumpnv_nt;
385       return taken ? Hexagon::J4_cmpgtu_t_jumpnv_t
386                    : Hexagon::J4_cmpgtu_t_jumpnv_nt;
387 
388     case Hexagon::C2_cmpgtui:
389       return taken ? Hexagon::J4_cmpgtui_t_jumpnv_t
390                    : Hexagon::J4_cmpgtui_t_jumpnv_nt;
391 
392     case Hexagon::C4_cmpneq:
393       return taken ? Hexagon::J4_cmpeq_f_jumpnv_t
394                    : Hexagon::J4_cmpeq_f_jumpnv_nt;
395 
396     case Hexagon::C4_cmplte:
397       if (secondRegNewified)
398         return taken ? Hexagon::J4_cmplt_f_jumpnv_t
399                      : Hexagon::J4_cmplt_f_jumpnv_nt;
400       return taken ? Hexagon::J4_cmpgt_f_jumpnv_t
401                    : Hexagon::J4_cmpgt_f_jumpnv_nt;
402 
403     case Hexagon::C4_cmplteu:
404       if (secondRegNewified)
405         return taken ? Hexagon::J4_cmpltu_f_jumpnv_t
406                      : Hexagon::J4_cmpltu_f_jumpnv_nt;
407       return taken ? Hexagon::J4_cmpgtu_f_jumpnv_t
408                    : Hexagon::J4_cmpgtu_f_jumpnv_nt;
409 
410     case Hexagon::C4_cmpltei:
411       if (reg >= 0)
412         return taken ? Hexagon::J4_cmpgti_f_jumpnv_t
413                      : Hexagon::J4_cmpgti_f_jumpnv_nt;
414       return taken ? Hexagon::J4_cmpgtn1_f_jumpnv_t
415                    : Hexagon::J4_cmpgtn1_f_jumpnv_nt;
416 
417     case Hexagon::C4_cmplteui:
418       return taken ? Hexagon::J4_cmpgtui_f_jumpnv_t
419                    : Hexagon::J4_cmpgtui_f_jumpnv_nt;
420 
421     default:
422        llvm_unreachable("Could not find matching New Value Jump instruction.");
423   }
424   // return *some value* to avoid compiler warning
425   return 0;
426 }
427 
isNewValueJumpCandidate(const MachineInstr & MI) const428 bool HexagonNewValueJump::isNewValueJumpCandidate(
429     const MachineInstr &MI) const {
430   switch (MI.getOpcode()) {
431   case Hexagon::C2_cmpeq:
432   case Hexagon::C2_cmpeqi:
433   case Hexagon::C2_cmpgt:
434   case Hexagon::C2_cmpgti:
435   case Hexagon::C2_cmpgtu:
436   case Hexagon::C2_cmpgtui:
437   case Hexagon::C4_cmpneq:
438   case Hexagon::C4_cmpneqi:
439   case Hexagon::C4_cmplte:
440   case Hexagon::C4_cmplteu:
441   case Hexagon::C4_cmpltei:
442   case Hexagon::C4_cmplteui:
443     return true;
444 
445   default:
446     return false;
447   }
448 }
449 
runOnMachineFunction(MachineFunction & MF)450 bool HexagonNewValueJump::runOnMachineFunction(MachineFunction &MF) {
451   LLVM_DEBUG(dbgs() << "********** Hexagon New Value Jump **********\n"
452                     << "********** Function: " << MF.getName() << "\n");
453 
454   if (skipFunction(MF.getFunction()))
455     return false;
456 
457   // If we move NewValueJump before register allocation we'll need live variable
458   // analysis here too.
459 
460   QII = static_cast<const HexagonInstrInfo *>(MF.getSubtarget().getInstrInfo());
461   QRI = static_cast<const HexagonRegisterInfo *>(
462       MF.getSubtarget().getRegisterInfo());
463   MBPI = &getAnalysis<MachineBranchProbabilityInfo>();
464 
465   if (DisableNewValueJumps ||
466       !MF.getSubtarget<HexagonSubtarget>().useNewValueJumps())
467     return false;
468 
469   int nvjCount = DbgNVJCount;
470   int nvjGenerated = 0;
471 
472   // Loop through all the bb's of the function
473   for (MachineFunction::iterator MBBb = MF.begin(), MBBe = MF.end();
474        MBBb != MBBe; ++MBBb) {
475     MachineBasicBlock *MBB = &*MBBb;
476 
477     LLVM_DEBUG(dbgs() << "** dumping bb ** " << MBB->getNumber() << "\n");
478     LLVM_DEBUG(MBB->dump());
479     LLVM_DEBUG(dbgs() << "\n"
480                       << "********** dumping instr bottom up **********\n");
481     bool foundJump    = false;
482     bool foundCompare = false;
483     bool invertPredicate = false;
484     unsigned predReg = 0; // predicate reg of the jump.
485     unsigned cmpReg1 = 0;
486     int cmpOp2 = 0;
487     MachineBasicBlock::iterator jmpPos;
488     MachineBasicBlock::iterator cmpPos;
489     MachineInstr *cmpInstr = nullptr, *jmpInstr = nullptr;
490     MachineBasicBlock *jmpTarget = nullptr;
491     bool afterRA = false;
492     bool isSecondOpReg = false;
493     bool isSecondOpNewified = false;
494     // Traverse the basic block - bottom up
495     for (MachineBasicBlock::iterator MII = MBB->end(), E = MBB->begin();
496          MII != E;) {
497       MachineInstr &MI = *--MII;
498       if (MI.isDebugInstr()) {
499         continue;
500       }
501 
502       if ((nvjCount == 0) || (nvjCount > -1 && nvjCount <= nvjGenerated))
503         break;
504 
505       LLVM_DEBUG(dbgs() << "Instr: "; MI.dump(); dbgs() << "\n");
506 
507       if (!foundJump && (MI.getOpcode() == Hexagon::J2_jumpt ||
508                          MI.getOpcode() == Hexagon::J2_jumptpt ||
509                          MI.getOpcode() == Hexagon::J2_jumpf ||
510                          MI.getOpcode() == Hexagon::J2_jumpfpt ||
511                          MI.getOpcode() == Hexagon::J2_jumptnewpt ||
512                          MI.getOpcode() == Hexagon::J2_jumptnew ||
513                          MI.getOpcode() == Hexagon::J2_jumpfnewpt ||
514                          MI.getOpcode() == Hexagon::J2_jumpfnew)) {
515         // This is where you would insert your compare and
516         // instr that feeds compare
517         jmpPos = MII;
518         jmpInstr = &MI;
519         predReg = MI.getOperand(0).getReg();
520         afterRA = Register::isPhysicalRegister(predReg);
521 
522         // If ifconverter had not messed up with the kill flags of the
523         // operands, the following check on the kill flag would suffice.
524         // if(!jmpInstr->getOperand(0).isKill()) break;
525 
526         // This predicate register is live out of BB
527         // this would only work if we can actually use Live
528         // variable analysis on phy regs - but LLVM does not
529         // provide LV analysis on phys regs.
530         //if(LVs.isLiveOut(predReg, *MBB)) break;
531 
532         // Get all the successors of this block - which will always
533         // be 2. Check if the predicate register is live-in in those
534         // successor. If yes, we can not delete the predicate -
535         // I am doing this only because LLVM does not provide LiveOut
536         // at the BB level.
537         bool predLive = false;
538         for (MachineBasicBlock::const_succ_iterator SI = MBB->succ_begin(),
539                                                     SIE = MBB->succ_end();
540              SI != SIE; ++SI) {
541           MachineBasicBlock *succMBB = *SI;
542           if (succMBB->isLiveIn(predReg))
543             predLive = true;
544         }
545         if (predLive)
546           break;
547 
548         if (!MI.getOperand(1).isMBB())
549           continue;
550         jmpTarget = MI.getOperand(1).getMBB();
551         foundJump = true;
552         if (MI.getOpcode() == Hexagon::J2_jumpf ||
553             MI.getOpcode() == Hexagon::J2_jumpfnewpt ||
554             MI.getOpcode() == Hexagon::J2_jumpfnew) {
555           invertPredicate = true;
556         }
557         continue;
558       }
559 
560       // No new value jump if there is a barrier. A barrier has to be in its
561       // own packet. A barrier has zero operands. We conservatively bail out
562       // here if we see any instruction with zero operands.
563       if (foundJump && MI.getNumOperands() == 0)
564         break;
565 
566       if (foundJump && !foundCompare && MI.getOperand(0).isReg() &&
567           MI.getOperand(0).getReg() == predReg) {
568         // Not all compares can be new value compare. Arch Spec: 7.6.1.1
569         if (isNewValueJumpCandidate(MI)) {
570           assert(
571               (MI.getDesc().isCompare()) &&
572               "Only compare instruction can be collapsed into New Value Jump");
573           isSecondOpReg = MI.getOperand(2).isReg();
574 
575           if (!canCompareBeNewValueJump(QII, QRI, MII, predReg, isSecondOpReg,
576                                         afterRA, jmpPos, MF))
577             break;
578 
579           cmpInstr = &MI;
580           cmpPos = MII;
581           foundCompare = true;
582 
583           // We need cmpReg1 and cmpOp2(imm or reg) while building
584           // new value jump instruction.
585           cmpReg1 = MI.getOperand(1).getReg();
586 
587           if (isSecondOpReg)
588             cmpOp2 = MI.getOperand(2).getReg();
589           else
590             cmpOp2 = MI.getOperand(2).getImm();
591           continue;
592         }
593       }
594 
595       if (foundCompare && foundJump) {
596         // If "common" checks fail, bail out on this BB.
597         if (!commonChecksToProhibitNewValueJump(afterRA, MII))
598           break;
599 
600         bool foundFeeder = false;
601         MachineBasicBlock::iterator feederPos = MII;
602         if (MI.getOperand(0).isReg() && MI.getOperand(0).isDef() &&
603             (MI.getOperand(0).getReg() == cmpReg1 ||
604              (isSecondOpReg &&
605               MI.getOperand(0).getReg() == (unsigned)cmpOp2))) {
606 
607           Register feederReg = MI.getOperand(0).getReg();
608 
609           // First try to see if we can get the feeder from the first operand
610           // of the compare. If we can not, and if secondOpReg is true
611           // (second operand of the compare is also register), try that one.
612           // TODO: Try to come up with some heuristic to figure out which
613           // feeder would benefit.
614 
615           if (feederReg == cmpReg1) {
616             if (!canBeFeederToNewValueJump(QII, QRI, MII, jmpPos, cmpPos, MF)) {
617               if (!isSecondOpReg)
618                 break;
619               else
620                 continue;
621             } else
622               foundFeeder = true;
623           }
624 
625           if (!foundFeeder && isSecondOpReg && feederReg == (unsigned)cmpOp2)
626             if (!canBeFeederToNewValueJump(QII, QRI, MII, jmpPos, cmpPos, MF))
627               break;
628 
629           if (isSecondOpReg) {
630             // In case of CMPLT, or CMPLTU, or EQ with the second register
631             // to newify, swap the operands.
632             unsigned COp = cmpInstr->getOpcode();
633             if ((COp == Hexagon::C2_cmpeq || COp == Hexagon::C4_cmpneq) &&
634                 (feederReg == (unsigned)cmpOp2)) {
635               unsigned tmp = cmpReg1;
636               cmpReg1 = cmpOp2;
637               cmpOp2 = tmp;
638             }
639 
640             // Now we have swapped the operands, all we need to check is,
641             // if the second operand (after swap) is the feeder.
642             // And if it is, make a note.
643             if (feederReg == (unsigned)cmpOp2)
644               isSecondOpNewified = true;
645           }
646 
647           // Now that we are moving feeder close the jump,
648           // make sure we are respecting the kill values of
649           // the operands of the feeder.
650 
651           auto TransferKills = [jmpPos,cmpPos] (MachineInstr &MI) {
652             for (MachineOperand &MO : MI.operands()) {
653               if (!MO.isReg() || !MO.isUse())
654                 continue;
655               Register UseR = MO.getReg();
656               for (auto I = std::next(MI.getIterator()); I != jmpPos; ++I) {
657                 if (I == cmpPos)
658                   continue;
659                 for (MachineOperand &Op : I->operands()) {
660                   if (!Op.isReg() || !Op.isUse() || !Op.isKill())
661                     continue;
662                   if (Op.getReg() != UseR)
663                     continue;
664                   // We found that there is kill of a use register
665                   // Set up a kill flag on the register
666                   Op.setIsKill(false);
667                   MO.setIsKill(true);
668                   return;
669                 }
670               }
671             }
672           };
673 
674           TransferKills(*feederPos);
675           TransferKills(*cmpPos);
676           bool MO1IsKill = cmpPos->killsRegister(cmpReg1, QRI);
677           bool MO2IsKill = isSecondOpReg && cmpPos->killsRegister(cmpOp2, QRI);
678 
679           MBB->splice(jmpPos, MI.getParent(), MI);
680           MBB->splice(jmpPos, MI.getParent(), cmpInstr);
681           DebugLoc dl = MI.getDebugLoc();
682           MachineInstr *NewMI;
683 
684           assert((isNewValueJumpCandidate(*cmpInstr)) &&
685                  "This compare is not a New Value Jump candidate.");
686           unsigned opc = getNewValueJumpOpcode(cmpInstr, cmpOp2,
687                                                isSecondOpNewified,
688                                                jmpTarget, MBPI);
689           if (invertPredicate)
690             opc = QII->getInvertedPredicatedOpcode(opc);
691 
692           if (isSecondOpReg)
693             NewMI = BuildMI(*MBB, jmpPos, dl, QII->get(opc))
694                         .addReg(cmpReg1, getKillRegState(MO1IsKill))
695                         .addReg(cmpOp2, getKillRegState(MO2IsKill))
696                         .addMBB(jmpTarget);
697 
698           else
699             NewMI = BuildMI(*MBB, jmpPos, dl, QII->get(opc))
700                         .addReg(cmpReg1, getKillRegState(MO1IsKill))
701                         .addImm(cmpOp2)
702                         .addMBB(jmpTarget);
703 
704           assert(NewMI && "New Value Jump Instruction Not created!");
705           (void)NewMI;
706           if (cmpInstr->getOperand(0).isReg() &&
707               cmpInstr->getOperand(0).isKill())
708             cmpInstr->getOperand(0).setIsKill(false);
709           if (cmpInstr->getOperand(1).isReg() &&
710               cmpInstr->getOperand(1).isKill())
711             cmpInstr->getOperand(1).setIsKill(false);
712           cmpInstr->eraseFromParent();
713           jmpInstr->eraseFromParent();
714           ++nvjGenerated;
715           ++NumNVJGenerated;
716           break;
717         }
718       }
719     }
720   }
721 
722   return true;
723 }
724 
createHexagonNewValueJump()725 FunctionPass *llvm::createHexagonNewValueJump() {
726   return new HexagonNewValueJump();
727 }
728