xref: /aosp_15_r20/external/llvm/lib/CodeGen/DetectDeadLanes.cpp (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker //===- DetectDeadLanes.cpp - SubRegister Lane Usage Analysis --*- C++ -*---===//
2*9880d681SAndroid Build Coastguard Worker //
3*9880d681SAndroid Build Coastguard Worker //                     The LLVM Compiler Infrastructure
4*9880d681SAndroid Build Coastguard Worker //
5*9880d681SAndroid Build Coastguard Worker // This file is distributed under the University of Illinois Open Source
6*9880d681SAndroid Build Coastguard Worker // License. See LICENSE.TXT for details.
7*9880d681SAndroid Build Coastguard Worker //
8*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
9*9880d681SAndroid Build Coastguard Worker //
10*9880d681SAndroid Build Coastguard Worker /// \file
11*9880d681SAndroid Build Coastguard Worker /// Analysis that tracks defined/used subregister lanes across COPY instructions
12*9880d681SAndroid Build Coastguard Worker /// and instructions that get lowered to a COPY (PHI, REG_SEQUENCE,
13*9880d681SAndroid Build Coastguard Worker /// INSERT_SUBREG, EXTRACT_SUBREG).
14*9880d681SAndroid Build Coastguard Worker /// The information is used to detect dead definitions and the usage of
15*9880d681SAndroid Build Coastguard Worker /// (completely) undefined values and mark the operands as such.
16*9880d681SAndroid Build Coastguard Worker /// This pass is necessary because the dead/undef status is not obvious anymore
17*9880d681SAndroid Build Coastguard Worker /// when subregisters are involved.
18*9880d681SAndroid Build Coastguard Worker ///
19*9880d681SAndroid Build Coastguard Worker /// Example:
20*9880d681SAndroid Build Coastguard Worker ///    %vreg0 = some definition
21*9880d681SAndroid Build Coastguard Worker ///    %vreg1 = IMPLICIT_DEF
22*9880d681SAndroid Build Coastguard Worker ///    %vreg2 = REG_SEQUENCE %vreg0, sub0, %vreg1, sub1
23*9880d681SAndroid Build Coastguard Worker ///    %vreg3 = EXTRACT_SUBREG %vreg2, sub1
24*9880d681SAndroid Build Coastguard Worker ///           = use %vreg3
25*9880d681SAndroid Build Coastguard Worker /// The %vreg0 definition is dead and %vreg3 contains an undefined value.
26*9880d681SAndroid Build Coastguard Worker //
27*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
28*9880d681SAndroid Build Coastguard Worker 
29*9880d681SAndroid Build Coastguard Worker #include <deque>
30*9880d681SAndroid Build Coastguard Worker #include <vector>
31*9880d681SAndroid Build Coastguard Worker 
32*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/BitVector.h"
33*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/SetVector.h"
34*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineFunctionPass.h"
35*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineRegisterInfo.h"
36*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/Passes.h"
37*9880d681SAndroid Build Coastguard Worker #include "llvm/InitializePasses.h"
38*9880d681SAndroid Build Coastguard Worker #include "llvm/Pass.h"
39*9880d681SAndroid Build Coastguard Worker #include "llvm/PassRegistry.h"
40*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/Debug.h"
41*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/raw_ostream.h"
42*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetInstrInfo.h"
43*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetRegisterInfo.h"
44*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetSubtargetInfo.h"
45*9880d681SAndroid Build Coastguard Worker 
46*9880d681SAndroid Build Coastguard Worker using namespace llvm;
47*9880d681SAndroid Build Coastguard Worker 
48*9880d681SAndroid Build Coastguard Worker #define DEBUG_TYPE "detect-dead-lanes"
49*9880d681SAndroid Build Coastguard Worker 
50*9880d681SAndroid Build Coastguard Worker namespace {
51*9880d681SAndroid Build Coastguard Worker 
52*9880d681SAndroid Build Coastguard Worker /// Contains a bitmask of which lanes of a given virtual register are
53*9880d681SAndroid Build Coastguard Worker /// defined and which ones are actually used.
54*9880d681SAndroid Build Coastguard Worker struct VRegInfo {
55*9880d681SAndroid Build Coastguard Worker   LaneBitmask UsedLanes;
56*9880d681SAndroid Build Coastguard Worker   LaneBitmask DefinedLanes;
57*9880d681SAndroid Build Coastguard Worker };
58*9880d681SAndroid Build Coastguard Worker 
59*9880d681SAndroid Build Coastguard Worker class DetectDeadLanes : public MachineFunctionPass {
60*9880d681SAndroid Build Coastguard Worker public:
61*9880d681SAndroid Build Coastguard Worker   bool runOnMachineFunction(MachineFunction &MF) override;
62*9880d681SAndroid Build Coastguard Worker 
63*9880d681SAndroid Build Coastguard Worker   static char ID;
DetectDeadLanes()64*9880d681SAndroid Build Coastguard Worker   DetectDeadLanes() : MachineFunctionPass(ID) {}
65*9880d681SAndroid Build Coastguard Worker 
getPassName() const66*9880d681SAndroid Build Coastguard Worker   const char *getPassName() const override { return "Detect Dead Lanes"; }
67*9880d681SAndroid Build Coastguard Worker 
getAnalysisUsage(AnalysisUsage & AU) const68*9880d681SAndroid Build Coastguard Worker   void getAnalysisUsage(AnalysisUsage &AU) const override {
69*9880d681SAndroid Build Coastguard Worker     AU.setPreservesCFG();
70*9880d681SAndroid Build Coastguard Worker     MachineFunctionPass::getAnalysisUsage(AU);
71*9880d681SAndroid Build Coastguard Worker   }
72*9880d681SAndroid Build Coastguard Worker 
73*9880d681SAndroid Build Coastguard Worker private:
74*9880d681SAndroid Build Coastguard Worker   /// Add used lane bits on the register used by operand \p MO. This translates
75*9880d681SAndroid Build Coastguard Worker   /// the bitmask based on the operands subregister, and puts the register into
76*9880d681SAndroid Build Coastguard Worker   /// the worklist if any new bits were added.
77*9880d681SAndroid Build Coastguard Worker   void addUsedLanesOnOperand(const MachineOperand &MO, LaneBitmask UsedLanes);
78*9880d681SAndroid Build Coastguard Worker 
79*9880d681SAndroid Build Coastguard Worker   /// Given a bitmask \p UsedLanes for the used lanes on a def output of a
80*9880d681SAndroid Build Coastguard Worker   /// COPY-like instruction determine the lanes used on the use operands
81*9880d681SAndroid Build Coastguard Worker   /// and call addUsedLanesOnOperand() for them.
82*9880d681SAndroid Build Coastguard Worker   void transferUsedLanesStep(const MachineInstr &MI, LaneBitmask UsedLanes);
83*9880d681SAndroid Build Coastguard Worker 
84*9880d681SAndroid Build Coastguard Worker   /// Given a use regiser operand \p Use and a mask of defined lanes, check
85*9880d681SAndroid Build Coastguard Worker   /// if the operand belongs to a lowersToCopies() instruction, transfer the
86*9880d681SAndroid Build Coastguard Worker   /// mask to the def and put the instruction into the worklist.
87*9880d681SAndroid Build Coastguard Worker   void transferDefinedLanesStep(const MachineOperand &Use,
88*9880d681SAndroid Build Coastguard Worker                                 LaneBitmask DefinedLanes);
89*9880d681SAndroid Build Coastguard Worker 
90*9880d681SAndroid Build Coastguard Worker   /// Given a mask \p DefinedLanes of lanes defined at operand \p OpNum
91*9880d681SAndroid Build Coastguard Worker   /// of COPY-like instruction, determine which lanes are defined at the output
92*9880d681SAndroid Build Coastguard Worker   /// operand \p Def.
93*9880d681SAndroid Build Coastguard Worker   LaneBitmask transferDefinedLanes(const MachineOperand &Def, unsigned OpNum,
94*9880d681SAndroid Build Coastguard Worker                                    LaneBitmask DefinedLanes) const;
95*9880d681SAndroid Build Coastguard Worker 
96*9880d681SAndroid Build Coastguard Worker   /// Given a mask \p UsedLanes used from the output of instruction \p MI
97*9880d681SAndroid Build Coastguard Worker   /// determine which lanes are used from operand \p MO of this instruction.
98*9880d681SAndroid Build Coastguard Worker   LaneBitmask transferUsedLanes(const MachineInstr &MI, LaneBitmask UsedLanes,
99*9880d681SAndroid Build Coastguard Worker                                 const MachineOperand &MO) const;
100*9880d681SAndroid Build Coastguard Worker 
101*9880d681SAndroid Build Coastguard Worker   bool runOnce(MachineFunction &MF);
102*9880d681SAndroid Build Coastguard Worker 
103*9880d681SAndroid Build Coastguard Worker   LaneBitmask determineInitialDefinedLanes(unsigned Reg);
104*9880d681SAndroid Build Coastguard Worker   LaneBitmask determineInitialUsedLanes(unsigned Reg);
105*9880d681SAndroid Build Coastguard Worker 
106*9880d681SAndroid Build Coastguard Worker   bool isUndefRegAtInput(const MachineOperand &MO,
107*9880d681SAndroid Build Coastguard Worker                          const VRegInfo &RegInfo) const;
108*9880d681SAndroid Build Coastguard Worker 
109*9880d681SAndroid Build Coastguard Worker   bool isUndefInput(const MachineOperand &MO, bool *CrossCopy) const;
110*9880d681SAndroid Build Coastguard Worker 
111*9880d681SAndroid Build Coastguard Worker   const MachineRegisterInfo *MRI;
112*9880d681SAndroid Build Coastguard Worker   const TargetRegisterInfo *TRI;
113*9880d681SAndroid Build Coastguard Worker 
PutInWorklist(unsigned RegIdx)114*9880d681SAndroid Build Coastguard Worker   void PutInWorklist(unsigned RegIdx) {
115*9880d681SAndroid Build Coastguard Worker     if (WorklistMembers.test(RegIdx))
116*9880d681SAndroid Build Coastguard Worker       return;
117*9880d681SAndroid Build Coastguard Worker     WorklistMembers.set(RegIdx);
118*9880d681SAndroid Build Coastguard Worker     Worklist.push_back(RegIdx);
119*9880d681SAndroid Build Coastguard Worker   }
120*9880d681SAndroid Build Coastguard Worker 
121*9880d681SAndroid Build Coastguard Worker   VRegInfo *VRegInfos;
122*9880d681SAndroid Build Coastguard Worker   /// Worklist containing virtreg indexes.
123*9880d681SAndroid Build Coastguard Worker   std::deque<unsigned> Worklist;
124*9880d681SAndroid Build Coastguard Worker   BitVector WorklistMembers;
125*9880d681SAndroid Build Coastguard Worker   /// This bitvector is set for each vreg index where the vreg is defined
126*9880d681SAndroid Build Coastguard Worker   /// by an instruction where lowersToCopies()==true.
127*9880d681SAndroid Build Coastguard Worker   BitVector DefinedByCopy;
128*9880d681SAndroid Build Coastguard Worker };
129*9880d681SAndroid Build Coastguard Worker 
130*9880d681SAndroid Build Coastguard Worker } // end anonymous namespace
131*9880d681SAndroid Build Coastguard Worker 
132*9880d681SAndroid Build Coastguard Worker char DetectDeadLanes::ID = 0;
133*9880d681SAndroid Build Coastguard Worker char &llvm::DetectDeadLanesID = DetectDeadLanes::ID;
134*9880d681SAndroid Build Coastguard Worker 
135*9880d681SAndroid Build Coastguard Worker INITIALIZE_PASS(DetectDeadLanes, "detect-dead-lanes", "Detect Dead Lanes",
136*9880d681SAndroid Build Coastguard Worker                 false, false)
137*9880d681SAndroid Build Coastguard Worker 
138*9880d681SAndroid Build Coastguard Worker /// Returns true if \p MI will get lowered to a series of COPY instructions.
139*9880d681SAndroid Build Coastguard Worker /// We call this a COPY-like instruction.
lowersToCopies(const MachineInstr & MI)140*9880d681SAndroid Build Coastguard Worker static bool lowersToCopies(const MachineInstr &MI) {
141*9880d681SAndroid Build Coastguard Worker   // Note: We could support instructions with MCInstrDesc::isRegSequenceLike(),
142*9880d681SAndroid Build Coastguard Worker   // isExtractSubRegLike(), isInsertSubregLike() in the future even though they
143*9880d681SAndroid Build Coastguard Worker   // are not lowered to a COPY.
144*9880d681SAndroid Build Coastguard Worker   switch (MI.getOpcode()) {
145*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::COPY:
146*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::PHI:
147*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::INSERT_SUBREG:
148*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::REG_SEQUENCE:
149*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::EXTRACT_SUBREG:
150*9880d681SAndroid Build Coastguard Worker     return true;
151*9880d681SAndroid Build Coastguard Worker   }
152*9880d681SAndroid Build Coastguard Worker   return false;
153*9880d681SAndroid Build Coastguard Worker }
154*9880d681SAndroid Build Coastguard Worker 
isCrossCopy(const MachineRegisterInfo & MRI,const MachineInstr & MI,const TargetRegisterClass * DstRC,const MachineOperand & MO)155*9880d681SAndroid Build Coastguard Worker static bool isCrossCopy(const MachineRegisterInfo &MRI,
156*9880d681SAndroid Build Coastguard Worker                         const MachineInstr &MI,
157*9880d681SAndroid Build Coastguard Worker                         const TargetRegisterClass *DstRC,
158*9880d681SAndroid Build Coastguard Worker                         const MachineOperand &MO) {
159*9880d681SAndroid Build Coastguard Worker   assert(lowersToCopies(MI));
160*9880d681SAndroid Build Coastguard Worker   unsigned SrcReg = MO.getReg();
161*9880d681SAndroid Build Coastguard Worker   const TargetRegisterClass *SrcRC = MRI.getRegClass(SrcReg);
162*9880d681SAndroid Build Coastguard Worker   if (DstRC == SrcRC)
163*9880d681SAndroid Build Coastguard Worker     return false;
164*9880d681SAndroid Build Coastguard Worker 
165*9880d681SAndroid Build Coastguard Worker   unsigned SrcSubIdx = MO.getSubReg();
166*9880d681SAndroid Build Coastguard Worker 
167*9880d681SAndroid Build Coastguard Worker   const TargetRegisterInfo &TRI = *MRI.getTargetRegisterInfo();
168*9880d681SAndroid Build Coastguard Worker   unsigned DstSubIdx = 0;
169*9880d681SAndroid Build Coastguard Worker   switch (MI.getOpcode()) {
170*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::INSERT_SUBREG:
171*9880d681SAndroid Build Coastguard Worker     if (MI.getOperandNo(&MO) == 2)
172*9880d681SAndroid Build Coastguard Worker       DstSubIdx = MI.getOperand(3).getImm();
173*9880d681SAndroid Build Coastguard Worker     break;
174*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::REG_SEQUENCE: {
175*9880d681SAndroid Build Coastguard Worker     unsigned OpNum = MI.getOperandNo(&MO);
176*9880d681SAndroid Build Coastguard Worker     DstSubIdx = MI.getOperand(OpNum+1).getImm();
177*9880d681SAndroid Build Coastguard Worker     break;
178*9880d681SAndroid Build Coastguard Worker   }
179*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::EXTRACT_SUBREG: {
180*9880d681SAndroid Build Coastguard Worker     unsigned SubReg = MI.getOperand(2).getImm();
181*9880d681SAndroid Build Coastguard Worker     SrcSubIdx = TRI.composeSubRegIndices(SubReg, SrcSubIdx);
182*9880d681SAndroid Build Coastguard Worker   }
183*9880d681SAndroid Build Coastguard Worker   }
184*9880d681SAndroid Build Coastguard Worker 
185*9880d681SAndroid Build Coastguard Worker   unsigned PreA, PreB; // Unused.
186*9880d681SAndroid Build Coastguard Worker   if (SrcSubIdx && DstSubIdx)
187*9880d681SAndroid Build Coastguard Worker     return !TRI.getCommonSuperRegClass(SrcRC, SrcSubIdx, DstRC, DstSubIdx, PreA,
188*9880d681SAndroid Build Coastguard Worker                                        PreB);
189*9880d681SAndroid Build Coastguard Worker   if (SrcSubIdx)
190*9880d681SAndroid Build Coastguard Worker     return !TRI.getMatchingSuperRegClass(SrcRC, DstRC, SrcSubIdx);
191*9880d681SAndroid Build Coastguard Worker   if (DstSubIdx)
192*9880d681SAndroid Build Coastguard Worker     return !TRI.getMatchingSuperRegClass(DstRC, SrcRC, DstSubIdx);
193*9880d681SAndroid Build Coastguard Worker   return !TRI.getCommonSubClass(SrcRC, DstRC);
194*9880d681SAndroid Build Coastguard Worker }
195*9880d681SAndroid Build Coastguard Worker 
addUsedLanesOnOperand(const MachineOperand & MO,LaneBitmask UsedLanes)196*9880d681SAndroid Build Coastguard Worker void DetectDeadLanes::addUsedLanesOnOperand(const MachineOperand &MO,
197*9880d681SAndroid Build Coastguard Worker                                             LaneBitmask UsedLanes) {
198*9880d681SAndroid Build Coastguard Worker   if (!MO.readsReg())
199*9880d681SAndroid Build Coastguard Worker     return;
200*9880d681SAndroid Build Coastguard Worker   unsigned MOReg = MO.getReg();
201*9880d681SAndroid Build Coastguard Worker   if (!TargetRegisterInfo::isVirtualRegister(MOReg))
202*9880d681SAndroid Build Coastguard Worker     return;
203*9880d681SAndroid Build Coastguard Worker 
204*9880d681SAndroid Build Coastguard Worker   unsigned MOSubReg = MO.getSubReg();
205*9880d681SAndroid Build Coastguard Worker   if (MOSubReg != 0)
206*9880d681SAndroid Build Coastguard Worker     UsedLanes = TRI->composeSubRegIndexLaneMask(MOSubReg, UsedLanes);
207*9880d681SAndroid Build Coastguard Worker   UsedLanes &= MRI->getMaxLaneMaskForVReg(MOReg);
208*9880d681SAndroid Build Coastguard Worker 
209*9880d681SAndroid Build Coastguard Worker   unsigned MORegIdx = TargetRegisterInfo::virtReg2Index(MOReg);
210*9880d681SAndroid Build Coastguard Worker   VRegInfo &MORegInfo = VRegInfos[MORegIdx];
211*9880d681SAndroid Build Coastguard Worker   LaneBitmask PrevUsedLanes = MORegInfo.UsedLanes;
212*9880d681SAndroid Build Coastguard Worker   // Any change at all?
213*9880d681SAndroid Build Coastguard Worker   if ((UsedLanes & ~PrevUsedLanes) == 0)
214*9880d681SAndroid Build Coastguard Worker     return;
215*9880d681SAndroid Build Coastguard Worker 
216*9880d681SAndroid Build Coastguard Worker   // Set UsedLanes and remember instruction for further propagation.
217*9880d681SAndroid Build Coastguard Worker   MORegInfo.UsedLanes = PrevUsedLanes | UsedLanes;
218*9880d681SAndroid Build Coastguard Worker   if (DefinedByCopy.test(MORegIdx))
219*9880d681SAndroid Build Coastguard Worker     PutInWorklist(MORegIdx);
220*9880d681SAndroid Build Coastguard Worker }
221*9880d681SAndroid Build Coastguard Worker 
transferUsedLanesStep(const MachineInstr & MI,LaneBitmask UsedLanes)222*9880d681SAndroid Build Coastguard Worker void DetectDeadLanes::transferUsedLanesStep(const MachineInstr &MI,
223*9880d681SAndroid Build Coastguard Worker                                             LaneBitmask UsedLanes) {
224*9880d681SAndroid Build Coastguard Worker   for (const MachineOperand &MO : MI.uses()) {
225*9880d681SAndroid Build Coastguard Worker     if (!MO.isReg() || !TargetRegisterInfo::isVirtualRegister(MO.getReg()))
226*9880d681SAndroid Build Coastguard Worker       continue;
227*9880d681SAndroid Build Coastguard Worker     LaneBitmask UsedOnMO = transferUsedLanes(MI, UsedLanes, MO);
228*9880d681SAndroid Build Coastguard Worker     addUsedLanesOnOperand(MO, UsedOnMO);
229*9880d681SAndroid Build Coastguard Worker   }
230*9880d681SAndroid Build Coastguard Worker }
231*9880d681SAndroid Build Coastguard Worker 
transferUsedLanes(const MachineInstr & MI,LaneBitmask UsedLanes,const MachineOperand & MO) const232*9880d681SAndroid Build Coastguard Worker LaneBitmask DetectDeadLanes::transferUsedLanes(const MachineInstr &MI,
233*9880d681SAndroid Build Coastguard Worker                                                LaneBitmask UsedLanes,
234*9880d681SAndroid Build Coastguard Worker                                                const MachineOperand &MO) const {
235*9880d681SAndroid Build Coastguard Worker   unsigned OpNum = MI.getOperandNo(&MO);
236*9880d681SAndroid Build Coastguard Worker   assert(lowersToCopies(MI) && DefinedByCopy[
237*9880d681SAndroid Build Coastguard Worker            TargetRegisterInfo::virtReg2Index(MI.getOperand(0).getReg())]);
238*9880d681SAndroid Build Coastguard Worker 
239*9880d681SAndroid Build Coastguard Worker   switch (MI.getOpcode()) {
240*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::COPY:
241*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::PHI:
242*9880d681SAndroid Build Coastguard Worker     return UsedLanes;
243*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::REG_SEQUENCE: {
244*9880d681SAndroid Build Coastguard Worker     assert(OpNum % 2 == 1);
245*9880d681SAndroid Build Coastguard Worker     unsigned SubIdx = MI.getOperand(OpNum + 1).getImm();
246*9880d681SAndroid Build Coastguard Worker     return TRI->reverseComposeSubRegIndexLaneMask(SubIdx, UsedLanes);
247*9880d681SAndroid Build Coastguard Worker   }
248*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::INSERT_SUBREG: {
249*9880d681SAndroid Build Coastguard Worker     unsigned SubIdx = MI.getOperand(3).getImm();
250*9880d681SAndroid Build Coastguard Worker     LaneBitmask MO2UsedLanes =
251*9880d681SAndroid Build Coastguard Worker         TRI->reverseComposeSubRegIndexLaneMask(SubIdx, UsedLanes);
252*9880d681SAndroid Build Coastguard Worker     if (OpNum == 2)
253*9880d681SAndroid Build Coastguard Worker       return MO2UsedLanes;
254*9880d681SAndroid Build Coastguard Worker 
255*9880d681SAndroid Build Coastguard Worker     const MachineOperand &Def = MI.getOperand(0);
256*9880d681SAndroid Build Coastguard Worker     unsigned DefReg = Def.getReg();
257*9880d681SAndroid Build Coastguard Worker     const TargetRegisterClass *RC = MRI->getRegClass(DefReg);
258*9880d681SAndroid Build Coastguard Worker     LaneBitmask MO1UsedLanes;
259*9880d681SAndroid Build Coastguard Worker     if (RC->CoveredBySubRegs)
260*9880d681SAndroid Build Coastguard Worker       MO1UsedLanes = UsedLanes & ~TRI->getSubRegIndexLaneMask(SubIdx);
261*9880d681SAndroid Build Coastguard Worker     else
262*9880d681SAndroid Build Coastguard Worker       MO1UsedLanes = RC->LaneMask;
263*9880d681SAndroid Build Coastguard Worker 
264*9880d681SAndroid Build Coastguard Worker     assert(OpNum == 1);
265*9880d681SAndroid Build Coastguard Worker     return MO1UsedLanes;
266*9880d681SAndroid Build Coastguard Worker   }
267*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::EXTRACT_SUBREG: {
268*9880d681SAndroid Build Coastguard Worker     assert(OpNum == 1);
269*9880d681SAndroid Build Coastguard Worker     unsigned SubIdx = MI.getOperand(2).getImm();
270*9880d681SAndroid Build Coastguard Worker     return TRI->composeSubRegIndexLaneMask(SubIdx, UsedLanes);
271*9880d681SAndroid Build Coastguard Worker   }
272*9880d681SAndroid Build Coastguard Worker   default:
273*9880d681SAndroid Build Coastguard Worker     llvm_unreachable("function must be called with COPY-like instruction");
274*9880d681SAndroid Build Coastguard Worker   }
275*9880d681SAndroid Build Coastguard Worker }
276*9880d681SAndroid Build Coastguard Worker 
transferDefinedLanesStep(const MachineOperand & Use,LaneBitmask DefinedLanes)277*9880d681SAndroid Build Coastguard Worker void DetectDeadLanes::transferDefinedLanesStep(const MachineOperand &Use,
278*9880d681SAndroid Build Coastguard Worker                                                LaneBitmask DefinedLanes) {
279*9880d681SAndroid Build Coastguard Worker   if (!Use.readsReg())
280*9880d681SAndroid Build Coastguard Worker     return;
281*9880d681SAndroid Build Coastguard Worker   // Check whether the operand writes a vreg and is part of a COPY-like
282*9880d681SAndroid Build Coastguard Worker   // instruction.
283*9880d681SAndroid Build Coastguard Worker   const MachineInstr &MI = *Use.getParent();
284*9880d681SAndroid Build Coastguard Worker   if (MI.getDesc().getNumDefs() != 1)
285*9880d681SAndroid Build Coastguard Worker     return;
286*9880d681SAndroid Build Coastguard Worker   // FIXME: PATCHPOINT instructions announce a Def that does not always exist,
287*9880d681SAndroid Build Coastguard Worker   // they really need to be modeled differently!
288*9880d681SAndroid Build Coastguard Worker   if (MI.getOpcode() == TargetOpcode::PATCHPOINT)
289*9880d681SAndroid Build Coastguard Worker     return;
290*9880d681SAndroid Build Coastguard Worker   const MachineOperand &Def = *MI.defs().begin();
291*9880d681SAndroid Build Coastguard Worker   unsigned DefReg = Def.getReg();
292*9880d681SAndroid Build Coastguard Worker   if (!TargetRegisterInfo::isVirtualRegister(DefReg))
293*9880d681SAndroid Build Coastguard Worker     return;
294*9880d681SAndroid Build Coastguard Worker   unsigned DefRegIdx = TargetRegisterInfo::virtReg2Index(DefReg);
295*9880d681SAndroid Build Coastguard Worker   if (!DefinedByCopy.test(DefRegIdx))
296*9880d681SAndroid Build Coastguard Worker     return;
297*9880d681SAndroid Build Coastguard Worker 
298*9880d681SAndroid Build Coastguard Worker   unsigned OpNum = MI.getOperandNo(&Use);
299*9880d681SAndroid Build Coastguard Worker   DefinedLanes =
300*9880d681SAndroid Build Coastguard Worker       TRI->reverseComposeSubRegIndexLaneMask(Use.getSubReg(), DefinedLanes);
301*9880d681SAndroid Build Coastguard Worker   DefinedLanes = transferDefinedLanes(Def, OpNum, DefinedLanes);
302*9880d681SAndroid Build Coastguard Worker 
303*9880d681SAndroid Build Coastguard Worker   VRegInfo &RegInfo = VRegInfos[DefRegIdx];
304*9880d681SAndroid Build Coastguard Worker   LaneBitmask PrevDefinedLanes = RegInfo.DefinedLanes;
305*9880d681SAndroid Build Coastguard Worker   // Any change at all?
306*9880d681SAndroid Build Coastguard Worker   if ((DefinedLanes & ~PrevDefinedLanes) == 0)
307*9880d681SAndroid Build Coastguard Worker     return;
308*9880d681SAndroid Build Coastguard Worker 
309*9880d681SAndroid Build Coastguard Worker   RegInfo.DefinedLanes = PrevDefinedLanes | DefinedLanes;
310*9880d681SAndroid Build Coastguard Worker   PutInWorklist(DefRegIdx);
311*9880d681SAndroid Build Coastguard Worker }
312*9880d681SAndroid Build Coastguard Worker 
transferDefinedLanes(const MachineOperand & Def,unsigned OpNum,LaneBitmask DefinedLanes) const313*9880d681SAndroid Build Coastguard Worker LaneBitmask DetectDeadLanes::transferDefinedLanes(const MachineOperand &Def,
314*9880d681SAndroid Build Coastguard Worker     unsigned OpNum, LaneBitmask DefinedLanes) const {
315*9880d681SAndroid Build Coastguard Worker   const MachineInstr &MI = *Def.getParent();
316*9880d681SAndroid Build Coastguard Worker   // Translate DefinedLanes if necessary.
317*9880d681SAndroid Build Coastguard Worker   switch (MI.getOpcode()) {
318*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::REG_SEQUENCE: {
319*9880d681SAndroid Build Coastguard Worker     unsigned SubIdx = MI.getOperand(OpNum + 1).getImm();
320*9880d681SAndroid Build Coastguard Worker     DefinedLanes = TRI->composeSubRegIndexLaneMask(SubIdx, DefinedLanes);
321*9880d681SAndroid Build Coastguard Worker     DefinedLanes &= TRI->getSubRegIndexLaneMask(SubIdx);
322*9880d681SAndroid Build Coastguard Worker     break;
323*9880d681SAndroid Build Coastguard Worker   }
324*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::INSERT_SUBREG: {
325*9880d681SAndroid Build Coastguard Worker     unsigned SubIdx = MI.getOperand(3).getImm();
326*9880d681SAndroid Build Coastguard Worker     if (OpNum == 2) {
327*9880d681SAndroid Build Coastguard Worker       DefinedLanes = TRI->composeSubRegIndexLaneMask(SubIdx, DefinedLanes);
328*9880d681SAndroid Build Coastguard Worker       DefinedLanes &= TRI->getSubRegIndexLaneMask(SubIdx);
329*9880d681SAndroid Build Coastguard Worker     } else {
330*9880d681SAndroid Build Coastguard Worker       assert(OpNum == 1 && "INSERT_SUBREG must have two operands");
331*9880d681SAndroid Build Coastguard Worker       // Ignore lanes defined by operand 2.
332*9880d681SAndroid Build Coastguard Worker       DefinedLanes &= ~TRI->getSubRegIndexLaneMask(SubIdx);
333*9880d681SAndroid Build Coastguard Worker     }
334*9880d681SAndroid Build Coastguard Worker     break;
335*9880d681SAndroid Build Coastguard Worker   }
336*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::EXTRACT_SUBREG: {
337*9880d681SAndroid Build Coastguard Worker     unsigned SubIdx = MI.getOperand(2).getImm();
338*9880d681SAndroid Build Coastguard Worker     assert(OpNum == 1 && "EXTRACT_SUBREG must have one register operand only");
339*9880d681SAndroid Build Coastguard Worker     DefinedLanes = TRI->reverseComposeSubRegIndexLaneMask(SubIdx, DefinedLanes);
340*9880d681SAndroid Build Coastguard Worker     break;
341*9880d681SAndroid Build Coastguard Worker   }
342*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::COPY:
343*9880d681SAndroid Build Coastguard Worker   case TargetOpcode::PHI:
344*9880d681SAndroid Build Coastguard Worker     break;
345*9880d681SAndroid Build Coastguard Worker   default:
346*9880d681SAndroid Build Coastguard Worker     llvm_unreachable("function must be called with COPY-like instruction");
347*9880d681SAndroid Build Coastguard Worker   }
348*9880d681SAndroid Build Coastguard Worker 
349*9880d681SAndroid Build Coastguard Worker   assert(Def.getSubReg() == 0 &&
350*9880d681SAndroid Build Coastguard Worker          "Should not have subregister defs in machine SSA phase");
351*9880d681SAndroid Build Coastguard Worker   DefinedLanes &= MRI->getMaxLaneMaskForVReg(Def.getReg());
352*9880d681SAndroid Build Coastguard Worker   return DefinedLanes;
353*9880d681SAndroid Build Coastguard Worker }
354*9880d681SAndroid Build Coastguard Worker 
determineInitialDefinedLanes(unsigned Reg)355*9880d681SAndroid Build Coastguard Worker LaneBitmask DetectDeadLanes::determineInitialDefinedLanes(unsigned Reg) {
356*9880d681SAndroid Build Coastguard Worker   // Live-In or unused registers have no definition but are considered fully
357*9880d681SAndroid Build Coastguard Worker   // defined.
358*9880d681SAndroid Build Coastguard Worker   if (!MRI->hasOneDef(Reg))
359*9880d681SAndroid Build Coastguard Worker     return ~0u;
360*9880d681SAndroid Build Coastguard Worker 
361*9880d681SAndroid Build Coastguard Worker   const MachineOperand &Def = *MRI->def_begin(Reg);
362*9880d681SAndroid Build Coastguard Worker   const MachineInstr &DefMI = *Def.getParent();
363*9880d681SAndroid Build Coastguard Worker   if (lowersToCopies(DefMI)) {
364*9880d681SAndroid Build Coastguard Worker     // Start optimisatically with no used or defined lanes for copy
365*9880d681SAndroid Build Coastguard Worker     // instructions. The following dataflow analysis will add more bits.
366*9880d681SAndroid Build Coastguard Worker     unsigned RegIdx = TargetRegisterInfo::virtReg2Index(Reg);
367*9880d681SAndroid Build Coastguard Worker     DefinedByCopy.set(RegIdx);
368*9880d681SAndroid Build Coastguard Worker     PutInWorklist(RegIdx);
369*9880d681SAndroid Build Coastguard Worker 
370*9880d681SAndroid Build Coastguard Worker     if (Def.isDead())
371*9880d681SAndroid Build Coastguard Worker       return 0;
372*9880d681SAndroid Build Coastguard Worker 
373*9880d681SAndroid Build Coastguard Worker     // COPY/PHI can copy across unrelated register classes (example: float/int)
374*9880d681SAndroid Build Coastguard Worker     // with incompatible subregister structure. Do not include these in the
375*9880d681SAndroid Build Coastguard Worker     // dataflow analysis since we cannot transfer lanemasks in a meaningful way.
376*9880d681SAndroid Build Coastguard Worker     const TargetRegisterClass *DefRC = MRI->getRegClass(Reg);
377*9880d681SAndroid Build Coastguard Worker 
378*9880d681SAndroid Build Coastguard Worker     // Determine initially DefinedLanes.
379*9880d681SAndroid Build Coastguard Worker     LaneBitmask DefinedLanes = 0;
380*9880d681SAndroid Build Coastguard Worker     for (const MachineOperand &MO : DefMI.uses()) {
381*9880d681SAndroid Build Coastguard Worker       if (!MO.isReg() || !MO.readsReg())
382*9880d681SAndroid Build Coastguard Worker         continue;
383*9880d681SAndroid Build Coastguard Worker       unsigned MOReg = MO.getReg();
384*9880d681SAndroid Build Coastguard Worker       if (!MOReg)
385*9880d681SAndroid Build Coastguard Worker         continue;
386*9880d681SAndroid Build Coastguard Worker 
387*9880d681SAndroid Build Coastguard Worker       LaneBitmask MODefinedLanes;
388*9880d681SAndroid Build Coastguard Worker       if (TargetRegisterInfo::isPhysicalRegister(MOReg)) {
389*9880d681SAndroid Build Coastguard Worker         MODefinedLanes = ~0u;
390*9880d681SAndroid Build Coastguard Worker       } else if (isCrossCopy(*MRI, DefMI, DefRC, MO)) {
391*9880d681SAndroid Build Coastguard Worker         MODefinedLanes = ~0u;
392*9880d681SAndroid Build Coastguard Worker       } else {
393*9880d681SAndroid Build Coastguard Worker         assert(TargetRegisterInfo::isVirtualRegister(MOReg));
394*9880d681SAndroid Build Coastguard Worker         if (MRI->hasOneDef(MOReg)) {
395*9880d681SAndroid Build Coastguard Worker           const MachineOperand &MODef = *MRI->def_begin(MOReg);
396*9880d681SAndroid Build Coastguard Worker           const MachineInstr &MODefMI = *MODef.getParent();
397*9880d681SAndroid Build Coastguard Worker           // Bits from copy-like operations will be added later.
398*9880d681SAndroid Build Coastguard Worker           if (lowersToCopies(MODefMI) || MODefMI.isImplicitDef())
399*9880d681SAndroid Build Coastguard Worker             continue;
400*9880d681SAndroid Build Coastguard Worker         }
401*9880d681SAndroid Build Coastguard Worker         unsigned MOSubReg = MO.getSubReg();
402*9880d681SAndroid Build Coastguard Worker         MODefinedLanes = MRI->getMaxLaneMaskForVReg(MOReg);
403*9880d681SAndroid Build Coastguard Worker         MODefinedLanes = TRI->reverseComposeSubRegIndexLaneMask(
404*9880d681SAndroid Build Coastguard Worker             MOSubReg, MODefinedLanes);
405*9880d681SAndroid Build Coastguard Worker       }
406*9880d681SAndroid Build Coastguard Worker 
407*9880d681SAndroid Build Coastguard Worker       unsigned OpNum = DefMI.getOperandNo(&MO);
408*9880d681SAndroid Build Coastguard Worker       DefinedLanes |= transferDefinedLanes(Def, OpNum, MODefinedLanes);
409*9880d681SAndroid Build Coastguard Worker     }
410*9880d681SAndroid Build Coastguard Worker     return DefinedLanes;
411*9880d681SAndroid Build Coastguard Worker   }
412*9880d681SAndroid Build Coastguard Worker   if (DefMI.isImplicitDef() || Def.isDead())
413*9880d681SAndroid Build Coastguard Worker     return 0;
414*9880d681SAndroid Build Coastguard Worker 
415*9880d681SAndroid Build Coastguard Worker   assert(Def.getSubReg() == 0 &&
416*9880d681SAndroid Build Coastguard Worker          "Should not have subregister defs in machine SSA phase");
417*9880d681SAndroid Build Coastguard Worker   return MRI->getMaxLaneMaskForVReg(Reg);
418*9880d681SAndroid Build Coastguard Worker }
419*9880d681SAndroid Build Coastguard Worker 
determineInitialUsedLanes(unsigned Reg)420*9880d681SAndroid Build Coastguard Worker LaneBitmask DetectDeadLanes::determineInitialUsedLanes(unsigned Reg) {
421*9880d681SAndroid Build Coastguard Worker   LaneBitmask UsedLanes = 0;
422*9880d681SAndroid Build Coastguard Worker   for (const MachineOperand &MO : MRI->use_nodbg_operands(Reg)) {
423*9880d681SAndroid Build Coastguard Worker     if (!MO.readsReg())
424*9880d681SAndroid Build Coastguard Worker       continue;
425*9880d681SAndroid Build Coastguard Worker 
426*9880d681SAndroid Build Coastguard Worker     const MachineInstr &UseMI = *MO.getParent();
427*9880d681SAndroid Build Coastguard Worker     if (UseMI.isKill())
428*9880d681SAndroid Build Coastguard Worker       continue;
429*9880d681SAndroid Build Coastguard Worker 
430*9880d681SAndroid Build Coastguard Worker     unsigned SubReg = MO.getSubReg();
431*9880d681SAndroid Build Coastguard Worker     if (lowersToCopies(UseMI)) {
432*9880d681SAndroid Build Coastguard Worker       assert(UseMI.getDesc().getNumDefs() == 1);
433*9880d681SAndroid Build Coastguard Worker       const MachineOperand &Def = *UseMI.defs().begin();
434*9880d681SAndroid Build Coastguard Worker       unsigned DefReg = Def.getReg();
435*9880d681SAndroid Build Coastguard Worker       // The used lanes of COPY-like instruction operands are determined by the
436*9880d681SAndroid Build Coastguard Worker       // following dataflow analysis.
437*9880d681SAndroid Build Coastguard Worker       if (TargetRegisterInfo::isVirtualRegister(DefReg)) {
438*9880d681SAndroid Build Coastguard Worker         // But ignore copies across incompatible register classes.
439*9880d681SAndroid Build Coastguard Worker         bool CrossCopy = false;
440*9880d681SAndroid Build Coastguard Worker         if (lowersToCopies(UseMI)) {
441*9880d681SAndroid Build Coastguard Worker           const TargetRegisterClass *DstRC = MRI->getRegClass(DefReg);
442*9880d681SAndroid Build Coastguard Worker           CrossCopy = isCrossCopy(*MRI, UseMI, DstRC, MO);
443*9880d681SAndroid Build Coastguard Worker           if (CrossCopy)
444*9880d681SAndroid Build Coastguard Worker             DEBUG(dbgs() << "Copy accross incompatible classes: " << UseMI);
445*9880d681SAndroid Build Coastguard Worker         }
446*9880d681SAndroid Build Coastguard Worker 
447*9880d681SAndroid Build Coastguard Worker         if (!CrossCopy)
448*9880d681SAndroid Build Coastguard Worker           continue;
449*9880d681SAndroid Build Coastguard Worker       }
450*9880d681SAndroid Build Coastguard Worker     }
451*9880d681SAndroid Build Coastguard Worker 
452*9880d681SAndroid Build Coastguard Worker     // Shortcut: All lanes are used.
453*9880d681SAndroid Build Coastguard Worker     if (SubReg == 0)
454*9880d681SAndroid Build Coastguard Worker       return MRI->getMaxLaneMaskForVReg(Reg);
455*9880d681SAndroid Build Coastguard Worker 
456*9880d681SAndroid Build Coastguard Worker     UsedLanes |= TRI->getSubRegIndexLaneMask(SubReg);
457*9880d681SAndroid Build Coastguard Worker   }
458*9880d681SAndroid Build Coastguard Worker   return UsedLanes;
459*9880d681SAndroid Build Coastguard Worker }
460*9880d681SAndroid Build Coastguard Worker 
isUndefRegAtInput(const MachineOperand & MO,const VRegInfo & RegInfo) const461*9880d681SAndroid Build Coastguard Worker bool DetectDeadLanes::isUndefRegAtInput(const MachineOperand &MO,
462*9880d681SAndroid Build Coastguard Worker                                         const VRegInfo &RegInfo) const {
463*9880d681SAndroid Build Coastguard Worker   unsigned SubReg = MO.getSubReg();
464*9880d681SAndroid Build Coastguard Worker   LaneBitmask Mask = TRI->getSubRegIndexLaneMask(SubReg);
465*9880d681SAndroid Build Coastguard Worker   return (RegInfo.DefinedLanes & RegInfo.UsedLanes & Mask) == 0;
466*9880d681SAndroid Build Coastguard Worker }
467*9880d681SAndroid Build Coastguard Worker 
isUndefInput(const MachineOperand & MO,bool * CrossCopy) const468*9880d681SAndroid Build Coastguard Worker bool DetectDeadLanes::isUndefInput(const MachineOperand &MO,
469*9880d681SAndroid Build Coastguard Worker                                    bool *CrossCopy) const {
470*9880d681SAndroid Build Coastguard Worker   if (!MO.isUse())
471*9880d681SAndroid Build Coastguard Worker     return false;
472*9880d681SAndroid Build Coastguard Worker   const MachineInstr &MI = *MO.getParent();
473*9880d681SAndroid Build Coastguard Worker   if (!lowersToCopies(MI))
474*9880d681SAndroid Build Coastguard Worker     return false;
475*9880d681SAndroid Build Coastguard Worker   const MachineOperand &Def = MI.getOperand(0);
476*9880d681SAndroid Build Coastguard Worker   unsigned DefReg = Def.getReg();
477*9880d681SAndroid Build Coastguard Worker   if (!TargetRegisterInfo::isVirtualRegister(DefReg))
478*9880d681SAndroid Build Coastguard Worker     return false;
479*9880d681SAndroid Build Coastguard Worker   unsigned DefRegIdx = TargetRegisterInfo::virtReg2Index(DefReg);
480*9880d681SAndroid Build Coastguard Worker   if (!DefinedByCopy.test(DefRegIdx))
481*9880d681SAndroid Build Coastguard Worker     return false;
482*9880d681SAndroid Build Coastguard Worker 
483*9880d681SAndroid Build Coastguard Worker   const VRegInfo &DefRegInfo = VRegInfos[DefRegIdx];
484*9880d681SAndroid Build Coastguard Worker   LaneBitmask UsedLanes = transferUsedLanes(MI, DefRegInfo.UsedLanes, MO);
485*9880d681SAndroid Build Coastguard Worker   if (UsedLanes != 0)
486*9880d681SAndroid Build Coastguard Worker     return false;
487*9880d681SAndroid Build Coastguard Worker 
488*9880d681SAndroid Build Coastguard Worker   unsigned MOReg = MO.getReg();
489*9880d681SAndroid Build Coastguard Worker   if (TargetRegisterInfo::isVirtualRegister(MOReg)) {
490*9880d681SAndroid Build Coastguard Worker     const TargetRegisterClass *DstRC = MRI->getRegClass(DefReg);
491*9880d681SAndroid Build Coastguard Worker     *CrossCopy = isCrossCopy(*MRI, MI, DstRC, MO);
492*9880d681SAndroid Build Coastguard Worker   }
493*9880d681SAndroid Build Coastguard Worker   return true;
494*9880d681SAndroid Build Coastguard Worker }
495*9880d681SAndroid Build Coastguard Worker 
runOnce(MachineFunction & MF)496*9880d681SAndroid Build Coastguard Worker bool DetectDeadLanes::runOnce(MachineFunction &MF) {
497*9880d681SAndroid Build Coastguard Worker   // First pass: Populate defs/uses of vregs with initial values
498*9880d681SAndroid Build Coastguard Worker   unsigned NumVirtRegs = MRI->getNumVirtRegs();
499*9880d681SAndroid Build Coastguard Worker   for (unsigned RegIdx = 0; RegIdx < NumVirtRegs; ++RegIdx) {
500*9880d681SAndroid Build Coastguard Worker     unsigned Reg = TargetRegisterInfo::index2VirtReg(RegIdx);
501*9880d681SAndroid Build Coastguard Worker 
502*9880d681SAndroid Build Coastguard Worker     // Determine used/defined lanes and add copy instructions to worklist.
503*9880d681SAndroid Build Coastguard Worker     VRegInfo &Info = VRegInfos[RegIdx];
504*9880d681SAndroid Build Coastguard Worker     Info.DefinedLanes = determineInitialDefinedLanes(Reg);
505*9880d681SAndroid Build Coastguard Worker     Info.UsedLanes = determineInitialUsedLanes(Reg);
506*9880d681SAndroid Build Coastguard Worker   }
507*9880d681SAndroid Build Coastguard Worker 
508*9880d681SAndroid Build Coastguard Worker   // Iterate as long as defined lanes/used lanes keep changing.
509*9880d681SAndroid Build Coastguard Worker   while (!Worklist.empty()) {
510*9880d681SAndroid Build Coastguard Worker     unsigned RegIdx = Worklist.front();
511*9880d681SAndroid Build Coastguard Worker     Worklist.pop_front();
512*9880d681SAndroid Build Coastguard Worker     WorklistMembers.reset(RegIdx);
513*9880d681SAndroid Build Coastguard Worker     VRegInfo &Info = VRegInfos[RegIdx];
514*9880d681SAndroid Build Coastguard Worker     unsigned Reg = TargetRegisterInfo::index2VirtReg(RegIdx);
515*9880d681SAndroid Build Coastguard Worker 
516*9880d681SAndroid Build Coastguard Worker     // Transfer UsedLanes to operands of DefMI (backwards dataflow).
517*9880d681SAndroid Build Coastguard Worker     MachineOperand &Def = *MRI->def_begin(Reg);
518*9880d681SAndroid Build Coastguard Worker     const MachineInstr &MI = *Def.getParent();
519*9880d681SAndroid Build Coastguard Worker     transferUsedLanesStep(MI, Info.UsedLanes);
520*9880d681SAndroid Build Coastguard Worker     // Transfer DefinedLanes to users of Reg (forward dataflow).
521*9880d681SAndroid Build Coastguard Worker     for (const MachineOperand &MO : MRI->use_nodbg_operands(Reg))
522*9880d681SAndroid Build Coastguard Worker       transferDefinedLanesStep(MO, Info.DefinedLanes);
523*9880d681SAndroid Build Coastguard Worker   }
524*9880d681SAndroid Build Coastguard Worker 
525*9880d681SAndroid Build Coastguard Worker   DEBUG(
526*9880d681SAndroid Build Coastguard Worker     dbgs() << "Defined/Used lanes:\n";
527*9880d681SAndroid Build Coastguard Worker     for (unsigned RegIdx = 0; RegIdx < NumVirtRegs; ++RegIdx) {
528*9880d681SAndroid Build Coastguard Worker       unsigned Reg = TargetRegisterInfo::index2VirtReg(RegIdx);
529*9880d681SAndroid Build Coastguard Worker       const VRegInfo &Info = VRegInfos[RegIdx];
530*9880d681SAndroid Build Coastguard Worker       dbgs() << PrintReg(Reg, nullptr)
531*9880d681SAndroid Build Coastguard Worker              << " Used: " << PrintLaneMask(Info.UsedLanes)
532*9880d681SAndroid Build Coastguard Worker              << " Def: " << PrintLaneMask(Info.DefinedLanes) << '\n';
533*9880d681SAndroid Build Coastguard Worker     }
534*9880d681SAndroid Build Coastguard Worker     dbgs() << "\n";
535*9880d681SAndroid Build Coastguard Worker   );
536*9880d681SAndroid Build Coastguard Worker 
537*9880d681SAndroid Build Coastguard Worker   bool Again = false;
538*9880d681SAndroid Build Coastguard Worker   // Mark operands as dead/unused.
539*9880d681SAndroid Build Coastguard Worker   for (MachineBasicBlock &MBB : MF) {
540*9880d681SAndroid Build Coastguard Worker     for (MachineInstr &MI : MBB) {
541*9880d681SAndroid Build Coastguard Worker       for (MachineOperand &MO : MI.operands()) {
542*9880d681SAndroid Build Coastguard Worker         if (!MO.isReg())
543*9880d681SAndroid Build Coastguard Worker           continue;
544*9880d681SAndroid Build Coastguard Worker         unsigned Reg = MO.getReg();
545*9880d681SAndroid Build Coastguard Worker         if (!TargetRegisterInfo::isVirtualRegister(Reg))
546*9880d681SAndroid Build Coastguard Worker           continue;
547*9880d681SAndroid Build Coastguard Worker         unsigned RegIdx = TargetRegisterInfo::virtReg2Index(Reg);
548*9880d681SAndroid Build Coastguard Worker         const VRegInfo &RegInfo = VRegInfos[RegIdx];
549*9880d681SAndroid Build Coastguard Worker         if (MO.isDef() && !MO.isDead() && RegInfo.UsedLanes == 0) {
550*9880d681SAndroid Build Coastguard Worker           DEBUG(dbgs() << "Marking operand '" << MO << "' as dead in " << MI);
551*9880d681SAndroid Build Coastguard Worker           MO.setIsDead();
552*9880d681SAndroid Build Coastguard Worker         }
553*9880d681SAndroid Build Coastguard Worker         if (MO.readsReg()) {
554*9880d681SAndroid Build Coastguard Worker           bool CrossCopy = false;
555*9880d681SAndroid Build Coastguard Worker           if (isUndefRegAtInput(MO, RegInfo)) {
556*9880d681SAndroid Build Coastguard Worker             DEBUG(dbgs() << "Marking operand '" << MO << "' as undef in "
557*9880d681SAndroid Build Coastguard Worker                   << MI);
558*9880d681SAndroid Build Coastguard Worker             MO.setIsUndef();
559*9880d681SAndroid Build Coastguard Worker           } else if (isUndefInput(MO, &CrossCopy)) {
560*9880d681SAndroid Build Coastguard Worker             DEBUG(dbgs() << "Marking operand '" << MO << "' as undef in "
561*9880d681SAndroid Build Coastguard Worker                   << MI);
562*9880d681SAndroid Build Coastguard Worker             MO.setIsUndef();
563*9880d681SAndroid Build Coastguard Worker             if (CrossCopy)
564*9880d681SAndroid Build Coastguard Worker               Again = true;
565*9880d681SAndroid Build Coastguard Worker           }
566*9880d681SAndroid Build Coastguard Worker         }
567*9880d681SAndroid Build Coastguard Worker       }
568*9880d681SAndroid Build Coastguard Worker     }
569*9880d681SAndroid Build Coastguard Worker   }
570*9880d681SAndroid Build Coastguard Worker 
571*9880d681SAndroid Build Coastguard Worker   return Again;
572*9880d681SAndroid Build Coastguard Worker }
573*9880d681SAndroid Build Coastguard Worker 
runOnMachineFunction(MachineFunction & MF)574*9880d681SAndroid Build Coastguard Worker bool DetectDeadLanes::runOnMachineFunction(MachineFunction &MF) {
575*9880d681SAndroid Build Coastguard Worker   // Don't bother if we won't track subregister liveness later.  This pass is
576*9880d681SAndroid Build Coastguard Worker   // required for correctness if subregister liveness is enabled because the
577*9880d681SAndroid Build Coastguard Worker   // register coalescer cannot deal with hidden dead defs. However without
578*9880d681SAndroid Build Coastguard Worker   // subregister liveness enabled, the expected benefits of this pass are small
579*9880d681SAndroid Build Coastguard Worker   // so we safe the compile time.
580*9880d681SAndroid Build Coastguard Worker   if (!MF.getSubtarget().enableSubRegLiveness()) {
581*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "Skipping Detect dead lanes pass\n");
582*9880d681SAndroid Build Coastguard Worker     return false;
583*9880d681SAndroid Build Coastguard Worker   }
584*9880d681SAndroid Build Coastguard Worker 
585*9880d681SAndroid Build Coastguard Worker   MRI = &MF.getRegInfo();
586*9880d681SAndroid Build Coastguard Worker   TRI = MRI->getTargetRegisterInfo();
587*9880d681SAndroid Build Coastguard Worker 
588*9880d681SAndroid Build Coastguard Worker   unsigned NumVirtRegs = MRI->getNumVirtRegs();
589*9880d681SAndroid Build Coastguard Worker   VRegInfos = new VRegInfo[NumVirtRegs];
590*9880d681SAndroid Build Coastguard Worker   WorklistMembers.resize(NumVirtRegs);
591*9880d681SAndroid Build Coastguard Worker   DefinedByCopy.resize(NumVirtRegs);
592*9880d681SAndroid Build Coastguard Worker 
593*9880d681SAndroid Build Coastguard Worker   bool Again;
594*9880d681SAndroid Build Coastguard Worker   do {
595*9880d681SAndroid Build Coastguard Worker     Again = runOnce(MF);
596*9880d681SAndroid Build Coastguard Worker   } while(Again);
597*9880d681SAndroid Build Coastguard Worker 
598*9880d681SAndroid Build Coastguard Worker   DefinedByCopy.clear();
599*9880d681SAndroid Build Coastguard Worker   WorklistMembers.clear();
600*9880d681SAndroid Build Coastguard Worker   delete[] VRegInfos;
601*9880d681SAndroid Build Coastguard Worker   return true;
602*9880d681SAndroid Build Coastguard Worker }
603