xref: /aosp_15_r20/external/llvm/lib/Target/Hexagon/HexagonOptAddrMode.cpp (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker //===--- HexagonOptAddrMode.cpp -------------------------------------------===//
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 // This implements a Hexagon-specific pass to optimize addressing mode for
10*9880d681SAndroid Build Coastguard Worker // load/store instructions.
11*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
12*9880d681SAndroid Build Coastguard Worker 
13*9880d681SAndroid Build Coastguard Worker #define DEBUG_TYPE "opt-addr-mode"
14*9880d681SAndroid Build Coastguard Worker 
15*9880d681SAndroid Build Coastguard Worker #include "HexagonTargetMachine.h"
16*9880d681SAndroid Build Coastguard Worker #include "RDFGraph.h"
17*9880d681SAndroid Build Coastguard Worker #include "RDFLiveness.h"
18*9880d681SAndroid Build Coastguard Worker 
19*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/DenseSet.h"
20*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineBasicBlock.h"
21*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineDominanceFrontier.h"
22*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineDominators.h"
23*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineFunction.h"
24*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineInstrBuilder.h"
25*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineRegisterInfo.h"
26*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/Passes.h"
27*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/CommandLine.h"
28*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/Debug.h"
29*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/raw_ostream.h"
30*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetInstrInfo.h"
31*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetMachine.h"
32*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetRegisterInfo.h"
33*9880d681SAndroid Build Coastguard Worker 
34*9880d681SAndroid Build Coastguard Worker static cl::opt<int> CodeGrowthLimit("hexagon-amode-growth-limit",
35*9880d681SAndroid Build Coastguard Worker   cl::Hidden, cl::init(0), cl::desc("Code growth limit for address mode "
36*9880d681SAndroid Build Coastguard Worker   "optimization"));
37*9880d681SAndroid Build Coastguard Worker 
38*9880d681SAndroid Build Coastguard Worker using namespace llvm;
39*9880d681SAndroid Build Coastguard Worker using namespace rdf;
40*9880d681SAndroid Build Coastguard Worker 
41*9880d681SAndroid Build Coastguard Worker namespace llvm {
42*9880d681SAndroid Build Coastguard Worker   FunctionPass *createHexagonOptAddrMode();
43*9880d681SAndroid Build Coastguard Worker   void initializeHexagonOptAddrModePass(PassRegistry &);
44*9880d681SAndroid Build Coastguard Worker }
45*9880d681SAndroid Build Coastguard Worker 
46*9880d681SAndroid Build Coastguard Worker namespace {
47*9880d681SAndroid Build Coastguard Worker class HexagonOptAddrMode : public MachineFunctionPass {
48*9880d681SAndroid Build Coastguard Worker public:
49*9880d681SAndroid Build Coastguard Worker   static char ID;
HexagonOptAddrMode()50*9880d681SAndroid Build Coastguard Worker   HexagonOptAddrMode()
51*9880d681SAndroid Build Coastguard Worker       : MachineFunctionPass(ID), HII(0), MDT(0), DFG(0), LV(0) {
52*9880d681SAndroid Build Coastguard Worker     PassRegistry &R = *PassRegistry::getPassRegistry();
53*9880d681SAndroid Build Coastguard Worker     initializeHexagonOptAddrModePass(R);
54*9880d681SAndroid Build Coastguard Worker   }
getPassName() const55*9880d681SAndroid Build Coastguard Worker   const char *getPassName() const override {
56*9880d681SAndroid Build Coastguard Worker     return "Optimize addressing mode of load/store";
57*9880d681SAndroid Build Coastguard Worker   }
getAnalysisUsage(AnalysisUsage & AU) const58*9880d681SAndroid Build Coastguard Worker   void getAnalysisUsage(AnalysisUsage &AU) const override {
59*9880d681SAndroid Build Coastguard Worker     MachineFunctionPass::getAnalysisUsage(AU);
60*9880d681SAndroid Build Coastguard Worker     AU.addRequired<MachineDominatorTree>();
61*9880d681SAndroid Build Coastguard Worker     AU.addRequired<MachineDominanceFrontier>();
62*9880d681SAndroid Build Coastguard Worker     AU.setPreservesAll();
63*9880d681SAndroid Build Coastguard Worker   }
64*9880d681SAndroid Build Coastguard Worker   bool runOnMachineFunction(MachineFunction &MF) override;
65*9880d681SAndroid Build Coastguard Worker 
66*9880d681SAndroid Build Coastguard Worker private:
67*9880d681SAndroid Build Coastguard Worker   typedef DenseSet<MachineInstr *> MISetType;
68*9880d681SAndroid Build Coastguard Worker   typedef DenseMap<MachineInstr *, bool> InstrEvalMap;
69*9880d681SAndroid Build Coastguard Worker   const HexagonInstrInfo *HII;
70*9880d681SAndroid Build Coastguard Worker   MachineDominatorTree *MDT;
71*9880d681SAndroid Build Coastguard Worker   DataFlowGraph *DFG;
72*9880d681SAndroid Build Coastguard Worker   DataFlowGraph::DefStackMap DefM;
73*9880d681SAndroid Build Coastguard Worker   std::map<RegisterRef, std::map<NodeId, NodeId>> RDefMap;
74*9880d681SAndroid Build Coastguard Worker   Liveness *LV;
75*9880d681SAndroid Build Coastguard Worker   MISetType Deleted;
76*9880d681SAndroid Build Coastguard Worker 
77*9880d681SAndroid Build Coastguard Worker   bool processBlock(NodeAddr<BlockNode *> BA);
78*9880d681SAndroid Build Coastguard Worker   bool xformUseMI(MachineInstr *TfrMI, MachineInstr *UseMI,
79*9880d681SAndroid Build Coastguard Worker                   NodeAddr<UseNode *> UseN, unsigned UseMOnum);
80*9880d681SAndroid Build Coastguard Worker   bool analyzeUses(unsigned DefR, const NodeList &UNodeList,
81*9880d681SAndroid Build Coastguard Worker                    InstrEvalMap &InstrEvalResult, short &SizeInc);
82*9880d681SAndroid Build Coastguard Worker   bool hasRepForm(MachineInstr *MI, unsigned TfrDefR);
83*9880d681SAndroid Build Coastguard Worker   bool canRemoveAddasl(NodeAddr<StmtNode *> AddAslSN, MachineInstr *MI,
84*9880d681SAndroid Build Coastguard Worker                        const NodeList &UNodeList);
85*9880d681SAndroid Build Coastguard Worker   void getAllRealUses(NodeAddr<StmtNode *> SN, NodeList &UNodeList);
86*9880d681SAndroid Build Coastguard Worker   bool allValidCandidates(NodeAddr<StmtNode *> SA, NodeList &UNodeList);
87*9880d681SAndroid Build Coastguard Worker   short getBaseWithLongOffset(const MachineInstr *MI) const;
88*9880d681SAndroid Build Coastguard Worker   void updateMap(NodeAddr<InstrNode *> IA);
89*9880d681SAndroid Build Coastguard Worker   bool constructDefMap(MachineBasicBlock *B);
90*9880d681SAndroid Build Coastguard Worker   bool changeStore(MachineInstr *OldMI, MachineOperand ImmOp,
91*9880d681SAndroid Build Coastguard Worker                    unsigned ImmOpNum);
92*9880d681SAndroid Build Coastguard Worker   bool changeLoad(MachineInstr *OldMI, MachineOperand ImmOp, unsigned ImmOpNum);
93*9880d681SAndroid Build Coastguard Worker   bool changeAddAsl(NodeAddr<UseNode *> AddAslUN, MachineInstr *AddAslMI,
94*9880d681SAndroid Build Coastguard Worker                     const MachineOperand &ImmOp, unsigned ImmOpNum);
95*9880d681SAndroid Build Coastguard Worker };
96*9880d681SAndroid Build Coastguard Worker }
97*9880d681SAndroid Build Coastguard Worker 
98*9880d681SAndroid Build Coastguard Worker char HexagonOptAddrMode::ID = 0;
99*9880d681SAndroid Build Coastguard Worker 
100*9880d681SAndroid Build Coastguard Worker INITIALIZE_PASS_BEGIN(HexagonOptAddrMode, "opt-amode",
101*9880d681SAndroid Build Coastguard Worker                       "Optimize addressing mode", false, false)
INITIALIZE_PASS_DEPENDENCY(MachineDominatorTree)102*9880d681SAndroid Build Coastguard Worker INITIALIZE_PASS_DEPENDENCY(MachineDominatorTree)
103*9880d681SAndroid Build Coastguard Worker INITIALIZE_PASS_DEPENDENCY(MachineDominanceFrontier)
104*9880d681SAndroid Build Coastguard Worker INITIALIZE_PASS_END(HexagonOptAddrMode, "opt-amode", "Optimize addressing mode",
105*9880d681SAndroid Build Coastguard Worker                     false, false)
106*9880d681SAndroid Build Coastguard Worker 
107*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::hasRepForm(MachineInstr *MI, unsigned TfrDefR) {
108*9880d681SAndroid Build Coastguard Worker   const MCInstrDesc &MID = MI->getDesc();
109*9880d681SAndroid Build Coastguard Worker 
110*9880d681SAndroid Build Coastguard Worker   if ((!MID.mayStore() && !MID.mayLoad()) || HII->isPredicated(*MI))
111*9880d681SAndroid Build Coastguard Worker     return false;
112*9880d681SAndroid Build Coastguard Worker 
113*9880d681SAndroid Build Coastguard Worker   if (MID.mayStore()) {
114*9880d681SAndroid Build Coastguard Worker     MachineOperand StOp = MI->getOperand(MI->getNumOperands() - 1);
115*9880d681SAndroid Build Coastguard Worker     if (StOp.isReg() && StOp.getReg() == TfrDefR)
116*9880d681SAndroid Build Coastguard Worker       return false;
117*9880d681SAndroid Build Coastguard Worker   }
118*9880d681SAndroid Build Coastguard Worker 
119*9880d681SAndroid Build Coastguard Worker   if (HII->getAddrMode(MI) == HexagonII::BaseRegOffset)
120*9880d681SAndroid Build Coastguard Worker     // Tranform to Absolute plus register offset.
121*9880d681SAndroid Build Coastguard Worker     return (HII->getBaseWithLongOffset(MI) >= 0);
122*9880d681SAndroid Build Coastguard Worker   else if (HII->getAddrMode(MI) == HexagonII::BaseImmOffset)
123*9880d681SAndroid Build Coastguard Worker     // Tranform to absolute addressing mode.
124*9880d681SAndroid Build Coastguard Worker     return (HII->getAbsoluteForm(MI) >= 0);
125*9880d681SAndroid Build Coastguard Worker 
126*9880d681SAndroid Build Coastguard Worker   return false;
127*9880d681SAndroid Build Coastguard Worker }
128*9880d681SAndroid Build Coastguard Worker 
129*9880d681SAndroid Build Coastguard Worker // Check if addasl instruction can be removed. This is possible only
130*9880d681SAndroid Build Coastguard Worker // if it's feeding to only load/store instructions with base + register
131*9880d681SAndroid Build Coastguard Worker // offset as these instruction can be tranformed to use 'absolute plus
132*9880d681SAndroid Build Coastguard Worker // shifted register offset'.
133*9880d681SAndroid Build Coastguard Worker // ex:
134*9880d681SAndroid Build Coastguard Worker // Rs = ##foo
135*9880d681SAndroid Build Coastguard Worker // Rx = addasl(Rs, Rt, #2)
136*9880d681SAndroid Build Coastguard Worker // Rd = memw(Rx + #28)
137*9880d681SAndroid Build Coastguard Worker // Above three instructions can be replaced with Rd = memw(Rt<<#2 + ##foo+28)
138*9880d681SAndroid Build Coastguard Worker 
canRemoveAddasl(NodeAddr<StmtNode * > AddAslSN,MachineInstr * MI,const NodeList & UNodeList)139*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::canRemoveAddasl(NodeAddr<StmtNode *> AddAslSN,
140*9880d681SAndroid Build Coastguard Worker                                          MachineInstr *MI,
141*9880d681SAndroid Build Coastguard Worker                                          const NodeList &UNodeList) {
142*9880d681SAndroid Build Coastguard Worker   // check offset size in addasl. if 'offset > 3' return false
143*9880d681SAndroid Build Coastguard Worker   const MachineOperand &OffsetOp = MI->getOperand(3);
144*9880d681SAndroid Build Coastguard Worker   if (!OffsetOp.isImm() || OffsetOp.getImm() > 3)
145*9880d681SAndroid Build Coastguard Worker     return false;
146*9880d681SAndroid Build Coastguard Worker 
147*9880d681SAndroid Build Coastguard Worker   unsigned OffsetReg = MI->getOperand(2).getReg();
148*9880d681SAndroid Build Coastguard Worker   RegisterRef OffsetRR;
149*9880d681SAndroid Build Coastguard Worker   NodeId OffsetRegRD = 0;
150*9880d681SAndroid Build Coastguard Worker   for (NodeAddr<UseNode *> UA : AddAslSN.Addr->members_if(DFG->IsUse, *DFG)) {
151*9880d681SAndroid Build Coastguard Worker     RegisterRef RR = UA.Addr->getRegRef();
152*9880d681SAndroid Build Coastguard Worker     if (OffsetReg == RR.Reg) {
153*9880d681SAndroid Build Coastguard Worker       OffsetRR = RR;
154*9880d681SAndroid Build Coastguard Worker       OffsetRegRD = UA.Addr->getReachingDef();
155*9880d681SAndroid Build Coastguard Worker     }
156*9880d681SAndroid Build Coastguard Worker   }
157*9880d681SAndroid Build Coastguard Worker 
158*9880d681SAndroid Build Coastguard Worker   for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
159*9880d681SAndroid Build Coastguard Worker     NodeAddr<UseNode *> UA = *I;
160*9880d681SAndroid Build Coastguard Worker     NodeAddr<InstrNode *> IA = UA.Addr->getOwner(*DFG);
161*9880d681SAndroid Build Coastguard Worker     if ((UA.Addr->getFlags() & NodeAttrs::PhiRef) ||
162*9880d681SAndroid Build Coastguard Worker         RDefMap[OffsetRR][IA.Id] != OffsetRegRD)
163*9880d681SAndroid Build Coastguard Worker       return false;
164*9880d681SAndroid Build Coastguard Worker 
165*9880d681SAndroid Build Coastguard Worker     MachineInstr *UseMI = NodeAddr<StmtNode *>(IA).Addr->getCode();
166*9880d681SAndroid Build Coastguard Worker     NodeAddr<DefNode *> OffsetRegDN = DFG->addr<DefNode *>(OffsetRegRD);
167*9880d681SAndroid Build Coastguard Worker     // Reaching Def to an offset register can't be a phi.
168*9880d681SAndroid Build Coastguard Worker     if ((OffsetRegDN.Addr->getFlags() & NodeAttrs::PhiRef) &&
169*9880d681SAndroid Build Coastguard Worker         MI->getParent() != UseMI->getParent())
170*9880d681SAndroid Build Coastguard Worker     return false;
171*9880d681SAndroid Build Coastguard Worker 
172*9880d681SAndroid Build Coastguard Worker     const MCInstrDesc &UseMID = UseMI->getDesc();
173*9880d681SAndroid Build Coastguard Worker     if ((!UseMID.mayLoad() && !UseMID.mayStore()) ||
174*9880d681SAndroid Build Coastguard Worker         HII->getAddrMode(UseMI) != HexagonII::BaseImmOffset ||
175*9880d681SAndroid Build Coastguard Worker         getBaseWithLongOffset(UseMI) < 0)
176*9880d681SAndroid Build Coastguard Worker       return false;
177*9880d681SAndroid Build Coastguard Worker 
178*9880d681SAndroid Build Coastguard Worker     // Addasl output can't be a store value.
179*9880d681SAndroid Build Coastguard Worker     if (UseMID.mayStore() && UseMI->getOperand(2).isReg() &&
180*9880d681SAndroid Build Coastguard Worker         UseMI->getOperand(2).getReg() == MI->getOperand(0).getReg())
181*9880d681SAndroid Build Coastguard Worker       return false;
182*9880d681SAndroid Build Coastguard Worker 
183*9880d681SAndroid Build Coastguard Worker     for (auto &Mo : UseMI->operands())
184*9880d681SAndroid Build Coastguard Worker       if (Mo.isFI())
185*9880d681SAndroid Build Coastguard Worker         return false;
186*9880d681SAndroid Build Coastguard Worker   }
187*9880d681SAndroid Build Coastguard Worker   return true;
188*9880d681SAndroid Build Coastguard Worker }
189*9880d681SAndroid Build Coastguard Worker 
allValidCandidates(NodeAddr<StmtNode * > SA,NodeList & UNodeList)190*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::allValidCandidates(NodeAddr<StmtNode *> SA,
191*9880d681SAndroid Build Coastguard Worker                                             NodeList &UNodeList) {
192*9880d681SAndroid Build Coastguard Worker   for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
193*9880d681SAndroid Build Coastguard Worker     NodeAddr<UseNode *> UN = *I;
194*9880d681SAndroid Build Coastguard Worker     RegisterRef UR = UN.Addr->getRegRef();
195*9880d681SAndroid Build Coastguard Worker     NodeSet Visited, Defs;
196*9880d681SAndroid Build Coastguard Worker     const auto &ReachingDefs = LV->getAllReachingDefsRec(UR, UN, Visited, Defs);
197*9880d681SAndroid Build Coastguard Worker     if (ReachingDefs.size() > 1) {
198*9880d681SAndroid Build Coastguard Worker       DEBUG({
199*9880d681SAndroid Build Coastguard Worker         dbgs() << "*** Multiple Reaching Defs found!!! ***\n";
200*9880d681SAndroid Build Coastguard Worker         for (auto DI : ReachingDefs) {
201*9880d681SAndroid Build Coastguard Worker           NodeAddr<UseNode *> DA = DFG->addr<UseNode *>(DI);
202*9880d681SAndroid Build Coastguard Worker           NodeAddr<StmtNode *> TempIA = DA.Addr->getOwner(*DFG);
203*9880d681SAndroid Build Coastguard Worker           dbgs() << "\t\t[Reaching Def]: "
204*9880d681SAndroid Build Coastguard Worker                  << Print<NodeAddr<InstrNode *>>(TempIA, *DFG) << "\n";
205*9880d681SAndroid Build Coastguard Worker         }
206*9880d681SAndroid Build Coastguard Worker       });
207*9880d681SAndroid Build Coastguard Worker       return false;
208*9880d681SAndroid Build Coastguard Worker     }
209*9880d681SAndroid Build Coastguard Worker   }
210*9880d681SAndroid Build Coastguard Worker   return true;
211*9880d681SAndroid Build Coastguard Worker }
212*9880d681SAndroid Build Coastguard Worker 
getAllRealUses(NodeAddr<StmtNode * > SA,NodeList & UNodeList)213*9880d681SAndroid Build Coastguard Worker void HexagonOptAddrMode::getAllRealUses(NodeAddr<StmtNode *> SA,
214*9880d681SAndroid Build Coastguard Worker                                         NodeList &UNodeList) {
215*9880d681SAndroid Build Coastguard Worker   for (NodeAddr<DefNode *> DA : SA.Addr->members_if(DFG->IsDef, *DFG)) {
216*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "\t\t[DefNode]: " << Print<NodeAddr<DefNode *>>(DA, *DFG)
217*9880d681SAndroid Build Coastguard Worker                  << "\n");
218*9880d681SAndroid Build Coastguard Worker     RegisterRef DR = DA.Addr->getRegRef();
219*9880d681SAndroid Build Coastguard Worker     auto UseSet = LV->getAllReachedUses(DR, DA);
220*9880d681SAndroid Build Coastguard Worker 
221*9880d681SAndroid Build Coastguard Worker     for (auto UI : UseSet) {
222*9880d681SAndroid Build Coastguard Worker       NodeAddr<UseNode *> UA = DFG->addr<UseNode *>(UI);
223*9880d681SAndroid Build Coastguard Worker       DEBUG({
224*9880d681SAndroid Build Coastguard Worker         NodeAddr<StmtNode *> TempIA = UA.Addr->getOwner(*DFG);
225*9880d681SAndroid Build Coastguard Worker         dbgs() << "\t\t\t[Reached Use]: "
226*9880d681SAndroid Build Coastguard Worker                << Print<NodeAddr<InstrNode *>>(TempIA, *DFG) << "\n";
227*9880d681SAndroid Build Coastguard Worker       });
228*9880d681SAndroid Build Coastguard Worker 
229*9880d681SAndroid Build Coastguard Worker       if (UA.Addr->getFlags() & NodeAttrs::PhiRef) {
230*9880d681SAndroid Build Coastguard Worker         NodeAddr<PhiNode *> PA = UA.Addr->getOwner(*DFG);
231*9880d681SAndroid Build Coastguard Worker         NodeId id = PA.Id;
232*9880d681SAndroid Build Coastguard Worker         const Liveness::RefMap &phiUse = LV->getRealUses(id);
233*9880d681SAndroid Build Coastguard Worker         DEBUG(dbgs() << "\t\t\t\tphi real Uses"
234*9880d681SAndroid Build Coastguard Worker                      << Print<Liveness::RefMap>(phiUse, *DFG) << "\n");
235*9880d681SAndroid Build Coastguard Worker         if (phiUse.size() > 0) {
236*9880d681SAndroid Build Coastguard Worker           for (auto I : phiUse) {
237*9880d681SAndroid Build Coastguard Worker             if (DR != I.first)
238*9880d681SAndroid Build Coastguard Worker               continue;
239*9880d681SAndroid Build Coastguard Worker             auto phiUseSet = I.second;
240*9880d681SAndroid Build Coastguard Worker             for (auto phiUI : phiUseSet) {
241*9880d681SAndroid Build Coastguard Worker               NodeAddr<UseNode *> phiUA = DFG->addr<UseNode *>(phiUI);
242*9880d681SAndroid Build Coastguard Worker               UNodeList.push_back(phiUA);
243*9880d681SAndroid Build Coastguard Worker             }
244*9880d681SAndroid Build Coastguard Worker           }
245*9880d681SAndroid Build Coastguard Worker         }
246*9880d681SAndroid Build Coastguard Worker       } else
247*9880d681SAndroid Build Coastguard Worker         UNodeList.push_back(UA);
248*9880d681SAndroid Build Coastguard Worker     }
249*9880d681SAndroid Build Coastguard Worker   }
250*9880d681SAndroid Build Coastguard Worker }
251*9880d681SAndroid Build Coastguard Worker 
analyzeUses(unsigned tfrDefR,const NodeList & UNodeList,InstrEvalMap & InstrEvalResult,short & SizeInc)252*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::analyzeUses(unsigned tfrDefR,
253*9880d681SAndroid Build Coastguard Worker                                      const NodeList &UNodeList,
254*9880d681SAndroid Build Coastguard Worker                                      InstrEvalMap &InstrEvalResult,
255*9880d681SAndroid Build Coastguard Worker                                      short &SizeInc) {
256*9880d681SAndroid Build Coastguard Worker   bool KeepTfr = false;
257*9880d681SAndroid Build Coastguard Worker   bool HasRepInstr = false;
258*9880d681SAndroid Build Coastguard Worker   InstrEvalResult.clear();
259*9880d681SAndroid Build Coastguard Worker 
260*9880d681SAndroid Build Coastguard Worker   for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
261*9880d681SAndroid Build Coastguard Worker     bool CanBeReplaced = false;
262*9880d681SAndroid Build Coastguard Worker     NodeAddr<UseNode *> UN = *I;
263*9880d681SAndroid Build Coastguard Worker     NodeAddr<StmtNode *> SN = UN.Addr->getOwner(*DFG);
264*9880d681SAndroid Build Coastguard Worker     MachineInstr *MI = SN.Addr->getCode();
265*9880d681SAndroid Build Coastguard Worker     const MCInstrDesc &MID = MI->getDesc();
266*9880d681SAndroid Build Coastguard Worker     if ((MID.mayLoad() || MID.mayStore())) {
267*9880d681SAndroid Build Coastguard Worker       if (!hasRepForm(MI, tfrDefR)) {
268*9880d681SAndroid Build Coastguard Worker         KeepTfr = true;
269*9880d681SAndroid Build Coastguard Worker         continue;
270*9880d681SAndroid Build Coastguard Worker       }
271*9880d681SAndroid Build Coastguard Worker       SizeInc++;
272*9880d681SAndroid Build Coastguard Worker       CanBeReplaced = true;
273*9880d681SAndroid Build Coastguard Worker     } else if (MI->getOpcode() == Hexagon::S2_addasl_rrri) {
274*9880d681SAndroid Build Coastguard Worker       NodeList AddaslUseList;
275*9880d681SAndroid Build Coastguard Worker 
276*9880d681SAndroid Build Coastguard Worker       DEBUG(dbgs() << "\nGetting ReachedUses for === " << *MI << "\n");
277*9880d681SAndroid Build Coastguard Worker       getAllRealUses(SN, AddaslUseList);
278*9880d681SAndroid Build Coastguard Worker       // Process phi nodes.
279*9880d681SAndroid Build Coastguard Worker       if (allValidCandidates(SN, AddaslUseList) &&
280*9880d681SAndroid Build Coastguard Worker           canRemoveAddasl(SN, MI, AddaslUseList)) {
281*9880d681SAndroid Build Coastguard Worker         SizeInc += AddaslUseList.size();
282*9880d681SAndroid Build Coastguard Worker         SizeInc -= 1; // Reduce size by 1 as addasl itself can be removed.
283*9880d681SAndroid Build Coastguard Worker         CanBeReplaced = true;
284*9880d681SAndroid Build Coastguard Worker       } else
285*9880d681SAndroid Build Coastguard Worker         SizeInc++;
286*9880d681SAndroid Build Coastguard Worker     } else
287*9880d681SAndroid Build Coastguard Worker       // Currently, only load/store and addasl are handled.
288*9880d681SAndroid Build Coastguard Worker       // Some other instructions to consider -
289*9880d681SAndroid Build Coastguard Worker       // A2_add -> A2_addi
290*9880d681SAndroid Build Coastguard Worker       // M4_mpyrr_addr -> M4_mpyrr_addi
291*9880d681SAndroid Build Coastguard Worker       KeepTfr = true;
292*9880d681SAndroid Build Coastguard Worker 
293*9880d681SAndroid Build Coastguard Worker     InstrEvalResult[MI] = CanBeReplaced;
294*9880d681SAndroid Build Coastguard Worker     HasRepInstr |= CanBeReplaced;
295*9880d681SAndroid Build Coastguard Worker   }
296*9880d681SAndroid Build Coastguard Worker 
297*9880d681SAndroid Build Coastguard Worker   // Reduce total size by 2 if original tfr can be deleted.
298*9880d681SAndroid Build Coastguard Worker   if (!KeepTfr)
299*9880d681SAndroid Build Coastguard Worker     SizeInc -= 2;
300*9880d681SAndroid Build Coastguard Worker 
301*9880d681SAndroid Build Coastguard Worker   return HasRepInstr;
302*9880d681SAndroid Build Coastguard Worker }
303*9880d681SAndroid Build Coastguard Worker 
changeLoad(MachineInstr * OldMI,MachineOperand ImmOp,unsigned ImmOpNum)304*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::changeLoad(MachineInstr *OldMI, MachineOperand ImmOp,
305*9880d681SAndroid Build Coastguard Worker                                     unsigned ImmOpNum) {
306*9880d681SAndroid Build Coastguard Worker   bool Changed = false;
307*9880d681SAndroid Build Coastguard Worker   MachineBasicBlock *BB = OldMI->getParent();
308*9880d681SAndroid Build Coastguard Worker   auto UsePos = MachineBasicBlock::iterator(OldMI);
309*9880d681SAndroid Build Coastguard Worker   MachineBasicBlock::instr_iterator InsertPt = UsePos.getInstrIterator();
310*9880d681SAndroid Build Coastguard Worker   ++InsertPt;
311*9880d681SAndroid Build Coastguard Worker   unsigned OpStart;
312*9880d681SAndroid Build Coastguard Worker   unsigned OpEnd = OldMI->getNumOperands();
313*9880d681SAndroid Build Coastguard Worker   MachineInstrBuilder MIB;
314*9880d681SAndroid Build Coastguard Worker 
315*9880d681SAndroid Build Coastguard Worker   if (ImmOpNum == 1) {
316*9880d681SAndroid Build Coastguard Worker     if (HII->getAddrMode(OldMI) == HexagonII::BaseRegOffset) {
317*9880d681SAndroid Build Coastguard Worker       short NewOpCode = HII->getBaseWithLongOffset(OldMI);
318*9880d681SAndroid Build Coastguard Worker       assert(NewOpCode >= 0 && "Invalid New opcode\n");
319*9880d681SAndroid Build Coastguard Worker       MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
320*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(0));
321*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(2));
322*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(3));
323*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(ImmOp);
324*9880d681SAndroid Build Coastguard Worker       OpStart = 4;
325*9880d681SAndroid Build Coastguard Worker       Changed = true;
326*9880d681SAndroid Build Coastguard Worker     } else if (HII->getAddrMode(OldMI) == HexagonII::BaseImmOffset) {
327*9880d681SAndroid Build Coastguard Worker       short NewOpCode = HII->getAbsoluteForm(OldMI);
328*9880d681SAndroid Build Coastguard Worker       assert(NewOpCode >= 0 && "Invalid New opcode\n");
329*9880d681SAndroid Build Coastguard Worker       MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode))
330*9880d681SAndroid Build Coastguard Worker                 .addOperand(OldMI->getOperand(0));
331*9880d681SAndroid Build Coastguard Worker       const GlobalValue *GV = ImmOp.getGlobal();
332*9880d681SAndroid Build Coastguard Worker       int64_t Offset = ImmOp.getOffset() + OldMI->getOperand(2).getImm();
333*9880d681SAndroid Build Coastguard Worker 
334*9880d681SAndroid Build Coastguard Worker       MIB.addGlobalAddress(GV, Offset, ImmOp.getTargetFlags());
335*9880d681SAndroid Build Coastguard Worker       OpStart = 3;
336*9880d681SAndroid Build Coastguard Worker       Changed = true;
337*9880d681SAndroid Build Coastguard Worker     } else
338*9880d681SAndroid Build Coastguard Worker       Changed = false;
339*9880d681SAndroid Build Coastguard Worker 
340*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
341*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[TO]: " << MIB << "\n");
342*9880d681SAndroid Build Coastguard Worker   } else if (ImmOpNum == 2 && OldMI->getOperand(3).getImm() == 0) {
343*9880d681SAndroid Build Coastguard Worker     short NewOpCode = HII->xformRegToImmOffset(OldMI);
344*9880d681SAndroid Build Coastguard Worker     assert(NewOpCode >= 0 && "Invalid New opcode\n");
345*9880d681SAndroid Build Coastguard Worker     MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
346*9880d681SAndroid Build Coastguard Worker     MIB.addOperand(OldMI->getOperand(0));
347*9880d681SAndroid Build Coastguard Worker     MIB.addOperand(OldMI->getOperand(1));
348*9880d681SAndroid Build Coastguard Worker     MIB.addOperand(ImmOp);
349*9880d681SAndroid Build Coastguard Worker     OpStart = 4;
350*9880d681SAndroid Build Coastguard Worker     Changed = true;
351*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
352*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[TO]: " << MIB << "\n");
353*9880d681SAndroid Build Coastguard Worker   }
354*9880d681SAndroid Build Coastguard Worker 
355*9880d681SAndroid Build Coastguard Worker   if (Changed)
356*9880d681SAndroid Build Coastguard Worker     for (unsigned i = OpStart; i < OpEnd; ++i)
357*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(i));
358*9880d681SAndroid Build Coastguard Worker 
359*9880d681SAndroid Build Coastguard Worker   return Changed;
360*9880d681SAndroid Build Coastguard Worker }
361*9880d681SAndroid Build Coastguard Worker 
changeStore(MachineInstr * OldMI,MachineOperand ImmOp,unsigned ImmOpNum)362*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::changeStore(MachineInstr *OldMI, MachineOperand ImmOp,
363*9880d681SAndroid Build Coastguard Worker                                      unsigned ImmOpNum) {
364*9880d681SAndroid Build Coastguard Worker   bool Changed = false;
365*9880d681SAndroid Build Coastguard Worker   unsigned OpStart;
366*9880d681SAndroid Build Coastguard Worker   unsigned OpEnd = OldMI->getNumOperands();
367*9880d681SAndroid Build Coastguard Worker   MachineBasicBlock *BB = OldMI->getParent();
368*9880d681SAndroid Build Coastguard Worker   auto UsePos = MachineBasicBlock::iterator(OldMI);
369*9880d681SAndroid Build Coastguard Worker   MachineBasicBlock::instr_iterator InsertPt = UsePos.getInstrIterator();
370*9880d681SAndroid Build Coastguard Worker   ++InsertPt;
371*9880d681SAndroid Build Coastguard Worker   MachineInstrBuilder MIB;
372*9880d681SAndroid Build Coastguard Worker   if (ImmOpNum == 0) {
373*9880d681SAndroid Build Coastguard Worker     if (HII->getAddrMode(OldMI) == HexagonII::BaseRegOffset) {
374*9880d681SAndroid Build Coastguard Worker       short NewOpCode = HII->getBaseWithLongOffset(OldMI);
375*9880d681SAndroid Build Coastguard Worker       assert(NewOpCode >= 0 && "Invalid New opcode\n");
376*9880d681SAndroid Build Coastguard Worker       MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
377*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(1));
378*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(2));
379*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(ImmOp);
380*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(3));
381*9880d681SAndroid Build Coastguard Worker       OpStart = 4;
382*9880d681SAndroid Build Coastguard Worker     } else if (HII->getAddrMode(OldMI) == HexagonII::BaseImmOffset) {
383*9880d681SAndroid Build Coastguard Worker       short NewOpCode = HII->getAbsoluteForm(OldMI);
384*9880d681SAndroid Build Coastguard Worker       assert(NewOpCode >= 0 && "Invalid New opcode\n");
385*9880d681SAndroid Build Coastguard Worker       MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
386*9880d681SAndroid Build Coastguard Worker       const GlobalValue *GV = ImmOp.getGlobal();
387*9880d681SAndroid Build Coastguard Worker       int64_t Offset = ImmOp.getOffset() + OldMI->getOperand(1).getImm();
388*9880d681SAndroid Build Coastguard Worker       MIB.addGlobalAddress(GV, Offset, ImmOp.getTargetFlags());
389*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(2));
390*9880d681SAndroid Build Coastguard Worker       OpStart = 3;
391*9880d681SAndroid Build Coastguard Worker     }
392*9880d681SAndroid Build Coastguard Worker     Changed = true;
393*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
394*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[TO]: " << MIB << "\n");
395*9880d681SAndroid Build Coastguard Worker   } else if (ImmOpNum == 1 && OldMI->getOperand(2).getImm() == 0) {
396*9880d681SAndroid Build Coastguard Worker     short NewOpCode = HII->xformRegToImmOffset(OldMI);
397*9880d681SAndroid Build Coastguard Worker     assert(NewOpCode >= 0 && "Invalid New opcode\n");
398*9880d681SAndroid Build Coastguard Worker     MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
399*9880d681SAndroid Build Coastguard Worker     MIB.addOperand(OldMI->getOperand(0));
400*9880d681SAndroid Build Coastguard Worker     MIB.addOperand(ImmOp);
401*9880d681SAndroid Build Coastguard Worker     MIB.addOperand(OldMI->getOperand(1));
402*9880d681SAndroid Build Coastguard Worker     OpStart = 2;
403*9880d681SAndroid Build Coastguard Worker     Changed = true;
404*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
405*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[TO]: " << MIB << "\n");
406*9880d681SAndroid Build Coastguard Worker   }
407*9880d681SAndroid Build Coastguard Worker   if (Changed)
408*9880d681SAndroid Build Coastguard Worker     for (unsigned i = OpStart; i < OpEnd; ++i)
409*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(OldMI->getOperand(i));
410*9880d681SAndroid Build Coastguard Worker 
411*9880d681SAndroid Build Coastguard Worker   return Changed;
412*9880d681SAndroid Build Coastguard Worker }
413*9880d681SAndroid Build Coastguard Worker 
getBaseWithLongOffset(const MachineInstr * MI) const414*9880d681SAndroid Build Coastguard Worker short HexagonOptAddrMode::getBaseWithLongOffset(const MachineInstr *MI) const {
415*9880d681SAndroid Build Coastguard Worker   if (HII->getAddrMode(MI) == HexagonII::BaseImmOffset) {
416*9880d681SAndroid Build Coastguard Worker     short TempOpCode = HII->getBaseWithRegOffset(MI);
417*9880d681SAndroid Build Coastguard Worker     return HII->getBaseWithLongOffset(TempOpCode);
418*9880d681SAndroid Build Coastguard Worker   } else
419*9880d681SAndroid Build Coastguard Worker     return HII->getBaseWithLongOffset(MI);
420*9880d681SAndroid Build Coastguard Worker }
421*9880d681SAndroid Build Coastguard Worker 
changeAddAsl(NodeAddr<UseNode * > AddAslUN,MachineInstr * AddAslMI,const MachineOperand & ImmOp,unsigned ImmOpNum)422*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::changeAddAsl(NodeAddr<UseNode *> AddAslUN,
423*9880d681SAndroid Build Coastguard Worker                                       MachineInstr *AddAslMI,
424*9880d681SAndroid Build Coastguard Worker                                       const MachineOperand &ImmOp,
425*9880d681SAndroid Build Coastguard Worker                                       unsigned ImmOpNum) {
426*9880d681SAndroid Build Coastguard Worker   NodeAddr<StmtNode *> SA = AddAslUN.Addr->getOwner(*DFG);
427*9880d681SAndroid Build Coastguard Worker 
428*9880d681SAndroid Build Coastguard Worker   DEBUG(dbgs() << "Processing addasl :" << *AddAslMI << "\n");
429*9880d681SAndroid Build Coastguard Worker 
430*9880d681SAndroid Build Coastguard Worker   NodeList UNodeList;
431*9880d681SAndroid Build Coastguard Worker   getAllRealUses(SA, UNodeList);
432*9880d681SAndroid Build Coastguard Worker 
433*9880d681SAndroid Build Coastguard Worker   for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
434*9880d681SAndroid Build Coastguard Worker     NodeAddr<UseNode *> UseUN = *I;
435*9880d681SAndroid Build Coastguard Worker     assert(!(UseUN.Addr->getFlags() & NodeAttrs::PhiRef) &&
436*9880d681SAndroid Build Coastguard Worker            "Can't transform this 'AddAsl' instruction!");
437*9880d681SAndroid Build Coastguard Worker 
438*9880d681SAndroid Build Coastguard Worker     NodeAddr<StmtNode *> UseIA = UseUN.Addr->getOwner(*DFG);
439*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[InstrNode]: " << Print<NodeAddr<InstrNode *>>(UseIA, *DFG)
440*9880d681SAndroid Build Coastguard Worker                  << "\n");
441*9880d681SAndroid Build Coastguard Worker     MachineInstr *UseMI = UseIA.Addr->getCode();
442*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[MI <BB#" << UseMI->getParent()->getNumber()
443*9880d681SAndroid Build Coastguard Worker                  << ">]: " << *UseMI << "\n");
444*9880d681SAndroid Build Coastguard Worker     const MCInstrDesc &UseMID = UseMI->getDesc();
445*9880d681SAndroid Build Coastguard Worker     assert(HII->getAddrMode(UseMI) == HexagonII::BaseImmOffset);
446*9880d681SAndroid Build Coastguard Worker 
447*9880d681SAndroid Build Coastguard Worker     auto UsePos = MachineBasicBlock::iterator(UseMI);
448*9880d681SAndroid Build Coastguard Worker     MachineBasicBlock::instr_iterator InsertPt = UsePos.getInstrIterator();
449*9880d681SAndroid Build Coastguard Worker     short NewOpCode = getBaseWithLongOffset(UseMI);
450*9880d681SAndroid Build Coastguard Worker     assert(NewOpCode >= 0 && "Invalid New opcode\n");
451*9880d681SAndroid Build Coastguard Worker 
452*9880d681SAndroid Build Coastguard Worker     unsigned OpStart;
453*9880d681SAndroid Build Coastguard Worker     unsigned OpEnd = UseMI->getNumOperands();
454*9880d681SAndroid Build Coastguard Worker 
455*9880d681SAndroid Build Coastguard Worker     MachineBasicBlock *BB = UseMI->getParent();
456*9880d681SAndroid Build Coastguard Worker     MachineInstrBuilder MIB =
457*9880d681SAndroid Build Coastguard Worker         BuildMI(*BB, InsertPt, UseMI->getDebugLoc(), HII->get(NewOpCode));
458*9880d681SAndroid Build Coastguard Worker     // change mem(Rs + # ) -> mem(Rt << # + ##)
459*9880d681SAndroid Build Coastguard Worker     if (UseMID.mayLoad()) {
460*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(UseMI->getOperand(0));
461*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(AddAslMI->getOperand(2));
462*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(AddAslMI->getOperand(3));
463*9880d681SAndroid Build Coastguard Worker       const GlobalValue *GV = ImmOp.getGlobal();
464*9880d681SAndroid Build Coastguard Worker       MIB.addGlobalAddress(GV, UseMI->getOperand(2).getImm(),
465*9880d681SAndroid Build Coastguard Worker                            ImmOp.getTargetFlags());
466*9880d681SAndroid Build Coastguard Worker       OpStart = 3;
467*9880d681SAndroid Build Coastguard Worker     } else if (UseMID.mayStore()) {
468*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(AddAslMI->getOperand(2));
469*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(AddAslMI->getOperand(3));
470*9880d681SAndroid Build Coastguard Worker       const GlobalValue *GV = ImmOp.getGlobal();
471*9880d681SAndroid Build Coastguard Worker       MIB.addGlobalAddress(GV, UseMI->getOperand(1).getImm(),
472*9880d681SAndroid Build Coastguard Worker                            ImmOp.getTargetFlags());
473*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(UseMI->getOperand(2));
474*9880d681SAndroid Build Coastguard Worker       OpStart = 3;
475*9880d681SAndroid Build Coastguard Worker     } else
476*9880d681SAndroid Build Coastguard Worker       llvm_unreachable("Unhandled instruction");
477*9880d681SAndroid Build Coastguard Worker 
478*9880d681SAndroid Build Coastguard Worker     for (unsigned i = OpStart; i < OpEnd; ++i)
479*9880d681SAndroid Build Coastguard Worker       MIB.addOperand(UseMI->getOperand(i));
480*9880d681SAndroid Build Coastguard Worker 
481*9880d681SAndroid Build Coastguard Worker     Deleted.insert(UseMI);
482*9880d681SAndroid Build Coastguard Worker   }
483*9880d681SAndroid Build Coastguard Worker 
484*9880d681SAndroid Build Coastguard Worker   return true;
485*9880d681SAndroid Build Coastguard Worker }
486*9880d681SAndroid Build Coastguard Worker 
xformUseMI(MachineInstr * TfrMI,MachineInstr * UseMI,NodeAddr<UseNode * > UseN,unsigned UseMOnum)487*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::xformUseMI(MachineInstr *TfrMI, MachineInstr *UseMI,
488*9880d681SAndroid Build Coastguard Worker                                     NodeAddr<UseNode *> UseN,
489*9880d681SAndroid Build Coastguard Worker                                     unsigned UseMOnum) {
490*9880d681SAndroid Build Coastguard Worker   const MachineOperand ImmOp = TfrMI->getOperand(1);
491*9880d681SAndroid Build Coastguard Worker   const MCInstrDesc &MID = UseMI->getDesc();
492*9880d681SAndroid Build Coastguard Worker   unsigned Changed = false;
493*9880d681SAndroid Build Coastguard Worker   if (MID.mayLoad())
494*9880d681SAndroid Build Coastguard Worker     Changed = changeLoad(UseMI, ImmOp, UseMOnum);
495*9880d681SAndroid Build Coastguard Worker   else if (MID.mayStore())
496*9880d681SAndroid Build Coastguard Worker     Changed = changeStore(UseMI, ImmOp, UseMOnum);
497*9880d681SAndroid Build Coastguard Worker   else if (UseMI->getOpcode() == Hexagon::S2_addasl_rrri)
498*9880d681SAndroid Build Coastguard Worker     Changed = changeAddAsl(UseN, UseMI, ImmOp, UseMOnum);
499*9880d681SAndroid Build Coastguard Worker 
500*9880d681SAndroid Build Coastguard Worker   if (Changed)
501*9880d681SAndroid Build Coastguard Worker     Deleted.insert(UseMI);
502*9880d681SAndroid Build Coastguard Worker 
503*9880d681SAndroid Build Coastguard Worker   return Changed;
504*9880d681SAndroid Build Coastguard Worker }
505*9880d681SAndroid Build Coastguard Worker 
processBlock(NodeAddr<BlockNode * > BA)506*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::processBlock(NodeAddr<BlockNode *> BA) {
507*9880d681SAndroid Build Coastguard Worker   bool Changed = false;
508*9880d681SAndroid Build Coastguard Worker 
509*9880d681SAndroid Build Coastguard Worker   for (auto IA : BA.Addr->members(*DFG)) {
510*9880d681SAndroid Build Coastguard Worker     if (!DFG->IsCode<NodeAttrs::Stmt>(IA))
511*9880d681SAndroid Build Coastguard Worker       continue;
512*9880d681SAndroid Build Coastguard Worker 
513*9880d681SAndroid Build Coastguard Worker     NodeAddr<StmtNode *> SA = IA;
514*9880d681SAndroid Build Coastguard Worker     MachineInstr *MI = SA.Addr->getCode();
515*9880d681SAndroid Build Coastguard Worker     if (MI->getOpcode() != Hexagon::A2_tfrsi ||
516*9880d681SAndroid Build Coastguard Worker         !MI->getOperand(1).isGlobal())
517*9880d681SAndroid Build Coastguard Worker       continue;
518*9880d681SAndroid Build Coastguard Worker 
519*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "[Analyzing A2_tfrsi]: " << *MI << "\n");
520*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "\t[InstrNode]: " << Print<NodeAddr<InstrNode *>>(IA, *DFG)
521*9880d681SAndroid Build Coastguard Worker                  << "\n");
522*9880d681SAndroid Build Coastguard Worker 
523*9880d681SAndroid Build Coastguard Worker     NodeList UNodeList;
524*9880d681SAndroid Build Coastguard Worker     getAllRealUses(SA, UNodeList);
525*9880d681SAndroid Build Coastguard Worker 
526*9880d681SAndroid Build Coastguard Worker     if (!allValidCandidates(SA, UNodeList))
527*9880d681SAndroid Build Coastguard Worker       continue;
528*9880d681SAndroid Build Coastguard Worker 
529*9880d681SAndroid Build Coastguard Worker     short SizeInc = 0;
530*9880d681SAndroid Build Coastguard Worker     unsigned DefR = MI->getOperand(0).getReg();
531*9880d681SAndroid Build Coastguard Worker     InstrEvalMap InstrEvalResult;
532*9880d681SAndroid Build Coastguard Worker 
533*9880d681SAndroid Build Coastguard Worker     // Analyze all uses and calculate increase in size. Perform the optimization
534*9880d681SAndroid Build Coastguard Worker     // only if there is no increase in size.
535*9880d681SAndroid Build Coastguard Worker     if (!analyzeUses(DefR, UNodeList, InstrEvalResult, SizeInc))
536*9880d681SAndroid Build Coastguard Worker       continue;
537*9880d681SAndroid Build Coastguard Worker     if (SizeInc > CodeGrowthLimit)
538*9880d681SAndroid Build Coastguard Worker       continue;
539*9880d681SAndroid Build Coastguard Worker 
540*9880d681SAndroid Build Coastguard Worker     bool KeepTfr = false;
541*9880d681SAndroid Build Coastguard Worker 
542*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "\t[Total reached uses] : " << UNodeList.size() << "\n");
543*9880d681SAndroid Build Coastguard Worker     DEBUG(dbgs() << "\t[Processing Reached Uses] ===\n");
544*9880d681SAndroid Build Coastguard Worker     for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
545*9880d681SAndroid Build Coastguard Worker       NodeAddr<UseNode *> UseN = *I;
546*9880d681SAndroid Build Coastguard Worker       assert(!(UseN.Addr->getFlags() & NodeAttrs::PhiRef) &&
547*9880d681SAndroid Build Coastguard Worker              "Found a PhiRef node as a real reached use!!");
548*9880d681SAndroid Build Coastguard Worker 
549*9880d681SAndroid Build Coastguard Worker       NodeAddr<StmtNode *> OwnerN = UseN.Addr->getOwner(*DFG);
550*9880d681SAndroid Build Coastguard Worker       MachineInstr *UseMI = OwnerN.Addr->getCode();
551*9880d681SAndroid Build Coastguard Worker       DEBUG(dbgs() << "\t\t[MI <BB#" << UseMI->getParent()->getNumber()
552*9880d681SAndroid Build Coastguard Worker                    << ">]: " << *UseMI << "\n");
553*9880d681SAndroid Build Coastguard Worker 
554*9880d681SAndroid Build Coastguard Worker       int UseMOnum = -1;
555*9880d681SAndroid Build Coastguard Worker       unsigned NumOperands = UseMI->getNumOperands();
556*9880d681SAndroid Build Coastguard Worker       for (unsigned j = 0; j < NumOperands - 1; ++j) {
557*9880d681SAndroid Build Coastguard Worker         const MachineOperand &op = UseMI->getOperand(j);
558*9880d681SAndroid Build Coastguard Worker         if (op.isReg() && op.isUse() && DefR == op.getReg())
559*9880d681SAndroid Build Coastguard Worker           UseMOnum = j;
560*9880d681SAndroid Build Coastguard Worker       }
561*9880d681SAndroid Build Coastguard Worker       assert(UseMOnum >= 0 && "Invalid reached use!");
562*9880d681SAndroid Build Coastguard Worker 
563*9880d681SAndroid Build Coastguard Worker       if (InstrEvalResult[UseMI])
564*9880d681SAndroid Build Coastguard Worker         // Change UseMI if replacement is possible.
565*9880d681SAndroid Build Coastguard Worker         Changed |= xformUseMI(MI, UseMI, UseN, UseMOnum);
566*9880d681SAndroid Build Coastguard Worker       else
567*9880d681SAndroid Build Coastguard Worker         KeepTfr = true;
568*9880d681SAndroid Build Coastguard Worker     }
569*9880d681SAndroid Build Coastguard Worker     if (!KeepTfr)
570*9880d681SAndroid Build Coastguard Worker       Deleted.insert(MI);
571*9880d681SAndroid Build Coastguard Worker   }
572*9880d681SAndroid Build Coastguard Worker   return Changed;
573*9880d681SAndroid Build Coastguard Worker }
574*9880d681SAndroid Build Coastguard Worker 
updateMap(NodeAddr<InstrNode * > IA)575*9880d681SAndroid Build Coastguard Worker void HexagonOptAddrMode::updateMap(NodeAddr<InstrNode *> IA) {
576*9880d681SAndroid Build Coastguard Worker   RegisterSet RRs;
577*9880d681SAndroid Build Coastguard Worker   for (NodeAddr<RefNode *> RA : IA.Addr->members(*DFG))
578*9880d681SAndroid Build Coastguard Worker     RRs.insert(RA.Addr->getRegRef());
579*9880d681SAndroid Build Coastguard Worker   bool Common = false;
580*9880d681SAndroid Build Coastguard Worker   for (auto &R : RDefMap) {
581*9880d681SAndroid Build Coastguard Worker     if (!RRs.count(R.first))
582*9880d681SAndroid Build Coastguard Worker       continue;
583*9880d681SAndroid Build Coastguard Worker     Common = true;
584*9880d681SAndroid Build Coastguard Worker     break;
585*9880d681SAndroid Build Coastguard Worker   }
586*9880d681SAndroid Build Coastguard Worker   if (!Common)
587*9880d681SAndroid Build Coastguard Worker     return;
588*9880d681SAndroid Build Coastguard Worker 
589*9880d681SAndroid Build Coastguard Worker   for (auto &R : RDefMap) {
590*9880d681SAndroid Build Coastguard Worker     auto F = DefM.find(R.first);
591*9880d681SAndroid Build Coastguard Worker     if (F == DefM.end() || F->second.empty())
592*9880d681SAndroid Build Coastguard Worker       continue;
593*9880d681SAndroid Build Coastguard Worker     R.second[IA.Id] = F->second.top()->Id;
594*9880d681SAndroid Build Coastguard Worker   }
595*9880d681SAndroid Build Coastguard Worker }
596*9880d681SAndroid Build Coastguard Worker 
constructDefMap(MachineBasicBlock * B)597*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::constructDefMap(MachineBasicBlock *B) {
598*9880d681SAndroid Build Coastguard Worker   bool Changed = false;
599*9880d681SAndroid Build Coastguard Worker   auto BA = DFG->getFunc().Addr->findBlock(B, *DFG);
600*9880d681SAndroid Build Coastguard Worker   DFG->markBlock(BA.Id, DefM);
601*9880d681SAndroid Build Coastguard Worker 
602*9880d681SAndroid Build Coastguard Worker   for (NodeAddr<InstrNode *> IA : BA.Addr->members(*DFG)) {
603*9880d681SAndroid Build Coastguard Worker     updateMap(IA);
604*9880d681SAndroid Build Coastguard Worker     DFG->pushDefs(IA, DefM);
605*9880d681SAndroid Build Coastguard Worker   }
606*9880d681SAndroid Build Coastguard Worker 
607*9880d681SAndroid Build Coastguard Worker   MachineDomTreeNode *N = MDT->getNode(B);
608*9880d681SAndroid Build Coastguard Worker   for (auto I : *N)
609*9880d681SAndroid Build Coastguard Worker     Changed |= constructDefMap(I->getBlock());
610*9880d681SAndroid Build Coastguard Worker 
611*9880d681SAndroid Build Coastguard Worker   DFG->releaseBlock(BA.Id, DefM);
612*9880d681SAndroid Build Coastguard Worker   return Changed;
613*9880d681SAndroid Build Coastguard Worker }
614*9880d681SAndroid Build Coastguard Worker 
runOnMachineFunction(MachineFunction & MF)615*9880d681SAndroid Build Coastguard Worker bool HexagonOptAddrMode::runOnMachineFunction(MachineFunction &MF) {
616*9880d681SAndroid Build Coastguard Worker   bool Changed = false;
617*9880d681SAndroid Build Coastguard Worker   auto &HST = MF.getSubtarget<HexagonSubtarget>();
618*9880d681SAndroid Build Coastguard Worker   auto &MRI = MF.getRegInfo();
619*9880d681SAndroid Build Coastguard Worker   HII = HST.getInstrInfo();
620*9880d681SAndroid Build Coastguard Worker   const auto &MDF = getAnalysis<MachineDominanceFrontier>();
621*9880d681SAndroid Build Coastguard Worker   MDT = &getAnalysis<MachineDominatorTree>();
622*9880d681SAndroid Build Coastguard Worker   const auto &TRI = *MF.getSubtarget().getRegisterInfo();
623*9880d681SAndroid Build Coastguard Worker   const TargetOperandInfo TOI(*HII);
624*9880d681SAndroid Build Coastguard Worker 
625*9880d681SAndroid Build Coastguard Worker   RegisterAliasInfo RAI(TRI);
626*9880d681SAndroid Build Coastguard Worker   DataFlowGraph G(MF, *HII, TRI, *MDT, MDF, RAI, TOI);
627*9880d681SAndroid Build Coastguard Worker   G.build();
628*9880d681SAndroid Build Coastguard Worker   DFG = &G;
629*9880d681SAndroid Build Coastguard Worker 
630*9880d681SAndroid Build Coastguard Worker   Liveness L(MRI, *DFG);
631*9880d681SAndroid Build Coastguard Worker   L.computePhiInfo();
632*9880d681SAndroid Build Coastguard Worker   LV = &L;
633*9880d681SAndroid Build Coastguard Worker 
634*9880d681SAndroid Build Coastguard Worker   constructDefMap(&DFG->getMF().front());
635*9880d681SAndroid Build Coastguard Worker 
636*9880d681SAndroid Build Coastguard Worker   Deleted.clear();
637*9880d681SAndroid Build Coastguard Worker   NodeAddr<FuncNode *> FA = DFG->getFunc();
638*9880d681SAndroid Build Coastguard Worker   DEBUG(dbgs() << "==== [RefMap#]=====:\n "
639*9880d681SAndroid Build Coastguard Worker                << Print<NodeAddr<FuncNode *>>(FA, *DFG) << "\n");
640*9880d681SAndroid Build Coastguard Worker 
641*9880d681SAndroid Build Coastguard Worker   for (NodeAddr<BlockNode *> BA : FA.Addr->members(*DFG))
642*9880d681SAndroid Build Coastguard Worker     Changed |= processBlock(BA);
643*9880d681SAndroid Build Coastguard Worker 
644*9880d681SAndroid Build Coastguard Worker   for (auto MI : Deleted)
645*9880d681SAndroid Build Coastguard Worker     MI->eraseFromParent();
646*9880d681SAndroid Build Coastguard Worker 
647*9880d681SAndroid Build Coastguard Worker   if (Changed) {
648*9880d681SAndroid Build Coastguard Worker     G.build();
649*9880d681SAndroid Build Coastguard Worker     L.computeLiveIns();
650*9880d681SAndroid Build Coastguard Worker     L.resetLiveIns();
651*9880d681SAndroid Build Coastguard Worker     L.resetKills();
652*9880d681SAndroid Build Coastguard Worker   }
653*9880d681SAndroid Build Coastguard Worker 
654*9880d681SAndroid Build Coastguard Worker   return Changed;
655*9880d681SAndroid Build Coastguard Worker }
656*9880d681SAndroid Build Coastguard Worker 
657*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
658*9880d681SAndroid Build Coastguard Worker //                         Public Constructor Functions
659*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
660*9880d681SAndroid Build Coastguard Worker 
createHexagonOptAddrMode()661*9880d681SAndroid Build Coastguard Worker FunctionPass *llvm::createHexagonOptAddrMode() {
662*9880d681SAndroid Build Coastguard Worker   return new HexagonOptAddrMode();
663*9880d681SAndroid Build Coastguard Worker }
664