blob: c7e5e55a6a715bb15628d73b5e4061f6594157c4 [file] [log] [blame]
Eugene Zelenko4d060b72017-07-29 00:56:56 +00001//===- HexagonOptAddrMode.cpp ---------------------------------------------===//
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9// This implements a Hexagon-specific pass to optimize addressing mode for
10// load/store instructions.
11//===----------------------------------------------------------------------===//
12
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000013#include "HexagonInstrInfo.h"
14#include "HexagonSubtarget.h"
15#include "MCTargetDesc/HexagonBaseInfo.h"
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000016#include "RDFGraph.h"
17#include "RDFLiveness.h"
Eugene Zelenko4d060b72017-07-29 00:56:56 +000018#include "RDFRegisters.h"
19#include "llvm/ADT/DenseMap.h"
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000020#include "llvm/ADT/DenseSet.h"
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000021#include "llvm/ADT/StringRef.h"
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000022#include "llvm/CodeGen/MachineBasicBlock.h"
23#include "llvm/CodeGen/MachineDominanceFrontier.h"
24#include "llvm/CodeGen/MachineDominators.h"
25#include "llvm/CodeGen/MachineFunction.h"
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000026#include "llvm/CodeGen/MachineFunctionPass.h"
27#include "llvm/CodeGen/MachineInstr.h"
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000028#include "llvm/CodeGen/MachineInstrBuilder.h"
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000029#include "llvm/CodeGen/MachineOperand.h"
30#include "llvm/MC/MCInstrDesc.h"
31#include "llvm/Pass.h"
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000032#include "llvm/Support/CommandLine.h"
33#include "llvm/Support/Debug.h"
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000034#include "llvm/Support/ErrorHandling.h"
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000035#include "llvm/Support/raw_ostream.h"
Eugene Zelenko4d060b72017-07-29 00:56:56 +000036#include "llvm/Target/TargetSubtargetInfo.h"
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000037#include <cassert>
38#include <cstdint>
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000039
Jakub Kuderski34327d22017-07-13 20:26:45 +000040#define DEBUG_TYPE "opt-addr-mode"
41
Eugene Zelenko4d060b72017-07-29 00:56:56 +000042using namespace llvm;
43using namespace rdf;
44
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000045static cl::opt<int> CodeGrowthLimit("hexagon-amode-growth-limit",
46 cl::Hidden, cl::init(0), cl::desc("Code growth limit for address mode "
47 "optimization"));
48
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000049namespace llvm {
Eugene Zelenko4d060b72017-07-29 00:56:56 +000050
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000051 FunctionPass *createHexagonOptAddrMode();
Krzysztof Parzyszek333b2bf2017-04-19 15:15:51 +000052 void initializeHexagonOptAddrModePass(PassRegistry&);
Eugene Zelenko4d060b72017-07-29 00:56:56 +000053
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000054} // end namespace llvm
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000055
56namespace {
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000057
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000058class HexagonOptAddrMode : public MachineFunctionPass {
59public:
60 static char ID;
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000061
Eugene Zelenko4d060b72017-07-29 00:56:56 +000062 HexagonOptAddrMode() : MachineFunctionPass(ID) {}
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000063
Mehdi Amini117296c2016-10-01 02:56:57 +000064 StringRef getPassName() const override {
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000065 return "Optimize addressing mode of load/store";
66 }
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000067
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000068 void getAnalysisUsage(AnalysisUsage &AU) const override {
69 MachineFunctionPass::getAnalysisUsage(AU);
70 AU.addRequired<MachineDominatorTree>();
71 AU.addRequired<MachineDominanceFrontier>();
72 AU.setPreservesAll();
73 }
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +000074
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000075 bool runOnMachineFunction(MachineFunction &MF) override;
76
77private:
Eugene Zelenko4d060b72017-07-29 00:56:56 +000078 using MISetType = DenseSet<MachineInstr *>;
79 using InstrEvalMap = DenseMap<MachineInstr *, bool>;
80
81 const HexagonInstrInfo *HII = nullptr;
82 MachineDominatorTree *MDT = nullptr;
83 DataFlowGraph *DFG = nullptr;
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000084 DataFlowGraph::DefStackMap DefM;
Eugene Zelenko4d060b72017-07-29 00:56:56 +000085 Liveness *LV = nullptr;
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000086 MISetType Deleted;
87
88 bool processBlock(NodeAddr<BlockNode *> BA);
89 bool xformUseMI(MachineInstr *TfrMI, MachineInstr *UseMI,
90 NodeAddr<UseNode *> UseN, unsigned UseMOnum);
91 bool analyzeUses(unsigned DefR, const NodeList &UNodeList,
92 InstrEvalMap &InstrEvalResult, short &SizeInc);
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +000093 bool hasRepForm(MachineInstr &MI, unsigned TfrDefR);
94 bool canRemoveAddasl(NodeAddr<StmtNode *> AddAslSN, MachineInstr &MI,
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000095 const NodeList &UNodeList);
96 void getAllRealUses(NodeAddr<StmtNode *> SN, NodeList &UNodeList);
97 bool allValidCandidates(NodeAddr<StmtNode *> SA, NodeList &UNodeList);
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +000098 short getBaseWithLongOffset(const MachineInstr &MI) const;
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +000099 bool changeStore(MachineInstr *OldMI, MachineOperand ImmOp,
100 unsigned ImmOpNum);
101 bool changeLoad(MachineInstr *OldMI, MachineOperand ImmOp, unsigned ImmOpNum);
102 bool changeAddAsl(NodeAddr<UseNode *> AddAslUN, MachineInstr *AddAslMI,
103 const MachineOperand &ImmOp, unsigned ImmOpNum);
104};
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +0000105
106} // end anonymous namespace
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000107
108char HexagonOptAddrMode::ID = 0;
109
Krzysztof Parzyszek333b2bf2017-04-19 15:15:51 +0000110INITIALIZE_PASS_BEGIN(HexagonOptAddrMode, "amode-opt",
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000111 "Optimize addressing mode", false, false)
112INITIALIZE_PASS_DEPENDENCY(MachineDominatorTree)
113INITIALIZE_PASS_DEPENDENCY(MachineDominanceFrontier)
Krzysztof Parzyszek333b2bf2017-04-19 15:15:51 +0000114INITIALIZE_PASS_END(HexagonOptAddrMode, "amode-opt", "Optimize addressing mode",
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000115 false, false)
116
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000117bool HexagonOptAddrMode::hasRepForm(MachineInstr &MI, unsigned TfrDefR) {
118 const MCInstrDesc &MID = MI.getDesc();
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000119
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000120 if ((!MID.mayStore() && !MID.mayLoad()) || HII->isPredicated(MI))
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000121 return false;
122
123 if (MID.mayStore()) {
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000124 MachineOperand StOp = MI.getOperand(MI.getNumOperands() - 1);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000125 if (StOp.isReg() && StOp.getReg() == TfrDefR)
126 return false;
127 }
128
129 if (HII->getAddrMode(MI) == HexagonII::BaseRegOffset)
130 // Tranform to Absolute plus register offset.
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000131 return (HII->changeAddrMode_rr_ur(MI) >= 0);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000132 else if (HII->getAddrMode(MI) == HexagonII::BaseImmOffset)
133 // Tranform to absolute addressing mode.
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000134 return (HII->changeAddrMode_io_abs(MI) >= 0);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000135
136 return false;
137}
138
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000139// Check if addasl instruction can be removed. This is possible only
140// if it's feeding to only load/store instructions with base + register
141// offset as these instruction can be tranformed to use 'absolute plus
142// shifted register offset'.
143// ex:
144// Rs = ##foo
145// Rx = addasl(Rs, Rt, #2)
146// Rd = memw(Rx + #28)
147// Above three instructions can be replaced with Rd = memw(Rt<<#2 + ##foo+28)
148
149bool HexagonOptAddrMode::canRemoveAddasl(NodeAddr<StmtNode *> AddAslSN,
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000150 MachineInstr &MI,
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000151 const NodeList &UNodeList) {
152 // check offset size in addasl. if 'offset > 3' return false
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000153 const MachineOperand &OffsetOp = MI.getOperand(3);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000154 if (!OffsetOp.isImm() || OffsetOp.getImm() > 3)
155 return false;
156
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000157 unsigned OffsetReg = MI.getOperand(2).getReg();
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000158 RegisterRef OffsetRR;
159 NodeId OffsetRegRD = 0;
160 for (NodeAddr<UseNode *> UA : AddAslSN.Addr->members_if(DFG->IsUse, *DFG)) {
Krzysztof Parzyszek445bd122016-10-14 17:57:55 +0000161 RegisterRef RR = UA.Addr->getRegRef(*DFG);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000162 if (OffsetReg == RR.Reg) {
163 OffsetRR = RR;
164 OffsetRegRD = UA.Addr->getReachingDef();
165 }
166 }
167
168 for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
169 NodeAddr<UseNode *> UA = *I;
170 NodeAddr<InstrNode *> IA = UA.Addr->getOwner(*DFG);
Krzysztof Parzyszek634f57e2017-04-19 15:14:30 +0000171 if (UA.Addr->getFlags() & NodeAttrs::PhiRef)
172 return false;
173 NodeAddr<RefNode*> AA = LV->getNearestAliasedRef(OffsetRR, IA);
174 if ((DFG->IsDef(AA) && AA.Id != OffsetRegRD) ||
175 AA.Addr->getReachingDef() != OffsetRegRD)
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000176 return false;
177
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000178 MachineInstr &UseMI = *NodeAddr<StmtNode *>(IA).Addr->getCode();
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000179 NodeAddr<DefNode *> OffsetRegDN = DFG->addr<DefNode *>(OffsetRegRD);
180 // Reaching Def to an offset register can't be a phi.
181 if ((OffsetRegDN.Addr->getFlags() & NodeAttrs::PhiRef) &&
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000182 MI.getParent() != UseMI.getParent())
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000183 return false;
184
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000185 const MCInstrDesc &UseMID = UseMI.getDesc();
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000186 if ((!UseMID.mayLoad() && !UseMID.mayStore()) ||
187 HII->getAddrMode(UseMI) != HexagonII::BaseImmOffset ||
188 getBaseWithLongOffset(UseMI) < 0)
189 return false;
190
191 // Addasl output can't be a store value.
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000192 if (UseMID.mayStore() && UseMI.getOperand(2).isReg() &&
193 UseMI.getOperand(2).getReg() == MI.getOperand(0).getReg())
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000194 return false;
195
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000196 for (auto &Mo : UseMI.operands())
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000197 if (Mo.isFI())
198 return false;
199 }
200 return true;
201}
202
203bool HexagonOptAddrMode::allValidCandidates(NodeAddr<StmtNode *> SA,
204 NodeList &UNodeList) {
205 for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
206 NodeAddr<UseNode *> UN = *I;
Krzysztof Parzyszek445bd122016-10-14 17:57:55 +0000207 RegisterRef UR = UN.Addr->getRegRef(*DFG);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000208 NodeSet Visited, Defs;
Krzysztof Parzyszekebabd992017-03-01 19:30:42 +0000209 const auto &P = LV->getAllReachingDefsRec(UR, UN, Visited, Defs);
210 if (!P.second) {
211 DEBUG({
212 dbgs() << "*** Unable to collect all reaching defs for use ***\n"
213 << PrintNode<UseNode*>(UN, *DFG) << '\n'
214 << "The program's complexity may exceed the limits.\n";
215 });
216 return false;
217 }
218 const auto &ReachingDefs = P.first;
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000219 if (ReachingDefs.size() > 1) {
Krzysztof Parzyszek4a3e2852016-05-23 17:31:30 +0000220 DEBUG({
221 dbgs() << "*** Multiple Reaching Defs found!!! ***\n";
222 for (auto DI : ReachingDefs) {
223 NodeAddr<UseNode *> DA = DFG->addr<UseNode *>(DI);
224 NodeAddr<StmtNode *> TempIA = DA.Addr->getOwner(*DFG);
225 dbgs() << "\t\t[Reaching Def]: "
226 << Print<NodeAddr<InstrNode *>>(TempIA, *DFG) << "\n";
227 }
228 });
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000229 return false;
230 }
231 }
232 return true;
233}
234
235void HexagonOptAddrMode::getAllRealUses(NodeAddr<StmtNode *> SA,
236 NodeList &UNodeList) {
237 for (NodeAddr<DefNode *> DA : SA.Addr->members_if(DFG->IsDef, *DFG)) {
238 DEBUG(dbgs() << "\t\t[DefNode]: " << Print<NodeAddr<DefNode *>>(DA, *DFG)
239 << "\n");
Krzysztof Parzyszek5226ba82017-02-16 18:45:23 +0000240 RegisterRef DR = DFG->getPRI().normalize(DA.Addr->getRegRef(*DFG));
Krzysztof Parzyszek7bb63ac2016-10-19 16:30:56 +0000241
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000242 auto UseSet = LV->getAllReachedUses(DR, DA);
243
244 for (auto UI : UseSet) {
245 NodeAddr<UseNode *> UA = DFG->addr<UseNode *>(UI);
Krzysztof Parzyszek4a3e2852016-05-23 17:31:30 +0000246 DEBUG({
247 NodeAddr<StmtNode *> TempIA = UA.Addr->getOwner(*DFG);
248 dbgs() << "\t\t\t[Reached Use]: "
249 << Print<NodeAddr<InstrNode *>>(TempIA, *DFG) << "\n";
250 });
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000251
252 if (UA.Addr->getFlags() & NodeAttrs::PhiRef) {
253 NodeAddr<PhiNode *> PA = UA.Addr->getOwner(*DFG);
254 NodeId id = PA.Id;
255 const Liveness::RefMap &phiUse = LV->getRealUses(id);
256 DEBUG(dbgs() << "\t\t\t\tphi real Uses"
257 << Print<Liveness::RefMap>(phiUse, *DFG) << "\n");
Eugene Zelenko26e8c7d2016-12-16 01:00:40 +0000258 if (!phiUse.empty()) {
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000259 for (auto I : phiUse) {
Krzysztof Parzyszek74b1f252017-04-14 17:25:13 +0000260 if (!DFG->getPRI().alias(RegisterRef(I.first), DR))
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000261 continue;
262 auto phiUseSet = I.second;
263 for (auto phiUI : phiUseSet) {
Krzysztof Parzyszek7bb63ac2016-10-19 16:30:56 +0000264 NodeAddr<UseNode *> phiUA = DFG->addr<UseNode *>(phiUI.first);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000265 UNodeList.push_back(phiUA);
266 }
267 }
268 }
269 } else
270 UNodeList.push_back(UA);
271 }
272 }
273}
274
275bool HexagonOptAddrMode::analyzeUses(unsigned tfrDefR,
276 const NodeList &UNodeList,
277 InstrEvalMap &InstrEvalResult,
278 short &SizeInc) {
279 bool KeepTfr = false;
280 bool HasRepInstr = false;
281 InstrEvalResult.clear();
282
283 for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
284 bool CanBeReplaced = false;
285 NodeAddr<UseNode *> UN = *I;
286 NodeAddr<StmtNode *> SN = UN.Addr->getOwner(*DFG);
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000287 MachineInstr &MI = *SN.Addr->getCode();
288 const MCInstrDesc &MID = MI.getDesc();
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000289 if ((MID.mayLoad() || MID.mayStore())) {
290 if (!hasRepForm(MI, tfrDefR)) {
291 KeepTfr = true;
292 continue;
293 }
294 SizeInc++;
295 CanBeReplaced = true;
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000296 } else if (MI.getOpcode() == Hexagon::S2_addasl_rrri) {
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000297 NodeList AddaslUseList;
298
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000299 DEBUG(dbgs() << "\nGetting ReachedUses for === " << MI << "\n");
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000300 getAllRealUses(SN, AddaslUseList);
301 // Process phi nodes.
302 if (allValidCandidates(SN, AddaslUseList) &&
303 canRemoveAddasl(SN, MI, AddaslUseList)) {
304 SizeInc += AddaslUseList.size();
305 SizeInc -= 1; // Reduce size by 1 as addasl itself can be removed.
306 CanBeReplaced = true;
307 } else
308 SizeInc++;
309 } else
310 // Currently, only load/store and addasl are handled.
311 // Some other instructions to consider -
312 // A2_add -> A2_addi
313 // M4_mpyrr_addr -> M4_mpyrr_addi
314 KeepTfr = true;
315
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000316 InstrEvalResult[&MI] = CanBeReplaced;
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000317 HasRepInstr |= CanBeReplaced;
318 }
319
320 // Reduce total size by 2 if original tfr can be deleted.
321 if (!KeepTfr)
322 SizeInc -= 2;
323
324 return HasRepInstr;
325}
326
327bool HexagonOptAddrMode::changeLoad(MachineInstr *OldMI, MachineOperand ImmOp,
328 unsigned ImmOpNum) {
329 bool Changed = false;
330 MachineBasicBlock *BB = OldMI->getParent();
331 auto UsePos = MachineBasicBlock::iterator(OldMI);
332 MachineBasicBlock::instr_iterator InsertPt = UsePos.getInstrIterator();
333 ++InsertPt;
334 unsigned OpStart;
335 unsigned OpEnd = OldMI->getNumOperands();
336 MachineInstrBuilder MIB;
337
338 if (ImmOpNum == 1) {
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000339 if (HII->getAddrMode(*OldMI) == HexagonII::BaseRegOffset) {
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000340 short NewOpCode = HII->changeAddrMode_rr_ur(*OldMI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000341 assert(NewOpCode >= 0 && "Invalid New opcode\n");
342 MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
Diana Picus116bbab2017-01-13 09:58:52 +0000343 MIB.add(OldMI->getOperand(0));
344 MIB.add(OldMI->getOperand(2));
345 MIB.add(OldMI->getOperand(3));
346 MIB.add(ImmOp);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000347 OpStart = 4;
348 Changed = true;
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000349 } else if (HII->getAddrMode(*OldMI) == HexagonII::BaseImmOffset) {
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000350 short NewOpCode = HII->changeAddrMode_io_abs(*OldMI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000351 assert(NewOpCode >= 0 && "Invalid New opcode\n");
352 MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode))
Diana Picus116bbab2017-01-13 09:58:52 +0000353 .add(OldMI->getOperand(0));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000354 const GlobalValue *GV = ImmOp.getGlobal();
355 int64_t Offset = ImmOp.getOffset() + OldMI->getOperand(2).getImm();
356
357 MIB.addGlobalAddress(GV, Offset, ImmOp.getTargetFlags());
358 OpStart = 3;
359 Changed = true;
360 } else
361 Changed = false;
362
363 DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
Krzysztof Parzyszeke4d0e192017-10-19 16:59:22 +0000364 DEBUG(dbgs() << "[TO]: " << *MIB << "\n");
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000365 } else if (ImmOpNum == 2 && OldMI->getOperand(3).getImm() == 0) {
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000366 short NewOpCode = HII->changeAddrMode_rr_io(*OldMI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000367 assert(NewOpCode >= 0 && "Invalid New opcode\n");
368 MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
Diana Picus116bbab2017-01-13 09:58:52 +0000369 MIB.add(OldMI->getOperand(0));
370 MIB.add(OldMI->getOperand(1));
371 MIB.add(ImmOp);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000372 OpStart = 4;
373 Changed = true;
374 DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
Krzysztof Parzyszeke4d0e192017-10-19 16:59:22 +0000375 DEBUG(dbgs() << "[TO]: " << *MIB << "\n");
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000376 }
377
378 if (Changed)
379 for (unsigned i = OpStart; i < OpEnd; ++i)
Diana Picus116bbab2017-01-13 09:58:52 +0000380 MIB.add(OldMI->getOperand(i));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000381
382 return Changed;
383}
384
385bool HexagonOptAddrMode::changeStore(MachineInstr *OldMI, MachineOperand ImmOp,
386 unsigned ImmOpNum) {
387 bool Changed = false;
388 unsigned OpStart;
389 unsigned OpEnd = OldMI->getNumOperands();
390 MachineBasicBlock *BB = OldMI->getParent();
391 auto UsePos = MachineBasicBlock::iterator(OldMI);
392 MachineBasicBlock::instr_iterator InsertPt = UsePos.getInstrIterator();
393 ++InsertPt;
394 MachineInstrBuilder MIB;
395 if (ImmOpNum == 0) {
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000396 if (HII->getAddrMode(*OldMI) == HexagonII::BaseRegOffset) {
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000397 short NewOpCode = HII->changeAddrMode_rr_ur(*OldMI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000398 assert(NewOpCode >= 0 && "Invalid New opcode\n");
399 MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
Diana Picus116bbab2017-01-13 09:58:52 +0000400 MIB.add(OldMI->getOperand(1));
401 MIB.add(OldMI->getOperand(2));
402 MIB.add(ImmOp);
403 MIB.add(OldMI->getOperand(3));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000404 OpStart = 4;
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000405 } else if (HII->getAddrMode(*OldMI) == HexagonII::BaseImmOffset) {
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000406 short NewOpCode = HII->changeAddrMode_io_abs(*OldMI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000407 assert(NewOpCode >= 0 && "Invalid New opcode\n");
408 MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
409 const GlobalValue *GV = ImmOp.getGlobal();
410 int64_t Offset = ImmOp.getOffset() + OldMI->getOperand(1).getImm();
411 MIB.addGlobalAddress(GV, Offset, ImmOp.getTargetFlags());
Diana Picus116bbab2017-01-13 09:58:52 +0000412 MIB.add(OldMI->getOperand(2));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000413 OpStart = 3;
414 }
415 Changed = true;
416 DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
Krzysztof Parzyszeke4d0e192017-10-19 16:59:22 +0000417 DEBUG(dbgs() << "[TO]: " << *MIB << "\n");
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000418 } else if (ImmOpNum == 1 && OldMI->getOperand(2).getImm() == 0) {
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000419 short NewOpCode = HII->changeAddrMode_rr_io(*OldMI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000420 assert(NewOpCode >= 0 && "Invalid New opcode\n");
421 MIB = BuildMI(*BB, InsertPt, OldMI->getDebugLoc(), HII->get(NewOpCode));
Diana Picus116bbab2017-01-13 09:58:52 +0000422 MIB.add(OldMI->getOperand(0));
423 MIB.add(ImmOp);
Krzysztof Parzyszeke4d0e192017-10-19 16:59:22 +0000424 OpStart = 3;
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000425 Changed = true;
426 DEBUG(dbgs() << "[Changing]: " << *OldMI << "\n");
Krzysztof Parzyszeke4d0e192017-10-19 16:59:22 +0000427 DEBUG(dbgs() << "[TO]: " << *MIB << "\n");
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000428 }
429 if (Changed)
430 for (unsigned i = OpStart; i < OpEnd; ++i)
Diana Picus116bbab2017-01-13 09:58:52 +0000431 MIB.add(OldMI->getOperand(i));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000432
433 return Changed;
434}
435
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000436short HexagonOptAddrMode::getBaseWithLongOffset(const MachineInstr &MI) const {
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000437 if (HII->getAddrMode(MI) == HexagonII::BaseImmOffset) {
Krzysztof Parzyszek7ae3ae92017-10-05 20:01:38 +0000438 short TempOpCode = HII->changeAddrMode_io_rr(MI);
439 return HII->changeAddrMode_rr_ur(TempOpCode);
440 }
441 return HII->changeAddrMode_rr_ur(MI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000442}
443
444bool HexagonOptAddrMode::changeAddAsl(NodeAddr<UseNode *> AddAslUN,
445 MachineInstr *AddAslMI,
446 const MachineOperand &ImmOp,
447 unsigned ImmOpNum) {
448 NodeAddr<StmtNode *> SA = AddAslUN.Addr->getOwner(*DFG);
449
450 DEBUG(dbgs() << "Processing addasl :" << *AddAslMI << "\n");
451
452 NodeList UNodeList;
453 getAllRealUses(SA, UNodeList);
454
455 for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
456 NodeAddr<UseNode *> UseUN = *I;
457 assert(!(UseUN.Addr->getFlags() & NodeAttrs::PhiRef) &&
458 "Can't transform this 'AddAsl' instruction!");
459
460 NodeAddr<StmtNode *> UseIA = UseUN.Addr->getOwner(*DFG);
461 DEBUG(dbgs() << "[InstrNode]: " << Print<NodeAddr<InstrNode *>>(UseIA, *DFG)
462 << "\n");
463 MachineInstr *UseMI = UseIA.Addr->getCode();
464 DEBUG(dbgs() << "[MI <BB#" << UseMI->getParent()->getNumber()
465 << ">]: " << *UseMI << "\n");
466 const MCInstrDesc &UseMID = UseMI->getDesc();
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000467 assert(HII->getAddrMode(*UseMI) == HexagonII::BaseImmOffset);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000468
469 auto UsePos = MachineBasicBlock::iterator(UseMI);
470 MachineBasicBlock::instr_iterator InsertPt = UsePos.getInstrIterator();
Krzysztof Parzyszekf0b34a52016-07-29 21:49:42 +0000471 short NewOpCode = getBaseWithLongOffset(*UseMI);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000472 assert(NewOpCode >= 0 && "Invalid New opcode\n");
473
474 unsigned OpStart;
475 unsigned OpEnd = UseMI->getNumOperands();
476
477 MachineBasicBlock *BB = UseMI->getParent();
478 MachineInstrBuilder MIB =
479 BuildMI(*BB, InsertPt, UseMI->getDebugLoc(), HII->get(NewOpCode));
480 // change mem(Rs + # ) -> mem(Rt << # + ##)
481 if (UseMID.mayLoad()) {
Diana Picus116bbab2017-01-13 09:58:52 +0000482 MIB.add(UseMI->getOperand(0));
483 MIB.add(AddAslMI->getOperand(2));
484 MIB.add(AddAslMI->getOperand(3));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000485 const GlobalValue *GV = ImmOp.getGlobal();
Krzysztof Parzyszek333b2bf2017-04-19 15:15:51 +0000486 MIB.addGlobalAddress(GV, UseMI->getOperand(2).getImm()+ImmOp.getOffset(),
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000487 ImmOp.getTargetFlags());
488 OpStart = 3;
489 } else if (UseMID.mayStore()) {
Diana Picus116bbab2017-01-13 09:58:52 +0000490 MIB.add(AddAslMI->getOperand(2));
491 MIB.add(AddAslMI->getOperand(3));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000492 const GlobalValue *GV = ImmOp.getGlobal();
Krzysztof Parzyszek333b2bf2017-04-19 15:15:51 +0000493 MIB.addGlobalAddress(GV, UseMI->getOperand(1).getImm()+ImmOp.getOffset(),
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000494 ImmOp.getTargetFlags());
Diana Picus116bbab2017-01-13 09:58:52 +0000495 MIB.add(UseMI->getOperand(2));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000496 OpStart = 3;
497 } else
498 llvm_unreachable("Unhandled instruction");
499
500 for (unsigned i = OpStart; i < OpEnd; ++i)
Diana Picus116bbab2017-01-13 09:58:52 +0000501 MIB.add(UseMI->getOperand(i));
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000502
503 Deleted.insert(UseMI);
504 }
505
506 return true;
507}
508
509bool HexagonOptAddrMode::xformUseMI(MachineInstr *TfrMI, MachineInstr *UseMI,
510 NodeAddr<UseNode *> UseN,
511 unsigned UseMOnum) {
512 const MachineOperand ImmOp = TfrMI->getOperand(1);
513 const MCInstrDesc &MID = UseMI->getDesc();
514 unsigned Changed = false;
515 if (MID.mayLoad())
516 Changed = changeLoad(UseMI, ImmOp, UseMOnum);
517 else if (MID.mayStore())
518 Changed = changeStore(UseMI, ImmOp, UseMOnum);
519 else if (UseMI->getOpcode() == Hexagon::S2_addasl_rrri)
520 Changed = changeAddAsl(UseN, UseMI, ImmOp, UseMOnum);
521
522 if (Changed)
523 Deleted.insert(UseMI);
524
525 return Changed;
526}
527
528bool HexagonOptAddrMode::processBlock(NodeAddr<BlockNode *> BA) {
529 bool Changed = false;
530
531 for (auto IA : BA.Addr->members(*DFG)) {
532 if (!DFG->IsCode<NodeAttrs::Stmt>(IA))
533 continue;
534
535 NodeAddr<StmtNode *> SA = IA;
536 MachineInstr *MI = SA.Addr->getCode();
537 if (MI->getOpcode() != Hexagon::A2_tfrsi ||
538 !MI->getOperand(1).isGlobal())
539 continue;
540
Krzysztof Parzyszek00894192017-06-29 15:55:59 +0000541 DEBUG(dbgs() << "[Analyzing " << HII->getName(MI->getOpcode()) << "]: "
542 << *MI << "\n\t[InstrNode]: "
543 << Print<NodeAddr<InstrNode *>>(IA, *DFG) << '\n');
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000544
545 NodeList UNodeList;
546 getAllRealUses(SA, UNodeList);
547
548 if (!allValidCandidates(SA, UNodeList))
549 continue;
550
551 short SizeInc = 0;
552 unsigned DefR = MI->getOperand(0).getReg();
553 InstrEvalMap InstrEvalResult;
554
555 // Analyze all uses and calculate increase in size. Perform the optimization
556 // only if there is no increase in size.
557 if (!analyzeUses(DefR, UNodeList, InstrEvalResult, SizeInc))
558 continue;
559 if (SizeInc > CodeGrowthLimit)
560 continue;
561
562 bool KeepTfr = false;
563
564 DEBUG(dbgs() << "\t[Total reached uses] : " << UNodeList.size() << "\n");
565 DEBUG(dbgs() << "\t[Processing Reached Uses] ===\n");
566 for (auto I = UNodeList.rbegin(), E = UNodeList.rend(); I != E; ++I) {
567 NodeAddr<UseNode *> UseN = *I;
568 assert(!(UseN.Addr->getFlags() & NodeAttrs::PhiRef) &&
569 "Found a PhiRef node as a real reached use!!");
570
571 NodeAddr<StmtNode *> OwnerN = UseN.Addr->getOwner(*DFG);
572 MachineInstr *UseMI = OwnerN.Addr->getCode();
Krzysztof Parzyszek4a3e2852016-05-23 17:31:30 +0000573 DEBUG(dbgs() << "\t\t[MI <BB#" << UseMI->getParent()->getNumber()
574 << ">]: " << *UseMI << "\n");
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000575
576 int UseMOnum = -1;
577 unsigned NumOperands = UseMI->getNumOperands();
578 for (unsigned j = 0; j < NumOperands - 1; ++j) {
579 const MachineOperand &op = UseMI->getOperand(j);
580 if (op.isReg() && op.isUse() && DefR == op.getReg())
581 UseMOnum = j;
582 }
583 assert(UseMOnum >= 0 && "Invalid reached use!");
584
585 if (InstrEvalResult[UseMI])
586 // Change UseMI if replacement is possible.
587 Changed |= xformUseMI(MI, UseMI, UseN, UseMOnum);
588 else
589 KeepTfr = true;
590 }
591 if (!KeepTfr)
592 Deleted.insert(MI);
593 }
594 return Changed;
595}
596
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000597bool HexagonOptAddrMode::runOnMachineFunction(MachineFunction &MF) {
Krzysztof Parzyszek643aaea2017-04-14 15:26:34 +0000598 if (skipFunction(*MF.getFunction()))
599 return false;
600
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000601 bool Changed = false;
602 auto &HST = MF.getSubtarget<HexagonSubtarget>();
603 auto &MRI = MF.getRegInfo();
604 HII = HST.getInstrInfo();
605 const auto &MDF = getAnalysis<MachineDominanceFrontier>();
606 MDT = &getAnalysis<MachineDominatorTree>();
607 const auto &TRI = *MF.getSubtarget().getRegisterInfo();
608 const TargetOperandInfo TOI(*HII);
609
Krzysztof Parzyszeka77fe4e2016-10-03 17:14:48 +0000610 DataFlowGraph G(MF, *HII, TRI, *MDT, MDF, TOI);
Krzysztof Parzyszek00894192017-06-29 15:55:59 +0000611 // Need to keep dead phis because we can propagate uses of registers into
612 // nodes dominated by those would-be phis.
613 G.build(BuildOptions::KeepDeadPhis);
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000614 DFG = &G;
615
616 Liveness L(MRI, *DFG);
617 L.computePhiInfo();
618 LV = &L;
619
Krzysztof Parzyszekf5cbac92016-04-29 15:49:13 +0000620 Deleted.clear();
621 NodeAddr<FuncNode *> FA = DFG->getFunc();
622 DEBUG(dbgs() << "==== [RefMap#]=====:\n "
623 << Print<NodeAddr<FuncNode *>>(FA, *DFG) << "\n");
624
625 for (NodeAddr<BlockNode *> BA : FA.Addr->members(*DFG))
626 Changed |= processBlock(BA);
627
628 for (auto MI : Deleted)
629 MI->eraseFromParent();
630
631 if (Changed) {
632 G.build();
633 L.computeLiveIns();
634 L.resetLiveIns();
635 L.resetKills();
636 }
637
638 return Changed;
639}
640
641//===----------------------------------------------------------------------===//
642// Public Constructor Functions
643//===----------------------------------------------------------------------===//
644
645FunctionPass *llvm::createHexagonOptAddrMode() {
646 return new HexagonOptAddrMode();
647}