1 //===-- ARMBaseInstrInfo.cpp - ARM Instruction Information ----------------===//
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 file contains the Base ARM implementation of the TargetInstrInfo class.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "ARMBaseInstrInfo.h"
14 #include "ARMBaseRegisterInfo.h"
15 #include "ARMConstantPoolValue.h"
16 #include "ARMFeatures.h"
17 #include "ARMHazardRecognizer.h"
18 #include "ARMMachineFunctionInfo.h"
19 #include "ARMSubtarget.h"
20 #include "MCTargetDesc/ARMAddressingModes.h"
21 #include "MCTargetDesc/ARMBaseInfo.h"
22 #include "MVETailPredUtils.h"
23 #include "llvm/ADT/DenseMap.h"
24 #include "llvm/ADT/STLExtras.h"
25 #include "llvm/ADT/SmallSet.h"
26 #include "llvm/ADT/SmallVector.h"
27 #include "llvm/ADT/Triple.h"
28 #include "llvm/CodeGen/LiveVariables.h"
29 #include "llvm/CodeGen/MachineBasicBlock.h"
30 #include "llvm/CodeGen/MachineConstantPool.h"
31 #include "llvm/CodeGen/MachineFrameInfo.h"
32 #include "llvm/CodeGen/MachineFunction.h"
33 #include "llvm/CodeGen/MachineInstr.h"
34 #include "llvm/CodeGen/MachineInstrBuilder.h"
35 #include "llvm/CodeGen/MachineMemOperand.h"
36 #include "llvm/CodeGen/MachineModuleInfo.h"
37 #include "llvm/CodeGen/MachineOperand.h"
38 #include "llvm/CodeGen/MachineRegisterInfo.h"
39 #include "llvm/CodeGen/MachineScheduler.h"
40 #include "llvm/CodeGen/MultiHazardRecognizer.h"
41 #include "llvm/CodeGen/ScoreboardHazardRecognizer.h"
42 #include "llvm/CodeGen/SelectionDAGNodes.h"
43 #include "llvm/CodeGen/TargetInstrInfo.h"
44 #include "llvm/CodeGen/TargetRegisterInfo.h"
45 #include "llvm/CodeGen/TargetSchedule.h"
46 #include "llvm/IR/Attributes.h"
47 #include "llvm/IR/Constants.h"
48 #include "llvm/IR/DebugLoc.h"
49 #include "llvm/IR/Function.h"
50 #include "llvm/IR/GlobalValue.h"
51 #include "llvm/MC/MCAsmInfo.h"
52 #include "llvm/MC/MCInstrDesc.h"
53 #include "llvm/MC/MCInstrItineraries.h"
54 #include "llvm/Support/BranchProbability.h"
55 #include "llvm/Support/Casting.h"
56 #include "llvm/Support/CommandLine.h"
57 #include "llvm/Support/Compiler.h"
58 #include "llvm/Support/Debug.h"
59 #include "llvm/Support/ErrorHandling.h"
60 #include "llvm/Support/raw_ostream.h"
61 #include "llvm/Target/TargetMachine.h"
62 #include <algorithm>
63 #include <cassert>
64 #include <cstdint>
65 #include <iterator>
66 #include <new>
67 #include <utility>
68 #include <vector>
69 
70 using namespace llvm;
71 
72 #define DEBUG_TYPE "arm-instrinfo"
73 
74 #define GET_INSTRINFO_CTOR_DTOR
75 #include "ARMGenInstrInfo.inc"
76 
77 static cl::opt<bool>
78 EnableARM3Addr("enable-arm-3-addr-conv", cl::Hidden,
79                cl::desc("Enable ARM 2-addr to 3-addr conv"));
80 
81 /// ARM_MLxEntry - Record information about MLA / MLS instructions.
82 struct ARM_MLxEntry {
83   uint16_t MLxOpc;     // MLA / MLS opcode
84   uint16_t MulOpc;     // Expanded multiplication opcode
85   uint16_t AddSubOpc;  // Expanded add / sub opcode
86   bool NegAcc;         // True if the acc is negated before the add / sub.
87   bool HasLane;        // True if instruction has an extra "lane" operand.
88 };
89 
90 static const ARM_MLxEntry ARM_MLxTable[] = {
91   // MLxOpc,          MulOpc,           AddSubOpc,       NegAcc, HasLane
92   // fp scalar ops
93   { ARM::VMLAS,       ARM::VMULS,       ARM::VADDS,      false,  false },
94   { ARM::VMLSS,       ARM::VMULS,       ARM::VSUBS,      false,  false },
95   { ARM::VMLAD,       ARM::VMULD,       ARM::VADDD,      false,  false },
96   { ARM::VMLSD,       ARM::VMULD,       ARM::VSUBD,      false,  false },
97   { ARM::VNMLAS,      ARM::VNMULS,      ARM::VSUBS,      true,   false },
98   { ARM::VNMLSS,      ARM::VMULS,       ARM::VSUBS,      true,   false },
99   { ARM::VNMLAD,      ARM::VNMULD,      ARM::VSUBD,      true,   false },
100   { ARM::VNMLSD,      ARM::VMULD,       ARM::VSUBD,      true,   false },
101 
102   // fp SIMD ops
103   { ARM::VMLAfd,      ARM::VMULfd,      ARM::VADDfd,     false,  false },
104   { ARM::VMLSfd,      ARM::VMULfd,      ARM::VSUBfd,     false,  false },
105   { ARM::VMLAfq,      ARM::VMULfq,      ARM::VADDfq,     false,  false },
106   { ARM::VMLSfq,      ARM::VMULfq,      ARM::VSUBfq,     false,  false },
107   { ARM::VMLAslfd,    ARM::VMULslfd,    ARM::VADDfd,     false,  true  },
108   { ARM::VMLSslfd,    ARM::VMULslfd,    ARM::VSUBfd,     false,  true  },
109   { ARM::VMLAslfq,    ARM::VMULslfq,    ARM::VADDfq,     false,  true  },
110   { ARM::VMLSslfq,    ARM::VMULslfq,    ARM::VSUBfq,     false,  true  },
111 };
112 
113 ARMBaseInstrInfo::ARMBaseInstrInfo(const ARMSubtarget& STI)
114   : ARMGenInstrInfo(ARM::ADJCALLSTACKDOWN, ARM::ADJCALLSTACKUP),
115     Subtarget(STI) {
116   for (unsigned i = 0, e = array_lengthof(ARM_MLxTable); i != e; ++i) {
117     if (!MLxEntryMap.insert(std::make_pair(ARM_MLxTable[i].MLxOpc, i)).second)
118       llvm_unreachable("Duplicated entries?");
119     MLxHazardOpcodes.insert(ARM_MLxTable[i].AddSubOpc);
120     MLxHazardOpcodes.insert(ARM_MLxTable[i].MulOpc);
121   }
122 }
123 
124 // Use a ScoreboardHazardRecognizer for prepass ARM scheduling. TargetInstrImpl
125 // currently defaults to no prepass hazard recognizer.
126 ScheduleHazardRecognizer *
127 ARMBaseInstrInfo::CreateTargetHazardRecognizer(const TargetSubtargetInfo *STI,
128                                                const ScheduleDAG *DAG) const {
129   if (usePreRAHazardRecognizer()) {
130     const InstrItineraryData *II =
131         static_cast<const ARMSubtarget *>(STI)->getInstrItineraryData();
132     return new ScoreboardHazardRecognizer(II, DAG, "pre-RA-sched");
133   }
134   return TargetInstrInfo::CreateTargetHazardRecognizer(STI, DAG);
135 }
136 
137 // Called during:
138 // - pre-RA scheduling
139 // - post-RA scheduling when FeatureUseMISched is set
140 ScheduleHazardRecognizer *ARMBaseInstrInfo::CreateTargetMIHazardRecognizer(
141     const InstrItineraryData *II, const ScheduleDAGMI *DAG) const {
142   MultiHazardRecognizer *MHR = new MultiHazardRecognizer();
143 
144   // We would like to restrict this hazard recognizer to only
145   // post-RA scheduling; we can tell that we're post-RA because we don't
146   // track VRegLiveness.
147   // Cortex-M7: TRM indicates that there is a single ITCM bank and two DTCM
148   //            banks banked on bit 2.  Assume that TCMs are in use.
149   if (Subtarget.isCortexM7() && !DAG->hasVRegLiveness())
150     MHR->AddHazardRecognizer(
151         std::make_unique<ARMBankConflictHazardRecognizer>(DAG, 0x4, true));
152 
153   // Not inserting ARMHazardRecognizerFPMLx because that would change
154   // legacy behavior
155 
156   auto BHR = TargetInstrInfo::CreateTargetMIHazardRecognizer(II, DAG);
157   MHR->AddHazardRecognizer(std::unique_ptr<ScheduleHazardRecognizer>(BHR));
158   return MHR;
159 }
160 
161 // Called during post-RA scheduling when FeatureUseMISched is not set
162 ScheduleHazardRecognizer *ARMBaseInstrInfo::
163 CreateTargetPostRAHazardRecognizer(const InstrItineraryData *II,
164                                    const ScheduleDAG *DAG) const {
165   MultiHazardRecognizer *MHR = new MultiHazardRecognizer();
166 
167   if (Subtarget.isThumb2() || Subtarget.hasVFP2Base())
168     MHR->AddHazardRecognizer(std::make_unique<ARMHazardRecognizerFPMLx>());
169 
170   auto BHR = TargetInstrInfo::CreateTargetPostRAHazardRecognizer(II, DAG);
171   if (BHR)
172     MHR->AddHazardRecognizer(std::unique_ptr<ScheduleHazardRecognizer>(BHR));
173   return MHR;
174 }
175 
176 MachineInstr *ARMBaseInstrInfo::convertToThreeAddress(
177     MachineFunction::iterator &MFI, MachineInstr &MI, LiveVariables *LV) const {
178   // FIXME: Thumb2 support.
179 
180   if (!EnableARM3Addr)
181     return nullptr;
182 
183   MachineFunction &MF = *MI.getParent()->getParent();
184   uint64_t TSFlags = MI.getDesc().TSFlags;
185   bool isPre = false;
186   switch ((TSFlags & ARMII::IndexModeMask) >> ARMII::IndexModeShift) {
187   default: return nullptr;
188   case ARMII::IndexModePre:
189     isPre = true;
190     break;
191   case ARMII::IndexModePost:
192     break;
193   }
194 
195   // Try splitting an indexed load/store to an un-indexed one plus an add/sub
196   // operation.
197   unsigned MemOpc = getUnindexedOpcode(MI.getOpcode());
198   if (MemOpc == 0)
199     return nullptr;
200 
201   MachineInstr *UpdateMI = nullptr;
202   MachineInstr *MemMI = nullptr;
203   unsigned AddrMode = (TSFlags & ARMII::AddrModeMask);
204   const MCInstrDesc &MCID = MI.getDesc();
205   unsigned NumOps = MCID.getNumOperands();
206   bool isLoad = !MI.mayStore();
207   const MachineOperand &WB = isLoad ? MI.getOperand(1) : MI.getOperand(0);
208   const MachineOperand &Base = MI.getOperand(2);
209   const MachineOperand &Offset = MI.getOperand(NumOps - 3);
210   Register WBReg = WB.getReg();
211   Register BaseReg = Base.getReg();
212   Register OffReg = Offset.getReg();
213   unsigned OffImm = MI.getOperand(NumOps - 2).getImm();
214   ARMCC::CondCodes Pred = (ARMCC::CondCodes)MI.getOperand(NumOps - 1).getImm();
215   switch (AddrMode) {
216   default: llvm_unreachable("Unknown indexed op!");
217   case ARMII::AddrMode2: {
218     bool isSub = ARM_AM::getAM2Op(OffImm) == ARM_AM::sub;
219     unsigned Amt = ARM_AM::getAM2Offset(OffImm);
220     if (OffReg == 0) {
221       if (ARM_AM::getSOImmVal(Amt) == -1)
222         // Can't encode it in a so_imm operand. This transformation will
223         // add more than 1 instruction. Abandon!
224         return nullptr;
225       UpdateMI = BuildMI(MF, MI.getDebugLoc(),
226                          get(isSub ? ARM::SUBri : ARM::ADDri), WBReg)
227                      .addReg(BaseReg)
228                      .addImm(Amt)
229                      .add(predOps(Pred))
230                      .add(condCodeOp());
231     } else if (Amt != 0) {
232       ARM_AM::ShiftOpc ShOpc = ARM_AM::getAM2ShiftOpc(OffImm);
233       unsigned SOOpc = ARM_AM::getSORegOpc(ShOpc, Amt);
234       UpdateMI = BuildMI(MF, MI.getDebugLoc(),
235                          get(isSub ? ARM::SUBrsi : ARM::ADDrsi), WBReg)
236                      .addReg(BaseReg)
237                      .addReg(OffReg)
238                      .addReg(0)
239                      .addImm(SOOpc)
240                      .add(predOps(Pred))
241                      .add(condCodeOp());
242     } else
243       UpdateMI = BuildMI(MF, MI.getDebugLoc(),
244                          get(isSub ? ARM::SUBrr : ARM::ADDrr), WBReg)
245                      .addReg(BaseReg)
246                      .addReg(OffReg)
247                      .add(predOps(Pred))
248                      .add(condCodeOp());
249     break;
250   }
251   case ARMII::AddrMode3 : {
252     bool isSub = ARM_AM::getAM3Op(OffImm) == ARM_AM::sub;
253     unsigned Amt = ARM_AM::getAM3Offset(OffImm);
254     if (OffReg == 0)
255       // Immediate is 8-bits. It's guaranteed to fit in a so_imm operand.
256       UpdateMI = BuildMI(MF, MI.getDebugLoc(),
257                          get(isSub ? ARM::SUBri : ARM::ADDri), WBReg)
258                      .addReg(BaseReg)
259                      .addImm(Amt)
260                      .add(predOps(Pred))
261                      .add(condCodeOp());
262     else
263       UpdateMI = BuildMI(MF, MI.getDebugLoc(),
264                          get(isSub ? ARM::SUBrr : ARM::ADDrr), WBReg)
265                      .addReg(BaseReg)
266                      .addReg(OffReg)
267                      .add(predOps(Pred))
268                      .add(condCodeOp());
269     break;
270   }
271   }
272 
273   std::vector<MachineInstr*> NewMIs;
274   if (isPre) {
275     if (isLoad)
276       MemMI =
277           BuildMI(MF, MI.getDebugLoc(), get(MemOpc), MI.getOperand(0).getReg())
278               .addReg(WBReg)
279               .addImm(0)
280               .addImm(Pred);
281     else
282       MemMI = BuildMI(MF, MI.getDebugLoc(), get(MemOpc))
283                   .addReg(MI.getOperand(1).getReg())
284                   .addReg(WBReg)
285                   .addReg(0)
286                   .addImm(0)
287                   .addImm(Pred);
288     NewMIs.push_back(MemMI);
289     NewMIs.push_back(UpdateMI);
290   } else {
291     if (isLoad)
292       MemMI =
293           BuildMI(MF, MI.getDebugLoc(), get(MemOpc), MI.getOperand(0).getReg())
294               .addReg(BaseReg)
295               .addImm(0)
296               .addImm(Pred);
297     else
298       MemMI = BuildMI(MF, MI.getDebugLoc(), get(MemOpc))
299                   .addReg(MI.getOperand(1).getReg())
300                   .addReg(BaseReg)
301                   .addReg(0)
302                   .addImm(0)
303                   .addImm(Pred);
304     if (WB.isDead())
305       UpdateMI->getOperand(0).setIsDead();
306     NewMIs.push_back(UpdateMI);
307     NewMIs.push_back(MemMI);
308   }
309 
310   // Transfer LiveVariables states, kill / dead info.
311   if (LV) {
312     for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
313       MachineOperand &MO = MI.getOperand(i);
314       if (MO.isReg() && Register::isVirtualRegister(MO.getReg())) {
315         Register Reg = MO.getReg();
316 
317         LiveVariables::VarInfo &VI = LV->getVarInfo(Reg);
318         if (MO.isDef()) {
319           MachineInstr *NewMI = (Reg == WBReg) ? UpdateMI : MemMI;
320           if (MO.isDead())
321             LV->addVirtualRegisterDead(Reg, *NewMI);
322         }
323         if (MO.isUse() && MO.isKill()) {
324           for (unsigned j = 0; j < 2; ++j) {
325             // Look at the two new MI's in reverse order.
326             MachineInstr *NewMI = NewMIs[j];
327             if (!NewMI->readsRegister(Reg))
328               continue;
329             LV->addVirtualRegisterKilled(Reg, *NewMI);
330             if (VI.removeKill(MI))
331               VI.Kills.push_back(NewMI);
332             break;
333           }
334         }
335       }
336     }
337   }
338 
339   MachineBasicBlock::iterator MBBI = MI.getIterator();
340   MFI->insert(MBBI, NewMIs[1]);
341   MFI->insert(MBBI, NewMIs[0]);
342   return NewMIs[0];
343 }
344 
345 // Branch analysis.
346 bool ARMBaseInstrInfo::analyzeBranch(MachineBasicBlock &MBB,
347                                      MachineBasicBlock *&TBB,
348                                      MachineBasicBlock *&FBB,
349                                      SmallVectorImpl<MachineOperand> &Cond,
350                                      bool AllowModify) const {
351   TBB = nullptr;
352   FBB = nullptr;
353 
354   MachineBasicBlock::instr_iterator I = MBB.instr_end();
355   if (I == MBB.instr_begin())
356     return false; // Empty blocks are easy.
357   --I;
358 
359   // Walk backwards from the end of the basic block until the branch is
360   // analyzed or we give up.
361   while (isPredicated(*I) || I->isTerminator() || I->isDebugValue()) {
362     // Flag to be raised on unanalyzeable instructions. This is useful in cases
363     // where we want to clean up on the end of the basic block before we bail
364     // out.
365     bool CantAnalyze = false;
366 
367     // Skip over DEBUG values, predicated nonterminators and speculation
368     // barrier terminators.
369     while (I->isDebugInstr() || !I->isTerminator() ||
370            isSpeculationBarrierEndBBOpcode(I->getOpcode()) ||
371            I->getOpcode() == ARM::t2DoLoopStartTP){
372       if (I == MBB.instr_begin())
373         return false;
374       --I;
375     }
376 
377     if (isIndirectBranchOpcode(I->getOpcode()) ||
378         isJumpTableBranchOpcode(I->getOpcode())) {
379       // Indirect branches and jump tables can't be analyzed, but we still want
380       // to clean up any instructions at the tail of the basic block.
381       CantAnalyze = true;
382     } else if (isUncondBranchOpcode(I->getOpcode())) {
383       TBB = I->getOperand(0).getMBB();
384     } else if (isCondBranchOpcode(I->getOpcode())) {
385       // Bail out if we encounter multiple conditional branches.
386       if (!Cond.empty())
387         return true;
388 
389       assert(!FBB && "FBB should have been null.");
390       FBB = TBB;
391       TBB = I->getOperand(0).getMBB();
392       Cond.push_back(I->getOperand(1));
393       Cond.push_back(I->getOperand(2));
394     } else if (I->isReturn()) {
395       // Returns can't be analyzed, but we should run cleanup.
396       CantAnalyze = true;
397     } else {
398       // We encountered other unrecognized terminator. Bail out immediately.
399       return true;
400     }
401 
402     // Cleanup code - to be run for unpredicated unconditional branches and
403     //                returns.
404     if (!isPredicated(*I) &&
405           (isUncondBranchOpcode(I->getOpcode()) ||
406            isIndirectBranchOpcode(I->getOpcode()) ||
407            isJumpTableBranchOpcode(I->getOpcode()) ||
408            I->isReturn())) {
409       // Forget any previous condition branch information - it no longer applies.
410       Cond.clear();
411       FBB = nullptr;
412 
413       // If we can modify the function, delete everything below this
414       // unconditional branch.
415       if (AllowModify) {
416         MachineBasicBlock::iterator DI = std::next(I);
417         while (DI != MBB.instr_end()) {
418           MachineInstr &InstToDelete = *DI;
419           ++DI;
420           // Speculation barriers must not be deleted.
421           if (isSpeculationBarrierEndBBOpcode(InstToDelete.getOpcode()))
422             continue;
423           InstToDelete.eraseFromParent();
424         }
425       }
426     }
427 
428     if (CantAnalyze) {
429       // We may not be able to analyze the block, but we could still have
430       // an unconditional branch as the last instruction in the block, which
431       // just branches to layout successor. If this is the case, then just
432       // remove it if we're allowed to make modifications.
433       if (AllowModify && !isPredicated(MBB.back()) &&
434           isUncondBranchOpcode(MBB.back().getOpcode()) &&
435           TBB && MBB.isLayoutSuccessor(TBB))
436         removeBranch(MBB);
437       return true;
438     }
439 
440     if (I == MBB.instr_begin())
441       return false;
442 
443     --I;
444   }
445 
446   // We made it past the terminators without bailing out - we must have
447   // analyzed this branch successfully.
448   return false;
449 }
450 
451 unsigned ARMBaseInstrInfo::removeBranch(MachineBasicBlock &MBB,
452                                         int *BytesRemoved) const {
453   assert(!BytesRemoved && "code size not handled");
454 
455   MachineBasicBlock::iterator I = MBB.getLastNonDebugInstr();
456   if (I == MBB.end())
457     return 0;
458 
459   if (!isUncondBranchOpcode(I->getOpcode()) &&
460       !isCondBranchOpcode(I->getOpcode()))
461     return 0;
462 
463   // Remove the branch.
464   I->eraseFromParent();
465 
466   I = MBB.end();
467 
468   if (I == MBB.begin()) return 1;
469   --I;
470   if (!isCondBranchOpcode(I->getOpcode()))
471     return 1;
472 
473   // Remove the branch.
474   I->eraseFromParent();
475   return 2;
476 }
477 
478 unsigned ARMBaseInstrInfo::insertBranch(MachineBasicBlock &MBB,
479                                         MachineBasicBlock *TBB,
480                                         MachineBasicBlock *FBB,
481                                         ArrayRef<MachineOperand> Cond,
482                                         const DebugLoc &DL,
483                                         int *BytesAdded) const {
484   assert(!BytesAdded && "code size not handled");
485   ARMFunctionInfo *AFI = MBB.getParent()->getInfo<ARMFunctionInfo>();
486   int BOpc   = !AFI->isThumbFunction()
487     ? ARM::B : (AFI->isThumb2Function() ? ARM::t2B : ARM::tB);
488   int BccOpc = !AFI->isThumbFunction()
489     ? ARM::Bcc : (AFI->isThumb2Function() ? ARM::t2Bcc : ARM::tBcc);
490   bool isThumb = AFI->isThumbFunction() || AFI->isThumb2Function();
491 
492   // Shouldn't be a fall through.
493   assert(TBB && "insertBranch must not be told to insert a fallthrough");
494   assert((Cond.size() == 2 || Cond.size() == 0) &&
495          "ARM branch conditions have two components!");
496 
497   // For conditional branches, we use addOperand to preserve CPSR flags.
498 
499   if (!FBB) {
500     if (Cond.empty()) { // Unconditional branch?
501       if (isThumb)
502         BuildMI(&MBB, DL, get(BOpc)).addMBB(TBB).add(predOps(ARMCC::AL));
503       else
504         BuildMI(&MBB, DL, get(BOpc)).addMBB(TBB);
505     } else
506       BuildMI(&MBB, DL, get(BccOpc))
507           .addMBB(TBB)
508           .addImm(Cond[0].getImm())
509           .add(Cond[1]);
510     return 1;
511   }
512 
513   // Two-way conditional branch.
514   BuildMI(&MBB, DL, get(BccOpc))
515       .addMBB(TBB)
516       .addImm(Cond[0].getImm())
517       .add(Cond[1]);
518   if (isThumb)
519     BuildMI(&MBB, DL, get(BOpc)).addMBB(FBB).add(predOps(ARMCC::AL));
520   else
521     BuildMI(&MBB, DL, get(BOpc)).addMBB(FBB);
522   return 2;
523 }
524 
525 bool ARMBaseInstrInfo::
526 reverseBranchCondition(SmallVectorImpl<MachineOperand> &Cond) const {
527   ARMCC::CondCodes CC = (ARMCC::CondCodes)(int)Cond[0].getImm();
528   Cond[0].setImm(ARMCC::getOppositeCondition(CC));
529   return false;
530 }
531 
532 bool ARMBaseInstrInfo::isPredicated(const MachineInstr &MI) const {
533   if (MI.isBundle()) {
534     MachineBasicBlock::const_instr_iterator I = MI.getIterator();
535     MachineBasicBlock::const_instr_iterator E = MI.getParent()->instr_end();
536     while (++I != E && I->isInsideBundle()) {
537       int PIdx = I->findFirstPredOperandIdx();
538       if (PIdx != -1 && I->getOperand(PIdx).getImm() != ARMCC::AL)
539         return true;
540     }
541     return false;
542   }
543 
544   int PIdx = MI.findFirstPredOperandIdx();
545   return PIdx != -1 && MI.getOperand(PIdx).getImm() != ARMCC::AL;
546 }
547 
548 std::string ARMBaseInstrInfo::createMIROperandComment(
549     const MachineInstr &MI, const MachineOperand &Op, unsigned OpIdx,
550     const TargetRegisterInfo *TRI) const {
551 
552   // First, let's see if there is a generic comment for this operand
553   std::string GenericComment =
554       TargetInstrInfo::createMIROperandComment(MI, Op, OpIdx, TRI);
555   if (!GenericComment.empty())
556     return GenericComment;
557 
558   // If not, check if we have an immediate operand.
559   if (Op.getType() != MachineOperand::MO_Immediate)
560     return std::string();
561 
562   // And print its corresponding condition code if the immediate is a
563   // predicate.
564   int FirstPredOp = MI.findFirstPredOperandIdx();
565   if (FirstPredOp != (int) OpIdx)
566     return std::string();
567 
568   std::string CC = "CC::";
569   CC += ARMCondCodeToString((ARMCC::CondCodes)Op.getImm());
570   return CC;
571 }
572 
573 bool ARMBaseInstrInfo::PredicateInstruction(
574     MachineInstr &MI, ArrayRef<MachineOperand> Pred) const {
575   unsigned Opc = MI.getOpcode();
576   if (isUncondBranchOpcode(Opc)) {
577     MI.setDesc(get(getMatchingCondBranchOpcode(Opc)));
578     MachineInstrBuilder(*MI.getParent()->getParent(), MI)
579       .addImm(Pred[0].getImm())
580       .addReg(Pred[1].getReg());
581     return true;
582   }
583 
584   int PIdx = MI.findFirstPredOperandIdx();
585   if (PIdx != -1) {
586     MachineOperand &PMO = MI.getOperand(PIdx);
587     PMO.setImm(Pred[0].getImm());
588     MI.getOperand(PIdx+1).setReg(Pred[1].getReg());
589 
590     // Thumb 1 arithmetic instructions do not set CPSR when executed inside an
591     // IT block. This affects how they are printed.
592     const MCInstrDesc &MCID = MI.getDesc();
593     if (MCID.TSFlags & ARMII::ThumbArithFlagSetting) {
594       assert(MCID.OpInfo[1].isOptionalDef() && "CPSR def isn't expected operand");
595       assert((MI.getOperand(1).isDead() ||
596               MI.getOperand(1).getReg() != ARM::CPSR) &&
597              "if conversion tried to stop defining used CPSR");
598       MI.getOperand(1).setReg(ARM::NoRegister);
599     }
600 
601     return true;
602   }
603   return false;
604 }
605 
606 bool ARMBaseInstrInfo::SubsumesPredicate(ArrayRef<MachineOperand> Pred1,
607                                          ArrayRef<MachineOperand> Pred2) const {
608   if (Pred1.size() > 2 || Pred2.size() > 2)
609     return false;
610 
611   ARMCC::CondCodes CC1 = (ARMCC::CondCodes)Pred1[0].getImm();
612   ARMCC::CondCodes CC2 = (ARMCC::CondCodes)Pred2[0].getImm();
613   if (CC1 == CC2)
614     return true;
615 
616   switch (CC1) {
617   default:
618     return false;
619   case ARMCC::AL:
620     return true;
621   case ARMCC::HS:
622     return CC2 == ARMCC::HI;
623   case ARMCC::LS:
624     return CC2 == ARMCC::LO || CC2 == ARMCC::EQ;
625   case ARMCC::GE:
626     return CC2 == ARMCC::GT;
627   case ARMCC::LE:
628     return CC2 == ARMCC::LT;
629   }
630 }
631 
632 bool ARMBaseInstrInfo::ClobbersPredicate(MachineInstr &MI,
633                                          std::vector<MachineOperand> &Pred,
634                                          bool SkipDead) const {
635   bool Found = false;
636   for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
637     const MachineOperand &MO = MI.getOperand(i);
638     bool ClobbersCPSR = MO.isRegMask() && MO.clobbersPhysReg(ARM::CPSR);
639     bool IsCPSR = MO.isReg() && MO.isDef() && MO.getReg() == ARM::CPSR;
640     if (ClobbersCPSR || IsCPSR) {
641 
642       // Filter out T1 instructions that have a dead CPSR,
643       // allowing IT blocks to be generated containing T1 instructions
644       const MCInstrDesc &MCID = MI.getDesc();
645       if (MCID.TSFlags & ARMII::ThumbArithFlagSetting && MO.isDead() &&
646           SkipDead)
647         continue;
648 
649       Pred.push_back(MO);
650       Found = true;
651     }
652   }
653 
654   return Found;
655 }
656 
657 bool ARMBaseInstrInfo::isCPSRDefined(const MachineInstr &MI) {
658   for (const auto &MO : MI.operands())
659     if (MO.isReg() && MO.getReg() == ARM::CPSR && MO.isDef() && !MO.isDead())
660       return true;
661   return false;
662 }
663 
664 static bool isEligibleForITBlock(const MachineInstr *MI) {
665   switch (MI->getOpcode()) {
666   default: return true;
667   case ARM::tADC:   // ADC (register) T1
668   case ARM::tADDi3: // ADD (immediate) T1
669   case ARM::tADDi8: // ADD (immediate) T2
670   case ARM::tADDrr: // ADD (register) T1
671   case ARM::tAND:   // AND (register) T1
672   case ARM::tASRri: // ASR (immediate) T1
673   case ARM::tASRrr: // ASR (register) T1
674   case ARM::tBIC:   // BIC (register) T1
675   case ARM::tEOR:   // EOR (register) T1
676   case ARM::tLSLri: // LSL (immediate) T1
677   case ARM::tLSLrr: // LSL (register) T1
678   case ARM::tLSRri: // LSR (immediate) T1
679   case ARM::tLSRrr: // LSR (register) T1
680   case ARM::tMUL:   // MUL T1
681   case ARM::tMVN:   // MVN (register) T1
682   case ARM::tORR:   // ORR (register) T1
683   case ARM::tROR:   // ROR (register) T1
684   case ARM::tRSB:   // RSB (immediate) T1
685   case ARM::tSBC:   // SBC (register) T1
686   case ARM::tSUBi3: // SUB (immediate) T1
687   case ARM::tSUBi8: // SUB (immediate) T2
688   case ARM::tSUBrr: // SUB (register) T1
689     return !ARMBaseInstrInfo::isCPSRDefined(*MI);
690   }
691 }
692 
693 /// isPredicable - Return true if the specified instruction can be predicated.
694 /// By default, this returns true for every instruction with a
695 /// PredicateOperand.
696 bool ARMBaseInstrInfo::isPredicable(const MachineInstr &MI) const {
697   if (!MI.isPredicable())
698     return false;
699 
700   if (MI.isBundle())
701     return false;
702 
703   if (!isEligibleForITBlock(&MI))
704     return false;
705 
706   const MachineFunction *MF = MI.getParent()->getParent();
707   const ARMFunctionInfo *AFI =
708       MF->getInfo<ARMFunctionInfo>();
709 
710   // Neon instructions in Thumb2 IT blocks are deprecated, see ARMARM.
711   // In their ARM encoding, they can't be encoded in a conditional form.
712   if ((MI.getDesc().TSFlags & ARMII::DomainMask) == ARMII::DomainNEON)
713     return false;
714 
715   // Make indirect control flow changes unpredicable when SLS mitigation is
716   // enabled.
717   const ARMSubtarget &ST = MF->getSubtarget<ARMSubtarget>();
718   if (ST.hardenSlsRetBr() && isIndirectControlFlowNotComingBack(MI))
719     return false;
720   if (ST.hardenSlsBlr() && isIndirectCall(MI))
721     return false;
722 
723   if (AFI->isThumb2Function()) {
724     if (getSubtarget().restrictIT())
725       return isV8EligibleForIT(&MI);
726   }
727 
728   return true;
729 }
730 
731 namespace llvm {
732 
733 template <> bool IsCPSRDead<MachineInstr>(const MachineInstr *MI) {
734   for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
735     const MachineOperand &MO = MI->getOperand(i);
736     if (!MO.isReg() || MO.isUndef() || MO.isUse())
737       continue;
738     if (MO.getReg() != ARM::CPSR)
739       continue;
740     if (!MO.isDead())
741       return false;
742   }
743   // all definitions of CPSR are dead
744   return true;
745 }
746 
747 } // end namespace llvm
748 
749 /// GetInstSize - Return the size of the specified MachineInstr.
750 ///
751 unsigned ARMBaseInstrInfo::getInstSizeInBytes(const MachineInstr &MI) const {
752   const MachineBasicBlock &MBB = *MI.getParent();
753   const MachineFunction *MF = MBB.getParent();
754   const MCAsmInfo *MAI = MF->getTarget().getMCAsmInfo();
755 
756   const MCInstrDesc &MCID = MI.getDesc();
757   if (MCID.getSize())
758     return MCID.getSize();
759 
760   switch (MI.getOpcode()) {
761   default:
762     // pseudo-instruction sizes are zero.
763     return 0;
764   case TargetOpcode::BUNDLE:
765     return getInstBundleLength(MI);
766   case ARM::MOVi16_ga_pcrel:
767   case ARM::MOVTi16_ga_pcrel:
768   case ARM::t2MOVi16_ga_pcrel:
769   case ARM::t2MOVTi16_ga_pcrel:
770     return 4;
771   case ARM::MOVi32imm:
772   case ARM::t2MOVi32imm:
773     return 8;
774   case ARM::CONSTPOOL_ENTRY:
775   case ARM::JUMPTABLE_INSTS:
776   case ARM::JUMPTABLE_ADDRS:
777   case ARM::JUMPTABLE_TBB:
778   case ARM::JUMPTABLE_TBH:
779     // If this machine instr is a constant pool entry, its size is recorded as
780     // operand #2.
781     return MI.getOperand(2).getImm();
782   case ARM::Int_eh_sjlj_longjmp:
783     return 16;
784   case ARM::tInt_eh_sjlj_longjmp:
785     return 10;
786   case ARM::tInt_WIN_eh_sjlj_longjmp:
787     return 12;
788   case ARM::Int_eh_sjlj_setjmp:
789   case ARM::Int_eh_sjlj_setjmp_nofp:
790     return 20;
791   case ARM::tInt_eh_sjlj_setjmp:
792   case ARM::t2Int_eh_sjlj_setjmp:
793   case ARM::t2Int_eh_sjlj_setjmp_nofp:
794     return 12;
795   case ARM::SPACE:
796     return MI.getOperand(1).getImm();
797   case ARM::INLINEASM:
798   case ARM::INLINEASM_BR: {
799     // If this machine instr is an inline asm, measure it.
800     unsigned Size = getInlineAsmLength(MI.getOperand(0).getSymbolName(), *MAI);
801     if (!MF->getInfo<ARMFunctionInfo>()->isThumbFunction())
802       Size = alignTo(Size, 4);
803     return Size;
804   }
805   case ARM::SpeculationBarrierISBDSBEndBB:
806   case ARM::t2SpeculationBarrierISBDSBEndBB:
807     // This gets lowered to 2 4-byte instructions.
808     return 8;
809   case ARM::SpeculationBarrierSBEndBB:
810   case ARM::t2SpeculationBarrierSBEndBB:
811     // This gets lowered to 1 4-byte instructions.
812     return 4;
813   }
814 }
815 
816 unsigned ARMBaseInstrInfo::getInstBundleLength(const MachineInstr &MI) const {
817   unsigned Size = 0;
818   MachineBasicBlock::const_instr_iterator I = MI.getIterator();
819   MachineBasicBlock::const_instr_iterator E = MI.getParent()->instr_end();
820   while (++I != E && I->isInsideBundle()) {
821     assert(!I->isBundle() && "No nested bundle!");
822     Size += getInstSizeInBytes(*I);
823   }
824   return Size;
825 }
826 
827 void ARMBaseInstrInfo::copyFromCPSR(MachineBasicBlock &MBB,
828                                     MachineBasicBlock::iterator I,
829                                     unsigned DestReg, bool KillSrc,
830                                     const ARMSubtarget &Subtarget) const {
831   unsigned Opc = Subtarget.isThumb()
832                      ? (Subtarget.isMClass() ? ARM::t2MRS_M : ARM::t2MRS_AR)
833                      : ARM::MRS;
834 
835   MachineInstrBuilder MIB =
836       BuildMI(MBB, I, I->getDebugLoc(), get(Opc), DestReg);
837 
838   // There is only 1 A/R class MRS instruction, and it always refers to
839   // APSR. However, there are lots of other possibilities on M-class cores.
840   if (Subtarget.isMClass())
841     MIB.addImm(0x800);
842 
843   MIB.add(predOps(ARMCC::AL))
844      .addReg(ARM::CPSR, RegState::Implicit | getKillRegState(KillSrc));
845 }
846 
847 void ARMBaseInstrInfo::copyToCPSR(MachineBasicBlock &MBB,
848                                   MachineBasicBlock::iterator I,
849                                   unsigned SrcReg, bool KillSrc,
850                                   const ARMSubtarget &Subtarget) const {
851   unsigned Opc = Subtarget.isThumb()
852                      ? (Subtarget.isMClass() ? ARM::t2MSR_M : ARM::t2MSR_AR)
853                      : ARM::MSR;
854 
855   MachineInstrBuilder MIB = BuildMI(MBB, I, I->getDebugLoc(), get(Opc));
856 
857   if (Subtarget.isMClass())
858     MIB.addImm(0x800);
859   else
860     MIB.addImm(8);
861 
862   MIB.addReg(SrcReg, getKillRegState(KillSrc))
863      .add(predOps(ARMCC::AL))
864      .addReg(ARM::CPSR, RegState::Implicit | RegState::Define);
865 }
866 
867 void llvm::addUnpredicatedMveVpredNOp(MachineInstrBuilder &MIB) {
868   MIB.addImm(ARMVCC::None);
869   MIB.addReg(0);
870 }
871 
872 void llvm::addUnpredicatedMveVpredROp(MachineInstrBuilder &MIB,
873                                       Register DestReg) {
874   addUnpredicatedMveVpredNOp(MIB);
875   MIB.addReg(DestReg, RegState::Undef);
876 }
877 
878 void llvm::addPredicatedMveVpredNOp(MachineInstrBuilder &MIB, unsigned Cond) {
879   MIB.addImm(Cond);
880   MIB.addReg(ARM::VPR, RegState::Implicit);
881 }
882 
883 void llvm::addPredicatedMveVpredROp(MachineInstrBuilder &MIB,
884                                     unsigned Cond, unsigned Inactive) {
885   addPredicatedMveVpredNOp(MIB, Cond);
886   MIB.addReg(Inactive);
887 }
888 
889 void ARMBaseInstrInfo::copyPhysReg(MachineBasicBlock &MBB,
890                                    MachineBasicBlock::iterator I,
891                                    const DebugLoc &DL, MCRegister DestReg,
892                                    MCRegister SrcReg, bool KillSrc) const {
893   bool GPRDest = ARM::GPRRegClass.contains(DestReg);
894   bool GPRSrc = ARM::GPRRegClass.contains(SrcReg);
895 
896   if (GPRDest && GPRSrc) {
897     BuildMI(MBB, I, DL, get(ARM::MOVr), DestReg)
898         .addReg(SrcReg, getKillRegState(KillSrc))
899         .add(predOps(ARMCC::AL))
900         .add(condCodeOp());
901     return;
902   }
903 
904   bool SPRDest = ARM::SPRRegClass.contains(DestReg);
905   bool SPRSrc = ARM::SPRRegClass.contains(SrcReg);
906 
907   unsigned Opc = 0;
908   if (SPRDest && SPRSrc)
909     Opc = ARM::VMOVS;
910   else if (GPRDest && SPRSrc)
911     Opc = ARM::VMOVRS;
912   else if (SPRDest && GPRSrc)
913     Opc = ARM::VMOVSR;
914   else if (ARM::DPRRegClass.contains(DestReg, SrcReg) && Subtarget.hasFP64())
915     Opc = ARM::VMOVD;
916   else if (ARM::QPRRegClass.contains(DestReg, SrcReg))
917     Opc = Subtarget.hasNEON() ? ARM::VORRq : ARM::MVE_VORR;
918 
919   if (Opc) {
920     MachineInstrBuilder MIB = BuildMI(MBB, I, DL, get(Opc), DestReg);
921     MIB.addReg(SrcReg, getKillRegState(KillSrc));
922     if (Opc == ARM::VORRq || Opc == ARM::MVE_VORR)
923       MIB.addReg(SrcReg, getKillRegState(KillSrc));
924     if (Opc == ARM::MVE_VORR)
925       addUnpredicatedMveVpredROp(MIB, DestReg);
926     else
927       MIB.add(predOps(ARMCC::AL));
928     return;
929   }
930 
931   // Handle register classes that require multiple instructions.
932   unsigned BeginIdx = 0;
933   unsigned SubRegs = 0;
934   int Spacing = 1;
935 
936   // Use VORRq when possible.
937   if (ARM::QQPRRegClass.contains(DestReg, SrcReg)) {
938     Opc = Subtarget.hasNEON() ? ARM::VORRq : ARM::MVE_VORR;
939     BeginIdx = ARM::qsub_0;
940     SubRegs = 2;
941   } else if (ARM::QQQQPRRegClass.contains(DestReg, SrcReg)) {
942     Opc = Subtarget.hasNEON() ? ARM::VORRq : ARM::MVE_VORR;
943     BeginIdx = ARM::qsub_0;
944     SubRegs = 4;
945   // Fall back to VMOVD.
946   } else if (ARM::DPairRegClass.contains(DestReg, SrcReg)) {
947     Opc = ARM::VMOVD;
948     BeginIdx = ARM::dsub_0;
949     SubRegs = 2;
950   } else if (ARM::DTripleRegClass.contains(DestReg, SrcReg)) {
951     Opc = ARM::VMOVD;
952     BeginIdx = ARM::dsub_0;
953     SubRegs = 3;
954   } else if (ARM::DQuadRegClass.contains(DestReg, SrcReg)) {
955     Opc = ARM::VMOVD;
956     BeginIdx = ARM::dsub_0;
957     SubRegs = 4;
958   } else if (ARM::GPRPairRegClass.contains(DestReg, SrcReg)) {
959     Opc = Subtarget.isThumb2() ? ARM::tMOVr : ARM::MOVr;
960     BeginIdx = ARM::gsub_0;
961     SubRegs = 2;
962   } else if (ARM::DPairSpcRegClass.contains(DestReg, SrcReg)) {
963     Opc = ARM::VMOVD;
964     BeginIdx = ARM::dsub_0;
965     SubRegs = 2;
966     Spacing = 2;
967   } else if (ARM::DTripleSpcRegClass.contains(DestReg, SrcReg)) {
968     Opc = ARM::VMOVD;
969     BeginIdx = ARM::dsub_0;
970     SubRegs = 3;
971     Spacing = 2;
972   } else if (ARM::DQuadSpcRegClass.contains(DestReg, SrcReg)) {
973     Opc = ARM::VMOVD;
974     BeginIdx = ARM::dsub_0;
975     SubRegs = 4;
976     Spacing = 2;
977   } else if (ARM::DPRRegClass.contains(DestReg, SrcReg) &&
978              !Subtarget.hasFP64()) {
979     Opc = ARM::VMOVS;
980     BeginIdx = ARM::ssub_0;
981     SubRegs = 2;
982   } else if (SrcReg == ARM::CPSR) {
983     copyFromCPSR(MBB, I, DestReg, KillSrc, Subtarget);
984     return;
985   } else if (DestReg == ARM::CPSR) {
986     copyToCPSR(MBB, I, SrcReg, KillSrc, Subtarget);
987     return;
988   } else if (DestReg == ARM::VPR) {
989     assert(ARM::GPRRegClass.contains(SrcReg));
990     BuildMI(MBB, I, I->getDebugLoc(), get(ARM::VMSR_P0), DestReg)
991         .addReg(SrcReg, getKillRegState(KillSrc))
992         .add(predOps(ARMCC::AL));
993     return;
994   } else if (SrcReg == ARM::VPR) {
995     assert(ARM::GPRRegClass.contains(DestReg));
996     BuildMI(MBB, I, I->getDebugLoc(), get(ARM::VMRS_P0), DestReg)
997         .addReg(SrcReg, getKillRegState(KillSrc))
998         .add(predOps(ARMCC::AL));
999     return;
1000   } else if (DestReg == ARM::FPSCR_NZCV) {
1001     assert(ARM::GPRRegClass.contains(SrcReg));
1002     BuildMI(MBB, I, I->getDebugLoc(), get(ARM::VMSR_FPSCR_NZCVQC), DestReg)
1003         .addReg(SrcReg, getKillRegState(KillSrc))
1004         .add(predOps(ARMCC::AL));
1005     return;
1006   } else if (SrcReg == ARM::FPSCR_NZCV) {
1007     assert(ARM::GPRRegClass.contains(DestReg));
1008     BuildMI(MBB, I, I->getDebugLoc(), get(ARM::VMRS_FPSCR_NZCVQC), DestReg)
1009         .addReg(SrcReg, getKillRegState(KillSrc))
1010         .add(predOps(ARMCC::AL));
1011     return;
1012   }
1013 
1014   assert(Opc && "Impossible reg-to-reg copy");
1015 
1016   const TargetRegisterInfo *TRI = &getRegisterInfo();
1017   MachineInstrBuilder Mov;
1018 
1019   // Copy register tuples backward when the first Dest reg overlaps with SrcReg.
1020   if (TRI->regsOverlap(SrcReg, TRI->getSubReg(DestReg, BeginIdx))) {
1021     BeginIdx = BeginIdx + ((SubRegs - 1) * Spacing);
1022     Spacing = -Spacing;
1023   }
1024 #ifndef NDEBUG
1025   SmallSet<unsigned, 4> DstRegs;
1026 #endif
1027   for (unsigned i = 0; i != SubRegs; ++i) {
1028     Register Dst = TRI->getSubReg(DestReg, BeginIdx + i * Spacing);
1029     Register Src = TRI->getSubReg(SrcReg, BeginIdx + i * Spacing);
1030     assert(Dst && Src && "Bad sub-register");
1031 #ifndef NDEBUG
1032     assert(!DstRegs.count(Src) && "destructive vector copy");
1033     DstRegs.insert(Dst);
1034 #endif
1035     Mov = BuildMI(MBB, I, I->getDebugLoc(), get(Opc), Dst).addReg(Src);
1036     // VORR (NEON or MVE) takes two source operands.
1037     if (Opc == ARM::VORRq || Opc == ARM::MVE_VORR) {
1038       Mov.addReg(Src);
1039     }
1040     // MVE VORR takes predicate operands in place of an ordinary condition.
1041     if (Opc == ARM::MVE_VORR)
1042       addUnpredicatedMveVpredROp(Mov, Dst);
1043     else
1044       Mov = Mov.add(predOps(ARMCC::AL));
1045     // MOVr can set CC.
1046     if (Opc == ARM::MOVr)
1047       Mov = Mov.add(condCodeOp());
1048   }
1049   // Add implicit super-register defs and kills to the last instruction.
1050   Mov->addRegisterDefined(DestReg, TRI);
1051   if (KillSrc)
1052     Mov->addRegisterKilled(SrcReg, TRI);
1053 }
1054 
1055 Optional<DestSourcePair>
1056 ARMBaseInstrInfo::isCopyInstrImpl(const MachineInstr &MI) const {
1057   // VMOVRRD is also a copy instruction but it requires
1058   // special way of handling. It is more complex copy version
1059   // and since that we are not considering it. For recognition
1060   // of such instruction isExtractSubregLike MI interface fuction
1061   // could be used.
1062   // VORRq is considered as a move only if two inputs are
1063   // the same register.
1064   if (!MI.isMoveReg() ||
1065       (MI.getOpcode() == ARM::VORRq &&
1066        MI.getOperand(1).getReg() != MI.getOperand(2).getReg()))
1067     return None;
1068   return DestSourcePair{MI.getOperand(0), MI.getOperand(1)};
1069 }
1070 
1071 Optional<ParamLoadedValue>
1072 ARMBaseInstrInfo::describeLoadedValue(const MachineInstr &MI,
1073                                       Register Reg) const {
1074   if (auto DstSrcPair = isCopyInstrImpl(MI)) {
1075     Register DstReg = DstSrcPair->Destination->getReg();
1076 
1077     // TODO: We don't handle cases where the forwarding reg is narrower/wider
1078     // than the copy registers. Consider for example:
1079     //
1080     //   s16 = VMOVS s0
1081     //   s17 = VMOVS s1
1082     //   call @callee(d0)
1083     //
1084     // We'd like to describe the call site value of d0 as d8, but this requires
1085     // gathering and merging the descriptions for the two VMOVS instructions.
1086     //
1087     // We also don't handle the reverse situation, where the forwarding reg is
1088     // narrower than the copy destination:
1089     //
1090     //   d8 = VMOVD d0
1091     //   call @callee(s1)
1092     //
1093     // We need to produce a fragment description (the call site value of s1 is
1094     // /not/ just d8).
1095     if (DstReg != Reg)
1096       return None;
1097   }
1098   return TargetInstrInfo::describeLoadedValue(MI, Reg);
1099 }
1100 
1101 const MachineInstrBuilder &
1102 ARMBaseInstrInfo::AddDReg(MachineInstrBuilder &MIB, unsigned Reg,
1103                           unsigned SubIdx, unsigned State,
1104                           const TargetRegisterInfo *TRI) const {
1105   if (!SubIdx)
1106     return MIB.addReg(Reg, State);
1107 
1108   if (Register::isPhysicalRegister(Reg))
1109     return MIB.addReg(TRI->getSubReg(Reg, SubIdx), State);
1110   return MIB.addReg(Reg, State, SubIdx);
1111 }
1112 
1113 void ARMBaseInstrInfo::
1114 storeRegToStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
1115                     Register SrcReg, bool isKill, int FI,
1116                     const TargetRegisterClass *RC,
1117                     const TargetRegisterInfo *TRI) const {
1118   MachineFunction &MF = *MBB.getParent();
1119   MachineFrameInfo &MFI = MF.getFrameInfo();
1120   Align Alignment = MFI.getObjectAlign(FI);
1121 
1122   MachineMemOperand *MMO = MF.getMachineMemOperand(
1123       MachinePointerInfo::getFixedStack(MF, FI), MachineMemOperand::MOStore,
1124       MFI.getObjectSize(FI), Alignment);
1125 
1126   switch (TRI->getSpillSize(*RC)) {
1127     case 2:
1128       if (ARM::HPRRegClass.hasSubClassEq(RC)) {
1129         BuildMI(MBB, I, DebugLoc(), get(ARM::VSTRH))
1130             .addReg(SrcReg, getKillRegState(isKill))
1131             .addFrameIndex(FI)
1132             .addImm(0)
1133             .addMemOperand(MMO)
1134             .add(predOps(ARMCC::AL));
1135       } else
1136         llvm_unreachable("Unknown reg class!");
1137       break;
1138     case 4:
1139       if (ARM::GPRRegClass.hasSubClassEq(RC)) {
1140         BuildMI(MBB, I, DebugLoc(), get(ARM::STRi12))
1141             .addReg(SrcReg, getKillRegState(isKill))
1142             .addFrameIndex(FI)
1143             .addImm(0)
1144             .addMemOperand(MMO)
1145             .add(predOps(ARMCC::AL));
1146       } else if (ARM::SPRRegClass.hasSubClassEq(RC)) {
1147         BuildMI(MBB, I, DebugLoc(), get(ARM::VSTRS))
1148             .addReg(SrcReg, getKillRegState(isKill))
1149             .addFrameIndex(FI)
1150             .addImm(0)
1151             .addMemOperand(MMO)
1152             .add(predOps(ARMCC::AL));
1153       } else if (ARM::VCCRRegClass.hasSubClassEq(RC)) {
1154         BuildMI(MBB, I, DebugLoc(), get(ARM::VSTR_P0_off))
1155             .addReg(SrcReg, getKillRegState(isKill))
1156             .addFrameIndex(FI)
1157             .addImm(0)
1158             .addMemOperand(MMO)
1159             .add(predOps(ARMCC::AL));
1160       } else
1161         llvm_unreachable("Unknown reg class!");
1162       break;
1163     case 8:
1164       if (ARM::DPRRegClass.hasSubClassEq(RC)) {
1165         BuildMI(MBB, I, DebugLoc(), get(ARM::VSTRD))
1166             .addReg(SrcReg, getKillRegState(isKill))
1167             .addFrameIndex(FI)
1168             .addImm(0)
1169             .addMemOperand(MMO)
1170             .add(predOps(ARMCC::AL));
1171       } else if (ARM::GPRPairRegClass.hasSubClassEq(RC)) {
1172         if (Subtarget.hasV5TEOps()) {
1173           MachineInstrBuilder MIB = BuildMI(MBB, I, DebugLoc(), get(ARM::STRD));
1174           AddDReg(MIB, SrcReg, ARM::gsub_0, getKillRegState(isKill), TRI);
1175           AddDReg(MIB, SrcReg, ARM::gsub_1, 0, TRI);
1176           MIB.addFrameIndex(FI).addReg(0).addImm(0).addMemOperand(MMO)
1177              .add(predOps(ARMCC::AL));
1178         } else {
1179           // Fallback to STM instruction, which has existed since the dawn of
1180           // time.
1181           MachineInstrBuilder MIB = BuildMI(MBB, I, DebugLoc(), get(ARM::STMIA))
1182                                         .addFrameIndex(FI)
1183                                         .addMemOperand(MMO)
1184                                         .add(predOps(ARMCC::AL));
1185           AddDReg(MIB, SrcReg, ARM::gsub_0, getKillRegState(isKill), TRI);
1186           AddDReg(MIB, SrcReg, ARM::gsub_1, 0, TRI);
1187         }
1188       } else
1189         llvm_unreachable("Unknown reg class!");
1190       break;
1191     case 16:
1192       if (ARM::DPairRegClass.hasSubClassEq(RC) && Subtarget.hasNEON()) {
1193         // Use aligned spills if the stack can be realigned.
1194         if (Alignment >= 16 && getRegisterInfo().canRealignStack(MF)) {
1195           BuildMI(MBB, I, DebugLoc(), get(ARM::VST1q64))
1196               .addFrameIndex(FI)
1197               .addImm(16)
1198               .addReg(SrcReg, getKillRegState(isKill))
1199               .addMemOperand(MMO)
1200               .add(predOps(ARMCC::AL));
1201         } else {
1202           BuildMI(MBB, I, DebugLoc(), get(ARM::VSTMQIA))
1203               .addReg(SrcReg, getKillRegState(isKill))
1204               .addFrameIndex(FI)
1205               .addMemOperand(MMO)
1206               .add(predOps(ARMCC::AL));
1207         }
1208       } else if (ARM::QPRRegClass.hasSubClassEq(RC) &&
1209                  Subtarget.hasMVEIntegerOps()) {
1210         auto MIB = BuildMI(MBB, I, DebugLoc(), get(ARM::MVE_VSTRWU32));
1211         MIB.addReg(SrcReg, getKillRegState(isKill))
1212           .addFrameIndex(FI)
1213           .addImm(0)
1214           .addMemOperand(MMO);
1215         addUnpredicatedMveVpredNOp(MIB);
1216       } else
1217         llvm_unreachable("Unknown reg class!");
1218       break;
1219     case 24:
1220       if (ARM::DTripleRegClass.hasSubClassEq(RC)) {
1221         // Use aligned spills if the stack can be realigned.
1222         if (Alignment >= 16 && getRegisterInfo().canRealignStack(MF) &&
1223             Subtarget.hasNEON()) {
1224           BuildMI(MBB, I, DebugLoc(), get(ARM::VST1d64TPseudo))
1225               .addFrameIndex(FI)
1226               .addImm(16)
1227               .addReg(SrcReg, getKillRegState(isKill))
1228               .addMemOperand(MMO)
1229               .add(predOps(ARMCC::AL));
1230         } else {
1231           MachineInstrBuilder MIB = BuildMI(MBB, I, DebugLoc(),
1232                                             get(ARM::VSTMDIA))
1233                                         .addFrameIndex(FI)
1234                                         .add(predOps(ARMCC::AL))
1235                                         .addMemOperand(MMO);
1236           MIB = AddDReg(MIB, SrcReg, ARM::dsub_0, getKillRegState(isKill), TRI);
1237           MIB = AddDReg(MIB, SrcReg, ARM::dsub_1, 0, TRI);
1238           AddDReg(MIB, SrcReg, ARM::dsub_2, 0, TRI);
1239         }
1240       } else
1241         llvm_unreachable("Unknown reg class!");
1242       break;
1243     case 32:
1244       if (ARM::QQPRRegClass.hasSubClassEq(RC) ||
1245           ARM::MQQPRRegClass.hasSubClassEq(RC) ||
1246           ARM::DQuadRegClass.hasSubClassEq(RC)) {
1247         if (Alignment >= 16 && getRegisterInfo().canRealignStack(MF) &&
1248             Subtarget.hasNEON()) {
1249           // FIXME: It's possible to only store part of the QQ register if the
1250           // spilled def has a sub-register index.
1251           BuildMI(MBB, I, DebugLoc(), get(ARM::VST1d64QPseudo))
1252               .addFrameIndex(FI)
1253               .addImm(16)
1254               .addReg(SrcReg, getKillRegState(isKill))
1255               .addMemOperand(MMO)
1256               .add(predOps(ARMCC::AL));
1257         } else if (Subtarget.hasMVEIntegerOps()) {
1258           BuildMI(MBB, I, DebugLoc(), get(ARM::MQQPRStore))
1259               .addReg(SrcReg, getKillRegState(isKill))
1260               .addFrameIndex(FI)
1261               .addMemOperand(MMO);
1262         } else {
1263           MachineInstrBuilder MIB = BuildMI(MBB, I, DebugLoc(),
1264                                             get(ARM::VSTMDIA))
1265                                         .addFrameIndex(FI)
1266                                         .add(predOps(ARMCC::AL))
1267                                         .addMemOperand(MMO);
1268           MIB = AddDReg(MIB, SrcReg, ARM::dsub_0, getKillRegState(isKill), TRI);
1269           MIB = AddDReg(MIB, SrcReg, ARM::dsub_1, 0, TRI);
1270           MIB = AddDReg(MIB, SrcReg, ARM::dsub_2, 0, TRI);
1271                 AddDReg(MIB, SrcReg, ARM::dsub_3, 0, TRI);
1272         }
1273       } else
1274         llvm_unreachable("Unknown reg class!");
1275       break;
1276     case 64:
1277       if (ARM::MQQQQPRRegClass.hasSubClassEq(RC) &&
1278           Subtarget.hasMVEIntegerOps()) {
1279         BuildMI(MBB, I, DebugLoc(), get(ARM::MQQQQPRStore))
1280             .addReg(SrcReg, getKillRegState(isKill))
1281             .addFrameIndex(FI)
1282             .addMemOperand(MMO);
1283       } else if (ARM::QQQQPRRegClass.hasSubClassEq(RC)) {
1284         MachineInstrBuilder MIB = BuildMI(MBB, I, DebugLoc(), get(ARM::VSTMDIA))
1285                                       .addFrameIndex(FI)
1286                                       .add(predOps(ARMCC::AL))
1287                                       .addMemOperand(MMO);
1288         MIB = AddDReg(MIB, SrcReg, ARM::dsub_0, getKillRegState(isKill), TRI);
1289         MIB = AddDReg(MIB, SrcReg, ARM::dsub_1, 0, TRI);
1290         MIB = AddDReg(MIB, SrcReg, ARM::dsub_2, 0, TRI);
1291         MIB = AddDReg(MIB, SrcReg, ARM::dsub_3, 0, TRI);
1292         MIB = AddDReg(MIB, SrcReg, ARM::dsub_4, 0, TRI);
1293         MIB = AddDReg(MIB, SrcReg, ARM::dsub_5, 0, TRI);
1294         MIB = AddDReg(MIB, SrcReg, ARM::dsub_6, 0, TRI);
1295               AddDReg(MIB, SrcReg, ARM::dsub_7, 0, TRI);
1296       } else
1297         llvm_unreachable("Unknown reg class!");
1298       break;
1299     default:
1300       llvm_unreachable("Unknown reg class!");
1301   }
1302 }
1303 
1304 unsigned ARMBaseInstrInfo::isStoreToStackSlot(const MachineInstr &MI,
1305                                               int &FrameIndex) const {
1306   switch (MI.getOpcode()) {
1307   default: break;
1308   case ARM::STRrs:
1309   case ARM::t2STRs: // FIXME: don't use t2STRs to access frame.
1310     if (MI.getOperand(1).isFI() && MI.getOperand(2).isReg() &&
1311         MI.getOperand(3).isImm() && MI.getOperand(2).getReg() == 0 &&
1312         MI.getOperand(3).getImm() == 0) {
1313       FrameIndex = MI.getOperand(1).getIndex();
1314       return MI.getOperand(0).getReg();
1315     }
1316     break;
1317   case ARM::STRi12:
1318   case ARM::t2STRi12:
1319   case ARM::tSTRspi:
1320   case ARM::VSTRD:
1321   case ARM::VSTRS:
1322   case ARM::VSTR_P0_off:
1323   case ARM::MVE_VSTRWU32:
1324     if (MI.getOperand(1).isFI() && MI.getOperand(2).isImm() &&
1325         MI.getOperand(2).getImm() == 0) {
1326       FrameIndex = MI.getOperand(1).getIndex();
1327       return MI.getOperand(0).getReg();
1328     }
1329     break;
1330   case ARM::VST1q64:
1331   case ARM::VST1d64TPseudo:
1332   case ARM::VST1d64QPseudo:
1333     if (MI.getOperand(0).isFI() && MI.getOperand(2).getSubReg() == 0) {
1334       FrameIndex = MI.getOperand(0).getIndex();
1335       return MI.getOperand(2).getReg();
1336     }
1337     break;
1338   case ARM::VSTMQIA:
1339     if (MI.getOperand(1).isFI() && MI.getOperand(0).getSubReg() == 0) {
1340       FrameIndex = MI.getOperand(1).getIndex();
1341       return MI.getOperand(0).getReg();
1342     }
1343     break;
1344   case ARM::MQQPRStore:
1345   case ARM::MQQQQPRStore:
1346     if (MI.getOperand(1).isFI()) {
1347       FrameIndex = MI.getOperand(1).getIndex();
1348       return MI.getOperand(0).getReg();
1349     }
1350     break;
1351   }
1352 
1353   return 0;
1354 }
1355 
1356 unsigned ARMBaseInstrInfo::isStoreToStackSlotPostFE(const MachineInstr &MI,
1357                                                     int &FrameIndex) const {
1358   SmallVector<const MachineMemOperand *, 1> Accesses;
1359   if (MI.mayStore() && hasStoreToStackSlot(MI, Accesses) &&
1360       Accesses.size() == 1) {
1361     FrameIndex =
1362         cast<FixedStackPseudoSourceValue>(Accesses.front()->getPseudoValue())
1363             ->getFrameIndex();
1364     return true;
1365   }
1366   return false;
1367 }
1368 
1369 void ARMBaseInstrInfo::
1370 loadRegFromStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
1371                      Register DestReg, int FI,
1372                      const TargetRegisterClass *RC,
1373                      const TargetRegisterInfo *TRI) const {
1374   DebugLoc DL;
1375   if (I != MBB.end()) DL = I->getDebugLoc();
1376   MachineFunction &MF = *MBB.getParent();
1377   MachineFrameInfo &MFI = MF.getFrameInfo();
1378   const Align Alignment = MFI.getObjectAlign(FI);
1379   MachineMemOperand *MMO = MF.getMachineMemOperand(
1380       MachinePointerInfo::getFixedStack(MF, FI), MachineMemOperand::MOLoad,
1381       MFI.getObjectSize(FI), Alignment);
1382 
1383   switch (TRI->getSpillSize(*RC)) {
1384   case 2:
1385     if (ARM::HPRRegClass.hasSubClassEq(RC)) {
1386       BuildMI(MBB, I, DL, get(ARM::VLDRH), DestReg)
1387           .addFrameIndex(FI)
1388           .addImm(0)
1389           .addMemOperand(MMO)
1390           .add(predOps(ARMCC::AL));
1391     } else
1392       llvm_unreachable("Unknown reg class!");
1393     break;
1394   case 4:
1395     if (ARM::GPRRegClass.hasSubClassEq(RC)) {
1396       BuildMI(MBB, I, DL, get(ARM::LDRi12), DestReg)
1397           .addFrameIndex(FI)
1398           .addImm(0)
1399           .addMemOperand(MMO)
1400           .add(predOps(ARMCC::AL));
1401     } else if (ARM::SPRRegClass.hasSubClassEq(RC)) {
1402       BuildMI(MBB, I, DL, get(ARM::VLDRS), DestReg)
1403           .addFrameIndex(FI)
1404           .addImm(0)
1405           .addMemOperand(MMO)
1406           .add(predOps(ARMCC::AL));
1407     } else if (ARM::VCCRRegClass.hasSubClassEq(RC)) {
1408       BuildMI(MBB, I, DL, get(ARM::VLDR_P0_off), DestReg)
1409           .addFrameIndex(FI)
1410           .addImm(0)
1411           .addMemOperand(MMO)
1412           .add(predOps(ARMCC::AL));
1413     } else
1414       llvm_unreachable("Unknown reg class!");
1415     break;
1416   case 8:
1417     if (ARM::DPRRegClass.hasSubClassEq(RC)) {
1418       BuildMI(MBB, I, DL, get(ARM::VLDRD), DestReg)
1419           .addFrameIndex(FI)
1420           .addImm(0)
1421           .addMemOperand(MMO)
1422           .add(predOps(ARMCC::AL));
1423     } else if (ARM::GPRPairRegClass.hasSubClassEq(RC)) {
1424       MachineInstrBuilder MIB;
1425 
1426       if (Subtarget.hasV5TEOps()) {
1427         MIB = BuildMI(MBB, I, DL, get(ARM::LDRD));
1428         AddDReg(MIB, DestReg, ARM::gsub_0, RegState::DefineNoRead, TRI);
1429         AddDReg(MIB, DestReg, ARM::gsub_1, RegState::DefineNoRead, TRI);
1430         MIB.addFrameIndex(FI).addReg(0).addImm(0).addMemOperand(MMO)
1431            .add(predOps(ARMCC::AL));
1432       } else {
1433         // Fallback to LDM instruction, which has existed since the dawn of
1434         // time.
1435         MIB = BuildMI(MBB, I, DL, get(ARM::LDMIA))
1436                   .addFrameIndex(FI)
1437                   .addMemOperand(MMO)
1438                   .add(predOps(ARMCC::AL));
1439         MIB = AddDReg(MIB, DestReg, ARM::gsub_0, RegState::DefineNoRead, TRI);
1440         MIB = AddDReg(MIB, DestReg, ARM::gsub_1, RegState::DefineNoRead, TRI);
1441       }
1442 
1443       if (Register::isPhysicalRegister(DestReg))
1444         MIB.addReg(DestReg, RegState::ImplicitDefine);
1445     } else
1446       llvm_unreachable("Unknown reg class!");
1447     break;
1448   case 16:
1449     if (ARM::DPairRegClass.hasSubClassEq(RC) && Subtarget.hasNEON()) {
1450       if (Alignment >= 16 && getRegisterInfo().canRealignStack(MF)) {
1451         BuildMI(MBB, I, DL, get(ARM::VLD1q64), DestReg)
1452             .addFrameIndex(FI)
1453             .addImm(16)
1454             .addMemOperand(MMO)
1455             .add(predOps(ARMCC::AL));
1456       } else {
1457         BuildMI(MBB, I, DL, get(ARM::VLDMQIA), DestReg)
1458             .addFrameIndex(FI)
1459             .addMemOperand(MMO)
1460             .add(predOps(ARMCC::AL));
1461       }
1462     } else if (ARM::QPRRegClass.hasSubClassEq(RC) &&
1463                Subtarget.hasMVEIntegerOps()) {
1464       auto MIB = BuildMI(MBB, I, DL, get(ARM::MVE_VLDRWU32), DestReg);
1465       MIB.addFrameIndex(FI)
1466         .addImm(0)
1467         .addMemOperand(MMO);
1468       addUnpredicatedMveVpredNOp(MIB);
1469     } else
1470       llvm_unreachable("Unknown reg class!");
1471     break;
1472   case 24:
1473     if (ARM::DTripleRegClass.hasSubClassEq(RC)) {
1474       if (Alignment >= 16 && getRegisterInfo().canRealignStack(MF) &&
1475           Subtarget.hasNEON()) {
1476         BuildMI(MBB, I, DL, get(ARM::VLD1d64TPseudo), DestReg)
1477             .addFrameIndex(FI)
1478             .addImm(16)
1479             .addMemOperand(MMO)
1480             .add(predOps(ARMCC::AL));
1481       } else {
1482         MachineInstrBuilder MIB = BuildMI(MBB, I, DL, get(ARM::VLDMDIA))
1483                                       .addFrameIndex(FI)
1484                                       .addMemOperand(MMO)
1485                                       .add(predOps(ARMCC::AL));
1486         MIB = AddDReg(MIB, DestReg, ARM::dsub_0, RegState::DefineNoRead, TRI);
1487         MIB = AddDReg(MIB, DestReg, ARM::dsub_1, RegState::DefineNoRead, TRI);
1488         MIB = AddDReg(MIB, DestReg, ARM::dsub_2, RegState::DefineNoRead, TRI);
1489         if (Register::isPhysicalRegister(DestReg))
1490           MIB.addReg(DestReg, RegState::ImplicitDefine);
1491       }
1492     } else
1493       llvm_unreachable("Unknown reg class!");
1494     break;
1495    case 32:
1496      if (ARM::QQPRRegClass.hasSubClassEq(RC) ||
1497          ARM::MQQPRRegClass.hasSubClassEq(RC) ||
1498          ARM::DQuadRegClass.hasSubClassEq(RC)) {
1499        if (Alignment >= 16 && getRegisterInfo().canRealignStack(MF) &&
1500            Subtarget.hasNEON()) {
1501          BuildMI(MBB, I, DL, get(ARM::VLD1d64QPseudo), DestReg)
1502              .addFrameIndex(FI)
1503              .addImm(16)
1504              .addMemOperand(MMO)
1505              .add(predOps(ARMCC::AL));
1506        } else if (Subtarget.hasMVEIntegerOps()) {
1507          BuildMI(MBB, I, DL, get(ARM::MQQPRLoad), DestReg)
1508              .addFrameIndex(FI)
1509              .addMemOperand(MMO);
1510        } else {
1511          MachineInstrBuilder MIB = BuildMI(MBB, I, DL, get(ARM::VLDMDIA))
1512                                        .addFrameIndex(FI)
1513                                        .add(predOps(ARMCC::AL))
1514                                        .addMemOperand(MMO);
1515          MIB = AddDReg(MIB, DestReg, ARM::dsub_0, RegState::DefineNoRead, TRI);
1516          MIB = AddDReg(MIB, DestReg, ARM::dsub_1, RegState::DefineNoRead, TRI);
1517          MIB = AddDReg(MIB, DestReg, ARM::dsub_2, RegState::DefineNoRead, TRI);
1518          MIB = AddDReg(MIB, DestReg, ARM::dsub_3, RegState::DefineNoRead, TRI);
1519          if (Register::isPhysicalRegister(DestReg))
1520            MIB.addReg(DestReg, RegState::ImplicitDefine);
1521        }
1522      } else
1523        llvm_unreachable("Unknown reg class!");
1524      break;
1525   case 64:
1526     if (ARM::MQQQQPRRegClass.hasSubClassEq(RC) &&
1527         Subtarget.hasMVEIntegerOps()) {
1528       BuildMI(MBB, I, DL, get(ARM::MQQQQPRLoad), DestReg)
1529           .addFrameIndex(FI)
1530           .addMemOperand(MMO);
1531     } else if (ARM::QQQQPRRegClass.hasSubClassEq(RC)) {
1532       MachineInstrBuilder MIB = BuildMI(MBB, I, DL, get(ARM::VLDMDIA))
1533                                     .addFrameIndex(FI)
1534                                     .add(predOps(ARMCC::AL))
1535                                     .addMemOperand(MMO);
1536       MIB = AddDReg(MIB, DestReg, ARM::dsub_0, RegState::DefineNoRead, TRI);
1537       MIB = AddDReg(MIB, DestReg, ARM::dsub_1, RegState::DefineNoRead, TRI);
1538       MIB = AddDReg(MIB, DestReg, ARM::dsub_2, RegState::DefineNoRead, TRI);
1539       MIB = AddDReg(MIB, DestReg, ARM::dsub_3, RegState::DefineNoRead, TRI);
1540       MIB = AddDReg(MIB, DestReg, ARM::dsub_4, RegState::DefineNoRead, TRI);
1541       MIB = AddDReg(MIB, DestReg, ARM::dsub_5, RegState::DefineNoRead, TRI);
1542       MIB = AddDReg(MIB, DestReg, ARM::dsub_6, RegState::DefineNoRead, TRI);
1543       MIB = AddDReg(MIB, DestReg, ARM::dsub_7, RegState::DefineNoRead, TRI);
1544       if (Register::isPhysicalRegister(DestReg))
1545         MIB.addReg(DestReg, RegState::ImplicitDefine);
1546     } else
1547       llvm_unreachable("Unknown reg class!");
1548     break;
1549   default:
1550     llvm_unreachable("Unknown regclass!");
1551   }
1552 }
1553 
1554 unsigned ARMBaseInstrInfo::isLoadFromStackSlot(const MachineInstr &MI,
1555                                                int &FrameIndex) const {
1556   switch (MI.getOpcode()) {
1557   default: break;
1558   case ARM::LDRrs:
1559   case ARM::t2LDRs:  // FIXME: don't use t2LDRs to access frame.
1560     if (MI.getOperand(1).isFI() && MI.getOperand(2).isReg() &&
1561         MI.getOperand(3).isImm() && MI.getOperand(2).getReg() == 0 &&
1562         MI.getOperand(3).getImm() == 0) {
1563       FrameIndex = MI.getOperand(1).getIndex();
1564       return MI.getOperand(0).getReg();
1565     }
1566     break;
1567   case ARM::LDRi12:
1568   case ARM::t2LDRi12:
1569   case ARM::tLDRspi:
1570   case ARM::VLDRD:
1571   case ARM::VLDRS:
1572   case ARM::VLDR_P0_off:
1573   case ARM::MVE_VLDRWU32:
1574     if (MI.getOperand(1).isFI() && MI.getOperand(2).isImm() &&
1575         MI.getOperand(2).getImm() == 0) {
1576       FrameIndex = MI.getOperand(1).getIndex();
1577       return MI.getOperand(0).getReg();
1578     }
1579     break;
1580   case ARM::VLD1q64:
1581   case ARM::VLD1d8TPseudo:
1582   case ARM::VLD1d16TPseudo:
1583   case ARM::VLD1d32TPseudo:
1584   case ARM::VLD1d64TPseudo:
1585   case ARM::VLD1d8QPseudo:
1586   case ARM::VLD1d16QPseudo:
1587   case ARM::VLD1d32QPseudo:
1588   case ARM::VLD1d64QPseudo:
1589     if (MI.getOperand(1).isFI() && MI.getOperand(0).getSubReg() == 0) {
1590       FrameIndex = MI.getOperand(1).getIndex();
1591       return MI.getOperand(0).getReg();
1592     }
1593     break;
1594   case ARM::VLDMQIA:
1595     if (MI.getOperand(1).isFI() && MI.getOperand(0).getSubReg() == 0) {
1596       FrameIndex = MI.getOperand(1).getIndex();
1597       return MI.getOperand(0).getReg();
1598     }
1599     break;
1600   case ARM::MQQPRLoad:
1601   case ARM::MQQQQPRLoad:
1602     if (MI.getOperand(1).isFI()) {
1603       FrameIndex = MI.getOperand(1).getIndex();
1604       return MI.getOperand(0).getReg();
1605     }
1606     break;
1607   }
1608 
1609   return 0;
1610 }
1611 
1612 unsigned ARMBaseInstrInfo::isLoadFromStackSlotPostFE(const MachineInstr &MI,
1613                                                      int &FrameIndex) const {
1614   SmallVector<const MachineMemOperand *, 1> Accesses;
1615   if (MI.mayLoad() && hasLoadFromStackSlot(MI, Accesses) &&
1616       Accesses.size() == 1) {
1617     FrameIndex =
1618         cast<FixedStackPseudoSourceValue>(Accesses.front()->getPseudoValue())
1619             ->getFrameIndex();
1620     return true;
1621   }
1622   return false;
1623 }
1624 
1625 /// Expands MEMCPY to either LDMIA/STMIA or LDMIA_UPD/STMID_UPD
1626 /// depending on whether the result is used.
1627 void ARMBaseInstrInfo::expandMEMCPY(MachineBasicBlock::iterator MI) const {
1628   bool isThumb1 = Subtarget.isThumb1Only();
1629   bool isThumb2 = Subtarget.isThumb2();
1630   const ARMBaseInstrInfo *TII = Subtarget.getInstrInfo();
1631 
1632   DebugLoc dl = MI->getDebugLoc();
1633   MachineBasicBlock *BB = MI->getParent();
1634 
1635   MachineInstrBuilder LDM, STM;
1636   if (isThumb1 || !MI->getOperand(1).isDead()) {
1637     MachineOperand LDWb(MI->getOperand(1));
1638     LDM = BuildMI(*BB, MI, dl, TII->get(isThumb2 ? ARM::t2LDMIA_UPD
1639                                                  : isThumb1 ? ARM::tLDMIA_UPD
1640                                                             : ARM::LDMIA_UPD))
1641               .add(LDWb);
1642   } else {
1643     LDM = BuildMI(*BB, MI, dl, TII->get(isThumb2 ? ARM::t2LDMIA : ARM::LDMIA));
1644   }
1645 
1646   if (isThumb1 || !MI->getOperand(0).isDead()) {
1647     MachineOperand STWb(MI->getOperand(0));
1648     STM = BuildMI(*BB, MI, dl, TII->get(isThumb2 ? ARM::t2STMIA_UPD
1649                                                  : isThumb1 ? ARM::tSTMIA_UPD
1650                                                             : ARM::STMIA_UPD))
1651               .add(STWb);
1652   } else {
1653     STM = BuildMI(*BB, MI, dl, TII->get(isThumb2 ? ARM::t2STMIA : ARM::STMIA));
1654   }
1655 
1656   MachineOperand LDBase(MI->getOperand(3));
1657   LDM.add(LDBase).add(predOps(ARMCC::AL));
1658 
1659   MachineOperand STBase(MI->getOperand(2));
1660   STM.add(STBase).add(predOps(ARMCC::AL));
1661 
1662   // Sort the scratch registers into ascending order.
1663   const TargetRegisterInfo &TRI = getRegisterInfo();
1664   SmallVector<unsigned, 6> ScratchRegs;
1665   for(unsigned I = 5; I < MI->getNumOperands(); ++I)
1666     ScratchRegs.push_back(MI->getOperand(I).getReg());
1667   llvm::sort(ScratchRegs,
1668              [&TRI](const unsigned &Reg1, const unsigned &Reg2) -> bool {
1669                return TRI.getEncodingValue(Reg1) <
1670                       TRI.getEncodingValue(Reg2);
1671              });
1672 
1673   for (const auto &Reg : ScratchRegs) {
1674     LDM.addReg(Reg, RegState::Define);
1675     STM.addReg(Reg, RegState::Kill);
1676   }
1677 
1678   BB->erase(MI);
1679 }
1680 
1681 bool ARMBaseInstrInfo::expandPostRAPseudo(MachineInstr &MI) const {
1682   if (MI.getOpcode() == TargetOpcode::LOAD_STACK_GUARD) {
1683     assert(getSubtarget().getTargetTriple().isOSBinFormatMachO() &&
1684            "LOAD_STACK_GUARD currently supported only for MachO.");
1685     expandLoadStackGuard(MI);
1686     MI.getParent()->erase(MI);
1687     return true;
1688   }
1689 
1690   if (MI.getOpcode() == ARM::MEMCPY) {
1691     expandMEMCPY(MI);
1692     return true;
1693   }
1694 
1695   // This hook gets to expand COPY instructions before they become
1696   // copyPhysReg() calls.  Look for VMOVS instructions that can legally be
1697   // widened to VMOVD.  We prefer the VMOVD when possible because it may be
1698   // changed into a VORR that can go down the NEON pipeline.
1699   if (!MI.isCopy() || Subtarget.dontWidenVMOVS() || !Subtarget.hasFP64())
1700     return false;
1701 
1702   // Look for a copy between even S-registers.  That is where we keep floats
1703   // when using NEON v2f32 instructions for f32 arithmetic.
1704   Register DstRegS = MI.getOperand(0).getReg();
1705   Register SrcRegS = MI.getOperand(1).getReg();
1706   if (!ARM::SPRRegClass.contains(DstRegS, SrcRegS))
1707     return false;
1708 
1709   const TargetRegisterInfo *TRI = &getRegisterInfo();
1710   unsigned DstRegD = TRI->getMatchingSuperReg(DstRegS, ARM::ssub_0,
1711                                               &ARM::DPRRegClass);
1712   unsigned SrcRegD = TRI->getMatchingSuperReg(SrcRegS, ARM::ssub_0,
1713                                               &ARM::DPRRegClass);
1714   if (!DstRegD || !SrcRegD)
1715     return false;
1716 
1717   // We want to widen this into a DstRegD = VMOVD SrcRegD copy.  This is only
1718   // legal if the COPY already defines the full DstRegD, and it isn't a
1719   // sub-register insertion.
1720   if (!MI.definesRegister(DstRegD, TRI) || MI.readsRegister(DstRegD, TRI))
1721     return false;
1722 
1723   // A dead copy shouldn't show up here, but reject it just in case.
1724   if (MI.getOperand(0).isDead())
1725     return false;
1726 
1727   // All clear, widen the COPY.
1728   LLVM_DEBUG(dbgs() << "widening:    " << MI);
1729   MachineInstrBuilder MIB(*MI.getParent()->getParent(), MI);
1730 
1731   // Get rid of the old implicit-def of DstRegD.  Leave it if it defines a Q-reg
1732   // or some other super-register.
1733   int ImpDefIdx = MI.findRegisterDefOperandIdx(DstRegD);
1734   if (ImpDefIdx != -1)
1735     MI.RemoveOperand(ImpDefIdx);
1736 
1737   // Change the opcode and operands.
1738   MI.setDesc(get(ARM::VMOVD));
1739   MI.getOperand(0).setReg(DstRegD);
1740   MI.getOperand(1).setReg(SrcRegD);
1741   MIB.add(predOps(ARMCC::AL));
1742 
1743   // We are now reading SrcRegD instead of SrcRegS.  This may upset the
1744   // register scavenger and machine verifier, so we need to indicate that we
1745   // are reading an undefined value from SrcRegD, but a proper value from
1746   // SrcRegS.
1747   MI.getOperand(1).setIsUndef();
1748   MIB.addReg(SrcRegS, RegState::Implicit);
1749 
1750   // SrcRegD may actually contain an unrelated value in the ssub_1
1751   // sub-register.  Don't kill it.  Only kill the ssub_0 sub-register.
1752   if (MI.getOperand(1).isKill()) {
1753     MI.getOperand(1).setIsKill(false);
1754     MI.addRegisterKilled(SrcRegS, TRI, true);
1755   }
1756 
1757   LLVM_DEBUG(dbgs() << "replaced by: " << MI);
1758   return true;
1759 }
1760 
1761 /// Create a copy of a const pool value. Update CPI to the new index and return
1762 /// the label UID.
1763 static unsigned duplicateCPV(MachineFunction &MF, unsigned &CPI) {
1764   MachineConstantPool *MCP = MF.getConstantPool();
1765   ARMFunctionInfo *AFI = MF.getInfo<ARMFunctionInfo>();
1766 
1767   const MachineConstantPoolEntry &MCPE = MCP->getConstants()[CPI];
1768   assert(MCPE.isMachineConstantPoolEntry() &&
1769          "Expecting a machine constantpool entry!");
1770   ARMConstantPoolValue *ACPV =
1771     static_cast<ARMConstantPoolValue*>(MCPE.Val.MachineCPVal);
1772 
1773   unsigned PCLabelId = AFI->createPICLabelUId();
1774   ARMConstantPoolValue *NewCPV = nullptr;
1775 
1776   // FIXME: The below assumes PIC relocation model and that the function
1777   // is Thumb mode (t1 or t2). PCAdjustment would be 8 for ARM mode PIC, and
1778   // zero for non-PIC in ARM or Thumb. The callers are all of thumb LDR
1779   // instructions, so that's probably OK, but is PIC always correct when
1780   // we get here?
1781   if (ACPV->isGlobalValue())
1782     NewCPV = ARMConstantPoolConstant::Create(
1783         cast<ARMConstantPoolConstant>(ACPV)->getGV(), PCLabelId, ARMCP::CPValue,
1784         4, ACPV->getModifier(), ACPV->mustAddCurrentAddress());
1785   else if (ACPV->isExtSymbol())
1786     NewCPV = ARMConstantPoolSymbol::
1787       Create(MF.getFunction().getContext(),
1788              cast<ARMConstantPoolSymbol>(ACPV)->getSymbol(), PCLabelId, 4);
1789   else if (ACPV->isBlockAddress())
1790     NewCPV = ARMConstantPoolConstant::
1791       Create(cast<ARMConstantPoolConstant>(ACPV)->getBlockAddress(), PCLabelId,
1792              ARMCP::CPBlockAddress, 4);
1793   else if (ACPV->isLSDA())
1794     NewCPV = ARMConstantPoolConstant::Create(&MF.getFunction(), PCLabelId,
1795                                              ARMCP::CPLSDA, 4);
1796   else if (ACPV->isMachineBasicBlock())
1797     NewCPV = ARMConstantPoolMBB::
1798       Create(MF.getFunction().getContext(),
1799              cast<ARMConstantPoolMBB>(ACPV)->getMBB(), PCLabelId, 4);
1800   else
1801     llvm_unreachable("Unexpected ARM constantpool value type!!");
1802   CPI = MCP->getConstantPoolIndex(NewCPV, MCPE.getAlign());
1803   return PCLabelId;
1804 }
1805 
1806 void ARMBaseInstrInfo::reMaterialize(MachineBasicBlock &MBB,
1807                                      MachineBasicBlock::iterator I,
1808                                      Register DestReg, unsigned SubIdx,
1809                                      const MachineInstr &Orig,
1810                                      const TargetRegisterInfo &TRI) const {
1811   unsigned Opcode = Orig.getOpcode();
1812   switch (Opcode) {
1813   default: {
1814     MachineInstr *MI = MBB.getParent()->CloneMachineInstr(&Orig);
1815     MI->substituteRegister(Orig.getOperand(0).getReg(), DestReg, SubIdx, TRI);
1816     MBB.insert(I, MI);
1817     break;
1818   }
1819   case ARM::tLDRpci_pic:
1820   case ARM::t2LDRpci_pic: {
1821     MachineFunction &MF = *MBB.getParent();
1822     unsigned CPI = Orig.getOperand(1).getIndex();
1823     unsigned PCLabelId = duplicateCPV(MF, CPI);
1824     BuildMI(MBB, I, Orig.getDebugLoc(), get(Opcode), DestReg)
1825         .addConstantPoolIndex(CPI)
1826         .addImm(PCLabelId)
1827         .cloneMemRefs(Orig);
1828     break;
1829   }
1830   }
1831 }
1832 
1833 MachineInstr &
1834 ARMBaseInstrInfo::duplicate(MachineBasicBlock &MBB,
1835     MachineBasicBlock::iterator InsertBefore,
1836     const MachineInstr &Orig) const {
1837   MachineInstr &Cloned = TargetInstrInfo::duplicate(MBB, InsertBefore, Orig);
1838   MachineBasicBlock::instr_iterator I = Cloned.getIterator();
1839   for (;;) {
1840     switch (I->getOpcode()) {
1841     case ARM::tLDRpci_pic:
1842     case ARM::t2LDRpci_pic: {
1843       MachineFunction &MF = *MBB.getParent();
1844       unsigned CPI = I->getOperand(1).getIndex();
1845       unsigned PCLabelId = duplicateCPV(MF, CPI);
1846       I->getOperand(1).setIndex(CPI);
1847       I->getOperand(2).setImm(PCLabelId);
1848       break;
1849     }
1850     }
1851     if (!I->isBundledWithSucc())
1852       break;
1853     ++I;
1854   }
1855   return Cloned;
1856 }
1857 
1858 bool ARMBaseInstrInfo::produceSameValue(const MachineInstr &MI0,
1859                                         const MachineInstr &MI1,
1860                                         const MachineRegisterInfo *MRI) const {
1861   unsigned Opcode = MI0.getOpcode();
1862   if (Opcode == ARM::t2LDRpci ||
1863       Opcode == ARM::t2LDRpci_pic ||
1864       Opcode == ARM::tLDRpci ||
1865       Opcode == ARM::tLDRpci_pic ||
1866       Opcode == ARM::LDRLIT_ga_pcrel ||
1867       Opcode == ARM::LDRLIT_ga_pcrel_ldr ||
1868       Opcode == ARM::tLDRLIT_ga_pcrel ||
1869       Opcode == ARM::MOV_ga_pcrel ||
1870       Opcode == ARM::MOV_ga_pcrel_ldr ||
1871       Opcode == ARM::t2MOV_ga_pcrel) {
1872     if (MI1.getOpcode() != Opcode)
1873       return false;
1874     if (MI0.getNumOperands() != MI1.getNumOperands())
1875       return false;
1876 
1877     const MachineOperand &MO0 = MI0.getOperand(1);
1878     const MachineOperand &MO1 = MI1.getOperand(1);
1879     if (MO0.getOffset() != MO1.getOffset())
1880       return false;
1881 
1882     if (Opcode == ARM::LDRLIT_ga_pcrel ||
1883         Opcode == ARM::LDRLIT_ga_pcrel_ldr ||
1884         Opcode == ARM::tLDRLIT_ga_pcrel ||
1885         Opcode == ARM::MOV_ga_pcrel ||
1886         Opcode == ARM::MOV_ga_pcrel_ldr ||
1887         Opcode == ARM::t2MOV_ga_pcrel)
1888       // Ignore the PC labels.
1889       return MO0.getGlobal() == MO1.getGlobal();
1890 
1891     const MachineFunction *MF = MI0.getParent()->getParent();
1892     const MachineConstantPool *MCP = MF->getConstantPool();
1893     int CPI0 = MO0.getIndex();
1894     int CPI1 = MO1.getIndex();
1895     const MachineConstantPoolEntry &MCPE0 = MCP->getConstants()[CPI0];
1896     const MachineConstantPoolEntry &MCPE1 = MCP->getConstants()[CPI1];
1897     bool isARMCP0 = MCPE0.isMachineConstantPoolEntry();
1898     bool isARMCP1 = MCPE1.isMachineConstantPoolEntry();
1899     if (isARMCP0 && isARMCP1) {
1900       ARMConstantPoolValue *ACPV0 =
1901         static_cast<ARMConstantPoolValue*>(MCPE0.Val.MachineCPVal);
1902       ARMConstantPoolValue *ACPV1 =
1903         static_cast<ARMConstantPoolValue*>(MCPE1.Val.MachineCPVal);
1904       return ACPV0->hasSameValue(ACPV1);
1905     } else if (!isARMCP0 && !isARMCP1) {
1906       return MCPE0.Val.ConstVal == MCPE1.Val.ConstVal;
1907     }
1908     return false;
1909   } else if (Opcode == ARM::PICLDR) {
1910     if (MI1.getOpcode() != Opcode)
1911       return false;
1912     if (MI0.getNumOperands() != MI1.getNumOperands())
1913       return false;
1914 
1915     Register Addr0 = MI0.getOperand(1).getReg();
1916     Register Addr1 = MI1.getOperand(1).getReg();
1917     if (Addr0 != Addr1) {
1918       if (!MRI || !Register::isVirtualRegister(Addr0) ||
1919           !Register::isVirtualRegister(Addr1))
1920         return false;
1921 
1922       // This assumes SSA form.
1923       MachineInstr *Def0 = MRI->getVRegDef(Addr0);
1924       MachineInstr *Def1 = MRI->getVRegDef(Addr1);
1925       // Check if the loaded value, e.g. a constantpool of a global address, are
1926       // the same.
1927       if (!produceSameValue(*Def0, *Def1, MRI))
1928         return false;
1929     }
1930 
1931     for (unsigned i = 3, e = MI0.getNumOperands(); i != e; ++i) {
1932       // %12 = PICLDR %11, 0, 14, %noreg
1933       const MachineOperand &MO0 = MI0.getOperand(i);
1934       const MachineOperand &MO1 = MI1.getOperand(i);
1935       if (!MO0.isIdenticalTo(MO1))
1936         return false;
1937     }
1938     return true;
1939   }
1940 
1941   return MI0.isIdenticalTo(MI1, MachineInstr::IgnoreVRegDefs);
1942 }
1943 
1944 /// areLoadsFromSameBasePtr - This is used by the pre-regalloc scheduler to
1945 /// determine if two loads are loading from the same base address. It should
1946 /// only return true if the base pointers are the same and the only differences
1947 /// between the two addresses is the offset. It also returns the offsets by
1948 /// reference.
1949 ///
1950 /// FIXME: remove this in favor of the MachineInstr interface once pre-RA-sched
1951 /// is permanently disabled.
1952 bool ARMBaseInstrInfo::areLoadsFromSameBasePtr(SDNode *Load1, SDNode *Load2,
1953                                                int64_t &Offset1,
1954                                                int64_t &Offset2) const {
1955   // Don't worry about Thumb: just ARM and Thumb2.
1956   if (Subtarget.isThumb1Only()) return false;
1957 
1958   if (!Load1->isMachineOpcode() || !Load2->isMachineOpcode())
1959     return false;
1960 
1961   switch (Load1->getMachineOpcode()) {
1962   default:
1963     return false;
1964   case ARM::LDRi12:
1965   case ARM::LDRBi12:
1966   case ARM::LDRD:
1967   case ARM::LDRH:
1968   case ARM::LDRSB:
1969   case ARM::LDRSH:
1970   case ARM::VLDRD:
1971   case ARM::VLDRS:
1972   case ARM::t2LDRi8:
1973   case ARM::t2LDRBi8:
1974   case ARM::t2LDRDi8:
1975   case ARM::t2LDRSHi8:
1976   case ARM::t2LDRi12:
1977   case ARM::t2LDRBi12:
1978   case ARM::t2LDRSHi12:
1979     break;
1980   }
1981 
1982   switch (Load2->getMachineOpcode()) {
1983   default:
1984     return false;
1985   case ARM::LDRi12:
1986   case ARM::LDRBi12:
1987   case ARM::LDRD:
1988   case ARM::LDRH:
1989   case ARM::LDRSB:
1990   case ARM::LDRSH:
1991   case ARM::VLDRD:
1992   case ARM::VLDRS:
1993   case ARM::t2LDRi8:
1994   case ARM::t2LDRBi8:
1995   case ARM::t2LDRSHi8:
1996   case ARM::t2LDRi12:
1997   case ARM::t2LDRBi12:
1998   case ARM::t2LDRSHi12:
1999     break;
2000   }
2001 
2002   // Check if base addresses and chain operands match.
2003   if (Load1->getOperand(0) != Load2->getOperand(0) ||
2004       Load1->getOperand(4) != Load2->getOperand(4))
2005     return false;
2006 
2007   // Index should be Reg0.
2008   if (Load1->getOperand(3) != Load2->getOperand(3))
2009     return false;
2010 
2011   // Determine the offsets.
2012   if (isa<ConstantSDNode>(Load1->getOperand(1)) &&
2013       isa<ConstantSDNode>(Load2->getOperand(1))) {
2014     Offset1 = cast<ConstantSDNode>(Load1->getOperand(1))->getSExtValue();
2015     Offset2 = cast<ConstantSDNode>(Load2->getOperand(1))->getSExtValue();
2016     return true;
2017   }
2018 
2019   return false;
2020 }
2021 
2022 /// shouldScheduleLoadsNear - This is a used by the pre-regalloc scheduler to
2023 /// determine (in conjunction with areLoadsFromSameBasePtr) if two loads should
2024 /// be scheduled togther. On some targets if two loads are loading from
2025 /// addresses in the same cache line, it's better if they are scheduled
2026 /// together. This function takes two integers that represent the load offsets
2027 /// from the common base address. It returns true if it decides it's desirable
2028 /// to schedule the two loads together. "NumLoads" is the number of loads that
2029 /// have already been scheduled after Load1.
2030 ///
2031 /// FIXME: remove this in favor of the MachineInstr interface once pre-RA-sched
2032 /// is permanently disabled.
2033 bool ARMBaseInstrInfo::shouldScheduleLoadsNear(SDNode *Load1, SDNode *Load2,
2034                                                int64_t Offset1, int64_t Offset2,
2035                                                unsigned NumLoads) const {
2036   // Don't worry about Thumb: just ARM and Thumb2.
2037   if (Subtarget.isThumb1Only()) return false;
2038 
2039   assert(Offset2 > Offset1);
2040 
2041   if ((Offset2 - Offset1) / 8 > 64)
2042     return false;
2043 
2044   // Check if the machine opcodes are different. If they are different
2045   // then we consider them to not be of the same base address,
2046   // EXCEPT in the case of Thumb2 byte loads where one is LDRBi8 and the other LDRBi12.
2047   // In this case, they are considered to be the same because they are different
2048   // encoding forms of the same basic instruction.
2049   if ((Load1->getMachineOpcode() != Load2->getMachineOpcode()) &&
2050       !((Load1->getMachineOpcode() == ARM::t2LDRBi8 &&
2051          Load2->getMachineOpcode() == ARM::t2LDRBi12) ||
2052         (Load1->getMachineOpcode() == ARM::t2LDRBi12 &&
2053          Load2->getMachineOpcode() == ARM::t2LDRBi8)))
2054     return false;  // FIXME: overly conservative?
2055 
2056   // Four loads in a row should be sufficient.
2057   if (NumLoads >= 3)
2058     return false;
2059 
2060   return true;
2061 }
2062 
2063 bool ARMBaseInstrInfo::isSchedulingBoundary(const MachineInstr &MI,
2064                                             const MachineBasicBlock *MBB,
2065                                             const MachineFunction &MF) const {
2066   // Debug info is never a scheduling boundary. It's necessary to be explicit
2067   // due to the special treatment of IT instructions below, otherwise a
2068   // dbg_value followed by an IT will result in the IT instruction being
2069   // considered a scheduling hazard, which is wrong. It should be the actual
2070   // instruction preceding the dbg_value instruction(s), just like it is
2071   // when debug info is not present.
2072   if (MI.isDebugInstr())
2073     return false;
2074 
2075   // Terminators and labels can't be scheduled around.
2076   if (MI.isTerminator() || MI.isPosition())
2077     return true;
2078 
2079   // INLINEASM_BR can jump to another block
2080   if (MI.getOpcode() == TargetOpcode::INLINEASM_BR)
2081     return true;
2082 
2083   // Treat the start of the IT block as a scheduling boundary, but schedule
2084   // t2IT along with all instructions following it.
2085   // FIXME: This is a big hammer. But the alternative is to add all potential
2086   // true and anti dependencies to IT block instructions as implicit operands
2087   // to the t2IT instruction. The added compile time and complexity does not
2088   // seem worth it.
2089   MachineBasicBlock::const_iterator I = MI;
2090   // Make sure to skip any debug instructions
2091   while (++I != MBB->end() && I->isDebugInstr())
2092     ;
2093   if (I != MBB->end() && I->getOpcode() == ARM::t2IT)
2094     return true;
2095 
2096   // Don't attempt to schedule around any instruction that defines
2097   // a stack-oriented pointer, as it's unlikely to be profitable. This
2098   // saves compile time, because it doesn't require every single
2099   // stack slot reference to depend on the instruction that does the
2100   // modification.
2101   // Calls don't actually change the stack pointer, even if they have imp-defs.
2102   // No ARM calling conventions change the stack pointer. (X86 calling
2103   // conventions sometimes do).
2104   if (!MI.isCall() && MI.definesRegister(ARM::SP))
2105     return true;
2106 
2107   return false;
2108 }
2109 
2110 bool ARMBaseInstrInfo::
2111 isProfitableToIfCvt(MachineBasicBlock &MBB,
2112                     unsigned NumCycles, unsigned ExtraPredCycles,
2113                     BranchProbability Probability) const {
2114   if (!NumCycles)
2115     return false;
2116 
2117   // If we are optimizing for size, see if the branch in the predecessor can be
2118   // lowered to cbn?z by the constant island lowering pass, and return false if
2119   // so. This results in a shorter instruction sequence.
2120   if (MBB.getParent()->getFunction().hasOptSize()) {
2121     MachineBasicBlock *Pred = *MBB.pred_begin();
2122     if (!Pred->empty()) {
2123       MachineInstr *LastMI = &*Pred->rbegin();
2124       if (LastMI->getOpcode() == ARM::t2Bcc) {
2125         const TargetRegisterInfo *TRI = &getRegisterInfo();
2126         MachineInstr *CmpMI = findCMPToFoldIntoCBZ(LastMI, TRI);
2127         if (CmpMI)
2128           return false;
2129       }
2130     }
2131   }
2132   return isProfitableToIfCvt(MBB, NumCycles, ExtraPredCycles,
2133                              MBB, 0, 0, Probability);
2134 }
2135 
2136 bool ARMBaseInstrInfo::
2137 isProfitableToIfCvt(MachineBasicBlock &TBB,
2138                     unsigned TCycles, unsigned TExtra,
2139                     MachineBasicBlock &FBB,
2140                     unsigned FCycles, unsigned FExtra,
2141                     BranchProbability Probability) const {
2142   if (!TCycles)
2143     return false;
2144 
2145   // In thumb code we often end up trading one branch for a IT block, and
2146   // if we are cloning the instruction can increase code size. Prevent
2147   // blocks with multiple predecesors from being ifcvted to prevent this
2148   // cloning.
2149   if (Subtarget.isThumb2() && TBB.getParent()->getFunction().hasMinSize()) {
2150     if (TBB.pred_size() != 1 || FBB.pred_size() != 1)
2151       return false;
2152   }
2153 
2154   // Attempt to estimate the relative costs of predication versus branching.
2155   // Here we scale up each component of UnpredCost to avoid precision issue when
2156   // scaling TCycles/FCycles by Probability.
2157   const unsigned ScalingUpFactor = 1024;
2158 
2159   unsigned PredCost = (TCycles + FCycles + TExtra + FExtra) * ScalingUpFactor;
2160   unsigned UnpredCost;
2161   if (!Subtarget.hasBranchPredictor()) {
2162     // When we don't have a branch predictor it's always cheaper to not take a
2163     // branch than take it, so we have to take that into account.
2164     unsigned NotTakenBranchCost = 1;
2165     unsigned TakenBranchCost = Subtarget.getMispredictionPenalty();
2166     unsigned TUnpredCycles, FUnpredCycles;
2167     if (!FCycles) {
2168       // Triangle: TBB is the fallthrough
2169       TUnpredCycles = TCycles + NotTakenBranchCost;
2170       FUnpredCycles = TakenBranchCost;
2171     } else {
2172       // Diamond: TBB is the block that is branched to, FBB is the fallthrough
2173       TUnpredCycles = TCycles + TakenBranchCost;
2174       FUnpredCycles = FCycles + NotTakenBranchCost;
2175       // The branch at the end of FBB will disappear when it's predicated, so
2176       // discount it from PredCost.
2177       PredCost -= 1 * ScalingUpFactor;
2178     }
2179     // The total cost is the cost of each path scaled by their probabilites
2180     unsigned TUnpredCost = Probability.scale(TUnpredCycles * ScalingUpFactor);
2181     unsigned FUnpredCost = Probability.getCompl().scale(FUnpredCycles * ScalingUpFactor);
2182     UnpredCost = TUnpredCost + FUnpredCost;
2183     // When predicating assume that the first IT can be folded away but later
2184     // ones cost one cycle each
2185     if (Subtarget.isThumb2() && TCycles + FCycles > 4) {
2186       PredCost += ((TCycles + FCycles - 4) / 4) * ScalingUpFactor;
2187     }
2188   } else {
2189     unsigned TUnpredCost = Probability.scale(TCycles * ScalingUpFactor);
2190     unsigned FUnpredCost =
2191       Probability.getCompl().scale(FCycles * ScalingUpFactor);
2192     UnpredCost = TUnpredCost + FUnpredCost;
2193     UnpredCost += 1 * ScalingUpFactor; // The branch itself
2194     UnpredCost += Subtarget.getMispredictionPenalty() * ScalingUpFactor / 10;
2195   }
2196 
2197   return PredCost <= UnpredCost;
2198 }
2199 
2200 unsigned
2201 ARMBaseInstrInfo::extraSizeToPredicateInstructions(const MachineFunction &MF,
2202                                                    unsigned NumInsts) const {
2203   // Thumb2 needs a 2-byte IT instruction to predicate up to 4 instructions.
2204   // ARM has a condition code field in every predicable instruction, using it
2205   // doesn't change code size.
2206   if (!Subtarget.isThumb2())
2207     return 0;
2208 
2209   // It's possible that the size of the IT is restricted to a single block.
2210   unsigned MaxInsts = Subtarget.restrictIT() ? 1 : 4;
2211   return divideCeil(NumInsts, MaxInsts) * 2;
2212 }
2213 
2214 unsigned
2215 ARMBaseInstrInfo::predictBranchSizeForIfCvt(MachineInstr &MI) const {
2216   // If this branch is likely to be folded into the comparison to form a
2217   // CB(N)Z, then removing it won't reduce code size at all, because that will
2218   // just replace the CB(N)Z with a CMP.
2219   if (MI.getOpcode() == ARM::t2Bcc &&
2220       findCMPToFoldIntoCBZ(&MI, &getRegisterInfo()))
2221     return 0;
2222 
2223   unsigned Size = getInstSizeInBytes(MI);
2224 
2225   // For Thumb2, all branches are 32-bit instructions during the if conversion
2226   // pass, but may be replaced with 16-bit instructions during size reduction.
2227   // Since the branches considered by if conversion tend to be forward branches
2228   // over small basic blocks, they are very likely to be in range for the
2229   // narrow instructions, so we assume the final code size will be half what it
2230   // currently is.
2231   if (Subtarget.isThumb2())
2232     Size /= 2;
2233 
2234   return Size;
2235 }
2236 
2237 bool
2238 ARMBaseInstrInfo::isProfitableToUnpredicate(MachineBasicBlock &TMBB,
2239                                             MachineBasicBlock &FMBB) const {
2240   // Reduce false anti-dependencies to let the target's out-of-order execution
2241   // engine do its thing.
2242   return Subtarget.isProfitableToUnpredicate();
2243 }
2244 
2245 /// getInstrPredicate - If instruction is predicated, returns its predicate
2246 /// condition, otherwise returns AL. It also returns the condition code
2247 /// register by reference.
2248 ARMCC::CondCodes llvm::getInstrPredicate(const MachineInstr &MI,
2249                                          Register &PredReg) {
2250   int PIdx = MI.findFirstPredOperandIdx();
2251   if (PIdx == -1) {
2252     PredReg = 0;
2253     return ARMCC::AL;
2254   }
2255 
2256   PredReg = MI.getOperand(PIdx+1).getReg();
2257   return (ARMCC::CondCodes)MI.getOperand(PIdx).getImm();
2258 }
2259 
2260 unsigned llvm::getMatchingCondBranchOpcode(unsigned Opc) {
2261   if (Opc == ARM::B)
2262     return ARM::Bcc;
2263   if (Opc == ARM::tB)
2264     return ARM::tBcc;
2265   if (Opc == ARM::t2B)
2266     return ARM::t2Bcc;
2267 
2268   llvm_unreachable("Unknown unconditional branch opcode!");
2269 }
2270 
2271 MachineInstr *ARMBaseInstrInfo::commuteInstructionImpl(MachineInstr &MI,
2272                                                        bool NewMI,
2273                                                        unsigned OpIdx1,
2274                                                        unsigned OpIdx2) const {
2275   switch (MI.getOpcode()) {
2276   case ARM::MOVCCr:
2277   case ARM::t2MOVCCr: {
2278     // MOVCC can be commuted by inverting the condition.
2279     Register PredReg;
2280     ARMCC::CondCodes CC = getInstrPredicate(MI, PredReg);
2281     // MOVCC AL can't be inverted. Shouldn't happen.
2282     if (CC == ARMCC::AL || PredReg != ARM::CPSR)
2283       return nullptr;
2284     MachineInstr *CommutedMI =
2285         TargetInstrInfo::commuteInstructionImpl(MI, NewMI, OpIdx1, OpIdx2);
2286     if (!CommutedMI)
2287       return nullptr;
2288     // After swapping the MOVCC operands, also invert the condition.
2289     CommutedMI->getOperand(CommutedMI->findFirstPredOperandIdx())
2290         .setImm(ARMCC::getOppositeCondition(CC));
2291     return CommutedMI;
2292   }
2293   }
2294   return TargetInstrInfo::commuteInstructionImpl(MI, NewMI, OpIdx1, OpIdx2);
2295 }
2296 
2297 /// Identify instructions that can be folded into a MOVCC instruction, and
2298 /// return the defining instruction.
2299 MachineInstr *
2300 ARMBaseInstrInfo::canFoldIntoMOVCC(Register Reg, const MachineRegisterInfo &MRI,
2301                                    const TargetInstrInfo *TII) const {
2302   if (!Reg.isVirtual())
2303     return nullptr;
2304   if (!MRI.hasOneNonDBGUse(Reg))
2305     return nullptr;
2306   MachineInstr *MI = MRI.getVRegDef(Reg);
2307   if (!MI)
2308     return nullptr;
2309   // Check if MI can be predicated and folded into the MOVCC.
2310   if (!isPredicable(*MI))
2311     return nullptr;
2312   // Check if MI has any non-dead defs or physreg uses. This also detects
2313   // predicated instructions which will be reading CPSR.
2314   for (unsigned i = 1, e = MI->getNumOperands(); i != e; ++i) {
2315     const MachineOperand &MO = MI->getOperand(i);
2316     // Reject frame index operands, PEI can't handle the predicated pseudos.
2317     if (MO.isFI() || MO.isCPI() || MO.isJTI())
2318       return nullptr;
2319     if (!MO.isReg())
2320       continue;
2321     // MI can't have any tied operands, that would conflict with predication.
2322     if (MO.isTied())
2323       return nullptr;
2324     if (Register::isPhysicalRegister(MO.getReg()))
2325       return nullptr;
2326     if (MO.isDef() && !MO.isDead())
2327       return nullptr;
2328   }
2329   bool DontMoveAcrossStores = true;
2330   if (!MI->isSafeToMove(/* AliasAnalysis = */ nullptr, DontMoveAcrossStores))
2331     return nullptr;
2332   return MI;
2333 }
2334 
2335 bool ARMBaseInstrInfo::analyzeSelect(const MachineInstr &MI,
2336                                      SmallVectorImpl<MachineOperand> &Cond,
2337                                      unsigned &TrueOp, unsigned &FalseOp,
2338                                      bool &Optimizable) const {
2339   assert((MI.getOpcode() == ARM::MOVCCr || MI.getOpcode() == ARM::t2MOVCCr) &&
2340          "Unknown select instruction");
2341   // MOVCC operands:
2342   // 0: Def.
2343   // 1: True use.
2344   // 2: False use.
2345   // 3: Condition code.
2346   // 4: CPSR use.
2347   TrueOp = 1;
2348   FalseOp = 2;
2349   Cond.push_back(MI.getOperand(3));
2350   Cond.push_back(MI.getOperand(4));
2351   // We can always fold a def.
2352   Optimizable = true;
2353   return false;
2354 }
2355 
2356 MachineInstr *
2357 ARMBaseInstrInfo::optimizeSelect(MachineInstr &MI,
2358                                  SmallPtrSetImpl<MachineInstr *> &SeenMIs,
2359                                  bool PreferFalse) const {
2360   assert((MI.getOpcode() == ARM::MOVCCr || MI.getOpcode() == ARM::t2MOVCCr) &&
2361          "Unknown select instruction");
2362   MachineRegisterInfo &MRI = MI.getParent()->getParent()->getRegInfo();
2363   MachineInstr *DefMI = canFoldIntoMOVCC(MI.getOperand(2).getReg(), MRI, this);
2364   bool Invert = !DefMI;
2365   if (!DefMI)
2366     DefMI = canFoldIntoMOVCC(MI.getOperand(1).getReg(), MRI, this);
2367   if (!DefMI)
2368     return nullptr;
2369 
2370   // Find new register class to use.
2371   MachineOperand FalseReg = MI.getOperand(Invert ? 2 : 1);
2372   Register DestReg = MI.getOperand(0).getReg();
2373   const TargetRegisterClass *PreviousClass = MRI.getRegClass(FalseReg.getReg());
2374   if (!MRI.constrainRegClass(DestReg, PreviousClass))
2375     return nullptr;
2376 
2377   // Create a new predicated version of DefMI.
2378   // Rfalse is the first use.
2379   MachineInstrBuilder NewMI =
2380       BuildMI(*MI.getParent(), MI, MI.getDebugLoc(), DefMI->getDesc(), DestReg);
2381 
2382   // Copy all the DefMI operands, excluding its (null) predicate.
2383   const MCInstrDesc &DefDesc = DefMI->getDesc();
2384   for (unsigned i = 1, e = DefDesc.getNumOperands();
2385        i != e && !DefDesc.OpInfo[i].isPredicate(); ++i)
2386     NewMI.add(DefMI->getOperand(i));
2387 
2388   unsigned CondCode = MI.getOperand(3).getImm();
2389   if (Invert)
2390     NewMI.addImm(ARMCC::getOppositeCondition(ARMCC::CondCodes(CondCode)));
2391   else
2392     NewMI.addImm(CondCode);
2393   NewMI.add(MI.getOperand(4));
2394 
2395   // DefMI is not the -S version that sets CPSR, so add an optional %noreg.
2396   if (NewMI->hasOptionalDef())
2397     NewMI.add(condCodeOp());
2398 
2399   // The output register value when the predicate is false is an implicit
2400   // register operand tied to the first def.
2401   // The tie makes the register allocator ensure the FalseReg is allocated the
2402   // same register as operand 0.
2403   FalseReg.setImplicit();
2404   NewMI.add(FalseReg);
2405   NewMI->tieOperands(0, NewMI->getNumOperands() - 1);
2406 
2407   // Update SeenMIs set: register newly created MI and erase removed DefMI.
2408   SeenMIs.insert(NewMI);
2409   SeenMIs.erase(DefMI);
2410 
2411   // If MI is inside a loop, and DefMI is outside the loop, then kill flags on
2412   // DefMI would be invalid when tranferred inside the loop.  Checking for a
2413   // loop is expensive, but at least remove kill flags if they are in different
2414   // BBs.
2415   if (DefMI->getParent() != MI.getParent())
2416     NewMI->clearKillInfo();
2417 
2418   // The caller will erase MI, but not DefMI.
2419   DefMI->eraseFromParent();
2420   return NewMI;
2421 }
2422 
2423 /// Map pseudo instructions that imply an 'S' bit onto real opcodes. Whether the
2424 /// instruction is encoded with an 'S' bit is determined by the optional CPSR
2425 /// def operand.
2426 ///
2427 /// This will go away once we can teach tblgen how to set the optional CPSR def
2428 /// operand itself.
2429 struct AddSubFlagsOpcodePair {
2430   uint16_t PseudoOpc;
2431   uint16_t MachineOpc;
2432 };
2433 
2434 static const AddSubFlagsOpcodePair AddSubFlagsOpcodeMap[] = {
2435   {ARM::ADDSri, ARM::ADDri},
2436   {ARM::ADDSrr, ARM::ADDrr},
2437   {ARM::ADDSrsi, ARM::ADDrsi},
2438   {ARM::ADDSrsr, ARM::ADDrsr},
2439 
2440   {ARM::SUBSri, ARM::SUBri},
2441   {ARM::SUBSrr, ARM::SUBrr},
2442   {ARM::SUBSrsi, ARM::SUBrsi},
2443   {ARM::SUBSrsr, ARM::SUBrsr},
2444 
2445   {ARM::RSBSri, ARM::RSBri},
2446   {ARM::RSBSrsi, ARM::RSBrsi},
2447   {ARM::RSBSrsr, ARM::RSBrsr},
2448 
2449   {ARM::tADDSi3, ARM::tADDi3},
2450   {ARM::tADDSi8, ARM::tADDi8},
2451   {ARM::tADDSrr, ARM::tADDrr},
2452   {ARM::tADCS, ARM::tADC},
2453 
2454   {ARM::tSUBSi3, ARM::tSUBi3},
2455   {ARM::tSUBSi8, ARM::tSUBi8},
2456   {ARM::tSUBSrr, ARM::tSUBrr},
2457   {ARM::tSBCS, ARM::tSBC},
2458   {ARM::tRSBS, ARM::tRSB},
2459   {ARM::tLSLSri, ARM::tLSLri},
2460 
2461   {ARM::t2ADDSri, ARM::t2ADDri},
2462   {ARM::t2ADDSrr, ARM::t2ADDrr},
2463   {ARM::t2ADDSrs, ARM::t2ADDrs},
2464 
2465   {ARM::t2SUBSri, ARM::t2SUBri},
2466   {ARM::t2SUBSrr, ARM::t2SUBrr},
2467   {ARM::t2SUBSrs, ARM::t2SUBrs},
2468 
2469   {ARM::t2RSBSri, ARM::t2RSBri},
2470   {ARM::t2RSBSrs, ARM::t2RSBrs},
2471 };
2472 
2473 unsigned llvm::convertAddSubFlagsOpcode(unsigned OldOpc) {
2474   for (unsigned i = 0, e = array_lengthof(AddSubFlagsOpcodeMap); i != e; ++i)
2475     if (OldOpc == AddSubFlagsOpcodeMap[i].PseudoOpc)
2476       return AddSubFlagsOpcodeMap[i].MachineOpc;
2477   return 0;
2478 }
2479 
2480 void llvm::emitARMRegPlusImmediate(MachineBasicBlock &MBB,
2481                                    MachineBasicBlock::iterator &MBBI,
2482                                    const DebugLoc &dl, Register DestReg,
2483                                    Register BaseReg, int NumBytes,
2484                                    ARMCC::CondCodes Pred, Register PredReg,
2485                                    const ARMBaseInstrInfo &TII,
2486                                    unsigned MIFlags) {
2487   if (NumBytes == 0 && DestReg != BaseReg) {
2488     BuildMI(MBB, MBBI, dl, TII.get(ARM::MOVr), DestReg)
2489         .addReg(BaseReg, RegState::Kill)
2490         .add(predOps(Pred, PredReg))
2491         .add(condCodeOp())
2492         .setMIFlags(MIFlags);
2493     return;
2494   }
2495 
2496   bool isSub = NumBytes < 0;
2497   if (isSub) NumBytes = -NumBytes;
2498 
2499   while (NumBytes) {
2500     unsigned RotAmt = ARM_AM::getSOImmValRotate(NumBytes);
2501     unsigned ThisVal = NumBytes & ARM_AM::rotr32(0xFF, RotAmt);
2502     assert(ThisVal && "Didn't extract field correctly");
2503 
2504     // We will handle these bits from offset, clear them.
2505     NumBytes &= ~ThisVal;
2506 
2507     assert(ARM_AM::getSOImmVal(ThisVal) != -1 && "Bit extraction didn't work?");
2508 
2509     // Build the new ADD / SUB.
2510     unsigned Opc = isSub ? ARM::SUBri : ARM::ADDri;
2511     BuildMI(MBB, MBBI, dl, TII.get(Opc), DestReg)
2512         .addReg(BaseReg, RegState::Kill)
2513         .addImm(ThisVal)
2514         .add(predOps(Pred, PredReg))
2515         .add(condCodeOp())
2516         .setMIFlags(MIFlags);
2517     BaseReg = DestReg;
2518   }
2519 }
2520 
2521 bool llvm::tryFoldSPUpdateIntoPushPop(const ARMSubtarget &Subtarget,
2522                                       MachineFunction &MF, MachineInstr *MI,
2523                                       unsigned NumBytes) {
2524   // This optimisation potentially adds lots of load and store
2525   // micro-operations, it's only really a great benefit to code-size.
2526   if (!Subtarget.hasMinSize())
2527     return false;
2528 
2529   // If only one register is pushed/popped, LLVM can use an LDR/STR
2530   // instead. We can't modify those so make sure we're dealing with an
2531   // instruction we understand.
2532   bool IsPop = isPopOpcode(MI->getOpcode());
2533   bool IsPush = isPushOpcode(MI->getOpcode());
2534   if (!IsPush && !IsPop)
2535     return false;
2536 
2537   bool IsVFPPushPop = MI->getOpcode() == ARM::VSTMDDB_UPD ||
2538                       MI->getOpcode() == ARM::VLDMDIA_UPD;
2539   bool IsT1PushPop = MI->getOpcode() == ARM::tPUSH ||
2540                      MI->getOpcode() == ARM::tPOP ||
2541                      MI->getOpcode() == ARM::tPOP_RET;
2542 
2543   assert((IsT1PushPop || (MI->getOperand(0).getReg() == ARM::SP &&
2544                           MI->getOperand(1).getReg() == ARM::SP)) &&
2545          "trying to fold sp update into non-sp-updating push/pop");
2546 
2547   // The VFP push & pop act on D-registers, so we can only fold an adjustment
2548   // by a multiple of 8 bytes in correctly. Similarly rN is 4-bytes. Don't try
2549   // if this is violated.
2550   if (NumBytes % (IsVFPPushPop ? 8 : 4) != 0)
2551     return false;
2552 
2553   // ARM and Thumb2 push/pop insts have explicit "sp, sp" operands (+
2554   // pred) so the list starts at 4. Thumb1 starts after the predicate.
2555   int RegListIdx = IsT1PushPop ? 2 : 4;
2556 
2557   // Calculate the space we'll need in terms of registers.
2558   unsigned RegsNeeded;
2559   const TargetRegisterClass *RegClass;
2560   if (IsVFPPushPop) {
2561     RegsNeeded = NumBytes / 8;
2562     RegClass = &ARM::DPRRegClass;
2563   } else {
2564     RegsNeeded = NumBytes / 4;
2565     RegClass = &ARM::GPRRegClass;
2566   }
2567 
2568   // We're going to have to strip all list operands off before
2569   // re-adding them since the order matters, so save the existing ones
2570   // for later.
2571   SmallVector<MachineOperand, 4> RegList;
2572 
2573   // We're also going to need the first register transferred by this
2574   // instruction, which won't necessarily be the first register in the list.
2575   unsigned FirstRegEnc = -1;
2576 
2577   const TargetRegisterInfo *TRI = MF.getRegInfo().getTargetRegisterInfo();
2578   for (int i = MI->getNumOperands() - 1; i >= RegListIdx; --i) {
2579     MachineOperand &MO = MI->getOperand(i);
2580     RegList.push_back(MO);
2581 
2582     if (MO.isReg() && !MO.isImplicit() &&
2583         TRI->getEncodingValue(MO.getReg()) < FirstRegEnc)
2584       FirstRegEnc = TRI->getEncodingValue(MO.getReg());
2585   }
2586 
2587   const MCPhysReg *CSRegs = TRI->getCalleeSavedRegs(&MF);
2588 
2589   // Now try to find enough space in the reglist to allocate NumBytes.
2590   for (int CurRegEnc = FirstRegEnc - 1; CurRegEnc >= 0 && RegsNeeded;
2591        --CurRegEnc) {
2592     unsigned CurReg = RegClass->getRegister(CurRegEnc);
2593     if (IsT1PushPop && CurRegEnc > TRI->getEncodingValue(ARM::R7))
2594       continue;
2595     if (!IsPop) {
2596       // Pushing any register is completely harmless, mark the register involved
2597       // as undef since we don't care about its value and must not restore it
2598       // during stack unwinding.
2599       RegList.push_back(MachineOperand::CreateReg(CurReg, false, false,
2600                                                   false, false, true));
2601       --RegsNeeded;
2602       continue;
2603     }
2604 
2605     // However, we can only pop an extra register if it's not live. For
2606     // registers live within the function we might clobber a return value
2607     // register; the other way a register can be live here is if it's
2608     // callee-saved.
2609     if (isCalleeSavedRegister(CurReg, CSRegs) ||
2610         MI->getParent()->computeRegisterLiveness(TRI, CurReg, MI) !=
2611         MachineBasicBlock::LQR_Dead) {
2612       // VFP pops don't allow holes in the register list, so any skip is fatal
2613       // for our transformation. GPR pops do, so we should just keep looking.
2614       if (IsVFPPushPop)
2615         return false;
2616       else
2617         continue;
2618     }
2619 
2620     // Mark the unimportant registers as <def,dead> in the POP.
2621     RegList.push_back(MachineOperand::CreateReg(CurReg, true, false, false,
2622                                                 true));
2623     --RegsNeeded;
2624   }
2625 
2626   if (RegsNeeded > 0)
2627     return false;
2628 
2629   // Finally we know we can profitably perform the optimisation so go
2630   // ahead: strip all existing registers off and add them back again
2631   // in the right order.
2632   for (int i = MI->getNumOperands() - 1; i >= RegListIdx; --i)
2633     MI->RemoveOperand(i);
2634 
2635   // Add the complete list back in.
2636   MachineInstrBuilder MIB(MF, &*MI);
2637   for (int i = RegList.size() - 1; i >= 0; --i)
2638     MIB.add(RegList[i]);
2639 
2640   return true;
2641 }
2642 
2643 bool llvm::rewriteARMFrameIndex(MachineInstr &MI, unsigned FrameRegIdx,
2644                                 Register FrameReg, int &Offset,
2645                                 const ARMBaseInstrInfo &TII) {
2646   unsigned Opcode = MI.getOpcode();
2647   const MCInstrDesc &Desc = MI.getDesc();
2648   unsigned AddrMode = (Desc.TSFlags & ARMII::AddrModeMask);
2649   bool isSub = false;
2650 
2651   // Memory operands in inline assembly always use AddrMode2.
2652   if (Opcode == ARM::INLINEASM || Opcode == ARM::INLINEASM_BR)
2653     AddrMode = ARMII::AddrMode2;
2654 
2655   if (Opcode == ARM::ADDri) {
2656     Offset += MI.getOperand(FrameRegIdx+1).getImm();
2657     if (Offset == 0) {
2658       // Turn it into a move.
2659       MI.setDesc(TII.get(ARM::MOVr));
2660       MI.getOperand(FrameRegIdx).ChangeToRegister(FrameReg, false);
2661       MI.RemoveOperand(FrameRegIdx+1);
2662       Offset = 0;
2663       return true;
2664     } else if (Offset < 0) {
2665       Offset = -Offset;
2666       isSub = true;
2667       MI.setDesc(TII.get(ARM::SUBri));
2668     }
2669 
2670     // Common case: small offset, fits into instruction.
2671     if (ARM_AM::getSOImmVal(Offset) != -1) {
2672       // Replace the FrameIndex with sp / fp
2673       MI.getOperand(FrameRegIdx).ChangeToRegister(FrameReg, false);
2674       MI.getOperand(FrameRegIdx+1).ChangeToImmediate(Offset);
2675       Offset = 0;
2676       return true;
2677     }
2678 
2679     // Otherwise, pull as much of the immedidate into this ADDri/SUBri
2680     // as possible.
2681     unsigned RotAmt = ARM_AM::getSOImmValRotate(Offset);
2682     unsigned ThisImmVal = Offset & ARM_AM::rotr32(0xFF, RotAmt);
2683 
2684     // We will handle these bits from offset, clear them.
2685     Offset &= ~ThisImmVal;
2686 
2687     // Get the properly encoded SOImmVal field.
2688     assert(ARM_AM::getSOImmVal(ThisImmVal) != -1 &&
2689            "Bit extraction didn't work?");
2690     MI.getOperand(FrameRegIdx+1).ChangeToImmediate(ThisImmVal);
2691  } else {
2692     unsigned ImmIdx = 0;
2693     int InstrOffs = 0;
2694     unsigned NumBits = 0;
2695     unsigned Scale = 1;
2696     switch (AddrMode) {
2697     case ARMII::AddrMode_i12:
2698       ImmIdx = FrameRegIdx + 1;
2699       InstrOffs = MI.getOperand(ImmIdx).getImm();
2700       NumBits = 12;
2701       break;
2702     case ARMII::AddrMode2:
2703       ImmIdx = FrameRegIdx+2;
2704       InstrOffs = ARM_AM::getAM2Offset(MI.getOperand(ImmIdx).getImm());
2705       if (ARM_AM::getAM2Op(MI.getOperand(ImmIdx).getImm()) == ARM_AM::sub)
2706         InstrOffs *= -1;
2707       NumBits = 12;
2708       break;
2709     case ARMII::AddrMode3:
2710       ImmIdx = FrameRegIdx+2;
2711       InstrOffs = ARM_AM::getAM3Offset(MI.getOperand(ImmIdx).getImm());
2712       if (ARM_AM::getAM3Op(MI.getOperand(ImmIdx).getImm()) == ARM_AM::sub)
2713         InstrOffs *= -1;
2714       NumBits = 8;
2715       break;
2716     case ARMII::AddrMode4:
2717     case ARMII::AddrMode6:
2718       // Can't fold any offset even if it's zero.
2719       return false;
2720     case ARMII::AddrMode5:
2721       ImmIdx = FrameRegIdx+1;
2722       InstrOffs = ARM_AM::getAM5Offset(MI.getOperand(ImmIdx).getImm());
2723       if (ARM_AM::getAM5Op(MI.getOperand(ImmIdx).getImm()) == ARM_AM::sub)
2724         InstrOffs *= -1;
2725       NumBits = 8;
2726       Scale = 4;
2727       break;
2728     case ARMII::AddrMode5FP16:
2729       ImmIdx = FrameRegIdx+1;
2730       InstrOffs = ARM_AM::getAM5Offset(MI.getOperand(ImmIdx).getImm());
2731       if (ARM_AM::getAM5Op(MI.getOperand(ImmIdx).getImm()) == ARM_AM::sub)
2732         InstrOffs *= -1;
2733       NumBits = 8;
2734       Scale = 2;
2735       break;
2736     case ARMII::AddrModeT2_i7:
2737     case ARMII::AddrModeT2_i7s2:
2738     case ARMII::AddrModeT2_i7s4:
2739       ImmIdx = FrameRegIdx+1;
2740       InstrOffs = MI.getOperand(ImmIdx).getImm();
2741       NumBits = 7;
2742       Scale = (AddrMode == ARMII::AddrModeT2_i7s2 ? 2 :
2743                AddrMode == ARMII::AddrModeT2_i7s4 ? 4 : 1);
2744       break;
2745     default:
2746       llvm_unreachable("Unsupported addressing mode!");
2747     }
2748 
2749     Offset += InstrOffs * Scale;
2750     assert((Offset & (Scale-1)) == 0 && "Can't encode this offset!");
2751     if (Offset < 0) {
2752       Offset = -Offset;
2753       isSub = true;
2754     }
2755 
2756     // Attempt to fold address comp. if opcode has offset bits
2757     if (NumBits > 0) {
2758       // Common case: small offset, fits into instruction.
2759       MachineOperand &ImmOp = MI.getOperand(ImmIdx);
2760       int ImmedOffset = Offset / Scale;
2761       unsigned Mask = (1 << NumBits) - 1;
2762       if ((unsigned)Offset <= Mask * Scale) {
2763         // Replace the FrameIndex with sp
2764         MI.getOperand(FrameRegIdx).ChangeToRegister(FrameReg, false);
2765         // FIXME: When addrmode2 goes away, this will simplify (like the
2766         // T2 version), as the LDR.i12 versions don't need the encoding
2767         // tricks for the offset value.
2768         if (isSub) {
2769           if (AddrMode == ARMII::AddrMode_i12)
2770             ImmedOffset = -ImmedOffset;
2771           else
2772             ImmedOffset |= 1 << NumBits;
2773         }
2774         ImmOp.ChangeToImmediate(ImmedOffset);
2775         Offset = 0;
2776         return true;
2777       }
2778 
2779       // Otherwise, it didn't fit. Pull in what we can to simplify the immed.
2780       ImmedOffset = ImmedOffset & Mask;
2781       if (isSub) {
2782         if (AddrMode == ARMII::AddrMode_i12)
2783           ImmedOffset = -ImmedOffset;
2784         else
2785           ImmedOffset |= 1 << NumBits;
2786       }
2787       ImmOp.ChangeToImmediate(ImmedOffset);
2788       Offset &= ~(Mask*Scale);
2789     }
2790   }
2791 
2792   Offset = (isSub) ? -Offset : Offset;
2793   return Offset == 0;
2794 }
2795 
2796 /// analyzeCompare - For a comparison instruction, return the source registers
2797 /// in SrcReg and SrcReg2 if having two register operands, and the value it
2798 /// compares against in CmpValue. Return true if the comparison instruction
2799 /// can be analyzed.
2800 bool ARMBaseInstrInfo::analyzeCompare(const MachineInstr &MI, Register &SrcReg,
2801                                       Register &SrcReg2, int64_t &CmpMask,
2802                                       int64_t &CmpValue) const {
2803   switch (MI.getOpcode()) {
2804   default: break;
2805   case ARM::CMPri:
2806   case ARM::t2CMPri:
2807   case ARM::tCMPi8:
2808     SrcReg = MI.getOperand(0).getReg();
2809     SrcReg2 = 0;
2810     CmpMask = ~0;
2811     CmpValue = MI.getOperand(1).getImm();
2812     return true;
2813   case ARM::CMPrr:
2814   case ARM::t2CMPrr:
2815   case ARM::tCMPr:
2816     SrcReg = MI.getOperand(0).getReg();
2817     SrcReg2 = MI.getOperand(1).getReg();
2818     CmpMask = ~0;
2819     CmpValue = 0;
2820     return true;
2821   case ARM::TSTri:
2822   case ARM::t2TSTri:
2823     SrcReg = MI.getOperand(0).getReg();
2824     SrcReg2 = 0;
2825     CmpMask = MI.getOperand(1).getImm();
2826     CmpValue = 0;
2827     return true;
2828   }
2829 
2830   return false;
2831 }
2832 
2833 /// isSuitableForMask - Identify a suitable 'and' instruction that
2834 /// operates on the given source register and applies the same mask
2835 /// as a 'tst' instruction. Provide a limited look-through for copies.
2836 /// When successful, MI will hold the found instruction.
2837 static bool isSuitableForMask(MachineInstr *&MI, Register SrcReg,
2838                               int CmpMask, bool CommonUse) {
2839   switch (MI->getOpcode()) {
2840     case ARM::ANDri:
2841     case ARM::t2ANDri:
2842       if (CmpMask != MI->getOperand(2).getImm())
2843         return false;
2844       if (SrcReg == MI->getOperand(CommonUse ? 1 : 0).getReg())
2845         return true;
2846       break;
2847   }
2848 
2849   return false;
2850 }
2851 
2852 /// getCmpToAddCondition - assume the flags are set by CMP(a,b), return
2853 /// the condition code if we modify the instructions such that flags are
2854 /// set by ADD(a,b,X).
2855 inline static ARMCC::CondCodes getCmpToAddCondition(ARMCC::CondCodes CC) {
2856   switch (CC) {
2857   default: return ARMCC::AL;
2858   case ARMCC::HS: return ARMCC::LO;
2859   case ARMCC::LO: return ARMCC::HS;
2860   case ARMCC::VS: return ARMCC::VS;
2861   case ARMCC::VC: return ARMCC::VC;
2862   }
2863 }
2864 
2865 /// isRedundantFlagInstr - check whether the first instruction, whose only
2866 /// purpose is to update flags, can be made redundant.
2867 /// CMPrr can be made redundant by SUBrr if the operands are the same.
2868 /// CMPri can be made redundant by SUBri if the operands are the same.
2869 /// CMPrr(r0, r1) can be made redundant by ADDr[ri](r0, r1, X).
2870 /// This function can be extended later on.
2871 inline static bool isRedundantFlagInstr(const MachineInstr *CmpI,
2872                                         Register SrcReg, Register SrcReg2,
2873                                         int64_t ImmValue,
2874                                         const MachineInstr *OI,
2875                                         bool &IsThumb1) {
2876   if ((CmpI->getOpcode() == ARM::CMPrr || CmpI->getOpcode() == ARM::t2CMPrr) &&
2877       (OI->getOpcode() == ARM::SUBrr || OI->getOpcode() == ARM::t2SUBrr) &&
2878       ((OI->getOperand(1).getReg() == SrcReg &&
2879         OI->getOperand(2).getReg() == SrcReg2) ||
2880        (OI->getOperand(1).getReg() == SrcReg2 &&
2881         OI->getOperand(2).getReg() == SrcReg))) {
2882     IsThumb1 = false;
2883     return true;
2884   }
2885 
2886   if (CmpI->getOpcode() == ARM::tCMPr && OI->getOpcode() == ARM::tSUBrr &&
2887       ((OI->getOperand(2).getReg() == SrcReg &&
2888         OI->getOperand(3).getReg() == SrcReg2) ||
2889        (OI->getOperand(2).getReg() == SrcReg2 &&
2890         OI->getOperand(3).getReg() == SrcReg))) {
2891     IsThumb1 = true;
2892     return true;
2893   }
2894 
2895   if ((CmpI->getOpcode() == ARM::CMPri || CmpI->getOpcode() == ARM::t2CMPri) &&
2896       (OI->getOpcode() == ARM::SUBri || OI->getOpcode() == ARM::t2SUBri) &&
2897       OI->getOperand(1).getReg() == SrcReg &&
2898       OI->getOperand(2).getImm() == ImmValue) {
2899     IsThumb1 = false;
2900     return true;
2901   }
2902 
2903   if (CmpI->getOpcode() == ARM::tCMPi8 &&
2904       (OI->getOpcode() == ARM::tSUBi8 || OI->getOpcode() == ARM::tSUBi3) &&
2905       OI->getOperand(2).getReg() == SrcReg &&
2906       OI->getOperand(3).getImm() == ImmValue) {
2907     IsThumb1 = true;
2908     return true;
2909   }
2910 
2911   if ((CmpI->getOpcode() == ARM::CMPrr || CmpI->getOpcode() == ARM::t2CMPrr) &&
2912       (OI->getOpcode() == ARM::ADDrr || OI->getOpcode() == ARM::t2ADDrr ||
2913        OI->getOpcode() == ARM::ADDri || OI->getOpcode() == ARM::t2ADDri) &&
2914       OI->getOperand(0).isReg() && OI->getOperand(1).isReg() &&
2915       OI->getOperand(0).getReg() == SrcReg &&
2916       OI->getOperand(1).getReg() == SrcReg2) {
2917     IsThumb1 = false;
2918     return true;
2919   }
2920 
2921   if (CmpI->getOpcode() == ARM::tCMPr &&
2922       (OI->getOpcode() == ARM::tADDi3 || OI->getOpcode() == ARM::tADDi8 ||
2923        OI->getOpcode() == ARM::tADDrr) &&
2924       OI->getOperand(0).getReg() == SrcReg &&
2925       OI->getOperand(2).getReg() == SrcReg2) {
2926     IsThumb1 = true;
2927     return true;
2928   }
2929 
2930   return false;
2931 }
2932 
2933 static bool isOptimizeCompareCandidate(MachineInstr *MI, bool &IsThumb1) {
2934   switch (MI->getOpcode()) {
2935   default: return false;
2936   case ARM::tLSLri:
2937   case ARM::tLSRri:
2938   case ARM::tLSLrr:
2939   case ARM::tLSRrr:
2940   case ARM::tSUBrr:
2941   case ARM::tADDrr:
2942   case ARM::tADDi3:
2943   case ARM::tADDi8:
2944   case ARM::tSUBi3:
2945   case ARM::tSUBi8:
2946   case ARM::tMUL:
2947   case ARM::tADC:
2948   case ARM::tSBC:
2949   case ARM::tRSB:
2950   case ARM::tAND:
2951   case ARM::tORR:
2952   case ARM::tEOR:
2953   case ARM::tBIC:
2954   case ARM::tMVN:
2955   case ARM::tASRri:
2956   case ARM::tASRrr:
2957   case ARM::tROR:
2958     IsThumb1 = true;
2959     LLVM_FALLTHROUGH;
2960   case ARM::RSBrr:
2961   case ARM::RSBri:
2962   case ARM::RSCrr:
2963   case ARM::RSCri:
2964   case ARM::ADDrr:
2965   case ARM::ADDri:
2966   case ARM::ADCrr:
2967   case ARM::ADCri:
2968   case ARM::SUBrr:
2969   case ARM::SUBri:
2970   case ARM::SBCrr:
2971   case ARM::SBCri:
2972   case ARM::t2RSBri:
2973   case ARM::t2ADDrr:
2974   case ARM::t2ADDri:
2975   case ARM::t2ADCrr:
2976   case ARM::t2ADCri:
2977   case ARM::t2SUBrr:
2978   case ARM::t2SUBri:
2979   case ARM::t2SBCrr:
2980   case ARM::t2SBCri:
2981   case ARM::ANDrr:
2982   case ARM::ANDri:
2983   case ARM::t2ANDrr:
2984   case ARM::t2ANDri:
2985   case ARM::ORRrr:
2986   case ARM::ORRri:
2987   case ARM::t2ORRrr:
2988   case ARM::t2ORRri:
2989   case ARM::EORrr:
2990   case ARM::EORri:
2991   case ARM::t2EORrr:
2992   case ARM::t2EORri:
2993   case ARM::t2LSRri:
2994   case ARM::t2LSRrr:
2995   case ARM::t2LSLri:
2996   case ARM::t2LSLrr:
2997     return true;
2998   }
2999 }
3000 
3001 /// optimizeCompareInstr - Convert the instruction supplying the argument to the
3002 /// comparison into one that sets the zero bit in the flags register;
3003 /// Remove a redundant Compare instruction if an earlier instruction can set the
3004 /// flags in the same way as Compare.
3005 /// E.g. SUBrr(r1,r2) and CMPrr(r1,r2). We also handle the case where two
3006 /// operands are swapped: SUBrr(r1,r2) and CMPrr(r2,r1), by updating the
3007 /// condition code of instructions which use the flags.
3008 bool ARMBaseInstrInfo::optimizeCompareInstr(
3009     MachineInstr &CmpInstr, Register SrcReg, Register SrcReg2, int64_t CmpMask,
3010     int64_t CmpValue, const MachineRegisterInfo *MRI) const {
3011   // Get the unique definition of SrcReg.
3012   MachineInstr *MI = MRI->getUniqueVRegDef(SrcReg);
3013   if (!MI) return false;
3014 
3015   // Masked compares sometimes use the same register as the corresponding 'and'.
3016   if (CmpMask != ~0) {
3017     if (!isSuitableForMask(MI, SrcReg, CmpMask, false) || isPredicated(*MI)) {
3018       MI = nullptr;
3019       for (MachineRegisterInfo::use_instr_iterator
3020            UI = MRI->use_instr_begin(SrcReg), UE = MRI->use_instr_end();
3021            UI != UE; ++UI) {
3022         if (UI->getParent() != CmpInstr.getParent())
3023           continue;
3024         MachineInstr *PotentialAND = &*UI;
3025         if (!isSuitableForMask(PotentialAND, SrcReg, CmpMask, true) ||
3026             isPredicated(*PotentialAND))
3027           continue;
3028         MI = PotentialAND;
3029         break;
3030       }
3031       if (!MI) return false;
3032     }
3033   }
3034 
3035   // Get ready to iterate backward from CmpInstr.
3036   MachineBasicBlock::iterator I = CmpInstr, E = MI,
3037                               B = CmpInstr.getParent()->begin();
3038 
3039   // Early exit if CmpInstr is at the beginning of the BB.
3040   if (I == B) return false;
3041 
3042   // There are two possible candidates which can be changed to set CPSR:
3043   // One is MI, the other is a SUB or ADD instruction.
3044   // For CMPrr(r1,r2), we are looking for SUB(r1,r2), SUB(r2,r1), or
3045   // ADDr[ri](r1, r2, X).
3046   // For CMPri(r1, CmpValue), we are looking for SUBri(r1, CmpValue).
3047   MachineInstr *SubAdd = nullptr;
3048   if (SrcReg2 != 0)
3049     // MI is not a candidate for CMPrr.
3050     MI = nullptr;
3051   else if (MI->getParent() != CmpInstr.getParent() || CmpValue != 0) {
3052     // Conservatively refuse to convert an instruction which isn't in the same
3053     // BB as the comparison.
3054     // For CMPri w/ CmpValue != 0, a SubAdd may still be a candidate.
3055     // Thus we cannot return here.
3056     if (CmpInstr.getOpcode() == ARM::CMPri ||
3057         CmpInstr.getOpcode() == ARM::t2CMPri ||
3058         CmpInstr.getOpcode() == ARM::tCMPi8)
3059       MI = nullptr;
3060     else
3061       return false;
3062   }
3063 
3064   bool IsThumb1 = false;
3065   if (MI && !isOptimizeCompareCandidate(MI, IsThumb1))
3066     return false;
3067 
3068   // We also want to do this peephole for cases like this: if (a*b == 0),
3069   // and optimise away the CMP instruction from the generated code sequence:
3070   // MULS, MOVS, MOVS, CMP. Here the MOVS instructions load the boolean values
3071   // resulting from the select instruction, but these MOVS instructions for
3072   // Thumb1 (V6M) are flag setting and are thus preventing this optimisation.
3073   // However, if we only have MOVS instructions in between the CMP and the
3074   // other instruction (the MULS in this example), then the CPSR is dead so we
3075   // can safely reorder the sequence into: MOVS, MOVS, MULS, CMP. We do this
3076   // reordering and then continue the analysis hoping we can eliminate the
3077   // CMP. This peephole works on the vregs, so is still in SSA form. As a
3078   // consequence, the movs won't redefine/kill the MUL operands which would
3079   // make this reordering illegal.
3080   const TargetRegisterInfo *TRI = &getRegisterInfo();
3081   if (MI && IsThumb1) {
3082     --I;
3083     if (I != E && !MI->readsRegister(ARM::CPSR, TRI)) {
3084       bool CanReorder = true;
3085       for (; I != E; --I) {
3086         if (I->getOpcode() != ARM::tMOVi8) {
3087           CanReorder = false;
3088           break;
3089         }
3090       }
3091       if (CanReorder) {
3092         MI = MI->removeFromParent();
3093         E = CmpInstr;
3094         CmpInstr.getParent()->insert(E, MI);
3095       }
3096     }
3097     I = CmpInstr;
3098     E = MI;
3099   }
3100 
3101   // Check that CPSR isn't set between the comparison instruction and the one we
3102   // want to change. At the same time, search for SubAdd.
3103   bool SubAddIsThumb1 = false;
3104   do {
3105     const MachineInstr &Instr = *--I;
3106 
3107     // Check whether CmpInstr can be made redundant by the current instruction.
3108     if (isRedundantFlagInstr(&CmpInstr, SrcReg, SrcReg2, CmpValue, &Instr,
3109                              SubAddIsThumb1)) {
3110       SubAdd = &*I;
3111       break;
3112     }
3113 
3114     // Allow E (which was initially MI) to be SubAdd but do not search before E.
3115     if (I == E)
3116       break;
3117 
3118     if (Instr.modifiesRegister(ARM::CPSR, TRI) ||
3119         Instr.readsRegister(ARM::CPSR, TRI))
3120       // This instruction modifies or uses CPSR after the one we want to
3121       // change. We can't do this transformation.
3122       return false;
3123 
3124     if (I == B) {
3125       // In some cases, we scan the use-list of an instruction for an AND;
3126       // that AND is in the same BB, but may not be scheduled before the
3127       // corresponding TST.  In that case, bail out.
3128       //
3129       // FIXME: We could try to reschedule the AND.
3130       return false;
3131     }
3132   } while (true);
3133 
3134   // Return false if no candidates exist.
3135   if (!MI && !SubAdd)
3136     return false;
3137 
3138   // If we found a SubAdd, use it as it will be closer to the CMP
3139   if (SubAdd) {
3140     MI = SubAdd;
3141     IsThumb1 = SubAddIsThumb1;
3142   }
3143 
3144   // We can't use a predicated instruction - it doesn't always write the flags.
3145   if (isPredicated(*MI))
3146     return false;
3147 
3148   // Scan forward for the use of CPSR
3149   // When checking against MI: if it's a conditional code that requires
3150   // checking of the V bit or C bit, then this is not safe to do.
3151   // It is safe to remove CmpInstr if CPSR is redefined or killed.
3152   // If we are done with the basic block, we need to check whether CPSR is
3153   // live-out.
3154   SmallVector<std::pair<MachineOperand*, ARMCC::CondCodes>, 4>
3155       OperandsToUpdate;
3156   bool isSafe = false;
3157   I = CmpInstr;
3158   E = CmpInstr.getParent()->end();
3159   while (!isSafe && ++I != E) {
3160     const MachineInstr &Instr = *I;
3161     for (unsigned IO = 0, EO = Instr.getNumOperands();
3162          !isSafe && IO != EO; ++IO) {
3163       const MachineOperand &MO = Instr.getOperand(IO);
3164       if (MO.isRegMask() && MO.clobbersPhysReg(ARM::CPSR)) {
3165         isSafe = true;
3166         break;
3167       }
3168       if (!MO.isReg() || MO.getReg() != ARM::CPSR)
3169         continue;
3170       if (MO.isDef()) {
3171         isSafe = true;
3172         break;
3173       }
3174       // Condition code is after the operand before CPSR except for VSELs.
3175       ARMCC::CondCodes CC;
3176       bool IsInstrVSel = true;
3177       switch (Instr.getOpcode()) {
3178       default:
3179         IsInstrVSel = false;
3180         CC = (ARMCC::CondCodes)Instr.getOperand(IO - 1).getImm();
3181         break;
3182       case ARM::VSELEQD:
3183       case ARM::VSELEQS:
3184       case ARM::VSELEQH:
3185         CC = ARMCC::EQ;
3186         break;
3187       case ARM::VSELGTD:
3188       case ARM::VSELGTS:
3189       case ARM::VSELGTH:
3190         CC = ARMCC::GT;
3191         break;
3192       case ARM::VSELGED:
3193       case ARM::VSELGES:
3194       case ARM::VSELGEH:
3195         CC = ARMCC::GE;
3196         break;
3197       case ARM::VSELVSD:
3198       case ARM::VSELVSS:
3199       case ARM::VSELVSH:
3200         CC = ARMCC::VS;
3201         break;
3202       }
3203 
3204       if (SubAdd) {
3205         // If we have SUB(r1, r2) and CMP(r2, r1), the condition code based
3206         // on CMP needs to be updated to be based on SUB.
3207         // If we have ADD(r1, r2, X) and CMP(r1, r2), the condition code also
3208         // needs to be modified.
3209         // Push the condition code operands to OperandsToUpdate.
3210         // If it is safe to remove CmpInstr, the condition code of these
3211         // operands will be modified.
3212         unsigned Opc = SubAdd->getOpcode();
3213         bool IsSub = Opc == ARM::SUBrr || Opc == ARM::t2SUBrr ||
3214                      Opc == ARM::SUBri || Opc == ARM::t2SUBri ||
3215                      Opc == ARM::tSUBrr || Opc == ARM::tSUBi3 ||
3216                      Opc == ARM::tSUBi8;
3217         unsigned OpI = Opc != ARM::tSUBrr ? 1 : 2;
3218         if (!IsSub ||
3219             (SrcReg2 != 0 && SubAdd->getOperand(OpI).getReg() == SrcReg2 &&
3220              SubAdd->getOperand(OpI + 1).getReg() == SrcReg)) {
3221           // VSel doesn't support condition code update.
3222           if (IsInstrVSel)
3223             return false;
3224           // Ensure we can swap the condition.
3225           ARMCC::CondCodes NewCC = (IsSub ? getSwappedCondition(CC) : getCmpToAddCondition(CC));
3226           if (NewCC == ARMCC::AL)
3227             return false;
3228           OperandsToUpdate.push_back(
3229               std::make_pair(&((*I).getOperand(IO - 1)), NewCC));
3230         }
3231       } else {
3232         // No SubAdd, so this is x = <op> y, z; cmp x, 0.
3233         switch (CC) {
3234         case ARMCC::EQ: // Z
3235         case ARMCC::NE: // Z
3236         case ARMCC::MI: // N
3237         case ARMCC::PL: // N
3238         case ARMCC::AL: // none
3239           // CPSR can be used multiple times, we should continue.
3240           break;
3241         case ARMCC::HS: // C
3242         case ARMCC::LO: // C
3243         case ARMCC::VS: // V
3244         case ARMCC::VC: // V
3245         case ARMCC::HI: // C Z
3246         case ARMCC::LS: // C Z
3247         case ARMCC::GE: // N V
3248         case ARMCC::LT: // N V
3249         case ARMCC::GT: // Z N V
3250         case ARMCC::LE: // Z N V
3251           // The instruction uses the V bit or C bit which is not safe.
3252           return false;
3253         }
3254       }
3255     }
3256   }
3257 
3258   // If CPSR is not killed nor re-defined, we should check whether it is
3259   // live-out. If it is live-out, do not optimize.
3260   if (!isSafe) {
3261     MachineBasicBlock *MBB = CmpInstr.getParent();
3262     for (MachineBasicBlock::succ_iterator SI = MBB->succ_begin(),
3263              SE = MBB->succ_end(); SI != SE; ++SI)
3264       if ((*SI)->isLiveIn(ARM::CPSR))
3265         return false;
3266   }
3267 
3268   // Toggle the optional operand to CPSR (if it exists - in Thumb1 we always
3269   // set CPSR so this is represented as an explicit output)
3270   if (!IsThumb1) {
3271     MI->getOperand(5).setReg(ARM::CPSR);
3272     MI->getOperand(5).setIsDef(true);
3273   }
3274   assert(!isPredicated(*MI) && "Can't use flags from predicated instruction");
3275   CmpInstr.eraseFromParent();
3276 
3277   // Modify the condition code of operands in OperandsToUpdate.
3278   // Since we have SUB(r1, r2) and CMP(r2, r1), the condition code needs to
3279   // be changed from r2 > r1 to r1 < r2, from r2 < r1 to r1 > r2, etc.
3280   for (unsigned i = 0, e = OperandsToUpdate.size(); i < e; i++)
3281     OperandsToUpdate[i].first->setImm(OperandsToUpdate[i].second);
3282 
3283   MI->clearRegisterDeads(ARM::CPSR);
3284 
3285   return true;
3286 }
3287 
3288 bool ARMBaseInstrInfo::shouldSink(const MachineInstr &MI) const {
3289   // Do not sink MI if it might be used to optimize a redundant compare.
3290   // We heuristically only look at the instruction immediately following MI to
3291   // avoid potentially searching the entire basic block.
3292   if (isPredicated(MI))
3293     return true;
3294   MachineBasicBlock::const_iterator Next = &MI;
3295   ++Next;
3296   Register SrcReg, SrcReg2;
3297   int64_t CmpMask, CmpValue;
3298   bool IsThumb1;
3299   if (Next != MI.getParent()->end() &&
3300       analyzeCompare(*Next, SrcReg, SrcReg2, CmpMask, CmpValue) &&
3301       isRedundantFlagInstr(&*Next, SrcReg, SrcReg2, CmpValue, &MI, IsThumb1))
3302     return false;
3303   return true;
3304 }
3305 
3306 bool ARMBaseInstrInfo::FoldImmediate(MachineInstr &UseMI, MachineInstr &DefMI,
3307                                      Register Reg,
3308                                      MachineRegisterInfo *MRI) const {
3309   // Fold large immediates into add, sub, or, xor.
3310   unsigned DefOpc = DefMI.getOpcode();
3311   if (DefOpc != ARM::t2MOVi32imm && DefOpc != ARM::MOVi32imm)
3312     return false;
3313   if (!DefMI.getOperand(1).isImm())
3314     // Could be t2MOVi32imm @xx
3315     return false;
3316 
3317   if (!MRI->hasOneNonDBGUse(Reg))
3318     return false;
3319 
3320   const MCInstrDesc &DefMCID = DefMI.getDesc();
3321   if (DefMCID.hasOptionalDef()) {
3322     unsigned NumOps = DefMCID.getNumOperands();
3323     const MachineOperand &MO = DefMI.getOperand(NumOps - 1);
3324     if (MO.getReg() == ARM::CPSR && !MO.isDead())
3325       // If DefMI defines CPSR and it is not dead, it's obviously not safe
3326       // to delete DefMI.
3327       return false;
3328   }
3329 
3330   const MCInstrDesc &UseMCID = UseMI.getDesc();
3331   if (UseMCID.hasOptionalDef()) {
3332     unsigned NumOps = UseMCID.getNumOperands();
3333     if (UseMI.getOperand(NumOps - 1).getReg() == ARM::CPSR)
3334       // If the instruction sets the flag, do not attempt this optimization
3335       // since it may change the semantics of the code.
3336       return false;
3337   }
3338 
3339   unsigned UseOpc = UseMI.getOpcode();
3340   unsigned NewUseOpc = 0;
3341   uint32_t ImmVal = (uint32_t)DefMI.getOperand(1).getImm();
3342   uint32_t SOImmValV1 = 0, SOImmValV2 = 0;
3343   bool Commute = false;
3344   switch (UseOpc) {
3345   default: return false;
3346   case ARM::SUBrr:
3347   case ARM::ADDrr:
3348   case ARM::ORRrr:
3349   case ARM::EORrr:
3350   case ARM::t2SUBrr:
3351   case ARM::t2ADDrr:
3352   case ARM::t2ORRrr:
3353   case ARM::t2EORrr: {
3354     Commute = UseMI.getOperand(2).getReg() != Reg;
3355     switch (UseOpc) {
3356     default: break;
3357     case ARM::ADDrr:
3358     case ARM::SUBrr:
3359       if (UseOpc == ARM::SUBrr && Commute)
3360         return false;
3361 
3362       // ADD/SUB are special because they're essentially the same operation, so
3363       // we can handle a larger range of immediates.
3364       if (ARM_AM::isSOImmTwoPartVal(ImmVal))
3365         NewUseOpc = UseOpc == ARM::ADDrr ? ARM::ADDri : ARM::SUBri;
3366       else if (ARM_AM::isSOImmTwoPartVal(-ImmVal)) {
3367         ImmVal = -ImmVal;
3368         NewUseOpc = UseOpc == ARM::ADDrr ? ARM::SUBri : ARM::ADDri;
3369       } else
3370         return false;
3371       SOImmValV1 = (uint32_t)ARM_AM::getSOImmTwoPartFirst(ImmVal);
3372       SOImmValV2 = (uint32_t)ARM_AM::getSOImmTwoPartSecond(ImmVal);
3373       break;
3374     case ARM::ORRrr:
3375     case ARM::EORrr:
3376       if (!ARM_AM::isSOImmTwoPartVal(ImmVal))
3377         return false;
3378       SOImmValV1 = (uint32_t)ARM_AM::getSOImmTwoPartFirst(ImmVal);
3379       SOImmValV2 = (uint32_t)ARM_AM::getSOImmTwoPartSecond(ImmVal);
3380       switch (UseOpc) {
3381       default: break;
3382       case ARM::ORRrr: NewUseOpc = ARM::ORRri; break;
3383       case ARM::EORrr: NewUseOpc = ARM::EORri; break;
3384       }
3385       break;
3386     case ARM::t2ADDrr:
3387     case ARM::t2SUBrr: {
3388       if (UseOpc == ARM::t2SUBrr && Commute)
3389         return false;
3390 
3391       // ADD/SUB are special because they're essentially the same operation, so
3392       // we can handle a larger range of immediates.
3393       const bool ToSP = DefMI.getOperand(0).getReg() == ARM::SP;
3394       const unsigned t2ADD = ToSP ? ARM::t2ADDspImm : ARM::t2ADDri;
3395       const unsigned t2SUB = ToSP ? ARM::t2SUBspImm : ARM::t2SUBri;
3396       if (ARM_AM::isT2SOImmTwoPartVal(ImmVal))
3397         NewUseOpc = UseOpc == ARM::t2ADDrr ? t2ADD : t2SUB;
3398       else if (ARM_AM::isT2SOImmTwoPartVal(-ImmVal)) {
3399         ImmVal = -ImmVal;
3400         NewUseOpc = UseOpc == ARM::t2ADDrr ? t2SUB : t2ADD;
3401       } else
3402         return false;
3403       SOImmValV1 = (uint32_t)ARM_AM::getT2SOImmTwoPartFirst(ImmVal);
3404       SOImmValV2 = (uint32_t)ARM_AM::getT2SOImmTwoPartSecond(ImmVal);
3405       break;
3406     }
3407     case ARM::t2ORRrr:
3408     case ARM::t2EORrr:
3409       if (!ARM_AM::isT2SOImmTwoPartVal(ImmVal))
3410         return false;
3411       SOImmValV1 = (uint32_t)ARM_AM::getT2SOImmTwoPartFirst(ImmVal);
3412       SOImmValV2 = (uint32_t)ARM_AM::getT2SOImmTwoPartSecond(ImmVal);
3413       switch (UseOpc) {
3414       default: break;
3415       case ARM::t2ORRrr: NewUseOpc = ARM::t2ORRri; break;
3416       case ARM::t2EORrr: NewUseOpc = ARM::t2EORri; break;
3417       }
3418       break;
3419     }
3420   }
3421   }
3422 
3423   unsigned OpIdx = Commute ? 2 : 1;
3424   Register Reg1 = UseMI.getOperand(OpIdx).getReg();
3425   bool isKill = UseMI.getOperand(OpIdx).isKill();
3426   const TargetRegisterClass *TRC = MRI->getRegClass(Reg);
3427   Register NewReg = MRI->createVirtualRegister(TRC);
3428   BuildMI(*UseMI.getParent(), UseMI, UseMI.getDebugLoc(), get(NewUseOpc),
3429           NewReg)
3430       .addReg(Reg1, getKillRegState(isKill))
3431       .addImm(SOImmValV1)
3432       .add(predOps(ARMCC::AL))
3433       .add(condCodeOp());
3434   UseMI.setDesc(get(NewUseOpc));
3435   UseMI.getOperand(1).setReg(NewReg);
3436   UseMI.getOperand(1).setIsKill();
3437   UseMI.getOperand(2).ChangeToImmediate(SOImmValV2);
3438   DefMI.eraseFromParent();
3439   // FIXME: t2ADDrr should be split, as different rulles apply when writing to SP.
3440   // Just as t2ADDri, that was split to [t2ADDri, t2ADDspImm].
3441   // Then the below code will not be needed, as the input/output register
3442   // classes will be rgpr or gprSP.
3443   // For now, we fix the UseMI operand explicitly here:
3444   switch(NewUseOpc){
3445     case ARM::t2ADDspImm:
3446     case ARM::t2SUBspImm:
3447     case ARM::t2ADDri:
3448     case ARM::t2SUBri:
3449       MRI->constrainRegClass(UseMI.getOperand(0).getReg(), TRC);
3450   }
3451   return true;
3452 }
3453 
3454 static unsigned getNumMicroOpsSwiftLdSt(const InstrItineraryData *ItinData,
3455                                         const MachineInstr &MI) {
3456   switch (MI.getOpcode()) {
3457   default: {
3458     const MCInstrDesc &Desc = MI.getDesc();
3459     int UOps = ItinData->getNumMicroOps(Desc.getSchedClass());
3460     assert(UOps >= 0 && "bad # UOps");
3461     return UOps;
3462   }
3463 
3464   case ARM::LDRrs:
3465   case ARM::LDRBrs:
3466   case ARM::STRrs:
3467   case ARM::STRBrs: {
3468     unsigned ShOpVal = MI.getOperand(3).getImm();
3469     bool isSub = ARM_AM::getAM2Op(ShOpVal) == ARM_AM::sub;
3470     unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
3471     if (!isSub &&
3472         (ShImm == 0 ||
3473          ((ShImm == 1 || ShImm == 2 || ShImm == 3) &&
3474           ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl)))
3475       return 1;
3476     return 2;
3477   }
3478 
3479   case ARM::LDRH:
3480   case ARM::STRH: {
3481     if (!MI.getOperand(2).getReg())
3482       return 1;
3483 
3484     unsigned ShOpVal = MI.getOperand(3).getImm();
3485     bool isSub = ARM_AM::getAM2Op(ShOpVal) == ARM_AM::sub;
3486     unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
3487     if (!isSub &&
3488         (ShImm == 0 ||
3489          ((ShImm == 1 || ShImm == 2 || ShImm == 3) &&
3490           ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl)))
3491       return 1;
3492     return 2;
3493   }
3494 
3495   case ARM::LDRSB:
3496   case ARM::LDRSH:
3497     return (ARM_AM::getAM3Op(MI.getOperand(3).getImm()) == ARM_AM::sub) ? 3 : 2;
3498 
3499   case ARM::LDRSB_POST:
3500   case ARM::LDRSH_POST: {
3501     Register Rt = MI.getOperand(0).getReg();
3502     Register Rm = MI.getOperand(3).getReg();
3503     return (Rt == Rm) ? 4 : 3;
3504   }
3505 
3506   case ARM::LDR_PRE_REG:
3507   case ARM::LDRB_PRE_REG: {
3508     Register Rt = MI.getOperand(0).getReg();
3509     Register Rm = MI.getOperand(3).getReg();
3510     if (Rt == Rm)
3511       return 3;
3512     unsigned ShOpVal = MI.getOperand(4).getImm();
3513     bool isSub = ARM_AM::getAM2Op(ShOpVal) == ARM_AM::sub;
3514     unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
3515     if (!isSub &&
3516         (ShImm == 0 ||
3517          ((ShImm == 1 || ShImm == 2 || ShImm == 3) &&
3518           ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl)))
3519       return 2;
3520     return 3;
3521   }
3522 
3523   case ARM::STR_PRE_REG:
3524   case ARM::STRB_PRE_REG: {
3525     unsigned ShOpVal = MI.getOperand(4).getImm();
3526     bool isSub = ARM_AM::getAM2Op(ShOpVal) == ARM_AM::sub;
3527     unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
3528     if (!isSub &&
3529         (ShImm == 0 ||
3530          ((ShImm == 1 || ShImm == 2 || ShImm == 3) &&
3531           ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl)))
3532       return 2;
3533     return 3;
3534   }
3535 
3536   case ARM::LDRH_PRE:
3537   case ARM::STRH_PRE: {
3538     Register Rt = MI.getOperand(0).getReg();
3539     Register Rm = MI.getOperand(3).getReg();
3540     if (!Rm)
3541       return 2;
3542     if (Rt == Rm)
3543       return 3;
3544     return (ARM_AM::getAM3Op(MI.getOperand(4).getImm()) == ARM_AM::sub) ? 3 : 2;
3545   }
3546 
3547   case ARM::LDR_POST_REG:
3548   case ARM::LDRB_POST_REG:
3549   case ARM::LDRH_POST: {
3550     Register Rt = MI.getOperand(0).getReg();
3551     Register Rm = MI.getOperand(3).getReg();
3552     return (Rt == Rm) ? 3 : 2;
3553   }
3554 
3555   case ARM::LDR_PRE_IMM:
3556   case ARM::LDRB_PRE_IMM:
3557   case ARM::LDR_POST_IMM:
3558   case ARM::LDRB_POST_IMM:
3559   case ARM::STRB_POST_IMM:
3560   case ARM::STRB_POST_REG:
3561   case ARM::STRB_PRE_IMM:
3562   case ARM::STRH_POST:
3563   case ARM::STR_POST_IMM:
3564   case ARM::STR_POST_REG:
3565   case ARM::STR_PRE_IMM:
3566     return 2;
3567 
3568   case ARM::LDRSB_PRE:
3569   case ARM::LDRSH_PRE: {
3570     Register Rm = MI.getOperand(3).getReg();
3571     if (Rm == 0)
3572       return 3;
3573     Register Rt = MI.getOperand(0).getReg();
3574     if (Rt == Rm)
3575       return 4;
3576     unsigned ShOpVal = MI.getOperand(4).getImm();
3577     bool isSub = ARM_AM::getAM2Op(ShOpVal) == ARM_AM::sub;
3578     unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
3579     if (!isSub &&
3580         (ShImm == 0 ||
3581          ((ShImm == 1 || ShImm == 2 || ShImm == 3) &&
3582           ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl)))
3583       return 3;
3584     return 4;
3585   }
3586 
3587   case ARM::LDRD: {
3588     Register Rt = MI.getOperand(0).getReg();
3589     Register Rn = MI.getOperand(2).getReg();
3590     Register Rm = MI.getOperand(3).getReg();
3591     if (Rm)
3592       return (ARM_AM::getAM3Op(MI.getOperand(4).getImm()) == ARM_AM::sub) ? 4
3593                                                                           : 3;
3594     return (Rt == Rn) ? 3 : 2;
3595   }
3596 
3597   case ARM::STRD: {
3598     Register Rm = MI.getOperand(3).getReg();
3599     if (Rm)
3600       return (ARM_AM::getAM3Op(MI.getOperand(4).getImm()) == ARM_AM::sub) ? 4
3601                                                                           : 3;
3602     return 2;
3603   }
3604 
3605   case ARM::LDRD_POST:
3606   case ARM::t2LDRD_POST:
3607     return 3;
3608 
3609   case ARM::STRD_POST:
3610   case ARM::t2STRD_POST:
3611     return 4;
3612 
3613   case ARM::LDRD_PRE: {
3614     Register Rt = MI.getOperand(0).getReg();
3615     Register Rn = MI.getOperand(3).getReg();
3616     Register Rm = MI.getOperand(4).getReg();
3617     if (Rm)
3618       return (ARM_AM::getAM3Op(MI.getOperand(5).getImm()) == ARM_AM::sub) ? 5
3619                                                                           : 4;
3620     return (Rt == Rn) ? 4 : 3;
3621   }
3622 
3623   case ARM::t2LDRD_PRE: {
3624     Register Rt = MI.getOperand(0).getReg();
3625     Register Rn = MI.getOperand(3).getReg();
3626     return (Rt == Rn) ? 4 : 3;
3627   }
3628 
3629   case ARM::STRD_PRE: {
3630     Register Rm = MI.getOperand(4).getReg();
3631     if (Rm)
3632       return (ARM_AM::getAM3Op(MI.getOperand(5).getImm()) == ARM_AM::sub) ? 5
3633                                                                           : 4;
3634     return 3;
3635   }
3636 
3637   case ARM::t2STRD_PRE:
3638     return 3;
3639 
3640   case ARM::t2LDR_POST:
3641   case ARM::t2LDRB_POST:
3642   case ARM::t2LDRB_PRE:
3643   case ARM::t2LDRSBi12:
3644   case ARM::t2LDRSBi8:
3645   case ARM::t2LDRSBpci:
3646   case ARM::t2LDRSBs:
3647   case ARM::t2LDRH_POST:
3648   case ARM::t2LDRH_PRE:
3649   case ARM::t2LDRSBT:
3650   case ARM::t2LDRSB_POST:
3651   case ARM::t2LDRSB_PRE:
3652   case ARM::t2LDRSH_POST:
3653   case ARM::t2LDRSH_PRE:
3654   case ARM::t2LDRSHi12:
3655   case ARM::t2LDRSHi8:
3656   case ARM::t2LDRSHpci:
3657   case ARM::t2LDRSHs:
3658     return 2;
3659 
3660   case ARM::t2LDRDi8: {
3661     Register Rt = MI.getOperand(0).getReg();
3662     Register Rn = MI.getOperand(2).getReg();
3663     return (Rt == Rn) ? 3 : 2;
3664   }
3665 
3666   case ARM::t2STRB_POST:
3667   case ARM::t2STRB_PRE:
3668   case ARM::t2STRBs:
3669   case ARM::t2STRDi8:
3670   case ARM::t2STRH_POST:
3671   case ARM::t2STRH_PRE:
3672   case ARM::t2STRHs:
3673   case ARM::t2STR_POST:
3674   case ARM::t2STR_PRE:
3675   case ARM::t2STRs:
3676     return 2;
3677   }
3678 }
3679 
3680 // Return the number of 32-bit words loaded by LDM or stored by STM. If this
3681 // can't be easily determined return 0 (missing MachineMemOperand).
3682 //
3683 // FIXME: The current MachineInstr design does not support relying on machine
3684 // mem operands to determine the width of a memory access. Instead, we expect
3685 // the target to provide this information based on the instruction opcode and
3686 // operands. However, using MachineMemOperand is the best solution now for
3687 // two reasons:
3688 //
3689 // 1) getNumMicroOps tries to infer LDM memory width from the total number of MI
3690 // operands. This is much more dangerous than using the MachineMemOperand
3691 // sizes because CodeGen passes can insert/remove optional machine operands. In
3692 // fact, it's totally incorrect for preRA passes and appears to be wrong for
3693 // postRA passes as well.
3694 //
3695 // 2) getNumLDMAddresses is only used by the scheduling machine model and any
3696 // machine model that calls this should handle the unknown (zero size) case.
3697 //
3698 // Long term, we should require a target hook that verifies MachineMemOperand
3699 // sizes during MC lowering. That target hook should be local to MC lowering
3700 // because we can't ensure that it is aware of other MI forms. Doing this will
3701 // ensure that MachineMemOperands are correctly propagated through all passes.
3702 unsigned ARMBaseInstrInfo::getNumLDMAddresses(const MachineInstr &MI) const {
3703   unsigned Size = 0;
3704   for (MachineInstr::mmo_iterator I = MI.memoperands_begin(),
3705                                   E = MI.memoperands_end();
3706        I != E; ++I) {
3707     Size += (*I)->getSize();
3708   }
3709   // FIXME: The scheduler currently can't handle values larger than 16. But
3710   // the values can actually go up to 32 for floating-point load/store
3711   // multiple (VLDMIA etc.). Also, the way this code is reasoning about memory
3712   // operations isn't right; we could end up with "extra" memory operands for
3713   // various reasons, like tail merge merging two memory operations.
3714   return std::min(Size / 4, 16U);
3715 }
3716 
3717 static unsigned getNumMicroOpsSingleIssuePlusExtras(unsigned Opc,
3718                                                     unsigned NumRegs) {
3719   unsigned UOps = 1 + NumRegs; // 1 for address computation.
3720   switch (Opc) {
3721   default:
3722     break;
3723   case ARM::VLDMDIA_UPD:
3724   case ARM::VLDMDDB_UPD:
3725   case ARM::VLDMSIA_UPD:
3726   case ARM::VLDMSDB_UPD:
3727   case ARM::VSTMDIA_UPD:
3728   case ARM::VSTMDDB_UPD:
3729   case ARM::VSTMSIA_UPD:
3730   case ARM::VSTMSDB_UPD:
3731   case ARM::LDMIA_UPD:
3732   case ARM::LDMDA_UPD:
3733   case ARM::LDMDB_UPD:
3734   case ARM::LDMIB_UPD:
3735   case ARM::STMIA_UPD:
3736   case ARM::STMDA_UPD:
3737   case ARM::STMDB_UPD:
3738   case ARM::STMIB_UPD:
3739   case ARM::tLDMIA_UPD:
3740   case ARM::tSTMIA_UPD:
3741   case ARM::t2LDMIA_UPD:
3742   case ARM::t2LDMDB_UPD:
3743   case ARM::t2STMIA_UPD:
3744   case ARM::t2STMDB_UPD:
3745     ++UOps; // One for base register writeback.
3746     break;
3747   case ARM::LDMIA_RET:
3748   case ARM::tPOP_RET:
3749   case ARM::t2LDMIA_RET:
3750     UOps += 2; // One for base reg wb, one for write to pc.
3751     break;
3752   }
3753   return UOps;
3754 }
3755 
3756 unsigned ARMBaseInstrInfo::getNumMicroOps(const InstrItineraryData *ItinData,
3757                                           const MachineInstr &MI) const {
3758   if (!ItinData || ItinData->isEmpty())
3759     return 1;
3760 
3761   const MCInstrDesc &Desc = MI.getDesc();
3762   unsigned Class = Desc.getSchedClass();
3763   int ItinUOps = ItinData->getNumMicroOps(Class);
3764   if (ItinUOps >= 0) {
3765     if (Subtarget.isSwift() && (Desc.mayLoad() || Desc.mayStore()))
3766       return getNumMicroOpsSwiftLdSt(ItinData, MI);
3767 
3768     return ItinUOps;
3769   }
3770 
3771   unsigned Opc = MI.getOpcode();
3772   switch (Opc) {
3773   default:
3774     llvm_unreachable("Unexpected multi-uops instruction!");
3775   case ARM::VLDMQIA:
3776   case ARM::VSTMQIA:
3777     return 2;
3778 
3779   // The number of uOps for load / store multiple are determined by the number
3780   // registers.
3781   //
3782   // On Cortex-A8, each pair of register loads / stores can be scheduled on the
3783   // same cycle. The scheduling for the first load / store must be done
3784   // separately by assuming the address is not 64-bit aligned.
3785   //
3786   // On Cortex-A9, the formula is simply (#reg / 2) + (#reg % 2). If the address
3787   // is not 64-bit aligned, then AGU would take an extra cycle.  For VFP / NEON
3788   // load / store multiple, the formula is (#reg / 2) + (#reg % 2) + 1.
3789   case ARM::VLDMDIA:
3790   case ARM::VLDMDIA_UPD:
3791   case ARM::VLDMDDB_UPD:
3792   case ARM::VLDMSIA:
3793   case ARM::VLDMSIA_UPD:
3794   case ARM::VLDMSDB_UPD:
3795   case ARM::VSTMDIA:
3796   case ARM::VSTMDIA_UPD:
3797   case ARM::VSTMDDB_UPD:
3798   case ARM::VSTMSIA:
3799   case ARM::VSTMSIA_UPD:
3800   case ARM::VSTMSDB_UPD: {
3801     unsigned NumRegs = MI.getNumOperands() - Desc.getNumOperands();
3802     return (NumRegs / 2) + (NumRegs % 2) + 1;
3803   }
3804 
3805   case ARM::LDMIA_RET:
3806   case ARM::LDMIA:
3807   case ARM::LDMDA:
3808   case ARM::LDMDB:
3809   case ARM::LDMIB:
3810   case ARM::LDMIA_UPD:
3811   case ARM::LDMDA_UPD:
3812   case ARM::LDMDB_UPD:
3813   case ARM::LDMIB_UPD:
3814   case ARM::STMIA:
3815   case ARM::STMDA:
3816   case ARM::STMDB:
3817   case ARM::STMIB:
3818   case ARM::STMIA_UPD:
3819   case ARM::STMDA_UPD:
3820   case ARM::STMDB_UPD:
3821   case ARM::STMIB_UPD:
3822   case ARM::tLDMIA:
3823   case ARM::tLDMIA_UPD:
3824   case ARM::tSTMIA_UPD:
3825   case ARM::tPOP_RET:
3826   case ARM::tPOP:
3827   case ARM::tPUSH:
3828   case ARM::t2LDMIA_RET:
3829   case ARM::t2LDMIA:
3830   case ARM::t2LDMDB:
3831   case ARM::t2LDMIA_UPD:
3832   case ARM::t2LDMDB_UPD:
3833   case ARM::t2STMIA:
3834   case ARM::t2STMDB:
3835   case ARM::t2STMIA_UPD:
3836   case ARM::t2STMDB_UPD: {
3837     unsigned NumRegs = MI.getNumOperands() - Desc.getNumOperands() + 1;
3838     switch (Subtarget.getLdStMultipleTiming()) {
3839     case ARMSubtarget::SingleIssuePlusExtras:
3840       return getNumMicroOpsSingleIssuePlusExtras(Opc, NumRegs);
3841     case ARMSubtarget::SingleIssue:
3842       // Assume the worst.
3843       return NumRegs;
3844     case ARMSubtarget::DoubleIssue: {
3845       if (NumRegs < 4)
3846         return 2;
3847       // 4 registers would be issued: 2, 2.
3848       // 5 registers would be issued: 2, 2, 1.
3849       unsigned UOps = (NumRegs / 2);
3850       if (NumRegs % 2)
3851         ++UOps;
3852       return UOps;
3853     }
3854     case ARMSubtarget::DoubleIssueCheckUnalignedAccess: {
3855       unsigned UOps = (NumRegs / 2);
3856       // If there are odd number of registers or if it's not 64-bit aligned,
3857       // then it takes an extra AGU (Address Generation Unit) cycle.
3858       if ((NumRegs % 2) || !MI.hasOneMemOperand() ||
3859           (*MI.memoperands_begin())->getAlign() < Align(8))
3860         ++UOps;
3861       return UOps;
3862       }
3863     }
3864   }
3865   }
3866   llvm_unreachable("Didn't find the number of microops");
3867 }
3868 
3869 int
3870 ARMBaseInstrInfo::getVLDMDefCycle(const InstrItineraryData *ItinData,
3871                                   const MCInstrDesc &DefMCID,
3872                                   unsigned DefClass,
3873                                   unsigned DefIdx, unsigned DefAlign) const {
3874   int RegNo = (int)(DefIdx+1) - DefMCID.getNumOperands() + 1;
3875   if (RegNo <= 0)
3876     // Def is the address writeback.
3877     return ItinData->getOperandCycle(DefClass, DefIdx);
3878 
3879   int DefCycle;
3880   if (Subtarget.isCortexA8() || Subtarget.isCortexA7()) {
3881     // (regno / 2) + (regno % 2) + 1
3882     DefCycle = RegNo / 2 + 1;
3883     if (RegNo % 2)
3884       ++DefCycle;
3885   } else if (Subtarget.isLikeA9() || Subtarget.isSwift()) {
3886     DefCycle = RegNo;
3887     bool isSLoad = false;
3888 
3889     switch (DefMCID.getOpcode()) {
3890     default: break;
3891     case ARM::VLDMSIA:
3892     case ARM::VLDMSIA_UPD:
3893     case ARM::VLDMSDB_UPD:
3894       isSLoad = true;
3895       break;
3896     }
3897 
3898     // If there are odd number of 'S' registers or if it's not 64-bit aligned,
3899     // then it takes an extra cycle.
3900     if ((isSLoad && (RegNo % 2)) || DefAlign < 8)
3901       ++DefCycle;
3902   } else {
3903     // Assume the worst.
3904     DefCycle = RegNo + 2;
3905   }
3906 
3907   return DefCycle;
3908 }
3909 
3910 int
3911 ARMBaseInstrInfo::getLDMDefCycle(const InstrItineraryData *ItinData,
3912                                  const MCInstrDesc &DefMCID,
3913                                  unsigned DefClass,
3914                                  unsigned DefIdx, unsigned DefAlign) const {
3915   int RegNo = (int)(DefIdx+1) - DefMCID.getNumOperands() + 1;
3916   if (RegNo <= 0)
3917     // Def is the address writeback.
3918     return ItinData->getOperandCycle(DefClass, DefIdx);
3919 
3920   int DefCycle;
3921   if (Subtarget.isCortexA8() || Subtarget.isCortexA7()) {
3922     // 4 registers would be issued: 1, 2, 1.
3923     // 5 registers would be issued: 1, 2, 2.
3924     DefCycle = RegNo / 2;
3925     if (DefCycle < 1)
3926       DefCycle = 1;
3927     // Result latency is issue cycle + 2: E2.
3928     DefCycle += 2;
3929   } else if (Subtarget.isLikeA9() || Subtarget.isSwift()) {
3930     DefCycle = (RegNo / 2);
3931     // If there are odd number of registers or if it's not 64-bit aligned,
3932     // then it takes an extra AGU (Address Generation Unit) cycle.
3933     if ((RegNo % 2) || DefAlign < 8)
3934       ++DefCycle;
3935     // Result latency is AGU cycles + 2.
3936     DefCycle += 2;
3937   } else {
3938     // Assume the worst.
3939     DefCycle = RegNo + 2;
3940   }
3941 
3942   return DefCycle;
3943 }
3944 
3945 int
3946 ARMBaseInstrInfo::getVSTMUseCycle(const InstrItineraryData *ItinData,
3947                                   const MCInstrDesc &UseMCID,
3948                                   unsigned UseClass,
3949                                   unsigned UseIdx, unsigned UseAlign) const {
3950   int RegNo = (int)(UseIdx+1) - UseMCID.getNumOperands() + 1;
3951   if (RegNo <= 0)
3952     return ItinData->getOperandCycle(UseClass, UseIdx);
3953 
3954   int UseCycle;
3955   if (Subtarget.isCortexA8() || Subtarget.isCortexA7()) {
3956     // (regno / 2) + (regno % 2) + 1
3957     UseCycle = RegNo / 2 + 1;
3958     if (RegNo % 2)
3959       ++UseCycle;
3960   } else if (Subtarget.isLikeA9() || Subtarget.isSwift()) {
3961     UseCycle = RegNo;
3962     bool isSStore = false;
3963 
3964     switch (UseMCID.getOpcode()) {
3965     default: break;
3966     case ARM::VSTMSIA:
3967     case ARM::VSTMSIA_UPD:
3968     case ARM::VSTMSDB_UPD:
3969       isSStore = true;
3970       break;
3971     }
3972 
3973     // If there are odd number of 'S' registers or if it's not 64-bit aligned,
3974     // then it takes an extra cycle.
3975     if ((isSStore && (RegNo % 2)) || UseAlign < 8)
3976       ++UseCycle;
3977   } else {
3978     // Assume the worst.
3979     UseCycle = RegNo + 2;
3980   }
3981 
3982   return UseCycle;
3983 }
3984 
3985 int
3986 ARMBaseInstrInfo::getSTMUseCycle(const InstrItineraryData *ItinData,
3987                                  const MCInstrDesc &UseMCID,
3988                                  unsigned UseClass,
3989                                  unsigned UseIdx, unsigned UseAlign) const {
3990   int RegNo = (int)(UseIdx+1) - UseMCID.getNumOperands() + 1;
3991   if (RegNo <= 0)
3992     return ItinData->getOperandCycle(UseClass, UseIdx);
3993 
3994   int UseCycle;
3995   if (Subtarget.isCortexA8() || Subtarget.isCortexA7()) {
3996     UseCycle = RegNo / 2;
3997     if (UseCycle < 2)
3998       UseCycle = 2;
3999     // Read in E3.
4000     UseCycle += 2;
4001   } else if (Subtarget.isLikeA9() || Subtarget.isSwift()) {
4002     UseCycle = (RegNo / 2);
4003     // If there are odd number of registers or if it's not 64-bit aligned,
4004     // then it takes an extra AGU (Address Generation Unit) cycle.
4005     if ((RegNo % 2) || UseAlign < 8)
4006       ++UseCycle;
4007   } else {
4008     // Assume the worst.
4009     UseCycle = 1;
4010   }
4011   return UseCycle;
4012 }
4013 
4014 int
4015 ARMBaseInstrInfo::getOperandLatency(const InstrItineraryData *ItinData,
4016                                     const MCInstrDesc &DefMCID,
4017                                     unsigned DefIdx, unsigned DefAlign,
4018                                     const MCInstrDesc &UseMCID,
4019                                     unsigned UseIdx, unsigned UseAlign) const {
4020   unsigned DefClass = DefMCID.getSchedClass();
4021   unsigned UseClass = UseMCID.getSchedClass();
4022 
4023   if (DefIdx < DefMCID.getNumDefs() && UseIdx < UseMCID.getNumOperands())
4024     return ItinData->getOperandLatency(DefClass, DefIdx, UseClass, UseIdx);
4025 
4026   // This may be a def / use of a variable_ops instruction, the operand
4027   // latency might be determinable dynamically. Let the target try to
4028   // figure it out.
4029   int DefCycle = -1;
4030   bool LdmBypass = false;
4031   switch (DefMCID.getOpcode()) {
4032   default:
4033     DefCycle = ItinData->getOperandCycle(DefClass, DefIdx);
4034     break;
4035 
4036   case ARM::VLDMDIA:
4037   case ARM::VLDMDIA_UPD:
4038   case ARM::VLDMDDB_UPD:
4039   case ARM::VLDMSIA:
4040   case ARM::VLDMSIA_UPD:
4041   case ARM::VLDMSDB_UPD:
4042     DefCycle = getVLDMDefCycle(ItinData, DefMCID, DefClass, DefIdx, DefAlign);
4043     break;
4044 
4045   case ARM::LDMIA_RET:
4046   case ARM::LDMIA:
4047   case ARM::LDMDA:
4048   case ARM::LDMDB:
4049   case ARM::LDMIB:
4050   case ARM::LDMIA_UPD:
4051   case ARM::LDMDA_UPD:
4052   case ARM::LDMDB_UPD:
4053   case ARM::LDMIB_UPD:
4054   case ARM::tLDMIA:
4055   case ARM::tLDMIA_UPD:
4056   case ARM::tPUSH:
4057   case ARM::t2LDMIA_RET:
4058   case ARM::t2LDMIA:
4059   case ARM::t2LDMDB:
4060   case ARM::t2LDMIA_UPD:
4061   case ARM::t2LDMDB_UPD:
4062     LdmBypass = true;
4063     DefCycle = getLDMDefCycle(ItinData, DefMCID, DefClass, DefIdx, DefAlign);
4064     break;
4065   }
4066 
4067   if (DefCycle == -1)
4068     // We can't seem to determine the result latency of the def, assume it's 2.
4069     DefCycle = 2;
4070 
4071   int UseCycle = -1;
4072   switch (UseMCID.getOpcode()) {
4073   default:
4074     UseCycle = ItinData->getOperandCycle(UseClass, UseIdx);
4075     break;
4076 
4077   case ARM::VSTMDIA:
4078   case ARM::VSTMDIA_UPD:
4079   case ARM::VSTMDDB_UPD:
4080   case ARM::VSTMSIA:
4081   case ARM::VSTMSIA_UPD:
4082   case ARM::VSTMSDB_UPD:
4083     UseCycle = getVSTMUseCycle(ItinData, UseMCID, UseClass, UseIdx, UseAlign);
4084     break;
4085 
4086   case ARM::STMIA:
4087   case ARM::STMDA:
4088   case ARM::STMDB:
4089   case ARM::STMIB:
4090   case ARM::STMIA_UPD:
4091   case ARM::STMDA_UPD:
4092   case ARM::STMDB_UPD:
4093   case ARM::STMIB_UPD:
4094   case ARM::tSTMIA_UPD:
4095   case ARM::tPOP_RET:
4096   case ARM::tPOP:
4097   case ARM::t2STMIA:
4098   case ARM::t2STMDB:
4099   case ARM::t2STMIA_UPD:
4100   case ARM::t2STMDB_UPD:
4101     UseCycle = getSTMUseCycle(ItinData, UseMCID, UseClass, UseIdx, UseAlign);
4102     break;
4103   }
4104 
4105   if (UseCycle == -1)
4106     // Assume it's read in the first stage.
4107     UseCycle = 1;
4108 
4109   UseCycle = DefCycle - UseCycle + 1;
4110   if (UseCycle > 0) {
4111     if (LdmBypass) {
4112       // It's a variable_ops instruction so we can't use DefIdx here. Just use
4113       // first def operand.
4114       if (ItinData->hasPipelineForwarding(DefClass, DefMCID.getNumOperands()-1,
4115                                           UseClass, UseIdx))
4116         --UseCycle;
4117     } else if (ItinData->hasPipelineForwarding(DefClass, DefIdx,
4118                                                UseClass, UseIdx)) {
4119       --UseCycle;
4120     }
4121   }
4122 
4123   return UseCycle;
4124 }
4125 
4126 static const MachineInstr *getBundledDefMI(const TargetRegisterInfo *TRI,
4127                                            const MachineInstr *MI, unsigned Reg,
4128                                            unsigned &DefIdx, unsigned &Dist) {
4129   Dist = 0;
4130 
4131   MachineBasicBlock::const_iterator I = MI; ++I;
4132   MachineBasicBlock::const_instr_iterator II = std::prev(I.getInstrIterator());
4133   assert(II->isInsideBundle() && "Empty bundle?");
4134 
4135   int Idx = -1;
4136   while (II->isInsideBundle()) {
4137     Idx = II->findRegisterDefOperandIdx(Reg, false, true, TRI);
4138     if (Idx != -1)
4139       break;
4140     --II;
4141     ++Dist;
4142   }
4143 
4144   assert(Idx != -1 && "Cannot find bundled definition!");
4145   DefIdx = Idx;
4146   return &*II;
4147 }
4148 
4149 static const MachineInstr *getBundledUseMI(const TargetRegisterInfo *TRI,
4150                                            const MachineInstr &MI, unsigned Reg,
4151                                            unsigned &UseIdx, unsigned &Dist) {
4152   Dist = 0;
4153 
4154   MachineBasicBlock::const_instr_iterator II = ++MI.getIterator();
4155   assert(II->isInsideBundle() && "Empty bundle?");
4156   MachineBasicBlock::const_instr_iterator E = MI.getParent()->instr_end();
4157 
4158   // FIXME: This doesn't properly handle multiple uses.
4159   int Idx = -1;
4160   while (II != E && II->isInsideBundle()) {
4161     Idx = II->findRegisterUseOperandIdx(Reg, false, TRI);
4162     if (Idx != -1)
4163       break;
4164     if (II->getOpcode() != ARM::t2IT)
4165       ++Dist;
4166     ++II;
4167   }
4168 
4169   if (Idx == -1) {
4170     Dist = 0;
4171     return nullptr;
4172   }
4173 
4174   UseIdx = Idx;
4175   return &*II;
4176 }
4177 
4178 /// Return the number of cycles to add to (or subtract from) the static
4179 /// itinerary based on the def opcode and alignment. The caller will ensure that
4180 /// adjusted latency is at least one cycle.
4181 static int adjustDefLatency(const ARMSubtarget &Subtarget,
4182                             const MachineInstr &DefMI,
4183                             const MCInstrDesc &DefMCID, unsigned DefAlign) {
4184   int Adjust = 0;
4185   if (Subtarget.isCortexA8() || Subtarget.isLikeA9() || Subtarget.isCortexA7()) {
4186     // FIXME: Shifter op hack: no shift (i.e. [r +/- r]) or [r + r << 2]
4187     // variants are one cycle cheaper.
4188     switch (DefMCID.getOpcode()) {
4189     default: break;
4190     case ARM::LDRrs:
4191     case ARM::LDRBrs: {
4192       unsigned ShOpVal = DefMI.getOperand(3).getImm();
4193       unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
4194       if (ShImm == 0 ||
4195           (ShImm == 2 && ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl))
4196         --Adjust;
4197       break;
4198     }
4199     case ARM::t2LDRs:
4200     case ARM::t2LDRBs:
4201     case ARM::t2LDRHs:
4202     case ARM::t2LDRSHs: {
4203       // Thumb2 mode: lsl only.
4204       unsigned ShAmt = DefMI.getOperand(3).getImm();
4205       if (ShAmt == 0 || ShAmt == 2)
4206         --Adjust;
4207       break;
4208     }
4209     }
4210   } else if (Subtarget.isSwift()) {
4211     // FIXME: Properly handle all of the latency adjustments for address
4212     // writeback.
4213     switch (DefMCID.getOpcode()) {
4214     default: break;
4215     case ARM::LDRrs:
4216     case ARM::LDRBrs: {
4217       unsigned ShOpVal = DefMI.getOperand(3).getImm();
4218       bool isSub = ARM_AM::getAM2Op(ShOpVal) == ARM_AM::sub;
4219       unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
4220       if (!isSub &&
4221           (ShImm == 0 ||
4222            ((ShImm == 1 || ShImm == 2 || ShImm == 3) &&
4223             ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl)))
4224         Adjust -= 2;
4225       else if (!isSub &&
4226                ShImm == 1 && ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsr)
4227         --Adjust;
4228       break;
4229     }
4230     case ARM::t2LDRs:
4231     case ARM::t2LDRBs:
4232     case ARM::t2LDRHs:
4233     case ARM::t2LDRSHs: {
4234       // Thumb2 mode: lsl only.
4235       unsigned ShAmt = DefMI.getOperand(3).getImm();
4236       if (ShAmt == 0 || ShAmt == 1 || ShAmt == 2 || ShAmt == 3)
4237         Adjust -= 2;
4238       break;
4239     }
4240     }
4241   }
4242 
4243   if (DefAlign < 8 && Subtarget.checkVLDnAccessAlignment()) {
4244     switch (DefMCID.getOpcode()) {
4245     default: break;
4246     case ARM::VLD1q8:
4247     case ARM::VLD1q16:
4248     case ARM::VLD1q32:
4249     case ARM::VLD1q64:
4250     case ARM::VLD1q8wb_fixed:
4251     case ARM::VLD1q16wb_fixed:
4252     case ARM::VLD1q32wb_fixed:
4253     case ARM::VLD1q64wb_fixed:
4254     case ARM::VLD1q8wb_register:
4255     case ARM::VLD1q16wb_register:
4256     case ARM::VLD1q32wb_register:
4257     case ARM::VLD1q64wb_register:
4258     case ARM::VLD2d8:
4259     case ARM::VLD2d16:
4260     case ARM::VLD2d32:
4261     case ARM::VLD2q8:
4262     case ARM::VLD2q16:
4263     case ARM::VLD2q32:
4264     case ARM::VLD2d8wb_fixed:
4265     case ARM::VLD2d16wb_fixed:
4266     case ARM::VLD2d32wb_fixed:
4267     case ARM::VLD2q8wb_fixed:
4268     case ARM::VLD2q16wb_fixed:
4269     case ARM::VLD2q32wb_fixed:
4270     case ARM::VLD2d8wb_register:
4271     case ARM::VLD2d16wb_register:
4272     case ARM::VLD2d32wb_register:
4273     case ARM::VLD2q8wb_register:
4274     case ARM::VLD2q16wb_register:
4275     case ARM::VLD2q32wb_register:
4276     case ARM::VLD3d8:
4277     case ARM::VLD3d16:
4278     case ARM::VLD3d32:
4279     case ARM::VLD1d64T:
4280     case ARM::VLD3d8_UPD:
4281     case ARM::VLD3d16_UPD:
4282     case ARM::VLD3d32_UPD:
4283     case ARM::VLD1d64Twb_fixed:
4284     case ARM::VLD1d64Twb_register:
4285     case ARM::VLD3q8_UPD:
4286     case ARM::VLD3q16_UPD:
4287     case ARM::VLD3q32_UPD:
4288     case ARM::VLD4d8:
4289     case ARM::VLD4d16:
4290     case ARM::VLD4d32:
4291     case ARM::VLD1d64Q:
4292     case ARM::VLD4d8_UPD:
4293     case ARM::VLD4d16_UPD:
4294     case ARM::VLD4d32_UPD:
4295     case ARM::VLD1d64Qwb_fixed:
4296     case ARM::VLD1d64Qwb_register:
4297     case ARM::VLD4q8_UPD:
4298     case ARM::VLD4q16_UPD:
4299     case ARM::VLD4q32_UPD:
4300     case ARM::VLD1DUPq8:
4301     case ARM::VLD1DUPq16:
4302     case ARM::VLD1DUPq32:
4303     case ARM::VLD1DUPq8wb_fixed:
4304     case ARM::VLD1DUPq16wb_fixed:
4305     case ARM::VLD1DUPq32wb_fixed:
4306     case ARM::VLD1DUPq8wb_register:
4307     case ARM::VLD1DUPq16wb_register:
4308     case ARM::VLD1DUPq32wb_register:
4309     case ARM::VLD2DUPd8:
4310     case ARM::VLD2DUPd16:
4311     case ARM::VLD2DUPd32:
4312     case ARM::VLD2DUPd8wb_fixed:
4313     case ARM::VLD2DUPd16wb_fixed:
4314     case ARM::VLD2DUPd32wb_fixed:
4315     case ARM::VLD2DUPd8wb_register:
4316     case ARM::VLD2DUPd16wb_register:
4317     case ARM::VLD2DUPd32wb_register:
4318     case ARM::VLD4DUPd8:
4319     case ARM::VLD4DUPd16:
4320     case ARM::VLD4DUPd32:
4321     case ARM::VLD4DUPd8_UPD:
4322     case ARM::VLD4DUPd16_UPD:
4323     case ARM::VLD4DUPd32_UPD:
4324     case ARM::VLD1LNd8:
4325     case ARM::VLD1LNd16:
4326     case ARM::VLD1LNd32:
4327     case ARM::VLD1LNd8_UPD:
4328     case ARM::VLD1LNd16_UPD:
4329     case ARM::VLD1LNd32_UPD:
4330     case ARM::VLD2LNd8:
4331     case ARM::VLD2LNd16:
4332     case ARM::VLD2LNd32:
4333     case ARM::VLD2LNq16:
4334     case ARM::VLD2LNq32:
4335     case ARM::VLD2LNd8_UPD:
4336     case ARM::VLD2LNd16_UPD:
4337     case ARM::VLD2LNd32_UPD:
4338     case ARM::VLD2LNq16_UPD:
4339     case ARM::VLD2LNq32_UPD:
4340     case ARM::VLD4LNd8:
4341     case ARM::VLD4LNd16:
4342     case ARM::VLD4LNd32:
4343     case ARM::VLD4LNq16:
4344     case ARM::VLD4LNq32:
4345     case ARM::VLD4LNd8_UPD:
4346     case ARM::VLD4LNd16_UPD:
4347     case ARM::VLD4LNd32_UPD:
4348     case ARM::VLD4LNq16_UPD:
4349     case ARM::VLD4LNq32_UPD:
4350       // If the address is not 64-bit aligned, the latencies of these
4351       // instructions increases by one.
4352       ++Adjust;
4353       break;
4354     }
4355   }
4356   return Adjust;
4357 }
4358 
4359 int ARMBaseInstrInfo::getOperandLatency(const InstrItineraryData *ItinData,
4360                                         const MachineInstr &DefMI,
4361                                         unsigned DefIdx,
4362                                         const MachineInstr &UseMI,
4363                                         unsigned UseIdx) const {
4364   // No operand latency. The caller may fall back to getInstrLatency.
4365   if (!ItinData || ItinData->isEmpty())
4366     return -1;
4367 
4368   const MachineOperand &DefMO = DefMI.getOperand(DefIdx);
4369   Register Reg = DefMO.getReg();
4370 
4371   const MachineInstr *ResolvedDefMI = &DefMI;
4372   unsigned DefAdj = 0;
4373   if (DefMI.isBundle())
4374     ResolvedDefMI =
4375         getBundledDefMI(&getRegisterInfo(), &DefMI, Reg, DefIdx, DefAdj);
4376   if (ResolvedDefMI->isCopyLike() || ResolvedDefMI->isInsertSubreg() ||
4377       ResolvedDefMI->isRegSequence() || ResolvedDefMI->isImplicitDef()) {
4378     return 1;
4379   }
4380 
4381   const MachineInstr *ResolvedUseMI = &UseMI;
4382   unsigned UseAdj = 0;
4383   if (UseMI.isBundle()) {
4384     ResolvedUseMI =
4385         getBundledUseMI(&getRegisterInfo(), UseMI, Reg, UseIdx, UseAdj);
4386     if (!ResolvedUseMI)
4387       return -1;
4388   }
4389 
4390   return getOperandLatencyImpl(
4391       ItinData, *ResolvedDefMI, DefIdx, ResolvedDefMI->getDesc(), DefAdj, DefMO,
4392       Reg, *ResolvedUseMI, UseIdx, ResolvedUseMI->getDesc(), UseAdj);
4393 }
4394 
4395 int ARMBaseInstrInfo::getOperandLatencyImpl(
4396     const InstrItineraryData *ItinData, const MachineInstr &DefMI,
4397     unsigned DefIdx, const MCInstrDesc &DefMCID, unsigned DefAdj,
4398     const MachineOperand &DefMO, unsigned Reg, const MachineInstr &UseMI,
4399     unsigned UseIdx, const MCInstrDesc &UseMCID, unsigned UseAdj) const {
4400   if (Reg == ARM::CPSR) {
4401     if (DefMI.getOpcode() == ARM::FMSTAT) {
4402       // fpscr -> cpsr stalls over 20 cycles on A8 (and earlier?)
4403       return Subtarget.isLikeA9() ? 1 : 20;
4404     }
4405 
4406     // CPSR set and branch can be paired in the same cycle.
4407     if (UseMI.isBranch())
4408       return 0;
4409 
4410     // Otherwise it takes the instruction latency (generally one).
4411     unsigned Latency = getInstrLatency(ItinData, DefMI);
4412 
4413     // For Thumb2 and -Os, prefer scheduling CPSR setting instruction close to
4414     // its uses. Instructions which are otherwise scheduled between them may
4415     // incur a code size penalty (not able to use the CPSR setting 16-bit
4416     // instructions).
4417     if (Latency > 0 && Subtarget.isThumb2()) {
4418       const MachineFunction *MF = DefMI.getParent()->getParent();
4419       // FIXME: Use Function::hasOptSize().
4420       if (MF->getFunction().hasFnAttribute(Attribute::OptimizeForSize))
4421         --Latency;
4422     }
4423     return Latency;
4424   }
4425 
4426   if (DefMO.isImplicit() || UseMI.getOperand(UseIdx).isImplicit())
4427     return -1;
4428 
4429   unsigned DefAlign = DefMI.hasOneMemOperand()
4430                           ? (*DefMI.memoperands_begin())->getAlign().value()
4431                           : 0;
4432   unsigned UseAlign = UseMI.hasOneMemOperand()
4433                           ? (*UseMI.memoperands_begin())->getAlign().value()
4434                           : 0;
4435 
4436   // Get the itinerary's latency if possible, and handle variable_ops.
4437   int Latency = getOperandLatency(ItinData, DefMCID, DefIdx, DefAlign, UseMCID,
4438                                   UseIdx, UseAlign);
4439   // Unable to find operand latency. The caller may resort to getInstrLatency.
4440   if (Latency < 0)
4441     return Latency;
4442 
4443   // Adjust for IT block position.
4444   int Adj = DefAdj + UseAdj;
4445 
4446   // Adjust for dynamic def-side opcode variants not captured by the itinerary.
4447   Adj += adjustDefLatency(Subtarget, DefMI, DefMCID, DefAlign);
4448   if (Adj >= 0 || (int)Latency > -Adj) {
4449     return Latency + Adj;
4450   }
4451   // Return the itinerary latency, which may be zero but not less than zero.
4452   return Latency;
4453 }
4454 
4455 int
4456 ARMBaseInstrInfo::getOperandLatency(const InstrItineraryData *ItinData,
4457                                     SDNode *DefNode, unsigned DefIdx,
4458                                     SDNode *UseNode, unsigned UseIdx) const {
4459   if (!DefNode->isMachineOpcode())
4460     return 1;
4461 
4462   const MCInstrDesc &DefMCID = get(DefNode->getMachineOpcode());
4463 
4464   if (isZeroCost(DefMCID.Opcode))
4465     return 0;
4466 
4467   if (!ItinData || ItinData->isEmpty())
4468     return DefMCID.mayLoad() ? 3 : 1;
4469 
4470   if (!UseNode->isMachineOpcode()) {
4471     int Latency = ItinData->getOperandCycle(DefMCID.getSchedClass(), DefIdx);
4472     int Adj = Subtarget.getPreISelOperandLatencyAdjustment();
4473     int Threshold = 1 + Adj;
4474     return Latency <= Threshold ? 1 : Latency - Adj;
4475   }
4476 
4477   const MCInstrDesc &UseMCID = get(UseNode->getMachineOpcode());
4478   auto *DefMN = cast<MachineSDNode>(DefNode);
4479   unsigned DefAlign = !DefMN->memoperands_empty()
4480                           ? (*DefMN->memoperands_begin())->getAlign().value()
4481                           : 0;
4482   auto *UseMN = cast<MachineSDNode>(UseNode);
4483   unsigned UseAlign = !UseMN->memoperands_empty()
4484                           ? (*UseMN->memoperands_begin())->getAlign().value()
4485                           : 0;
4486   int Latency = getOperandLatency(ItinData, DefMCID, DefIdx, DefAlign,
4487                                   UseMCID, UseIdx, UseAlign);
4488 
4489   if (Latency > 1 &&
4490       (Subtarget.isCortexA8() || Subtarget.isLikeA9() ||
4491        Subtarget.isCortexA7())) {
4492     // FIXME: Shifter op hack: no shift (i.e. [r +/- r]) or [r + r << 2]
4493     // variants are one cycle cheaper.
4494     switch (DefMCID.getOpcode()) {
4495     default: break;
4496     case ARM::LDRrs:
4497     case ARM::LDRBrs: {
4498       unsigned ShOpVal =
4499         cast<ConstantSDNode>(DefNode->getOperand(2))->getZExtValue();
4500       unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
4501       if (ShImm == 0 ||
4502           (ShImm == 2 && ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl))
4503         --Latency;
4504       break;
4505     }
4506     case ARM::t2LDRs:
4507     case ARM::t2LDRBs:
4508     case ARM::t2LDRHs:
4509     case ARM::t2LDRSHs: {
4510       // Thumb2 mode: lsl only.
4511       unsigned ShAmt =
4512         cast<ConstantSDNode>(DefNode->getOperand(2))->getZExtValue();
4513       if (ShAmt == 0 || ShAmt == 2)
4514         --Latency;
4515       break;
4516     }
4517     }
4518   } else if (DefIdx == 0 && Latency > 2 && Subtarget.isSwift()) {
4519     // FIXME: Properly handle all of the latency adjustments for address
4520     // writeback.
4521     switch (DefMCID.getOpcode()) {
4522     default: break;
4523     case ARM::LDRrs:
4524     case ARM::LDRBrs: {
4525       unsigned ShOpVal =
4526         cast<ConstantSDNode>(DefNode->getOperand(2))->getZExtValue();
4527       unsigned ShImm = ARM_AM::getAM2Offset(ShOpVal);
4528       if (ShImm == 0 ||
4529           ((ShImm == 1 || ShImm == 2 || ShImm == 3) &&
4530            ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsl))
4531         Latency -= 2;
4532       else if (ShImm == 1 && ARM_AM::getAM2ShiftOpc(ShOpVal) == ARM_AM::lsr)
4533         --Latency;
4534       break;
4535     }
4536     case ARM::t2LDRs:
4537     case ARM::t2LDRBs:
4538     case ARM::t2LDRHs:
4539     case ARM::t2LDRSHs:
4540       // Thumb2 mode: lsl 0-3 only.
4541       Latency -= 2;
4542       break;
4543     }
4544   }
4545 
4546   if (DefAlign < 8 && Subtarget.checkVLDnAccessAlignment())
4547     switch (DefMCID.getOpcode()) {
4548     default: break;
4549     case ARM::VLD1q8:
4550     case ARM::VLD1q16:
4551     case ARM::VLD1q32:
4552     case ARM::VLD1q64:
4553     case ARM::VLD1q8wb_register:
4554     case ARM::VLD1q16wb_register:
4555     case ARM::VLD1q32wb_register:
4556     case ARM::VLD1q64wb_register:
4557     case ARM::VLD1q8wb_fixed:
4558     case ARM::VLD1q16wb_fixed:
4559     case ARM::VLD1q32wb_fixed:
4560     case ARM::VLD1q64wb_fixed:
4561     case ARM::VLD2d8:
4562     case ARM::VLD2d16:
4563     case ARM::VLD2d32:
4564     case ARM::VLD2q8Pseudo:
4565     case ARM::VLD2q16Pseudo:
4566     case ARM::VLD2q32Pseudo:
4567     case ARM::VLD2d8wb_fixed:
4568     case ARM::VLD2d16wb_fixed:
4569     case ARM::VLD2d32wb_fixed:
4570     case ARM::VLD2q8PseudoWB_fixed:
4571     case ARM::VLD2q16PseudoWB_fixed:
4572     case ARM::VLD2q32PseudoWB_fixed:
4573     case ARM::VLD2d8wb_register:
4574     case ARM::VLD2d16wb_register:
4575     case ARM::VLD2d32wb_register:
4576     case ARM::VLD2q8PseudoWB_register:
4577     case ARM::VLD2q16PseudoWB_register:
4578     case ARM::VLD2q32PseudoWB_register:
4579     case ARM::VLD3d8Pseudo:
4580     case ARM::VLD3d16Pseudo:
4581     case ARM::VLD3d32Pseudo:
4582     case ARM::VLD1d8TPseudo:
4583     case ARM::VLD1d16TPseudo:
4584     case ARM::VLD1d32TPseudo:
4585     case ARM::VLD1d64TPseudo:
4586     case ARM::VLD1d64TPseudoWB_fixed:
4587     case ARM::VLD1d64TPseudoWB_register:
4588     case ARM::VLD3d8Pseudo_UPD:
4589     case ARM::VLD3d16Pseudo_UPD:
4590     case ARM::VLD3d32Pseudo_UPD:
4591     case ARM::VLD3q8Pseudo_UPD:
4592     case ARM::VLD3q16Pseudo_UPD:
4593     case ARM::VLD3q32Pseudo_UPD:
4594     case ARM::VLD3q8oddPseudo:
4595     case ARM::VLD3q16oddPseudo:
4596     case ARM::VLD3q32oddPseudo:
4597     case ARM::VLD3q8oddPseudo_UPD:
4598     case ARM::VLD3q16oddPseudo_UPD:
4599     case ARM::VLD3q32oddPseudo_UPD:
4600     case ARM::VLD4d8Pseudo:
4601     case ARM::VLD4d16Pseudo:
4602     case ARM::VLD4d32Pseudo:
4603     case ARM::VLD1d8QPseudo:
4604     case ARM::VLD1d16QPseudo:
4605     case ARM::VLD1d32QPseudo:
4606     case ARM::VLD1d64QPseudo:
4607     case ARM::VLD1d64QPseudoWB_fixed:
4608     case ARM::VLD1d64QPseudoWB_register:
4609     case ARM::VLD1q8HighQPseudo:
4610     case ARM::VLD1q8LowQPseudo_UPD:
4611     case ARM::VLD1q8HighTPseudo:
4612     case ARM::VLD1q8LowTPseudo_UPD:
4613     case ARM::VLD1q16HighQPseudo:
4614     case ARM::VLD1q16LowQPseudo_UPD:
4615     case ARM::VLD1q16HighTPseudo:
4616     case ARM::VLD1q16LowTPseudo_UPD:
4617     case ARM::VLD1q32HighQPseudo:
4618     case ARM::VLD1q32LowQPseudo_UPD:
4619     case ARM::VLD1q32HighTPseudo:
4620     case ARM::VLD1q32LowTPseudo_UPD:
4621     case ARM::VLD1q64HighQPseudo:
4622     case ARM::VLD1q64LowQPseudo_UPD:
4623     case ARM::VLD1q64HighTPseudo:
4624     case ARM::VLD1q64LowTPseudo_UPD:
4625     case ARM::VLD4d8Pseudo_UPD:
4626     case ARM::VLD4d16Pseudo_UPD:
4627     case ARM::VLD4d32Pseudo_UPD:
4628     case ARM::VLD4q8Pseudo_UPD:
4629     case ARM::VLD4q16Pseudo_UPD:
4630     case ARM::VLD4q32Pseudo_UPD:
4631     case ARM::VLD4q8oddPseudo:
4632     case ARM::VLD4q16oddPseudo:
4633     case ARM::VLD4q32oddPseudo:
4634     case ARM::VLD4q8oddPseudo_UPD:
4635     case ARM::VLD4q16oddPseudo_UPD:
4636     case ARM::VLD4q32oddPseudo_UPD:
4637     case ARM::VLD1DUPq8:
4638     case ARM::VLD1DUPq16:
4639     case ARM::VLD1DUPq32:
4640     case ARM::VLD1DUPq8wb_fixed:
4641     case ARM::VLD1DUPq16wb_fixed:
4642     case ARM::VLD1DUPq32wb_fixed:
4643     case ARM::VLD1DUPq8wb_register:
4644     case ARM::VLD1DUPq16wb_register:
4645     case ARM::VLD1DUPq32wb_register:
4646     case ARM::VLD2DUPd8:
4647     case ARM::VLD2DUPd16:
4648     case ARM::VLD2DUPd32:
4649     case ARM::VLD2DUPd8wb_fixed:
4650     case ARM::VLD2DUPd16wb_fixed:
4651     case ARM::VLD2DUPd32wb_fixed:
4652     case ARM::VLD2DUPd8wb_register:
4653     case ARM::VLD2DUPd16wb_register:
4654     case ARM::VLD2DUPd32wb_register:
4655     case ARM::VLD2DUPq8EvenPseudo:
4656     case ARM::VLD2DUPq8OddPseudo:
4657     case ARM::VLD2DUPq16EvenPseudo:
4658     case ARM::VLD2DUPq16OddPseudo:
4659     case ARM::VLD2DUPq32EvenPseudo:
4660     case ARM::VLD2DUPq32OddPseudo:
4661     case ARM::VLD3DUPq8EvenPseudo:
4662     case ARM::VLD3DUPq8OddPseudo:
4663     case ARM::VLD3DUPq16EvenPseudo:
4664     case ARM::VLD3DUPq16OddPseudo:
4665     case ARM::VLD3DUPq32EvenPseudo:
4666     case ARM::VLD3DUPq32OddPseudo:
4667     case ARM::VLD4DUPd8Pseudo:
4668     case ARM::VLD4DUPd16Pseudo:
4669     case ARM::VLD4DUPd32Pseudo:
4670     case ARM::VLD4DUPd8Pseudo_UPD:
4671     case ARM::VLD4DUPd16Pseudo_UPD:
4672     case ARM::VLD4DUPd32Pseudo_UPD:
4673     case ARM::VLD4DUPq8EvenPseudo:
4674     case ARM::VLD4DUPq8OddPseudo:
4675     case ARM::VLD4DUPq16EvenPseudo:
4676     case ARM::VLD4DUPq16OddPseudo:
4677     case ARM::VLD4DUPq32EvenPseudo:
4678     case ARM::VLD4DUPq32OddPseudo:
4679     case ARM::VLD1LNq8Pseudo:
4680     case ARM::VLD1LNq16Pseudo:
4681     case ARM::VLD1LNq32Pseudo:
4682     case ARM::VLD1LNq8Pseudo_UPD:
4683     case ARM::VLD1LNq16Pseudo_UPD:
4684     case ARM::VLD1LNq32Pseudo_UPD:
4685     case ARM::VLD2LNd8Pseudo:
4686     case ARM::VLD2LNd16Pseudo:
4687     case ARM::VLD2LNd32Pseudo:
4688     case ARM::VLD2LNq16Pseudo:
4689     case ARM::VLD2LNq32Pseudo:
4690     case ARM::VLD2LNd8Pseudo_UPD:
4691     case ARM::VLD2LNd16Pseudo_UPD:
4692     case ARM::VLD2LNd32Pseudo_UPD:
4693     case ARM::VLD2LNq16Pseudo_UPD:
4694     case ARM::VLD2LNq32Pseudo_UPD:
4695     case ARM::VLD4LNd8Pseudo:
4696     case ARM::VLD4LNd16Pseudo:
4697     case ARM::VLD4LNd32Pseudo:
4698     case ARM::VLD4LNq16Pseudo:
4699     case ARM::VLD4LNq32Pseudo:
4700     case ARM::VLD4LNd8Pseudo_UPD:
4701     case ARM::VLD4LNd16Pseudo_UPD:
4702     case ARM::VLD4LNd32Pseudo_UPD:
4703     case ARM::VLD4LNq16Pseudo_UPD:
4704     case ARM::VLD4LNq32Pseudo_UPD:
4705       // If the address is not 64-bit aligned, the latencies of these
4706       // instructions increases by one.
4707       ++Latency;
4708       break;
4709     }
4710 
4711   return Latency;
4712 }
4713 
4714 unsigned ARMBaseInstrInfo::getPredicationCost(const MachineInstr &MI) const {
4715   if (MI.isCopyLike() || MI.isInsertSubreg() || MI.isRegSequence() ||
4716       MI.isImplicitDef())
4717     return 0;
4718 
4719   if (MI.isBundle())
4720     return 0;
4721 
4722   const MCInstrDesc &MCID = MI.getDesc();
4723 
4724   if (MCID.isCall() || (MCID.hasImplicitDefOfPhysReg(ARM::CPSR) &&
4725                         !Subtarget.cheapPredicableCPSRDef())) {
4726     // When predicated, CPSR is an additional source operand for CPSR updating
4727     // instructions, this apparently increases their latencies.
4728     return 1;
4729   }
4730   return 0;
4731 }
4732 
4733 unsigned ARMBaseInstrInfo::getInstrLatency(const InstrItineraryData *ItinData,
4734                                            const MachineInstr &MI,
4735                                            unsigned *PredCost) const {
4736   if (MI.isCopyLike() || MI.isInsertSubreg() || MI.isRegSequence() ||
4737       MI.isImplicitDef())
4738     return 1;
4739 
4740   // An instruction scheduler typically runs on unbundled instructions, however
4741   // other passes may query the latency of a bundled instruction.
4742   if (MI.isBundle()) {
4743     unsigned Latency = 0;
4744     MachineBasicBlock::const_instr_iterator I = MI.getIterator();
4745     MachineBasicBlock::const_instr_iterator E = MI.getParent()->instr_end();
4746     while (++I != E && I->isInsideBundle()) {
4747       if (I->getOpcode() != ARM::t2IT)
4748         Latency += getInstrLatency(ItinData, *I, PredCost);
4749     }
4750     return Latency;
4751   }
4752 
4753   const MCInstrDesc &MCID = MI.getDesc();
4754   if (PredCost && (MCID.isCall() || (MCID.hasImplicitDefOfPhysReg(ARM::CPSR) &&
4755                                      !Subtarget.cheapPredicableCPSRDef()))) {
4756     // When predicated, CPSR is an additional source operand for CPSR updating
4757     // instructions, this apparently increases their latencies.
4758     *PredCost = 1;
4759   }
4760   // Be sure to call getStageLatency for an empty itinerary in case it has a
4761   // valid MinLatency property.
4762   if (!ItinData)
4763     return MI.mayLoad() ? 3 : 1;
4764 
4765   unsigned Class = MCID.getSchedClass();
4766 
4767   // For instructions with variable uops, use uops as latency.
4768   if (!ItinData->isEmpty() && ItinData->getNumMicroOps(Class) < 0)
4769     return getNumMicroOps(ItinData, MI);
4770 
4771   // For the common case, fall back on the itinerary's latency.
4772   unsigned Latency = ItinData->getStageLatency(Class);
4773 
4774   // Adjust for dynamic def-side opcode variants not captured by the itinerary.
4775   unsigned DefAlign =
4776       MI.hasOneMemOperand() ? (*MI.memoperands_begin())->getAlign().value() : 0;
4777   int Adj = adjustDefLatency(Subtarget, MI, MCID, DefAlign);
4778   if (Adj >= 0 || (int)Latency > -Adj) {
4779     return Latency + Adj;
4780   }
4781   return Latency;
4782 }
4783 
4784 int ARMBaseInstrInfo::getInstrLatency(const InstrItineraryData *ItinData,
4785                                       SDNode *Node) const {
4786   if (!Node->isMachineOpcode())
4787     return 1;
4788 
4789   if (!ItinData || ItinData->isEmpty())
4790     return 1;
4791 
4792   unsigned Opcode = Node->getMachineOpcode();
4793   switch (Opcode) {
4794   default:
4795     return ItinData->getStageLatency(get(Opcode).getSchedClass());
4796   case ARM::VLDMQIA:
4797   case ARM::VSTMQIA:
4798     return 2;
4799   }
4800 }
4801 
4802 bool ARMBaseInstrInfo::hasHighOperandLatency(const TargetSchedModel &SchedModel,
4803                                              const MachineRegisterInfo *MRI,
4804                                              const MachineInstr &DefMI,
4805                                              unsigned DefIdx,
4806                                              const MachineInstr &UseMI,
4807                                              unsigned UseIdx) const {
4808   unsigned DDomain = DefMI.getDesc().TSFlags & ARMII::DomainMask;
4809   unsigned UDomain = UseMI.getDesc().TSFlags & ARMII::DomainMask;
4810   if (Subtarget.nonpipelinedVFP() &&
4811       (DDomain == ARMII::DomainVFP || UDomain == ARMII::DomainVFP))
4812     return true;
4813 
4814   // Hoist VFP / NEON instructions with 4 or higher latency.
4815   unsigned Latency =
4816       SchedModel.computeOperandLatency(&DefMI, DefIdx, &UseMI, UseIdx);
4817   if (Latency <= 3)
4818     return false;
4819   return DDomain == ARMII::DomainVFP || DDomain == ARMII::DomainNEON ||
4820          UDomain == ARMII::DomainVFP || UDomain == ARMII::DomainNEON;
4821 }
4822 
4823 bool ARMBaseInstrInfo::hasLowDefLatency(const TargetSchedModel &SchedModel,
4824                                         const MachineInstr &DefMI,
4825                                         unsigned DefIdx) const {
4826   const InstrItineraryData *ItinData = SchedModel.getInstrItineraries();
4827   if (!ItinData || ItinData->isEmpty())
4828     return false;
4829 
4830   unsigned DDomain = DefMI.getDesc().TSFlags & ARMII::DomainMask;
4831   if (DDomain == ARMII::DomainGeneral) {
4832     unsigned DefClass = DefMI.getDesc().getSchedClass();
4833     int DefCycle = ItinData->getOperandCycle(DefClass, DefIdx);
4834     return (DefCycle != -1 && DefCycle <= 2);
4835   }
4836   return false;
4837 }
4838 
4839 bool ARMBaseInstrInfo::verifyInstruction(const MachineInstr &MI,
4840                                          StringRef &ErrInfo) const {
4841   if (convertAddSubFlagsOpcode(MI.getOpcode())) {
4842     ErrInfo = "Pseudo flag setting opcodes only exist in Selection DAG";
4843     return false;
4844   }
4845   if (MI.getOpcode() == ARM::tMOVr && !Subtarget.hasV6Ops()) {
4846     // Make sure we don't generate a lo-lo mov that isn't supported.
4847     if (!ARM::hGPRRegClass.contains(MI.getOperand(0).getReg()) &&
4848         !ARM::hGPRRegClass.contains(MI.getOperand(1).getReg())) {
4849       ErrInfo = "Non-flag-setting Thumb1 mov is v6-only";
4850       return false;
4851     }
4852   }
4853   if (MI.getOpcode() == ARM::tPUSH ||
4854       MI.getOpcode() == ARM::tPOP ||
4855       MI.getOpcode() == ARM::tPOP_RET) {
4856     for (int i = 2, e = MI.getNumOperands(); i < e; ++i) {
4857       if (MI.getOperand(i).isImplicit() ||
4858           !MI.getOperand(i).isReg())
4859         continue;
4860       Register Reg = MI.getOperand(i).getReg();
4861       if (Reg < ARM::R0 || Reg > ARM::R7) {
4862         if (!(MI.getOpcode() == ARM::tPUSH && Reg == ARM::LR) &&
4863             !(MI.getOpcode() == ARM::tPOP_RET && Reg == ARM::PC)) {
4864           ErrInfo = "Unsupported register in Thumb1 push/pop";
4865           return false;
4866         }
4867       }
4868     }
4869   }
4870   if (MI.getOpcode() == ARM::MVE_VMOV_q_rr) {
4871     assert(MI.getOperand(4).isImm() && MI.getOperand(5).isImm());
4872     if ((MI.getOperand(4).getImm() != 2 && MI.getOperand(4).getImm() != 3) ||
4873         MI.getOperand(4).getImm() != MI.getOperand(5).getImm() + 2) {
4874       ErrInfo = "Incorrect array index for MVE_VMOV_q_rr";
4875       return false;
4876     }
4877   }
4878   return true;
4879 }
4880 
4881 // LoadStackGuard has so far only been implemented for MachO. Different code
4882 // sequence is needed for other targets.
4883 void ARMBaseInstrInfo::expandLoadStackGuardBase(MachineBasicBlock::iterator MI,
4884                                                 unsigned LoadImmOpc,
4885                                                 unsigned LoadOpc) const {
4886   assert(!Subtarget.isROPI() && !Subtarget.isRWPI() &&
4887          "ROPI/RWPI not currently supported with stack guard");
4888 
4889   MachineBasicBlock &MBB = *MI->getParent();
4890   DebugLoc DL = MI->getDebugLoc();
4891   Register Reg = MI->getOperand(0).getReg();
4892   const GlobalValue *GV =
4893       cast<GlobalValue>((*MI->memoperands_begin())->getValue());
4894   MachineInstrBuilder MIB;
4895 
4896   BuildMI(MBB, MI, DL, get(LoadImmOpc), Reg)
4897       .addGlobalAddress(GV, 0, ARMII::MO_NONLAZY);
4898 
4899   if (Subtarget.isGVIndirectSymbol(GV)) {
4900     MIB = BuildMI(MBB, MI, DL, get(LoadOpc), Reg);
4901     MIB.addReg(Reg, RegState::Kill).addImm(0);
4902     auto Flags = MachineMemOperand::MOLoad |
4903                  MachineMemOperand::MODereferenceable |
4904                  MachineMemOperand::MOInvariant;
4905     MachineMemOperand *MMO = MBB.getParent()->getMachineMemOperand(
4906         MachinePointerInfo::getGOT(*MBB.getParent()), Flags, 4, Align(4));
4907     MIB.addMemOperand(MMO).add(predOps(ARMCC::AL));
4908   }
4909 
4910   MIB = BuildMI(MBB, MI, DL, get(LoadOpc), Reg);
4911   MIB.addReg(Reg, RegState::Kill)
4912       .addImm(0)
4913       .cloneMemRefs(*MI)
4914       .add(predOps(ARMCC::AL));
4915 }
4916 
4917 bool
4918 ARMBaseInstrInfo::isFpMLxInstruction(unsigned Opcode, unsigned &MulOpc,
4919                                      unsigned &AddSubOpc,
4920                                      bool &NegAcc, bool &HasLane) const {
4921   DenseMap<unsigned, unsigned>::const_iterator I = MLxEntryMap.find(Opcode);
4922   if (I == MLxEntryMap.end())
4923     return false;
4924 
4925   const ARM_MLxEntry &Entry = ARM_MLxTable[I->second];
4926   MulOpc = Entry.MulOpc;
4927   AddSubOpc = Entry.AddSubOpc;
4928   NegAcc = Entry.NegAcc;
4929   HasLane = Entry.HasLane;
4930   return true;
4931 }
4932 
4933 //===----------------------------------------------------------------------===//
4934 // Execution domains.
4935 //===----------------------------------------------------------------------===//
4936 //
4937 // Some instructions go down the NEON pipeline, some go down the VFP pipeline,
4938 // and some can go down both.  The vmov instructions go down the VFP pipeline,
4939 // but they can be changed to vorr equivalents that are executed by the NEON
4940 // pipeline.
4941 //
4942 // We use the following execution domain numbering:
4943 //
4944 enum ARMExeDomain {
4945   ExeGeneric = 0,
4946   ExeVFP = 1,
4947   ExeNEON = 2
4948 };
4949 
4950 //
4951 // Also see ARMInstrFormats.td and Domain* enums in ARMBaseInfo.h
4952 //
4953 std::pair<uint16_t, uint16_t>
4954 ARMBaseInstrInfo::getExecutionDomain(const MachineInstr &MI) const {
4955   // If we don't have access to NEON instructions then we won't be able
4956   // to swizzle anything to the NEON domain. Check to make sure.
4957   if (Subtarget.hasNEON()) {
4958     // VMOVD, VMOVRS and VMOVSR are VFP instructions, but can be changed to NEON
4959     // if they are not predicated.
4960     if (MI.getOpcode() == ARM::VMOVD && !isPredicated(MI))
4961       return std::make_pair(ExeVFP, (1 << ExeVFP) | (1 << ExeNEON));
4962 
4963     // CortexA9 is particularly picky about mixing the two and wants these
4964     // converted.
4965     if (Subtarget.useNEONForFPMovs() && !isPredicated(MI) &&
4966         (MI.getOpcode() == ARM::VMOVRS || MI.getOpcode() == ARM::VMOVSR ||
4967          MI.getOpcode() == ARM::VMOVS))
4968       return std::make_pair(ExeVFP, (1 << ExeVFP) | (1 << ExeNEON));
4969   }
4970   // No other instructions can be swizzled, so just determine their domain.
4971   unsigned Domain = MI.getDesc().TSFlags & ARMII::DomainMask;
4972 
4973   if (Domain & ARMII::DomainNEON)
4974     return std::make_pair(ExeNEON, 0);
4975 
4976   // Certain instructions can go either way on Cortex-A8.
4977   // Treat them as NEON instructions.
4978   if ((Domain & ARMII::DomainNEONA8) && Subtarget.isCortexA8())
4979     return std::make_pair(ExeNEON, 0);
4980 
4981   if (Domain & ARMII::DomainVFP)
4982     return std::make_pair(ExeVFP, 0);
4983 
4984   return std::make_pair(ExeGeneric, 0);
4985 }
4986 
4987 static unsigned getCorrespondingDRegAndLane(const TargetRegisterInfo *TRI,
4988                                             unsigned SReg, unsigned &Lane) {
4989   unsigned DReg = TRI->getMatchingSuperReg(SReg, ARM::ssub_0, &ARM::DPRRegClass);
4990   Lane = 0;
4991 
4992   if (DReg != ARM::NoRegister)
4993    return DReg;
4994 
4995   Lane = 1;
4996   DReg = TRI->getMatchingSuperReg(SReg, ARM::ssub_1, &ARM::DPRRegClass);
4997 
4998   assert(DReg && "S-register with no D super-register?");
4999   return DReg;
5000 }
5001 
5002 /// getImplicitSPRUseForDPRUse - Given a use of a DPR register and lane,
5003 /// set ImplicitSReg to a register number that must be marked as implicit-use or
5004 /// zero if no register needs to be defined as implicit-use.
5005 ///
5006 /// If the function cannot determine if an SPR should be marked implicit use or
5007 /// not, it returns false.
5008 ///
5009 /// This function handles cases where an instruction is being modified from taking
5010 /// an SPR to a DPR[Lane]. A use of the DPR is being added, which may conflict
5011 /// with an earlier def of an SPR corresponding to DPR[Lane^1] (i.e. the other
5012 /// lane of the DPR).
5013 ///
5014 /// If the other SPR is defined, an implicit-use of it should be added. Else,
5015 /// (including the case where the DPR itself is defined), it should not.
5016 ///
5017 static bool getImplicitSPRUseForDPRUse(const TargetRegisterInfo *TRI,
5018                                        MachineInstr &MI, unsigned DReg,
5019                                        unsigned Lane, unsigned &ImplicitSReg) {
5020   // If the DPR is defined or used already, the other SPR lane will be chained
5021   // correctly, so there is nothing to be done.
5022   if (MI.definesRegister(DReg, TRI) || MI.readsRegister(DReg, TRI)) {
5023     ImplicitSReg = 0;
5024     return true;
5025   }
5026 
5027   // Otherwise we need to go searching to see if the SPR is set explicitly.
5028   ImplicitSReg = TRI->getSubReg(DReg,
5029                                 (Lane & 1) ? ARM::ssub_0 : ARM::ssub_1);
5030   MachineBasicBlock::LivenessQueryResult LQR =
5031       MI.getParent()->computeRegisterLiveness(TRI, ImplicitSReg, MI);
5032 
5033   if (LQR == MachineBasicBlock::LQR_Live)
5034     return true;
5035   else if (LQR == MachineBasicBlock::LQR_Unknown)
5036     return false;
5037 
5038   // If the register is known not to be live, there is no need to add an
5039   // implicit-use.
5040   ImplicitSReg = 0;
5041   return true;
5042 }
5043 
5044 void ARMBaseInstrInfo::setExecutionDomain(MachineInstr &MI,
5045                                           unsigned Domain) const {
5046   unsigned DstReg, SrcReg, DReg;
5047   unsigned Lane;
5048   MachineInstrBuilder MIB(*MI.getParent()->getParent(), MI);
5049   const TargetRegisterInfo *TRI = &getRegisterInfo();
5050   switch (MI.getOpcode()) {
5051   default:
5052     llvm_unreachable("cannot handle opcode!");
5053     break;
5054   case ARM::VMOVD:
5055     if (Domain != ExeNEON)
5056       break;
5057 
5058     // Zap the predicate operands.
5059     assert(!isPredicated(MI) && "Cannot predicate a VORRd");
5060 
5061     // Make sure we've got NEON instructions.
5062     assert(Subtarget.hasNEON() && "VORRd requires NEON");
5063 
5064     // Source instruction is %DDst = VMOVD %DSrc, 14, %noreg (; implicits)
5065     DstReg = MI.getOperand(0).getReg();
5066     SrcReg = MI.getOperand(1).getReg();
5067 
5068     for (unsigned i = MI.getDesc().getNumOperands(); i; --i)
5069       MI.RemoveOperand(i - 1);
5070 
5071     // Change to a %DDst = VORRd %DSrc, %DSrc, 14, %noreg (; implicits)
5072     MI.setDesc(get(ARM::VORRd));
5073     MIB.addReg(DstReg, RegState::Define)
5074         .addReg(SrcReg)
5075         .addReg(SrcReg)
5076         .add(predOps(ARMCC::AL));
5077     break;
5078   case ARM::VMOVRS:
5079     if (Domain != ExeNEON)
5080       break;
5081     assert(!isPredicated(MI) && "Cannot predicate a VGETLN");
5082 
5083     // Source instruction is %RDst = VMOVRS %SSrc, 14, %noreg (; implicits)
5084     DstReg = MI.getOperand(0).getReg();
5085     SrcReg = MI.getOperand(1).getReg();
5086 
5087     for (unsigned i = MI.getDesc().getNumOperands(); i; --i)
5088       MI.RemoveOperand(i - 1);
5089 
5090     DReg = getCorrespondingDRegAndLane(TRI, SrcReg, Lane);
5091 
5092     // Convert to %RDst = VGETLNi32 %DSrc, Lane, 14, %noreg (; imps)
5093     // Note that DSrc has been widened and the other lane may be undef, which
5094     // contaminates the entire register.
5095     MI.setDesc(get(ARM::VGETLNi32));
5096     MIB.addReg(DstReg, RegState::Define)
5097         .addReg(DReg, RegState::Undef)
5098         .addImm(Lane)
5099         .add(predOps(ARMCC::AL));
5100 
5101     // The old source should be an implicit use, otherwise we might think it
5102     // was dead before here.
5103     MIB.addReg(SrcReg, RegState::Implicit);
5104     break;
5105   case ARM::VMOVSR: {
5106     if (Domain != ExeNEON)
5107       break;
5108     assert(!isPredicated(MI) && "Cannot predicate a VSETLN");
5109 
5110     // Source instruction is %SDst = VMOVSR %RSrc, 14, %noreg (; implicits)
5111     DstReg = MI.getOperand(0).getReg();
5112     SrcReg = MI.getOperand(1).getReg();
5113 
5114     DReg = getCorrespondingDRegAndLane(TRI, DstReg, Lane);
5115 
5116     unsigned ImplicitSReg;
5117     if (!getImplicitSPRUseForDPRUse(TRI, MI, DReg, Lane, ImplicitSReg))
5118       break;
5119 
5120     for (unsigned i = MI.getDesc().getNumOperands(); i; --i)
5121       MI.RemoveOperand(i - 1);
5122 
5123     // Convert to %DDst = VSETLNi32 %DDst, %RSrc, Lane, 14, %noreg (; imps)
5124     // Again DDst may be undefined at the beginning of this instruction.
5125     MI.setDesc(get(ARM::VSETLNi32));
5126     MIB.addReg(DReg, RegState::Define)
5127         .addReg(DReg, getUndefRegState(!MI.readsRegister(DReg, TRI)))
5128         .addReg(SrcReg)
5129         .addImm(Lane)
5130         .add(predOps(ARMCC::AL));
5131 
5132     // The narrower destination must be marked as set to keep previous chains
5133     // in place.
5134     MIB.addReg(DstReg, RegState::Define | RegState::Implicit);
5135     if (ImplicitSReg != 0)
5136       MIB.addReg(ImplicitSReg, RegState::Implicit);
5137     break;
5138     }
5139     case ARM::VMOVS: {
5140       if (Domain != ExeNEON)
5141         break;
5142 
5143       // Source instruction is %SDst = VMOVS %SSrc, 14, %noreg (; implicits)
5144       DstReg = MI.getOperand(0).getReg();
5145       SrcReg = MI.getOperand(1).getReg();
5146 
5147       unsigned DstLane = 0, SrcLane = 0, DDst, DSrc;
5148       DDst = getCorrespondingDRegAndLane(TRI, DstReg, DstLane);
5149       DSrc = getCorrespondingDRegAndLane(TRI, SrcReg, SrcLane);
5150 
5151       unsigned ImplicitSReg;
5152       if (!getImplicitSPRUseForDPRUse(TRI, MI, DSrc, SrcLane, ImplicitSReg))
5153         break;
5154 
5155       for (unsigned i = MI.getDesc().getNumOperands(); i; --i)
5156         MI.RemoveOperand(i - 1);
5157 
5158       if (DSrc == DDst) {
5159         // Destination can be:
5160         //     %DDst = VDUPLN32d %DDst, Lane, 14, %noreg (; implicits)
5161         MI.setDesc(get(ARM::VDUPLN32d));
5162         MIB.addReg(DDst, RegState::Define)
5163             .addReg(DDst, getUndefRegState(!MI.readsRegister(DDst, TRI)))
5164             .addImm(SrcLane)
5165             .add(predOps(ARMCC::AL));
5166 
5167         // Neither the source or the destination are naturally represented any
5168         // more, so add them in manually.
5169         MIB.addReg(DstReg, RegState::Implicit | RegState::Define);
5170         MIB.addReg(SrcReg, RegState::Implicit);
5171         if (ImplicitSReg != 0)
5172           MIB.addReg(ImplicitSReg, RegState::Implicit);
5173         break;
5174       }
5175 
5176       // In general there's no single instruction that can perform an S <-> S
5177       // move in NEON space, but a pair of VEXT instructions *can* do the
5178       // job. It turns out that the VEXTs needed will only use DSrc once, with
5179       // the position based purely on the combination of lane-0 and lane-1
5180       // involved. For example
5181       //     vmov s0, s2 -> vext.32 d0, d0, d1, #1  vext.32 d0, d0, d0, #1
5182       //     vmov s1, s3 -> vext.32 d0, d1, d0, #1  vext.32 d0, d0, d0, #1
5183       //     vmov s0, s3 -> vext.32 d0, d0, d0, #1  vext.32 d0, d1, d0, #1
5184       //     vmov s1, s2 -> vext.32 d0, d0, d0, #1  vext.32 d0, d0, d1, #1
5185       //
5186       // Pattern of the MachineInstrs is:
5187       //     %DDst = VEXTd32 %DSrc1, %DSrc2, Lane, 14, %noreg (;implicits)
5188       MachineInstrBuilder NewMIB;
5189       NewMIB = BuildMI(*MI.getParent(), MI, MI.getDebugLoc(), get(ARM::VEXTd32),
5190                        DDst);
5191 
5192       // On the first instruction, both DSrc and DDst may be undef if present.
5193       // Specifically when the original instruction didn't have them as an
5194       // <imp-use>.
5195       unsigned CurReg = SrcLane == 1 && DstLane == 1 ? DSrc : DDst;
5196       bool CurUndef = !MI.readsRegister(CurReg, TRI);
5197       NewMIB.addReg(CurReg, getUndefRegState(CurUndef));
5198 
5199       CurReg = SrcLane == 0 && DstLane == 0 ? DSrc : DDst;
5200       CurUndef = !MI.readsRegister(CurReg, TRI);
5201       NewMIB.addReg(CurReg, getUndefRegState(CurUndef))
5202             .addImm(1)
5203             .add(predOps(ARMCC::AL));
5204 
5205       if (SrcLane == DstLane)
5206         NewMIB.addReg(SrcReg, RegState::Implicit);
5207 
5208       MI.setDesc(get(ARM::VEXTd32));
5209       MIB.addReg(DDst, RegState::Define);
5210 
5211       // On the second instruction, DDst has definitely been defined above, so
5212       // it is not undef. DSrc, if present, can be undef as above.
5213       CurReg = SrcLane == 1 && DstLane == 0 ? DSrc : DDst;
5214       CurUndef = CurReg == DSrc && !MI.readsRegister(CurReg, TRI);
5215       MIB.addReg(CurReg, getUndefRegState(CurUndef));
5216 
5217       CurReg = SrcLane == 0 && DstLane == 1 ? DSrc : DDst;
5218       CurUndef = CurReg == DSrc && !MI.readsRegister(CurReg, TRI);
5219       MIB.addReg(CurReg, getUndefRegState(CurUndef))
5220          .addImm(1)
5221          .add(predOps(ARMCC::AL));
5222 
5223       if (SrcLane != DstLane)
5224         MIB.addReg(SrcReg, RegState::Implicit);
5225 
5226       // As before, the original destination is no longer represented, add it
5227       // implicitly.
5228       MIB.addReg(DstReg, RegState::Define | RegState::Implicit);
5229       if (ImplicitSReg != 0)
5230         MIB.addReg(ImplicitSReg, RegState::Implicit);
5231       break;
5232     }
5233   }
5234 }
5235 
5236 //===----------------------------------------------------------------------===//
5237 // Partial register updates
5238 //===----------------------------------------------------------------------===//
5239 //
5240 // Swift renames NEON registers with 64-bit granularity.  That means any
5241 // instruction writing an S-reg implicitly reads the containing D-reg.  The
5242 // problem is mostly avoided by translating f32 operations to v2f32 operations
5243 // on D-registers, but f32 loads are still a problem.
5244 //
5245 // These instructions can load an f32 into a NEON register:
5246 //
5247 // VLDRS - Only writes S, partial D update.
5248 // VLD1LNd32 - Writes all D-regs, explicit partial D update, 2 uops.
5249 // VLD1DUPd32 - Writes all D-regs, no partial reg update, 2 uops.
5250 //
5251 // FCONSTD can be used as a dependency-breaking instruction.
5252 unsigned ARMBaseInstrInfo::getPartialRegUpdateClearance(
5253     const MachineInstr &MI, unsigned OpNum,
5254     const TargetRegisterInfo *TRI) const {
5255   auto PartialUpdateClearance = Subtarget.getPartialUpdateClearance();
5256   if (!PartialUpdateClearance)
5257     return 0;
5258 
5259   assert(TRI && "Need TRI instance");
5260 
5261   const MachineOperand &MO = MI.getOperand(OpNum);
5262   if (MO.readsReg())
5263     return 0;
5264   Register Reg = MO.getReg();
5265   int UseOp = -1;
5266 
5267   switch (MI.getOpcode()) {
5268   // Normal instructions writing only an S-register.
5269   case ARM::VLDRS:
5270   case ARM::FCONSTS:
5271   case ARM::VMOVSR:
5272   case ARM::VMOVv8i8:
5273   case ARM::VMOVv4i16:
5274   case ARM::VMOVv2i32:
5275   case ARM::VMOVv2f32:
5276   case ARM::VMOVv1i64:
5277     UseOp = MI.findRegisterUseOperandIdx(Reg, false, TRI);
5278     break;
5279 
5280     // Explicitly reads the dependency.
5281   case ARM::VLD1LNd32:
5282     UseOp = 3;
5283     break;
5284   default:
5285     return 0;
5286   }
5287 
5288   // If this instruction actually reads a value from Reg, there is no unwanted
5289   // dependency.
5290   if (UseOp != -1 && MI.getOperand(UseOp).readsReg())
5291     return 0;
5292 
5293   // We must be able to clobber the whole D-reg.
5294   if (Register::isVirtualRegister(Reg)) {
5295     // Virtual register must be a def undef foo:ssub_0 operand.
5296     if (!MO.getSubReg() || MI.readsVirtualRegister(Reg))
5297       return 0;
5298   } else if (ARM::SPRRegClass.contains(Reg)) {
5299     // Physical register: MI must define the full D-reg.
5300     unsigned DReg = TRI->getMatchingSuperReg(Reg, ARM::ssub_0,
5301                                              &ARM::DPRRegClass);
5302     if (!DReg || !MI.definesRegister(DReg, TRI))
5303       return 0;
5304   }
5305 
5306   // MI has an unwanted D-register dependency.
5307   // Avoid defs in the previous N instructrions.
5308   return PartialUpdateClearance;
5309 }
5310 
5311 // Break a partial register dependency after getPartialRegUpdateClearance
5312 // returned non-zero.
5313 void ARMBaseInstrInfo::breakPartialRegDependency(
5314     MachineInstr &MI, unsigned OpNum, const TargetRegisterInfo *TRI) const {
5315   assert(OpNum < MI.getDesc().getNumDefs() && "OpNum is not a def");
5316   assert(TRI && "Need TRI instance");
5317 
5318   const MachineOperand &MO = MI.getOperand(OpNum);
5319   Register Reg = MO.getReg();
5320   assert(Register::isPhysicalRegister(Reg) &&
5321          "Can't break virtual register dependencies.");
5322   unsigned DReg = Reg;
5323 
5324   // If MI defines an S-reg, find the corresponding D super-register.
5325   if (ARM::SPRRegClass.contains(Reg)) {
5326     DReg = ARM::D0 + (Reg - ARM::S0) / 2;
5327     assert(TRI->isSuperRegister(Reg, DReg) && "Register enums broken");
5328   }
5329 
5330   assert(ARM::DPRRegClass.contains(DReg) && "Can only break D-reg deps");
5331   assert(MI.definesRegister(DReg, TRI) && "MI doesn't clobber full D-reg");
5332 
5333   // FIXME: In some cases, VLDRS can be changed to a VLD1DUPd32 which defines
5334   // the full D-register by loading the same value to both lanes.  The
5335   // instruction is micro-coded with 2 uops, so don't do this until we can
5336   // properly schedule micro-coded instructions.  The dispatcher stalls cause
5337   // too big regressions.
5338 
5339   // Insert the dependency-breaking FCONSTD before MI.
5340   // 96 is the encoding of 0.5, but the actual value doesn't matter here.
5341   BuildMI(*MI.getParent(), MI, MI.getDebugLoc(), get(ARM::FCONSTD), DReg)
5342       .addImm(96)
5343       .add(predOps(ARMCC::AL));
5344   MI.addRegisterKilled(DReg, TRI, true);
5345 }
5346 
5347 bool ARMBaseInstrInfo::hasNOP() const {
5348   return Subtarget.getFeatureBits()[ARM::HasV6KOps];
5349 }
5350 
5351 bool ARMBaseInstrInfo::isSwiftFastImmShift(const MachineInstr *MI) const {
5352   if (MI->getNumOperands() < 4)
5353     return true;
5354   unsigned ShOpVal = MI->getOperand(3).getImm();
5355   unsigned ShImm = ARM_AM::getSORegOffset(ShOpVal);
5356   // Swift supports faster shifts for: lsl 2, lsl 1, and lsr 1.
5357   if ((ShImm == 1 && ARM_AM::getSORegShOp(ShOpVal) == ARM_AM::lsr) ||
5358       ((ShImm == 1 || ShImm == 2) &&
5359        ARM_AM::getSORegShOp(ShOpVal) == ARM_AM::lsl))
5360     return true;
5361 
5362   return false;
5363 }
5364 
5365 bool ARMBaseInstrInfo::getRegSequenceLikeInputs(
5366     const MachineInstr &MI, unsigned DefIdx,
5367     SmallVectorImpl<RegSubRegPairAndIdx> &InputRegs) const {
5368   assert(DefIdx < MI.getDesc().getNumDefs() && "Invalid definition index");
5369   assert(MI.isRegSequenceLike() && "Invalid kind of instruction");
5370 
5371   switch (MI.getOpcode()) {
5372   case ARM::VMOVDRR:
5373     // dX = VMOVDRR rY, rZ
5374     // is the same as:
5375     // dX = REG_SEQUENCE rY, ssub_0, rZ, ssub_1
5376     // Populate the InputRegs accordingly.
5377     // rY
5378     const MachineOperand *MOReg = &MI.getOperand(1);
5379     if (!MOReg->isUndef())
5380       InputRegs.push_back(RegSubRegPairAndIdx(MOReg->getReg(),
5381                                               MOReg->getSubReg(), ARM::ssub_0));
5382     // rZ
5383     MOReg = &MI.getOperand(2);
5384     if (!MOReg->isUndef())
5385       InputRegs.push_back(RegSubRegPairAndIdx(MOReg->getReg(),
5386                                               MOReg->getSubReg(), ARM::ssub_1));
5387     return true;
5388   }
5389   llvm_unreachable("Target dependent opcode missing");
5390 }
5391 
5392 bool ARMBaseInstrInfo::getExtractSubregLikeInputs(
5393     const MachineInstr &MI, unsigned DefIdx,
5394     RegSubRegPairAndIdx &InputReg) const {
5395   assert(DefIdx < MI.getDesc().getNumDefs() && "Invalid definition index");
5396   assert(MI.isExtractSubregLike() && "Invalid kind of instruction");
5397 
5398   switch (MI.getOpcode()) {
5399   case ARM::VMOVRRD:
5400     // rX, rY = VMOVRRD dZ
5401     // is the same as:
5402     // rX = EXTRACT_SUBREG dZ, ssub_0
5403     // rY = EXTRACT_SUBREG dZ, ssub_1
5404     const MachineOperand &MOReg = MI.getOperand(2);
5405     if (MOReg.isUndef())
5406       return false;
5407     InputReg.Reg = MOReg.getReg();
5408     InputReg.SubReg = MOReg.getSubReg();
5409     InputReg.SubIdx = DefIdx == 0 ? ARM::ssub_0 : ARM::ssub_1;
5410     return true;
5411   }
5412   llvm_unreachable("Target dependent opcode missing");
5413 }
5414 
5415 bool ARMBaseInstrInfo::getInsertSubregLikeInputs(
5416     const MachineInstr &MI, unsigned DefIdx, RegSubRegPair &BaseReg,
5417     RegSubRegPairAndIdx &InsertedReg) const {
5418   assert(DefIdx < MI.getDesc().getNumDefs() && "Invalid definition index");
5419   assert(MI.isInsertSubregLike() && "Invalid kind of instruction");
5420 
5421   switch (MI.getOpcode()) {
5422   case ARM::VSETLNi32:
5423   case ARM::MVE_VMOV_to_lane_32:
5424     // dX = VSETLNi32 dY, rZ, imm
5425     // qX = MVE_VMOV_to_lane_32 qY, rZ, imm
5426     const MachineOperand &MOBaseReg = MI.getOperand(1);
5427     const MachineOperand &MOInsertedReg = MI.getOperand(2);
5428     if (MOInsertedReg.isUndef())
5429       return false;
5430     const MachineOperand &MOIndex = MI.getOperand(3);
5431     BaseReg.Reg = MOBaseReg.getReg();
5432     BaseReg.SubReg = MOBaseReg.getSubReg();
5433 
5434     InsertedReg.Reg = MOInsertedReg.getReg();
5435     InsertedReg.SubReg = MOInsertedReg.getSubReg();
5436     InsertedReg.SubIdx = ARM::ssub_0 + MOIndex.getImm();
5437     return true;
5438   }
5439   llvm_unreachable("Target dependent opcode missing");
5440 }
5441 
5442 std::pair<unsigned, unsigned>
5443 ARMBaseInstrInfo::decomposeMachineOperandsTargetFlags(unsigned TF) const {
5444   const unsigned Mask = ARMII::MO_OPTION_MASK;
5445   return std::make_pair(TF & Mask, TF & ~Mask);
5446 }
5447 
5448 ArrayRef<std::pair<unsigned, const char *>>
5449 ARMBaseInstrInfo::getSerializableDirectMachineOperandTargetFlags() const {
5450   using namespace ARMII;
5451 
5452   static const std::pair<unsigned, const char *> TargetFlags[] = {
5453       {MO_LO16, "arm-lo16"}, {MO_HI16, "arm-hi16"}};
5454   return makeArrayRef(TargetFlags);
5455 }
5456 
5457 ArrayRef<std::pair<unsigned, const char *>>
5458 ARMBaseInstrInfo::getSerializableBitmaskMachineOperandTargetFlags() const {
5459   using namespace ARMII;
5460 
5461   static const std::pair<unsigned, const char *> TargetFlags[] = {
5462       {MO_COFFSTUB, "arm-coffstub"},
5463       {MO_GOT, "arm-got"},
5464       {MO_SBREL, "arm-sbrel"},
5465       {MO_DLLIMPORT, "arm-dllimport"},
5466       {MO_SECREL, "arm-secrel"},
5467       {MO_NONLAZY, "arm-nonlazy"}};
5468   return makeArrayRef(TargetFlags);
5469 }
5470 
5471 Optional<RegImmPair> ARMBaseInstrInfo::isAddImmediate(const MachineInstr &MI,
5472                                                       Register Reg) const {
5473   int Sign = 1;
5474   unsigned Opcode = MI.getOpcode();
5475   int64_t Offset = 0;
5476 
5477   // TODO: Handle cases where Reg is a super- or sub-register of the
5478   // destination register.
5479   const MachineOperand &Op0 = MI.getOperand(0);
5480   if (!Op0.isReg() || Reg != Op0.getReg())
5481     return None;
5482 
5483   // We describe SUBri or ADDri instructions.
5484   if (Opcode == ARM::SUBri)
5485     Sign = -1;
5486   else if (Opcode != ARM::ADDri)
5487     return None;
5488 
5489   // TODO: Third operand can be global address (usually some string). Since
5490   //       strings can be relocated we cannot calculate their offsets for
5491   //       now.
5492   if (!MI.getOperand(1).isReg() || !MI.getOperand(2).isImm())
5493     return None;
5494 
5495   Offset = MI.getOperand(2).getImm() * Sign;
5496   return RegImmPair{MI.getOperand(1).getReg(), Offset};
5497 }
5498 
5499 bool llvm::registerDefinedBetween(unsigned Reg,
5500                                   MachineBasicBlock::iterator From,
5501                                   MachineBasicBlock::iterator To,
5502                                   const TargetRegisterInfo *TRI) {
5503   for (auto I = From; I != To; ++I)
5504     if (I->modifiesRegister(Reg, TRI))
5505       return true;
5506   return false;
5507 }
5508 
5509 MachineInstr *llvm::findCMPToFoldIntoCBZ(MachineInstr *Br,
5510                                          const TargetRegisterInfo *TRI) {
5511   // Search backwards to the instruction that defines CSPR. This may or not
5512   // be a CMP, we check that after this loop. If we find another instruction
5513   // that reads cpsr, we return nullptr.
5514   MachineBasicBlock::iterator CmpMI = Br;
5515   while (CmpMI != Br->getParent()->begin()) {
5516     --CmpMI;
5517     if (CmpMI->modifiesRegister(ARM::CPSR, TRI))
5518       break;
5519     if (CmpMI->readsRegister(ARM::CPSR, TRI))
5520       break;
5521   }
5522 
5523   // Check that this inst is a CMP r[0-7], #0 and that the register
5524   // is not redefined between the cmp and the br.
5525   if (CmpMI->getOpcode() != ARM::tCMPi8 && CmpMI->getOpcode() != ARM::t2CMPri)
5526     return nullptr;
5527   Register Reg = CmpMI->getOperand(0).getReg();
5528   Register PredReg;
5529   ARMCC::CondCodes Pred = getInstrPredicate(*CmpMI, PredReg);
5530   if (Pred != ARMCC::AL || CmpMI->getOperand(1).getImm() != 0)
5531     return nullptr;
5532   if (!isARMLowRegister(Reg))
5533     return nullptr;
5534   if (registerDefinedBetween(Reg, CmpMI->getNextNode(), Br, TRI))
5535     return nullptr;
5536 
5537   return &*CmpMI;
5538 }
5539 
5540 unsigned llvm::ConstantMaterializationCost(unsigned Val,
5541                                            const ARMSubtarget *Subtarget,
5542                                            bool ForCodesize) {
5543   if (Subtarget->isThumb()) {
5544     if (Val <= 255) // MOV
5545       return ForCodesize ? 2 : 1;
5546     if (Subtarget->hasV6T2Ops() && (Val <= 0xffff ||                    // MOV
5547                                     ARM_AM::getT2SOImmVal(Val) != -1 || // MOVW
5548                                     ARM_AM::getT2SOImmVal(~Val) != -1)) // MVN
5549       return ForCodesize ? 4 : 1;
5550     if (Val <= 510) // MOV + ADDi8
5551       return ForCodesize ? 4 : 2;
5552     if (~Val <= 255) // MOV + MVN
5553       return ForCodesize ? 4 : 2;
5554     if (ARM_AM::isThumbImmShiftedVal(Val)) // MOV + LSL
5555       return ForCodesize ? 4 : 2;
5556   } else {
5557     if (ARM_AM::getSOImmVal(Val) != -1) // MOV
5558       return ForCodesize ? 4 : 1;
5559     if (ARM_AM::getSOImmVal(~Val) != -1) // MVN
5560       return ForCodesize ? 4 : 1;
5561     if (Subtarget->hasV6T2Ops() && Val <= 0xffff) // MOVW
5562       return ForCodesize ? 4 : 1;
5563     if (ARM_AM::isSOImmTwoPartVal(Val)) // two instrs
5564       return ForCodesize ? 8 : 2;
5565     if (ARM_AM::isSOImmTwoPartValNeg(Val)) // two instrs
5566       return ForCodesize ? 8 : 2;
5567   }
5568   if (Subtarget->useMovt()) // MOVW + MOVT
5569     return ForCodesize ? 8 : 2;
5570   return ForCodesize ? 8 : 3; // Literal pool load
5571 }
5572 
5573 bool llvm::HasLowerConstantMaterializationCost(unsigned Val1, unsigned Val2,
5574                                                const ARMSubtarget *Subtarget,
5575                                                bool ForCodesize) {
5576   // Check with ForCodesize
5577   unsigned Cost1 = ConstantMaterializationCost(Val1, Subtarget, ForCodesize);
5578   unsigned Cost2 = ConstantMaterializationCost(Val2, Subtarget, ForCodesize);
5579   if (Cost1 < Cost2)
5580     return true;
5581   if (Cost1 > Cost2)
5582     return false;
5583 
5584   // If they are equal, try with !ForCodesize
5585   return ConstantMaterializationCost(Val1, Subtarget, !ForCodesize) <
5586          ConstantMaterializationCost(Val2, Subtarget, !ForCodesize);
5587 }
5588 
5589 /// Constants defining how certain sequences should be outlined.
5590 /// This encompasses how an outlined function should be called, and what kind of
5591 /// frame should be emitted for that outlined function.
5592 ///
5593 /// \p MachineOutlinerTailCall implies that the function is being created from
5594 /// a sequence of instructions ending in a return.
5595 ///
5596 /// That is,
5597 ///
5598 /// I1                                OUTLINED_FUNCTION:
5599 /// I2    --> B OUTLINED_FUNCTION     I1
5600 /// BX LR                             I2
5601 ///                                   BX LR
5602 ///
5603 /// +-------------------------+--------+-----+
5604 /// |                         | Thumb2 | ARM |
5605 /// +-------------------------+--------+-----+
5606 /// | Call overhead in Bytes  |      4 |   4 |
5607 /// | Frame overhead in Bytes |      0 |   0 |
5608 /// | Stack fixup required    |     No |  No |
5609 /// +-------------------------+--------+-----+
5610 ///
5611 /// \p MachineOutlinerThunk implies that the function is being created from
5612 /// a sequence of instructions ending in a call. The outlined function is
5613 /// called with a BL instruction, and the outlined function tail-calls the
5614 /// original call destination.
5615 ///
5616 /// That is,
5617 ///
5618 /// I1                                OUTLINED_FUNCTION:
5619 /// I2   --> BL OUTLINED_FUNCTION     I1
5620 /// BL f                              I2
5621 ///                                   B f
5622 ///
5623 /// +-------------------------+--------+-----+
5624 /// |                         | Thumb2 | ARM |
5625 /// +-------------------------+--------+-----+
5626 /// | Call overhead in Bytes  |      4 |   4 |
5627 /// | Frame overhead in Bytes |      0 |   0 |
5628 /// | Stack fixup required    |     No |  No |
5629 /// +-------------------------+--------+-----+
5630 ///
5631 /// \p MachineOutlinerNoLRSave implies that the function should be called using
5632 /// a BL instruction, but doesn't require LR to be saved and restored. This
5633 /// happens when LR is known to be dead.
5634 ///
5635 /// That is,
5636 ///
5637 /// I1                                OUTLINED_FUNCTION:
5638 /// I2 --> BL OUTLINED_FUNCTION       I1
5639 /// I3                                I2
5640 ///                                   I3
5641 ///                                   BX LR
5642 ///
5643 /// +-------------------------+--------+-----+
5644 /// |                         | Thumb2 | ARM |
5645 /// +-------------------------+--------+-----+
5646 /// | Call overhead in Bytes  |      4 |   4 |
5647 /// | Frame overhead in Bytes |      4 |   4 |
5648 /// | Stack fixup required    |     No |  No |
5649 /// +-------------------------+--------+-----+
5650 ///
5651 /// \p MachineOutlinerRegSave implies that the function should be called with a
5652 /// save and restore of LR to an available register. This allows us to avoid
5653 /// stack fixups. Note that this outlining variant is compatible with the
5654 /// NoLRSave case.
5655 ///
5656 /// That is,
5657 ///
5658 /// I1     Save LR                    OUTLINED_FUNCTION:
5659 /// I2 --> BL OUTLINED_FUNCTION       I1
5660 /// I3     Restore LR                 I2
5661 ///                                   I3
5662 ///                                   BX LR
5663 ///
5664 /// +-------------------------+--------+-----+
5665 /// |                         | Thumb2 | ARM |
5666 /// +-------------------------+--------+-----+
5667 /// | Call overhead in Bytes  |      8 |  12 |
5668 /// | Frame overhead in Bytes |      2 |   4 |
5669 /// | Stack fixup required    |     No |  No |
5670 /// +-------------------------+--------+-----+
5671 ///
5672 /// \p MachineOutlinerDefault implies that the function should be called with
5673 /// a save and restore of LR to the stack.
5674 ///
5675 /// That is,
5676 ///
5677 /// I1     Save LR                    OUTLINED_FUNCTION:
5678 /// I2 --> BL OUTLINED_FUNCTION       I1
5679 /// I3     Restore LR                 I2
5680 ///                                   I3
5681 ///                                   BX LR
5682 ///
5683 /// +-------------------------+--------+-----+
5684 /// |                         | Thumb2 | ARM |
5685 /// +-------------------------+--------+-----+
5686 /// | Call overhead in Bytes  |      8 |  12 |
5687 /// | Frame overhead in Bytes |      2 |   4 |
5688 /// | Stack fixup required    |    Yes | Yes |
5689 /// +-------------------------+--------+-----+
5690 
5691 enum MachineOutlinerClass {
5692   MachineOutlinerTailCall,
5693   MachineOutlinerThunk,
5694   MachineOutlinerNoLRSave,
5695   MachineOutlinerRegSave,
5696   MachineOutlinerDefault
5697 };
5698 
5699 enum MachineOutlinerMBBFlags {
5700   LRUnavailableSomewhere = 0x2,
5701   HasCalls = 0x4,
5702   UnsafeRegsDead = 0x8
5703 };
5704 
5705 struct OutlinerCosts {
5706   const int CallTailCall;
5707   const int FrameTailCall;
5708   const int CallThunk;
5709   const int FrameThunk;
5710   const int CallNoLRSave;
5711   const int FrameNoLRSave;
5712   const int CallRegSave;
5713   const int FrameRegSave;
5714   const int CallDefault;
5715   const int FrameDefault;
5716   const int SaveRestoreLROnStack;
5717 
5718   OutlinerCosts(const ARMSubtarget &target)
5719       : CallTailCall(target.isThumb() ? 4 : 4),
5720         FrameTailCall(target.isThumb() ? 0 : 0),
5721         CallThunk(target.isThumb() ? 4 : 4),
5722         FrameThunk(target.isThumb() ? 0 : 0),
5723         CallNoLRSave(target.isThumb() ? 4 : 4),
5724         FrameNoLRSave(target.isThumb() ? 4 : 4),
5725         CallRegSave(target.isThumb() ? 8 : 12),
5726         FrameRegSave(target.isThumb() ? 2 : 4),
5727         CallDefault(target.isThumb() ? 8 : 12),
5728         FrameDefault(target.isThumb() ? 2 : 4),
5729         SaveRestoreLROnStack(target.isThumb() ? 8 : 8) {}
5730 };
5731 
5732 unsigned
5733 ARMBaseInstrInfo::findRegisterToSaveLRTo(const outliner::Candidate &C) const {
5734   assert(C.LRUWasSet && "LRU wasn't set?");
5735   MachineFunction *MF = C.getMF();
5736   const ARMBaseRegisterInfo *ARI = static_cast<const ARMBaseRegisterInfo *>(
5737       MF->getSubtarget().getRegisterInfo());
5738 
5739   BitVector regsReserved = ARI->getReservedRegs(*MF);
5740   // Check if there is an available register across the sequence that we can
5741   // use.
5742   for (unsigned Reg : ARM::rGPRRegClass) {
5743     if (!(Reg < regsReserved.size() && regsReserved.test(Reg)) &&
5744         Reg != ARM::LR &&  // LR is not reserved, but don't use it.
5745         Reg != ARM::R12 && // R12 is not guaranteed to be preserved.
5746         C.LRU.available(Reg) && C.UsedInSequence.available(Reg))
5747       return Reg;
5748   }
5749 
5750   // No suitable register. Return 0.
5751   return 0u;
5752 }
5753 
5754 // Compute liveness of LR at the point after the interval [I, E), which
5755 // denotes a *backward* iteration through instructions. Used only for return
5756 // basic blocks, which do not end with a tail call.
5757 static bool isLRAvailable(const TargetRegisterInfo &TRI,
5758                           MachineBasicBlock::reverse_iterator I,
5759                           MachineBasicBlock::reverse_iterator E) {
5760   // At the end of the function LR dead.
5761   bool Live = false;
5762   for (; I != E; ++I) {
5763     const MachineInstr &MI = *I;
5764 
5765     // Check defs of LR.
5766     if (MI.modifiesRegister(ARM::LR, &TRI))
5767       Live = false;
5768 
5769     // Check uses of LR.
5770     unsigned Opcode = MI.getOpcode();
5771     if (Opcode == ARM::BX_RET || Opcode == ARM::MOVPCLR ||
5772         Opcode == ARM::SUBS_PC_LR || Opcode == ARM::tBX_RET ||
5773         Opcode == ARM::tBXNS_RET) {
5774       // These instructions use LR, but it's not an (explicit or implicit)
5775       // operand.
5776       Live = true;
5777       continue;
5778     }
5779     if (MI.readsRegister(ARM::LR, &TRI))
5780       Live = true;
5781   }
5782   return !Live;
5783 }
5784 
5785 outliner::OutlinedFunction ARMBaseInstrInfo::getOutliningCandidateInfo(
5786     std::vector<outliner::Candidate> &RepeatedSequenceLocs) const {
5787   outliner::Candidate &FirstCand = RepeatedSequenceLocs[0];
5788   unsigned SequenceSize =
5789       std::accumulate(FirstCand.front(), std::next(FirstCand.back()), 0,
5790                       [this](unsigned Sum, const MachineInstr &MI) {
5791                         return Sum + getInstSizeInBytes(MI);
5792                       });
5793 
5794   // Properties about candidate MBBs that hold for all of them.
5795   unsigned FlagsSetInAll = 0xF;
5796 
5797   // Compute liveness information for each candidate, and set FlagsSetInAll.
5798   const TargetRegisterInfo &TRI = getRegisterInfo();
5799   std::for_each(
5800       RepeatedSequenceLocs.begin(), RepeatedSequenceLocs.end(),
5801       [&FlagsSetInAll](outliner::Candidate &C) { FlagsSetInAll &= C.Flags; });
5802 
5803   // According to the ARM Procedure Call Standard, the following are
5804   // undefined on entry/exit from a function call:
5805   //
5806   // * Register R12(IP),
5807   // * Condition codes (and thus the CPSR register)
5808   //
5809   // Since we control the instructions which are part of the outlined regions
5810   // we don't need to be fully compliant with the AAPCS, but we have to
5811   // guarantee that if a veneer is inserted at link time the code is still
5812   // correct.  Because of this, we can't outline any sequence of instructions
5813   // where one of these registers is live into/across it. Thus, we need to
5814   // delete those candidates.
5815   auto CantGuaranteeValueAcrossCall = [&TRI](outliner::Candidate &C) {
5816     // If the unsafe registers in this block are all dead, then we don't need
5817     // to compute liveness here.
5818     if (C.Flags & UnsafeRegsDead)
5819       return false;
5820     C.initLRU(TRI);
5821     LiveRegUnits LRU = C.LRU;
5822     return (!LRU.available(ARM::R12) || !LRU.available(ARM::CPSR));
5823   };
5824 
5825   // Are there any candidates where those registers are live?
5826   if (!(FlagsSetInAll & UnsafeRegsDead)) {
5827     // Erase every candidate that violates the restrictions above. (It could be
5828     // true that we have viable candidates, so it's not worth bailing out in
5829     // the case that, say, 1 out of 20 candidates violate the restructions.)
5830     llvm::erase_if(RepeatedSequenceLocs, CantGuaranteeValueAcrossCall);
5831 
5832     // If the sequence doesn't have enough candidates left, then we're done.
5833     if (RepeatedSequenceLocs.size() < 2)
5834       return outliner::OutlinedFunction();
5835   }
5836 
5837   // At this point, we have only "safe" candidates to outline. Figure out
5838   // frame + call instruction information.
5839 
5840   unsigned LastInstrOpcode = RepeatedSequenceLocs[0].back()->getOpcode();
5841 
5842   // Helper lambda which sets call information for every candidate.
5843   auto SetCandidateCallInfo =
5844       [&RepeatedSequenceLocs](unsigned CallID, unsigned NumBytesForCall) {
5845         for (outliner::Candidate &C : RepeatedSequenceLocs)
5846           C.setCallInfo(CallID, NumBytesForCall);
5847       };
5848 
5849   OutlinerCosts Costs(Subtarget);
5850   unsigned FrameID = MachineOutlinerDefault;
5851   unsigned NumBytesToCreateFrame = Costs.FrameDefault;
5852 
5853   // If the last instruction in any candidate is a terminator, then we should
5854   // tail call all of the candidates.
5855   if (RepeatedSequenceLocs[0].back()->isTerminator()) {
5856     FrameID = MachineOutlinerTailCall;
5857     NumBytesToCreateFrame = Costs.FrameTailCall;
5858     SetCandidateCallInfo(MachineOutlinerTailCall, Costs.CallTailCall);
5859   } else if (LastInstrOpcode == ARM::BL || LastInstrOpcode == ARM::BLX ||
5860              LastInstrOpcode == ARM::BLX_noip || LastInstrOpcode == ARM::tBL ||
5861              LastInstrOpcode == ARM::tBLXr ||
5862              LastInstrOpcode == ARM::tBLXr_noip ||
5863              LastInstrOpcode == ARM::tBLXi) {
5864     FrameID = MachineOutlinerThunk;
5865     NumBytesToCreateFrame = Costs.FrameThunk;
5866     SetCandidateCallInfo(MachineOutlinerThunk, Costs.CallThunk);
5867   } else {
5868     // We need to decide how to emit calls + frames. We can always emit the same
5869     // frame if we don't need to save to the stack. If we have to save to the
5870     // stack, then we need a different frame.
5871     unsigned NumBytesNoStackCalls = 0;
5872     std::vector<outliner::Candidate> CandidatesWithoutStackFixups;
5873 
5874     for (outliner::Candidate &C : RepeatedSequenceLocs) {
5875       C.initLRU(TRI);
5876       // LR liveness is overestimated in return blocks, unless they end with a
5877       // tail call.
5878       const auto Last = C.getMBB()->rbegin();
5879       const bool LRIsAvailable =
5880           C.getMBB()->isReturnBlock() && !Last->isCall()
5881               ? isLRAvailable(TRI, Last,
5882                               (MachineBasicBlock::reverse_iterator)C.front())
5883               : C.LRU.available(ARM::LR);
5884       if (LRIsAvailable) {
5885         FrameID = MachineOutlinerNoLRSave;
5886         NumBytesNoStackCalls += Costs.CallNoLRSave;
5887         C.setCallInfo(MachineOutlinerNoLRSave, Costs.CallNoLRSave);
5888         CandidatesWithoutStackFixups.push_back(C);
5889       }
5890 
5891       // Is an unused register available? If so, we won't modify the stack, so
5892       // we can outline with the same frame type as those that don't save LR.
5893       else if (findRegisterToSaveLRTo(C)) {
5894         FrameID = MachineOutlinerRegSave;
5895         NumBytesNoStackCalls += Costs.CallRegSave;
5896         C.setCallInfo(MachineOutlinerRegSave, Costs.CallRegSave);
5897         CandidatesWithoutStackFixups.push_back(C);
5898       }
5899 
5900       // Is SP used in the sequence at all? If not, we don't have to modify
5901       // the stack, so we are guaranteed to get the same frame.
5902       else if (C.UsedInSequence.available(ARM::SP)) {
5903         NumBytesNoStackCalls += Costs.CallDefault;
5904         C.setCallInfo(MachineOutlinerDefault, Costs.CallDefault);
5905         CandidatesWithoutStackFixups.push_back(C);
5906       }
5907 
5908       // If we outline this, we need to modify the stack. Pretend we don't
5909       // outline this by saving all of its bytes.
5910       else
5911         NumBytesNoStackCalls += SequenceSize;
5912     }
5913 
5914     // If there are no places where we have to save LR, then note that we don't
5915     // have to update the stack. Otherwise, give every candidate the default
5916     // call type
5917     if (NumBytesNoStackCalls <=
5918         RepeatedSequenceLocs.size() * Costs.CallDefault) {
5919       RepeatedSequenceLocs = CandidatesWithoutStackFixups;
5920       FrameID = MachineOutlinerNoLRSave;
5921     } else
5922       SetCandidateCallInfo(MachineOutlinerDefault, Costs.CallDefault);
5923   }
5924 
5925   // Does every candidate's MBB contain a call?  If so, then we might have a
5926   // call in the range.
5927   if (FlagsSetInAll & MachineOutlinerMBBFlags::HasCalls) {
5928     // check if the range contains a call.  These require a save + restore of
5929     // the link register.
5930     if (std::any_of(FirstCand.front(), FirstCand.back(),
5931                     [](const MachineInstr &MI) { return MI.isCall(); }))
5932       NumBytesToCreateFrame += Costs.SaveRestoreLROnStack;
5933 
5934     // Handle the last instruction separately.  If it is tail call, then the
5935     // last instruction is a call, we don't want to save + restore in this
5936     // case.  However, it could be possible that the last instruction is a
5937     // call without it being valid to tail call this sequence.  We should
5938     // consider this as well.
5939     else if (FrameID != MachineOutlinerThunk &&
5940              FrameID != MachineOutlinerTailCall && FirstCand.back()->isCall())
5941       NumBytesToCreateFrame += Costs.SaveRestoreLROnStack;
5942   }
5943 
5944   return outliner::OutlinedFunction(RepeatedSequenceLocs, SequenceSize,
5945                                     NumBytesToCreateFrame, FrameID);
5946 }
5947 
5948 bool ARMBaseInstrInfo::checkAndUpdateStackOffset(MachineInstr *MI,
5949                                                  int64_t Fixup,
5950                                                  bool Updt) const {
5951   int SPIdx = MI->findRegisterUseOperandIdx(ARM::SP);
5952   unsigned AddrMode = (MI->getDesc().TSFlags & ARMII::AddrModeMask);
5953   if (SPIdx < 0)
5954     // No SP operand
5955     return true;
5956   else if (SPIdx != 1 && (AddrMode != ARMII::AddrModeT2_i8s4 || SPIdx != 2))
5957     // If SP is not the base register we can't do much
5958     return false;
5959 
5960   // Stack might be involved but addressing mode doesn't handle any offset.
5961   // Rq: AddrModeT1_[1|2|4] don't operate on SP
5962   if (AddrMode == ARMII::AddrMode1        // Arithmetic instructions
5963       || AddrMode == ARMII::AddrMode4     // Load/Store Multiple
5964       || AddrMode == ARMII::AddrMode6     // Neon Load/Store Multiple
5965       || AddrMode == ARMII::AddrModeT2_so // SP can't be used as based register
5966       || AddrMode == ARMII::AddrModeT2_pc // PCrel access
5967       || AddrMode == ARMII::AddrMode2     // Used by PRE and POST indexed LD/ST
5968       || AddrMode == ARMII::AddrModeT2_i7 // v8.1-M MVE
5969       || AddrMode == ARMII::AddrModeT2_i7s2 // v8.1-M MVE
5970       || AddrMode == ARMII::AddrModeT2_i7s4 // v8.1-M sys regs VLDR/VSTR
5971       || AddrMode == ARMII::AddrModeNone)
5972     return false;
5973 
5974   unsigned NumOps = MI->getDesc().getNumOperands();
5975   unsigned ImmIdx = NumOps - 3;
5976 
5977   const MachineOperand &Offset = MI->getOperand(ImmIdx);
5978   assert(Offset.isImm() && "Is not an immediate");
5979   int64_t OffVal = Offset.getImm();
5980 
5981   if (OffVal < 0)
5982     // Don't override data if the are below SP.
5983     return false;
5984 
5985   unsigned NumBits = 0;
5986   unsigned Scale = 1;
5987 
5988   switch (AddrMode) {
5989   case ARMII::AddrMode3:
5990     if (ARM_AM::getAM3Op(OffVal) == ARM_AM::sub)
5991       return false;
5992     OffVal = ARM_AM::getAM3Offset(OffVal);
5993     NumBits = 8;
5994     break;
5995   case ARMII::AddrMode5:
5996     if (ARM_AM::getAM5Op(OffVal) == ARM_AM::sub)
5997       return false;
5998     OffVal = ARM_AM::getAM5Offset(OffVal);
5999     NumBits = 8;
6000     Scale = 4;
6001     break;
6002   case ARMII::AddrMode5FP16:
6003     if (ARM_AM::getAM5FP16Op(OffVal) == ARM_AM::sub)
6004       return false;
6005     OffVal = ARM_AM::getAM5FP16Offset(OffVal);
6006     NumBits = 8;
6007     Scale = 2;
6008     break;
6009   case ARMII::AddrModeT2_i8:
6010     NumBits = 8;
6011     break;
6012   case ARMII::AddrModeT2_i8s4:
6013     // FIXME: Values are already scaled in this addressing mode.
6014     assert((Fixup & 3) == 0 && "Can't encode this offset!");
6015     NumBits = 10;
6016     break;
6017   case ARMII::AddrModeT2_ldrex:
6018     NumBits = 8;
6019     Scale = 4;
6020     break;
6021   case ARMII::AddrModeT2_i12:
6022   case ARMII::AddrMode_i12:
6023     NumBits = 12;
6024     break;
6025   case ARMII::AddrModeT1_s: // SP-relative LD/ST
6026     NumBits = 8;
6027     Scale = 4;
6028     break;
6029   default:
6030     llvm_unreachable("Unsupported addressing mode!");
6031   }
6032   // Make sure the offset is encodable for instructions that scale the
6033   // immediate.
6034   assert(((OffVal * Scale + Fixup) & (Scale - 1)) == 0 &&
6035          "Can't encode this offset!");
6036   OffVal += Fixup / Scale;
6037 
6038   unsigned Mask = (1 << NumBits) - 1;
6039 
6040   if (OffVal <= Mask) {
6041     if (Updt)
6042       MI->getOperand(ImmIdx).setImm(OffVal);
6043     return true;
6044   }
6045 
6046   return false;
6047 
6048 }
6049 
6050 bool ARMBaseInstrInfo::isFunctionSafeToOutlineFrom(
6051     MachineFunction &MF, bool OutlineFromLinkOnceODRs) const {
6052   const Function &F = MF.getFunction();
6053 
6054   // Can F be deduplicated by the linker? If it can, don't outline from it.
6055   if (!OutlineFromLinkOnceODRs && F.hasLinkOnceODRLinkage())
6056     return false;
6057 
6058   // Don't outline from functions with section markings; the program could
6059   // expect that all the code is in the named section.
6060   // FIXME: Allow outlining from multiple functions with the same section
6061   // marking.
6062   if (F.hasSection())
6063     return false;
6064 
6065   // FIXME: Thumb1 outlining is not handled
6066   if (MF.getInfo<ARMFunctionInfo>()->isThumb1OnlyFunction())
6067     return false;
6068 
6069   // It's safe to outline from MF.
6070   return true;
6071 }
6072 
6073 bool ARMBaseInstrInfo::isMBBSafeToOutlineFrom(MachineBasicBlock &MBB,
6074                                               unsigned &Flags) const {
6075   // Check if LR is available through all of the MBB. If it's not, then set
6076   // a flag.
6077   assert(MBB.getParent()->getRegInfo().tracksLiveness() &&
6078          "Suitable Machine Function for outlining must track liveness");
6079 
6080   LiveRegUnits LRU(getRegisterInfo());
6081 
6082   std::for_each(MBB.rbegin(), MBB.rend(),
6083                 [&LRU](MachineInstr &MI) { LRU.accumulate(MI); });
6084 
6085   // Check if each of the unsafe registers are available...
6086   bool R12AvailableInBlock = LRU.available(ARM::R12);
6087   bool CPSRAvailableInBlock = LRU.available(ARM::CPSR);
6088 
6089   // If all of these are dead (and not live out), we know we don't have to check
6090   // them later.
6091   if (R12AvailableInBlock && CPSRAvailableInBlock)
6092     Flags |= MachineOutlinerMBBFlags::UnsafeRegsDead;
6093 
6094   // Now, add the live outs to the set.
6095   LRU.addLiveOuts(MBB);
6096 
6097   // If any of these registers is available in the MBB, but also a live out of
6098   // the block, then we know outlining is unsafe.
6099   if (R12AvailableInBlock && !LRU.available(ARM::R12))
6100     return false;
6101   if (CPSRAvailableInBlock && !LRU.available(ARM::CPSR))
6102     return false;
6103 
6104   // Check if there's a call inside this MachineBasicBlock.  If there is, then
6105   // set a flag.
6106   if (any_of(MBB, [](MachineInstr &MI) { return MI.isCall(); }))
6107     Flags |= MachineOutlinerMBBFlags::HasCalls;
6108 
6109   // LR liveness is overestimated in return blocks.
6110 
6111   bool LRIsAvailable =
6112       MBB.isReturnBlock() && !MBB.back().isCall()
6113           ? isLRAvailable(getRegisterInfo(), MBB.rbegin(), MBB.rend())
6114           : LRU.available(ARM::LR);
6115   if (!LRIsAvailable)
6116     Flags |= MachineOutlinerMBBFlags::LRUnavailableSomewhere;
6117 
6118   return true;
6119 }
6120 
6121 outliner::InstrType
6122 ARMBaseInstrInfo::getOutliningType(MachineBasicBlock::iterator &MIT,
6123                                    unsigned Flags) const {
6124   MachineInstr &MI = *MIT;
6125   const TargetRegisterInfo *TRI = &getRegisterInfo();
6126 
6127   // Be conservative with inline ASM
6128   if (MI.isInlineAsm())
6129     return outliner::InstrType::Illegal;
6130 
6131   // Don't allow debug values to impact outlining type.
6132   if (MI.isDebugInstr() || MI.isIndirectDebugValue())
6133     return outliner::InstrType::Invisible;
6134 
6135   // At this point, KILL or IMPLICIT_DEF instructions don't really tell us much
6136   // so we can go ahead and skip over them.
6137   if (MI.isKill() || MI.isImplicitDef())
6138     return outliner::InstrType::Invisible;
6139 
6140   // PIC instructions contain labels, outlining them would break offset
6141   // computing.  unsigned Opc = MI.getOpcode();
6142   unsigned Opc = MI.getOpcode();
6143   if (Opc == ARM::tPICADD || Opc == ARM::PICADD || Opc == ARM::PICSTR ||
6144       Opc == ARM::PICSTRB || Opc == ARM::PICSTRH || Opc == ARM::PICLDR ||
6145       Opc == ARM::PICLDRB || Opc == ARM::PICLDRH || Opc == ARM::PICLDRSB ||
6146       Opc == ARM::PICLDRSH || Opc == ARM::t2LDRpci_pic ||
6147       Opc == ARM::t2MOVi16_ga_pcrel || Opc == ARM::t2MOVTi16_ga_pcrel ||
6148       Opc == ARM::t2MOV_ga_pcrel)
6149     return outliner::InstrType::Illegal;
6150 
6151   // Be conservative with ARMv8.1 MVE instructions.
6152   if (Opc == ARM::t2BF_LabelPseudo || Opc == ARM::t2DoLoopStart ||
6153       Opc == ARM::t2DoLoopStartTP || Opc == ARM::t2WhileLoopStart ||
6154       Opc == ARM::t2WhileLoopStartLR || Opc == ARM::t2WhileLoopStartTP ||
6155       Opc == ARM::t2LoopDec || Opc == ARM::t2LoopEnd ||
6156       Opc == ARM::t2LoopEndDec)
6157     return outliner::InstrType::Illegal;
6158 
6159   const MCInstrDesc &MCID = MI.getDesc();
6160   uint64_t MIFlags = MCID.TSFlags;
6161   if ((MIFlags & ARMII::DomainMask) == ARMII::DomainMVE)
6162     return outliner::InstrType::Illegal;
6163 
6164   // Is this a terminator for a basic block?
6165   if (MI.isTerminator()) {
6166     // Don't outline if the branch is not unconditional.
6167     if (isPredicated(MI))
6168       return outliner::InstrType::Illegal;
6169 
6170     // Is this the end of a function?
6171     if (MI.getParent()->succ_empty())
6172       return outliner::InstrType::Legal;
6173 
6174     // It's not, so don't outline it.
6175     return outliner::InstrType::Illegal;
6176   }
6177 
6178   // Make sure none of the operands are un-outlinable.
6179   for (const MachineOperand &MOP : MI.operands()) {
6180     if (MOP.isCPI() || MOP.isJTI() || MOP.isCFIIndex() || MOP.isFI() ||
6181         MOP.isTargetIndex())
6182       return outliner::InstrType::Illegal;
6183   }
6184 
6185   // Don't outline if link register or program counter value are used.
6186   if (MI.readsRegister(ARM::LR, TRI) || MI.readsRegister(ARM::PC, TRI))
6187     return outliner::InstrType::Illegal;
6188 
6189   if (MI.isCall()) {
6190     // Get the function associated with the call.  Look at each operand and find
6191     // the one that represents the calle and get its name.
6192     const Function *Callee = nullptr;
6193     for (const MachineOperand &MOP : MI.operands()) {
6194       if (MOP.isGlobal()) {
6195         Callee = dyn_cast<Function>(MOP.getGlobal());
6196         break;
6197       }
6198     }
6199 
6200     // Dont't outline calls to "mcount" like functions, in particular Linux
6201     // kernel function tracing relies on it.
6202     if (Callee &&
6203         (Callee->getName() == "\01__gnu_mcount_nc" ||
6204          Callee->getName() == "\01mcount" || Callee->getName() == "__mcount"))
6205       return outliner::InstrType::Illegal;
6206 
6207     // If we don't know anything about the callee, assume it depends on the
6208     // stack layout of the caller. In that case, it's only legal to outline
6209     // as a tail-call. Explicitly list the call instructions we know about so
6210     // we don't get unexpected results with call pseudo-instructions.
6211     auto UnknownCallOutlineType = outliner::InstrType::Illegal;
6212     if (Opc == ARM::BL || Opc == ARM::tBL || Opc == ARM::BLX ||
6213         Opc == ARM::BLX_noip || Opc == ARM::tBLXr || Opc == ARM::tBLXr_noip ||
6214         Opc == ARM::tBLXi)
6215       UnknownCallOutlineType = outliner::InstrType::LegalTerminator;
6216 
6217     if (!Callee)
6218       return UnknownCallOutlineType;
6219 
6220     // We have a function we have information about.  Check if it's something we
6221     // can safely outline.
6222     MachineFunction *MF = MI.getParent()->getParent();
6223     MachineFunction *CalleeMF = MF->getMMI().getMachineFunction(*Callee);
6224 
6225     // We don't know what's going on with the callee at all.  Don't touch it.
6226     if (!CalleeMF)
6227       return UnknownCallOutlineType;
6228 
6229     // Check if we know anything about the callee saves on the function. If we
6230     // don't, then don't touch it, since that implies that we haven't computed
6231     // anything about its stack frame yet.
6232     MachineFrameInfo &MFI = CalleeMF->getFrameInfo();
6233     if (!MFI.isCalleeSavedInfoValid() || MFI.getStackSize() > 0 ||
6234         MFI.getNumObjects() > 0)
6235       return UnknownCallOutlineType;
6236 
6237     // At this point, we can say that CalleeMF ought to not pass anything on the
6238     // stack. Therefore, we can outline it.
6239     return outliner::InstrType::Legal;
6240   }
6241 
6242   // Since calls are handled, don't touch LR or PC
6243   if (MI.modifiesRegister(ARM::LR, TRI) || MI.modifiesRegister(ARM::PC, TRI))
6244     return outliner::InstrType::Illegal;
6245 
6246   // Does this use the stack?
6247   if (MI.modifiesRegister(ARM::SP, TRI) || MI.readsRegister(ARM::SP, TRI)) {
6248     // True if there is no chance that any outlined candidate from this range
6249     // could require stack fixups. That is, both
6250     // * LR is available in the range (No save/restore around call)
6251     // * The range doesn't include calls (No save/restore in outlined frame)
6252     // are true.
6253     // FIXME: This is very restrictive; the flags check the whole block,
6254     // not just the bit we will try to outline.
6255     bool MightNeedStackFixUp =
6256         (Flags & (MachineOutlinerMBBFlags::LRUnavailableSomewhere |
6257                   MachineOutlinerMBBFlags::HasCalls));
6258 
6259     if (!MightNeedStackFixUp)
6260       return outliner::InstrType::Legal;
6261 
6262     // Any modification of SP will break our code to save/restore LR.
6263     // FIXME: We could handle some instructions which add a constant offset to
6264     // SP, with a bit more work.
6265     if (MI.modifiesRegister(ARM::SP, TRI))
6266       return outliner::InstrType::Illegal;
6267 
6268     // At this point, we have a stack instruction that we might need to fix up.
6269     // up. We'll handle it if it's a load or store.
6270     if (checkAndUpdateStackOffset(&MI, Subtarget.getStackAlignment().value(),
6271                                   false))
6272       return outliner::InstrType::Legal;
6273 
6274     // We can't fix it up, so don't outline it.
6275     return outliner::InstrType::Illegal;
6276   }
6277 
6278   // Be conservative with IT blocks.
6279   if (MI.readsRegister(ARM::ITSTATE, TRI) ||
6280       MI.modifiesRegister(ARM::ITSTATE, TRI))
6281     return outliner::InstrType::Illegal;
6282 
6283   // Don't outline positions.
6284   if (MI.isPosition())
6285     return outliner::InstrType::Illegal;
6286 
6287   return outliner::InstrType::Legal;
6288 }
6289 
6290 void ARMBaseInstrInfo::fixupPostOutline(MachineBasicBlock &MBB) const {
6291   for (MachineInstr &MI : MBB) {
6292     checkAndUpdateStackOffset(&MI, Subtarget.getStackAlignment().value(), true);
6293   }
6294 }
6295 
6296 void ARMBaseInstrInfo::saveLROnStack(MachineBasicBlock &MBB,
6297                                      MachineBasicBlock::iterator It) const {
6298   unsigned Opc = Subtarget.isThumb() ? ARM::t2STR_PRE : ARM::STR_PRE_IMM;
6299   int Align = -Subtarget.getStackAlignment().value();
6300   BuildMI(MBB, It, DebugLoc(), get(Opc), ARM::SP)
6301     .addReg(ARM::LR, RegState::Kill)
6302     .addReg(ARM::SP)
6303     .addImm(Align)
6304     .add(predOps(ARMCC::AL));
6305 }
6306 
6307 void ARMBaseInstrInfo::emitCFIForLRSaveOnStack(
6308     MachineBasicBlock &MBB, MachineBasicBlock::iterator It) const {
6309   MachineFunction &MF = *MBB.getParent();
6310   const MCRegisterInfo *MRI = Subtarget.getRegisterInfo();
6311   unsigned DwarfLR = MRI->getDwarfRegNum(ARM::LR, true);
6312   int Align = Subtarget.getStackAlignment().value();
6313   // Add a CFI saying the stack was moved down.
6314   int64_t StackPosEntry =
6315       MF.addFrameInst(MCCFIInstruction::cfiDefCfaOffset(nullptr, Align));
6316   BuildMI(MBB, It, DebugLoc(), get(ARM::CFI_INSTRUCTION))
6317       .addCFIIndex(StackPosEntry)
6318       .setMIFlags(MachineInstr::FrameSetup);
6319 
6320   // Add a CFI saying that the LR that we want to find is now higher than
6321   // before.
6322   int64_t LRPosEntry =
6323       MF.addFrameInst(MCCFIInstruction::createOffset(nullptr, DwarfLR, -Align));
6324   BuildMI(MBB, It, DebugLoc(), get(ARM::CFI_INSTRUCTION))
6325       .addCFIIndex(LRPosEntry)
6326       .setMIFlags(MachineInstr::FrameSetup);
6327 }
6328 
6329 void ARMBaseInstrInfo::emitCFIForLRSaveToReg(MachineBasicBlock &MBB,
6330                                              MachineBasicBlock::iterator It,
6331                                              Register Reg) const {
6332   MachineFunction &MF = *MBB.getParent();
6333   const MCRegisterInfo *MRI = Subtarget.getRegisterInfo();
6334   unsigned DwarfLR = MRI->getDwarfRegNum(ARM::LR, true);
6335   unsigned DwarfReg = MRI->getDwarfRegNum(Reg, true);
6336 
6337   int64_t LRPosEntry = MF.addFrameInst(
6338       MCCFIInstruction::createRegister(nullptr, DwarfLR, DwarfReg));
6339   BuildMI(MBB, It, DebugLoc(), get(ARM::CFI_INSTRUCTION))
6340       .addCFIIndex(LRPosEntry)
6341       .setMIFlags(MachineInstr::FrameSetup);
6342 }
6343 
6344 void ARMBaseInstrInfo::restoreLRFromStack(
6345     MachineBasicBlock &MBB, MachineBasicBlock::iterator It) const {
6346   unsigned Opc = Subtarget.isThumb() ? ARM::t2LDR_POST : ARM::LDR_POST_IMM;
6347   MachineInstrBuilder MIB = BuildMI(MBB, It, DebugLoc(), get(Opc), ARM::LR)
6348     .addReg(ARM::SP, RegState::Define)
6349     .addReg(ARM::SP);
6350   if (!Subtarget.isThumb())
6351     MIB.addReg(0);
6352   MIB.addImm(Subtarget.getStackAlignment().value()).add(predOps(ARMCC::AL));
6353 }
6354 
6355 void ARMBaseInstrInfo::emitCFIForLRRestoreFromStack(
6356     MachineBasicBlock &MBB, MachineBasicBlock::iterator It) const {
6357   // Now stack has moved back up...
6358   MachineFunction &MF = *MBB.getParent();
6359   const MCRegisterInfo *MRI = Subtarget.getRegisterInfo();
6360   unsigned DwarfLR = MRI->getDwarfRegNum(ARM::LR, true);
6361   int64_t StackPosEntry =
6362       MF.addFrameInst(MCCFIInstruction::cfiDefCfaOffset(nullptr, 0));
6363   BuildMI(MBB, It, DebugLoc(), get(ARM::CFI_INSTRUCTION))
6364       .addCFIIndex(StackPosEntry)
6365       .setMIFlags(MachineInstr::FrameDestroy);
6366 
6367   // ... and we have restored LR.
6368   int64_t LRPosEntry =
6369       MF.addFrameInst(MCCFIInstruction::createRestore(nullptr, DwarfLR));
6370   BuildMI(MBB, It, DebugLoc(), get(ARM::CFI_INSTRUCTION))
6371       .addCFIIndex(LRPosEntry)
6372       .setMIFlags(MachineInstr::FrameDestroy);
6373 }
6374 
6375 void ARMBaseInstrInfo::emitCFIForLRRestoreFromReg(
6376     MachineBasicBlock &MBB, MachineBasicBlock::iterator It) const {
6377   MachineFunction &MF = *MBB.getParent();
6378   const MCRegisterInfo *MRI = Subtarget.getRegisterInfo();
6379   unsigned DwarfLR = MRI->getDwarfRegNum(ARM::LR, true);
6380 
6381   int64_t LRPosEntry =
6382       MF.addFrameInst(MCCFIInstruction::createRestore(nullptr, DwarfLR));
6383   BuildMI(MBB, It, DebugLoc(), get(ARM::CFI_INSTRUCTION))
6384       .addCFIIndex(LRPosEntry)
6385       .setMIFlags(MachineInstr::FrameDestroy);
6386 }
6387 
6388 void ARMBaseInstrInfo::buildOutlinedFrame(
6389     MachineBasicBlock &MBB, MachineFunction &MF,
6390     const outliner::OutlinedFunction &OF) const {
6391   // For thunk outlining, rewrite the last instruction from a call to a
6392   // tail-call.
6393   if (OF.FrameConstructionID == MachineOutlinerThunk) {
6394     MachineInstr *Call = &*--MBB.instr_end();
6395     bool isThumb = Subtarget.isThumb();
6396     unsigned FuncOp = isThumb ? 2 : 0;
6397     unsigned Opc = Call->getOperand(FuncOp).isReg()
6398                        ? isThumb ? ARM::tTAILJMPr : ARM::TAILJMPr
6399                        : isThumb ? Subtarget.isTargetMachO() ? ARM::tTAILJMPd
6400                                                              : ARM::tTAILJMPdND
6401                                  : ARM::TAILJMPd;
6402     MachineInstrBuilder MIB = BuildMI(MBB, MBB.end(), DebugLoc(), get(Opc))
6403                                   .add(Call->getOperand(FuncOp));
6404     if (isThumb && !Call->getOperand(FuncOp).isReg())
6405       MIB.add(predOps(ARMCC::AL));
6406     Call->eraseFromParent();
6407   }
6408 
6409   // Is there a call in the outlined range?
6410   auto IsNonTailCall = [](MachineInstr &MI) {
6411     return MI.isCall() && !MI.isReturn();
6412   };
6413   if (llvm::any_of(MBB.instrs(), IsNonTailCall)) {
6414     MachineBasicBlock::iterator It = MBB.begin();
6415     MachineBasicBlock::iterator Et = MBB.end();
6416 
6417     if (OF.FrameConstructionID == MachineOutlinerTailCall ||
6418         OF.FrameConstructionID == MachineOutlinerThunk)
6419       Et = std::prev(MBB.end());
6420 
6421     // We have to save and restore LR, we need to add it to the liveins if it
6422     // is not already part of the set.  This is suffient since outlined
6423     // functions only have one block.
6424     if (!MBB.isLiveIn(ARM::LR))
6425       MBB.addLiveIn(ARM::LR);
6426 
6427     // Insert a save before the outlined region
6428     saveLROnStack(MBB, It);
6429     emitCFIForLRSaveOnStack(MBB, It);
6430 
6431     // Fix up the instructions in the range, since we're going to modify the
6432     // stack.
6433     assert(OF.FrameConstructionID != MachineOutlinerDefault &&
6434            "Can only fix up stack references once");
6435     fixupPostOutline(MBB);
6436 
6437     // Insert a restore before the terminator for the function.  Restore LR.
6438     restoreLRFromStack(MBB, Et);
6439     emitCFIForLRRestoreFromStack(MBB, Et);
6440   }
6441 
6442   // If this is a tail call outlined function, then there's already a return.
6443   if (OF.FrameConstructionID == MachineOutlinerTailCall ||
6444       OF.FrameConstructionID == MachineOutlinerThunk)
6445     return;
6446 
6447   // Here we have to insert the return ourselves.  Get the correct opcode from
6448   // current feature set.
6449   BuildMI(MBB, MBB.end(), DebugLoc(), get(Subtarget.getReturnOpcode()))
6450       .add(predOps(ARMCC::AL));
6451 
6452   // Did we have to modify the stack by saving the link register?
6453   if (OF.FrameConstructionID != MachineOutlinerDefault &&
6454       OF.Candidates[0].CallConstructionID != MachineOutlinerDefault)
6455     return;
6456 
6457   // We modified the stack.
6458   // Walk over the basic block and fix up all the stack accesses.
6459   fixupPostOutline(MBB);
6460 }
6461 
6462 MachineBasicBlock::iterator ARMBaseInstrInfo::insertOutlinedCall(
6463     Module &M, MachineBasicBlock &MBB, MachineBasicBlock::iterator &It,
6464     MachineFunction &MF, const outliner::Candidate &C) const {
6465   MachineInstrBuilder MIB;
6466   MachineBasicBlock::iterator CallPt;
6467   unsigned Opc;
6468   bool isThumb = Subtarget.isThumb();
6469 
6470   // Are we tail calling?
6471   if (C.CallConstructionID == MachineOutlinerTailCall) {
6472     // If yes, then we can just branch to the label.
6473     Opc = isThumb
6474               ? Subtarget.isTargetMachO() ? ARM::tTAILJMPd : ARM::tTAILJMPdND
6475               : ARM::TAILJMPd;
6476     MIB = BuildMI(MF, DebugLoc(), get(Opc))
6477               .addGlobalAddress(M.getNamedValue(MF.getName()));
6478     if (isThumb)
6479       MIB.add(predOps(ARMCC::AL));
6480     It = MBB.insert(It, MIB);
6481     return It;
6482   }
6483 
6484   // Create the call instruction.
6485   Opc = isThumb ? ARM::tBL : ARM::BL;
6486   MachineInstrBuilder CallMIB = BuildMI(MF, DebugLoc(), get(Opc));
6487   if (isThumb)
6488     CallMIB.add(predOps(ARMCC::AL));
6489   CallMIB.addGlobalAddress(M.getNamedValue(MF.getName()));
6490 
6491   if (C.CallConstructionID == MachineOutlinerNoLRSave ||
6492       C.CallConstructionID == MachineOutlinerThunk) {
6493     // No, so just insert the call.
6494     It = MBB.insert(It, CallMIB);
6495     return It;
6496   }
6497 
6498   const ARMFunctionInfo &AFI = *C.getMF()->getInfo<ARMFunctionInfo>();
6499   // Can we save to a register?
6500   if (C.CallConstructionID == MachineOutlinerRegSave) {
6501     unsigned Reg = findRegisterToSaveLRTo(C);
6502     assert(Reg != 0 && "No callee-saved register available?");
6503 
6504     // Save and restore LR from that register.
6505     copyPhysReg(MBB, It, DebugLoc(), Reg, ARM::LR, true);
6506     if (!AFI.isLRSpilled())
6507       emitCFIForLRSaveToReg(MBB, It, Reg);
6508     CallPt = MBB.insert(It, CallMIB);
6509     copyPhysReg(MBB, It, DebugLoc(), ARM::LR, Reg, true);
6510     if (!AFI.isLRSpilled())
6511       emitCFIForLRRestoreFromReg(MBB, It);
6512     It--;
6513     return CallPt;
6514   }
6515   // We have the default case. Save and restore from SP.
6516   if (!MBB.isLiveIn(ARM::LR))
6517     MBB.addLiveIn(ARM::LR);
6518   saveLROnStack(MBB, It);
6519   if (!AFI.isLRSpilled())
6520     emitCFIForLRSaveOnStack(MBB, It);
6521   CallPt = MBB.insert(It, CallMIB);
6522   restoreLRFromStack(MBB, It);
6523   if (!AFI.isLRSpilled())
6524     emitCFIForLRRestoreFromStack(MBB, It);
6525   It--;
6526   return CallPt;
6527 }
6528 
6529 bool ARMBaseInstrInfo::shouldOutlineFromFunctionByDefault(
6530     MachineFunction &MF) const {
6531   return Subtarget.isMClass() && MF.getFunction().hasMinSize();
6532 }
6533 
6534 bool ARMBaseInstrInfo::isReallyTriviallyReMaterializable(const MachineInstr &MI,
6535                                                          AAResults *AA) const {
6536   // Try hard to rematerialize any VCTPs because if we spill P0, it will block
6537   // the tail predication conversion. This means that the element count
6538   // register has to be live for longer, but that has to be better than
6539   // spill/restore and VPT predication.
6540   return isVCTP(&MI) && !isPredicated(MI);
6541 }
6542 
6543 unsigned llvm::getBLXOpcode(const MachineFunction &MF) {
6544   return (MF.getSubtarget<ARMSubtarget>().hardenSlsBlr()) ? ARM::BLX_noip
6545                                                           : ARM::BLX;
6546 }
6547 
6548 unsigned llvm::gettBLXrOpcode(const MachineFunction &MF) {
6549   return (MF.getSubtarget<ARMSubtarget>().hardenSlsBlr()) ? ARM::tBLXr_noip
6550                                                           : ARM::tBLXr;
6551 }
6552 
6553 unsigned llvm::getBLXpredOpcode(const MachineFunction &MF) {
6554   return (MF.getSubtarget<ARMSubtarget>().hardenSlsBlr()) ? ARM::BLX_pred_noip
6555                                                           : ARM::BLX_pred;
6556 }
6557 
6558