blob: b8c282a4e56d79ceba12aea52da4a42c83f2e2a3 [file] [log] [blame]
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001//===- ConstantHoisting.cpp - Prepare code for expensive constants --------===//
2//
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//
10// This pass identifies expensive constants to hoist and coalesces them to
11// better prepare it for SelectionDAG-based code generation. This works around
12// the limitations of the basic-block-at-a-time approach.
13//
14// First it scans all instructions for integer constants and calculates its
15// cost. If the constant can be folded into the instruction (the cost is
16// TCC_Free) or the cost is just a simple operation (TCC_BASIC), then we don't
17// consider it expensive and leave it alone. This is the default behavior and
18// the default implementation of getIntImmCost will always return TCC_Free.
19//
20// If the cost is more than TCC_BASIC, then the integer constant can't be folded
21// into the instruction and it might be beneficial to hoist the constant.
22// Similar constants are coalesced to reduce register pressure and
23// materialization code.
24//
25// When a constant is hoisted, it is also hidden behind a bitcast to force it to
26// be live-out of the basic block. Otherwise the constant would be just
27// duplicated and each basic block would have its own copy in the SelectionDAG.
28// The SelectionDAG recognizes such constants as opaque and doesn't perform
29// certain transformations on them, which would create a new expensive constant.
30//
31// This optimization is only applied to integer constants in instructions and
Juergen Ributzkaf0dff492014-03-21 06:04:45 +000032// simple (this means not nested) constant cast expressions. For example:
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000033// %0 = load i64* inttoptr (i64 big_constant to i64*)
34//===----------------------------------------------------------------------===//
35
36#define DEBUG_TYPE "consthoist"
37#include "llvm/Transforms/Scalar.h"
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000038#include "llvm/ADT/SmallSet.h"
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +000039#include "llvm/ADT/SmallVector.h"
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000040#include "llvm/ADT/Statistic.h"
41#include "llvm/Analysis/TargetTransformInfo.h"
42#include "llvm/IR/Constants.h"
43#include "llvm/IR/Dominators.h"
44#include "llvm/IR/IntrinsicInst.h"
45#include "llvm/Pass.h"
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000046#include "llvm/Support/Debug.h"
47
48using namespace llvm;
49
50STATISTIC(NumConstantsHoisted, "Number of constants hoisted");
51STATISTIC(NumConstantsRebased, "Number of constants rebased");
52
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000053namespace {
Juergen Ributzka5429c062014-03-21 06:04:36 +000054struct ConstantUser;
55struct RebasedConstantInfo;
56
57typedef SmallVector<ConstantUser, 8> ConstantUseListType;
58typedef SmallVector<RebasedConstantInfo, 4> RebasedConstantListType;
59
60/// \brief Keeps track of the user of a constant and the operand index where the
61/// constant is used.
62struct ConstantUser {
63 Instruction *Inst;
64 unsigned OpndIdx;
65
66 ConstantUser(Instruction *Inst, unsigned Idx) : Inst(Inst), OpndIdx(Idx) { }
67};
68
Juergen Ributzkaf0dff492014-03-21 06:04:45 +000069/// \brief Keeps track of a constant candidate and its uses.
Juergen Ributzka6dab5202014-03-20 19:55:52 +000070struct ConstantCandidate {
Juergen Ributzka6dab5202014-03-20 19:55:52 +000071 ConstantUseListType Uses;
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +000072 ConstantInt *ConstInt;
73 unsigned CumulativeCost;
74
75 ConstantCandidate(ConstantInt *ConstInt)
76 : ConstInt(ConstInt), CumulativeCost(0) { }
Juergen Ributzka5429c062014-03-21 06:04:36 +000077
78 /// \brief Add the user to the use list and update the cost.
79 void addUser(Instruction *Inst, unsigned Idx, unsigned Cost) {
80 CumulativeCost += Cost;
81 Uses.push_back(ConstantUser(Inst, Idx));
82 }
Juergen Ributzka6dab5202014-03-20 19:55:52 +000083};
84
Juergen Ributzka5429c062014-03-21 06:04:36 +000085/// \brief This represents a constant that has been rebased with respect to a
86/// base constant. The difference to the base constant is recorded in Offset.
87struct RebasedConstantInfo {
88 ConstantUseListType Uses;
89 Constant *Offset;
Juergen Ributzka500abd42014-03-21 06:04:39 +000090 mutable BasicBlock *IDom;
Juergen Ributzka5429c062014-03-21 06:04:36 +000091
92 RebasedConstantInfo(ConstantUseListType &&Uses, Constant *Offset)
Juergen Ributzka500abd42014-03-21 06:04:39 +000093 : Uses(Uses), Offset(Offset), IDom(nullptr) { }
Juergen Ributzka5429c062014-03-21 06:04:36 +000094};
95
96/// \brief A base constant and all its rebased constants.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000097struct ConstantInfo {
98 ConstantInt *BaseConstant;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000099 RebasedConstantListType RebasedConstants;
100};
101
Juergen Ributzka5429c062014-03-21 06:04:36 +0000102/// \brief The constant hoisting pass.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000103class ConstantHoisting : public FunctionPass {
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000104 typedef DenseMap<ConstantInt *, unsigned> ConstCandMapType;
105 typedef std::vector<ConstantCandidate> ConstCandVecType;
106
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000107 const TargetTransformInfo *TTI;
108 DominatorTree *DT;
Juergen Ributzka5429c062014-03-21 06:04:36 +0000109 BasicBlock *Entry;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000110
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000111 /// Keeps track of constant candidates found in the function.
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000112 ConstCandVecType ConstCandVec;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000113
Juergen Ributzka5429c062014-03-21 06:04:36 +0000114 /// Keep track of cast instructions we already cloned.
115 SmallDenseMap<Instruction *, Instruction *> ClonedCastMap;
116
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000117 /// These are the final constants we decided to hoist.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000118 SmallVector<ConstantInfo, 8> ConstantVec;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000119public:
120 static char ID; // Pass identification, replacement for typeid
Juergen Ributzka5429c062014-03-21 06:04:36 +0000121 ConstantHoisting() : FunctionPass(ID), TTI(0), DT(0), Entry(0) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000122 initializeConstantHoistingPass(*PassRegistry::getPassRegistry());
123 }
124
Juergen Ributzka5429c062014-03-21 06:04:36 +0000125 bool runOnFunction(Function &Fn) override;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000126
Craig Topper3e4c6972014-03-05 09:10:37 +0000127 const char *getPassName() const override { return "Constant Hoisting"; }
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000128
Craig Topper3e4c6972014-03-05 09:10:37 +0000129 void getAnalysisUsage(AnalysisUsage &AU) const override {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000130 AU.setPreservesCFG();
131 AU.addRequired<DominatorTreeWrapperPass>();
132 AU.addRequired<TargetTransformInfo>();
133 }
134
135private:
Juergen Ributzka5429c062014-03-21 06:04:36 +0000136 /// \brief Initialize the pass.
137 void setup(Function &Fn) {
138 DT = &getAnalysis<DominatorTreeWrapperPass>().getDomTree();
139 TTI = &getAnalysis<TargetTransformInfo>();
140 Entry = &Fn.getEntryBlock();
141 }
142
143 /// \brief Cleanup.
144 void cleanup() {
145 ConstantVec.clear();
146 ClonedCastMap.clear();
147 ConstCandVec.clear();
Juergen Ributzka5429c062014-03-21 06:04:36 +0000148
149 TTI = nullptr;
150 DT = nullptr;
151 Entry = nullptr;
152 }
153
Juergen Ributzka500abd42014-03-21 06:04:39 +0000154 /// \brief Find the common dominator of all uses and cache the result for
155 /// future lookup.
156 BasicBlock *getIDom(const RebasedConstantInfo &RCI) const {
157 if (RCI.IDom)
158 return RCI.IDom;
159 RCI.IDom = findIDomOfAllUses(RCI.Uses);
160 assert(RCI.IDom && "Invalid IDom.");
161 return RCI.IDom;
162 }
163
164 BasicBlock *findIDomOfAllUses(const ConstantUseListType &Uses) const;
Juergen Ributzka5429c062014-03-21 06:04:36 +0000165 Instruction *findMatInsertPt(Instruction *Inst, unsigned Idx = ~0U) const;
166 Instruction *findConstantInsertionPoint(const ConstantInfo &ConstInfo) const;
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000167 void collectConstantCandidates(ConstCandMapType &ConstCandMap,
168 Instruction *Inst, unsigned Idx,
Juergen Ributzka5429c062014-03-21 06:04:36 +0000169 ConstantInt *ConstInt);
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000170 void collectConstantCandidates(ConstCandMapType &ConstCandMap,
171 Instruction *Inst);
Juergen Ributzka5429c062014-03-21 06:04:36 +0000172 void collectConstantCandidates(Function &Fn);
Juergen Ributzkab8489b32014-03-21 06:04:33 +0000173 void findAndMakeBaseConstant(ConstCandVecType::iterator S,
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000174 ConstCandVecType::iterator E);
Juergen Ributzkab8489b32014-03-21 06:04:33 +0000175 void findBaseConstants();
Juergen Ributzka5429c062014-03-21 06:04:36 +0000176 void emitBaseConstants(Instruction *Base, Constant *Offset,
177 const ConstantUser &ConstUser);
178 bool emitBaseConstants();
179 void deleteDeadCastInst() const;
180 bool optimizeConstants(Function &Fn);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000181};
182}
183
184char ConstantHoisting::ID = 0;
185INITIALIZE_PASS_BEGIN(ConstantHoisting, "consthoist", "Constant Hoisting",
186 false, false)
187INITIALIZE_PASS_DEPENDENCY(DominatorTreeWrapperPass)
188INITIALIZE_AG_DEPENDENCY(TargetTransformInfo)
189INITIALIZE_PASS_END(ConstantHoisting, "consthoist", "Constant Hoisting",
190 false, false)
191
192FunctionPass *llvm::createConstantHoistingPass() {
193 return new ConstantHoisting();
194}
195
196/// \brief Perform the constant hoisting optimization for the given function.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000197bool ConstantHoisting::runOnFunction(Function &Fn) {
198 DEBUG(dbgs() << "********** Begin Constant Hoisting **********\n");
199 DEBUG(dbgs() << "********** Function: " << Fn.getName() << '\n');
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000200
Juergen Ributzka5429c062014-03-21 06:04:36 +0000201 setup(Fn);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000202
Juergen Ributzka5429c062014-03-21 06:04:36 +0000203 bool MadeChange = optimizeConstants(Fn);
204
205 if (MadeChange) {
206 DEBUG(dbgs() << "********** Function after Constant Hoisting: "
207 << Fn.getName() << '\n');
208 DEBUG(dbgs() << Fn);
209 }
210 DEBUG(dbgs() << "********** End Constant Hoisting **********\n");
211
212 cleanup();
213
214 return MadeChange;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000215}
216
Juergen Ributzka500abd42014-03-21 06:04:39 +0000217/// \brief Find nearest common dominator of all uses.
218/// FIXME: Replace this with NearestCommonDominator once it is in common code.
219BasicBlock *
220ConstantHoisting::findIDomOfAllUses(const ConstantUseListType &Uses) const {
221 // Collect all basic blocks.
222 SmallPtrSet<BasicBlock *, 8> BBs;
223 for (auto const &U : Uses)
224 BBs.insert(findMatInsertPt(U.Inst, U.OpndIdx)->getParent());
225
226 if (BBs.count(Entry))
227 return Entry;
228
229 while (BBs.size() >= 2) {
230 BasicBlock *BB, *BB1, *BB2;
231 BB1 = *BBs.begin();
232 BB2 = *std::next(BBs.begin());
233 BB = DT->findNearestCommonDominator(BB1, BB2);
234 if (BB == Entry)
235 return Entry;
236 BBs.erase(BB1);
237 BBs.erase(BB2);
238 BBs.insert(BB);
239 }
240 assert((BBs.size() == 1) && "Expected only one element.");
241 return *BBs.begin();
242}
Juergen Ributzka5429c062014-03-21 06:04:36 +0000243
244/// \brief Find the constant materialization insertion point.
245Instruction *ConstantHoisting::findMatInsertPt(Instruction *Inst,
246 unsigned Idx) const {
247 // The simple and common case.
248 if (!isa<PHINode>(Inst) && !isa<LandingPadInst>(Inst))
249 return Inst;
250
251 // We can't insert directly before a phi node or landing pad. Insert before
252 // the terminator of the incoming or dominating block.
253 assert(Entry != Inst->getParent() && "PHI or landing pad in entry block!");
254 if (Idx != ~0U && isa<PHINode>(Inst))
255 return cast<PHINode>(Inst)->getIncomingBlock(Idx)->getTerminator();
256
257 BasicBlock *IDom = DT->getNode(Inst->getParent())->getIDom()->getBlock();
258 return IDom->getTerminator();
259}
260
261/// \brief Find an insertion point that dominates all uses.
262Instruction *ConstantHoisting::
263findConstantInsertionPoint(const ConstantInfo &ConstInfo) const {
264 assert(!ConstInfo.RebasedConstants.empty() && "Invalid constant info entry.");
Juergen Ributzka500abd42014-03-21 06:04:39 +0000265 // Collect all IDoms.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000266 SmallPtrSet<BasicBlock *, 8> BBs;
267 for (auto const &RCI : ConstInfo.RebasedConstants)
Juergen Ributzka500abd42014-03-21 06:04:39 +0000268 BBs.insert(getIDom(RCI));
269
270 assert(!BBs.empty() && "No dominators!?");
Juergen Ributzka5429c062014-03-21 06:04:36 +0000271
272 if (BBs.count(Entry))
273 return &Entry->front();
274
275 while (BBs.size() >= 2) {
276 BasicBlock *BB, *BB1, *BB2;
277 BB1 = *BBs.begin();
278 BB2 = *std::next(BBs.begin());
279 BB = DT->findNearestCommonDominator(BB1, BB2);
280 if (BB == Entry)
281 return &Entry->front();
282 BBs.erase(BB1);
283 BBs.erase(BB2);
284 BBs.insert(BB);
285 }
286 assert((BBs.size() == 1) && "Expected only one element.");
287 Instruction &FirstInst = (*BBs.begin())->front();
288 return findMatInsertPt(&FirstInst);
289}
290
291
292/// \brief Record constant integer ConstInt for instruction Inst at operand
293/// index Idx.
294///
Juergen Ributzkaf0dff492014-03-21 06:04:45 +0000295/// The operand at index Idx is not necessarily the constant integer itself. It
Juergen Ributzka5429c062014-03-21 06:04:36 +0000296/// could also be a cast instruction or a constant expression that uses the
297// constant integer.
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000298void ConstantHoisting::collectConstantCandidates(ConstCandMapType &ConstCandMap,
299 Instruction *Inst,
Juergen Ributzka5429c062014-03-21 06:04:36 +0000300 unsigned Idx,
301 ConstantInt *ConstInt) {
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000302 unsigned Cost;
Juergen Ributzka5429c062014-03-21 06:04:36 +0000303 // Ask the target about the cost of materializing the constant for the given
Juergen Ributzkaf0dff492014-03-21 06:04:45 +0000304 // instruction and operand index.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000305 if (auto IntrInst = dyn_cast<IntrinsicInst>(Inst))
Juergen Ributzkaf0dff492014-03-21 06:04:45 +0000306 Cost = TTI->getIntImmCost(IntrInst->getIntrinsicID(), Idx,
Juergen Ributzka5429c062014-03-21 06:04:36 +0000307 ConstInt->getValue(), ConstInt->getType());
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000308 else
Juergen Ributzkaf0dff492014-03-21 06:04:45 +0000309 Cost = TTI->getIntImmCost(Inst->getOpcode(), Idx, ConstInt->getValue(),
Juergen Ributzka5429c062014-03-21 06:04:36 +0000310 ConstInt->getType());
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000311
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000312 // Ignore cheap integer constants.
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000313 if (Cost > TargetTransformInfo::TCC_Basic) {
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000314 ConstCandMapType::iterator Itr;
315 bool Inserted;
Juergen Ributzka5429c062014-03-21 06:04:36 +0000316 std::tie(Itr, Inserted) = ConstCandMap.insert(std::make_pair(ConstInt, 0));
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000317 if (Inserted) {
Juergen Ributzka5429c062014-03-21 06:04:36 +0000318 ConstCandVec.push_back(ConstantCandidate(ConstInt));
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000319 Itr->second = ConstCandVec.size() - 1;
320 }
Juergen Ributzka5429c062014-03-21 06:04:36 +0000321 ConstCandVec[Itr->second].addUser(Inst, Idx, Cost);
322 DEBUG(if (isa<ConstantInt>(Inst->getOperand(Idx)))
323 dbgs() << "Collect constant " << *ConstInt << " from " << *Inst
324 << " with cost " << Cost << '\n';
325 else
326 dbgs() << "Collect constant " << *ConstInt << " indirectly from "
327 << *Inst << " via " << *Inst->getOperand(Idx) << " with cost "
328 << Cost << '\n';
329 );
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000330 }
331}
332
Juergen Ributzka5429c062014-03-21 06:04:36 +0000333/// \brief Scan the instruction for expensive integer constants and record them
334/// in the constant candidate vector.
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000335void ConstantHoisting::collectConstantCandidates(ConstCandMapType &ConstCandMap,
336 Instruction *Inst) {
Juergen Ributzka5429c062014-03-21 06:04:36 +0000337 // Skip all cast instructions. They are visited indirectly later on.
338 if (Inst->isCast())
339 return;
340
341 // Can't handle inline asm. Skip it.
342 if (auto Call = dyn_cast<CallInst>(Inst))
343 if (isa<InlineAsm>(Call->getCalledValue()))
344 return;
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000345
346 // Scan all operands.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000347 for (unsigned Idx = 0, E = Inst->getNumOperands(); Idx != E; ++Idx) {
348 Value *Opnd = Inst->getOperand(Idx);
349
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000350 // Visit constant integers.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000351 if (auto ConstInt = dyn_cast<ConstantInt>(Opnd)) {
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000352 collectConstantCandidates(ConstCandMap, Inst, Idx, ConstInt);
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000353 continue;
354 }
Juergen Ributzka5429c062014-03-21 06:04:36 +0000355
356 // Visit cast instructions that have constant integers.
357 if (auto CastInst = dyn_cast<Instruction>(Opnd)) {
358 // Only visit cast instructions, which have been skipped. All other
359 // instructions should have already been visited.
360 if (!CastInst->isCast())
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000361 continue;
362
Juergen Ributzka5429c062014-03-21 06:04:36 +0000363 if (auto *ConstInt = dyn_cast<ConstantInt>(CastInst->getOperand(0))) {
364 // Pretend the constant is directly used by the instruction and ignore
365 // the cast instruction.
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000366 collectConstantCandidates(ConstCandMap, Inst, Idx, ConstInt);
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000367 continue;
368 }
369 }
Juergen Ributzka5429c062014-03-21 06:04:36 +0000370
371 // Visit constant expressions that have constant integers.
372 if (auto ConstExpr = dyn_cast<ConstantExpr>(Opnd)) {
373 // Only visit constant cast expressions.
374 if (!ConstExpr->isCast())
375 continue;
376
377 if (auto ConstInt = dyn_cast<ConstantInt>(ConstExpr->getOperand(0))) {
378 // Pretend the constant is directly used by the instruction and ignore
379 // the constant expression.
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000380 collectConstantCandidates(ConstCandMap, Inst, Idx, ConstInt);
Juergen Ributzka5429c062014-03-21 06:04:36 +0000381 continue;
382 }
383 }
384 } // end of for all operands
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000385}
386
387/// \brief Collect all integer constants in the function that cannot be folded
388/// into an instruction itself.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000389void ConstantHoisting::collectConstantCandidates(Function &Fn) {
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000390 ConstCandMapType ConstCandMap;
Juergen Ributzka5429c062014-03-21 06:04:36 +0000391 for (Function::iterator BB : Fn)
392 for (BasicBlock::iterator Inst : *BB)
Juergen Ributzka7be410f2014-03-25 21:21:10 +0000393 collectConstantCandidates(ConstCandMap, Inst);
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000394}
395
396/// \brief Find the base constant within the given range and rebase all other
397/// constants with respect to the base constant.
Juergen Ributzkab8489b32014-03-21 06:04:33 +0000398void ConstantHoisting::findAndMakeBaseConstant(ConstCandVecType::iterator S,
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000399 ConstCandVecType::iterator E) {
Juergen Ributzka5429c062014-03-21 06:04:36 +0000400 auto MaxCostItr = S;
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000401 unsigned NumUses = 0;
402 // Use the constant that has the maximum cost as base constant.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000403 for (auto ConstCand = S; ConstCand != E; ++ConstCand) {
404 NumUses += ConstCand->Uses.size();
405 if (ConstCand->CumulativeCost > MaxCostItr->CumulativeCost)
406 MaxCostItr = ConstCand;
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000407 }
408
409 // Don't hoist constants that have only one use.
410 if (NumUses <= 1)
411 return;
412
Juergen Ributzka5429c062014-03-21 06:04:36 +0000413 ConstantInfo ConstInfo;
414 ConstInfo.BaseConstant = MaxCostItr->ConstInt;
415 Type *Ty = ConstInfo.BaseConstant->getType();
416
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000417 // Rebase the constants with respect to the base constant.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000418 for (auto ConstCand = S; ConstCand != E; ++ConstCand) {
419 APInt Diff = ConstCand->ConstInt->getValue() -
420 ConstInfo.BaseConstant->getValue();
421 Constant *Offset = Diff == 0 ? nullptr : ConstantInt::get(Ty, Diff);
422 ConstInfo.RebasedConstants.push_back(
423 RebasedConstantInfo(std::move(ConstCand->Uses), Offset));
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000424 }
Juergen Ributzka5429c062014-03-21 06:04:36 +0000425 ConstantVec.push_back(ConstInfo);
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000426}
427
Juergen Ributzka5429c062014-03-21 06:04:36 +0000428/// \brief Finds and combines constant candidates that can be easily
429/// rematerialized with an add from a common base constant.
Juergen Ributzkab8489b32014-03-21 06:04:33 +0000430void ConstantHoisting::findBaseConstants() {
Juergen Ributzka5429c062014-03-21 06:04:36 +0000431 // Sort the constants by value and type. This invalidates the mapping!
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000432 std::sort(ConstCandVec.begin(), ConstCandVec.end(),
433 [](const ConstantCandidate &LHS, const ConstantCandidate &RHS) {
434 if (LHS.ConstInt->getType() != RHS.ConstInt->getType())
435 return LHS.ConstInt->getType()->getBitWidth() <
436 RHS.ConstInt->getType()->getBitWidth();
437 return LHS.ConstInt->getValue().ult(RHS.ConstInt->getValue());
Juergen Ributzka46357932014-03-20 20:17:13 +0000438 });
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000439
Juergen Ributzka5429c062014-03-21 06:04:36 +0000440 // Simple linear scan through the sorted constant candidate vector for viable
441 // merge candidates.
442 auto MinValItr = ConstCandVec.begin();
443 for (auto CC = std::next(ConstCandVec.begin()), E = ConstCandVec.end();
444 CC != E; ++CC) {
445 if (MinValItr->ConstInt->getType() == CC->ConstInt->getType()) {
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000446 // Check if the constant is in range of an add with immediate.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000447 APInt Diff = CC->ConstInt->getValue() - MinValItr->ConstInt->getValue();
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000448 if ((Diff.getBitWidth() <= 64) &&
449 TTI->isLegalAddImmediate(Diff.getSExtValue()))
450 continue;
451 }
452 // We either have now a different constant type or the constant is not in
453 // range of an add with immediate anymore.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000454 findAndMakeBaseConstant(MinValItr, CC);
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000455 // Start a new base constant search.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000456 MinValItr = CC;
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000457 }
458 // Finalize the last base constant search.
Juergen Ributzkab8489b32014-03-21 06:04:33 +0000459 findAndMakeBaseConstant(MinValItr, ConstCandVec.end());
Juergen Ributzka46357932014-03-20 20:17:13 +0000460}
461
Juergen Ributzkae802d502014-03-22 01:49:27 +0000462/// \brief Updates the operand at Idx in instruction Inst with the result of
463/// instruction Mat. If the instruction is a PHI node then special
464/// handling for duplicate values form the same incomming basic block is
465/// required.
466/// \return The update will always succeed, but the return value indicated if
467/// Mat was used for the update or not.
468static bool updateOperand(Instruction *Inst, unsigned Idx, Instruction *Mat) {
469 if (auto PHI = dyn_cast<PHINode>(Inst)) {
470 // Check if any previous operand of the PHI node has the same incoming basic
471 // block. This is a very odd case that happens when the incoming basic block
472 // has a switch statement. In this case use the same value as the previous
473 // operand(s), otherwise we will fail verification due to different values.
474 // The values are actually the same, but the variable names are different
475 // and the verifier doesn't like that.
476 BasicBlock *IncomingBB = PHI->getIncomingBlock(Idx);
477 for (unsigned i = 0; i < Idx; ++i) {
478 if (PHI->getIncomingBlock(i) == IncomingBB) {
479 Value *IncomingVal = PHI->getIncomingValue(i);
480 Inst->setOperand(Idx, IncomingVal);
481 return false;
482 }
483 }
484 }
485
486 Inst->setOperand(Idx, Mat);
487 return true;
488}
489
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000490/// \brief Emit materialization code for all rebased constants and update their
491/// users.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000492void ConstantHoisting::emitBaseConstants(Instruction *Base, Constant *Offset,
493 const ConstantUser &ConstUser) {
494 Instruction *Mat = Base;
495 if (Offset) {
496 Instruction *InsertionPt = findMatInsertPt(ConstUser.Inst,
497 ConstUser.OpndIdx);
498 Mat = BinaryOperator::Create(Instruction::Add, Base, Offset,
499 "const_mat", InsertionPt);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000500
Juergen Ributzka5429c062014-03-21 06:04:36 +0000501 DEBUG(dbgs() << "Materialize constant (" << *Base->getOperand(0)
502 << " + " << *Offset << ") in BB "
503 << Mat->getParent()->getName() << '\n' << *Mat << '\n');
504 Mat->setDebugLoc(ConstUser.Inst->getDebugLoc());
505 }
506 Value *Opnd = ConstUser.Inst->getOperand(ConstUser.OpndIdx);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000507
Juergen Ributzka5429c062014-03-21 06:04:36 +0000508 // Visit constant integer.
509 if (isa<ConstantInt>(Opnd)) {
510 DEBUG(dbgs() << "Update: " << *ConstUser.Inst << '\n');
Juergen Ributzkae802d502014-03-22 01:49:27 +0000511 if (!updateOperand(ConstUser.Inst, ConstUser.OpndIdx, Mat) && Offset)
512 Mat->eraseFromParent();
Juergen Ributzka5429c062014-03-21 06:04:36 +0000513 DEBUG(dbgs() << "To : " << *ConstUser.Inst << '\n');
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000514 return;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000515 }
Juergen Ributzka6dab5202014-03-20 19:55:52 +0000516
Juergen Ributzka5429c062014-03-21 06:04:36 +0000517 // Visit cast instruction.
518 if (auto CastInst = dyn_cast<Instruction>(Opnd)) {
519 assert(CastInst->isCast() && "Expected an cast instruction!");
520 // Check if we already have visited this cast instruction before to avoid
521 // unnecessary cloning.
522 Instruction *&ClonedCastInst = ClonedCastMap[CastInst];
523 if (!ClonedCastInst) {
524 ClonedCastInst = CastInst->clone();
525 ClonedCastInst->setOperand(0, Mat);
526 ClonedCastInst->insertAfter(CastInst);
527 // Use the same debug location as the original cast instruction.
528 ClonedCastInst->setDebugLoc(CastInst->getDebugLoc());
529 DEBUG(dbgs() << "Clone instruction: " << *ClonedCastInst << '\n'
530 << "To : " << *CastInst << '\n');
Juergen Ributzka46357932014-03-20 20:17:13 +0000531 }
Juergen Ributzka5429c062014-03-21 06:04:36 +0000532
533 DEBUG(dbgs() << "Update: " << *ConstUser.Inst << '\n');
Juergen Ributzkae802d502014-03-22 01:49:27 +0000534 updateOperand(ConstUser.Inst, ConstUser.OpndIdx, ClonedCastInst);
Juergen Ributzka5429c062014-03-21 06:04:36 +0000535 DEBUG(dbgs() << "To : " << *ConstUser.Inst << '\n');
536 return;
Juergen Ributzka46357932014-03-20 20:17:13 +0000537 }
Juergen Ributzka5429c062014-03-21 06:04:36 +0000538
539 // Visit constant expression.
540 if (auto ConstExpr = dyn_cast<ConstantExpr>(Opnd)) {
541 Instruction *ConstExprInst = ConstExpr->getAsInstruction();
542 ConstExprInst->setOperand(0, Mat);
543 ConstExprInst->insertBefore(findMatInsertPt(ConstUser.Inst,
544 ConstUser.OpndIdx));
545
546 // Use the same debug location as the instruction we are about to update.
547 ConstExprInst->setDebugLoc(ConstUser.Inst->getDebugLoc());
548
549 DEBUG(dbgs() << "Create instruction: " << *ConstExprInst << '\n'
550 << "From : " << *ConstExpr << '\n');
551 DEBUG(dbgs() << "Update: " << *ConstUser.Inst << '\n');
Juergen Ributzkae802d502014-03-22 01:49:27 +0000552 if (!updateOperand(ConstUser.Inst, ConstUser.OpndIdx, ConstExprInst)) {
553 ConstExprInst->eraseFromParent();
554 if (Offset)
555 Mat->eraseFromParent();
556 }
Juergen Ributzka5429c062014-03-21 06:04:36 +0000557 DEBUG(dbgs() << "To : " << *ConstUser.Inst << '\n');
558 return;
Juergen Ributzka4c8a0252014-02-08 00:20:45 +0000559 }
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000560}
561
562/// \brief Hoist and hide the base constant behind a bitcast and emit
563/// materialization code for derived constants.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000564bool ConstantHoisting::emitBaseConstants() {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000565 bool MadeChange = false;
Juergen Ributzka5429c062014-03-21 06:04:36 +0000566 for (auto const &ConstInfo : ConstantVec) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000567 // Hoist and hide the base constant behind a bitcast.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000568 Instruction *IP = findConstantInsertionPoint(ConstInfo);
569 IntegerType *Ty = ConstInfo.BaseConstant->getType();
570 Instruction *Base =
571 new BitCastInst(ConstInfo.BaseConstant, Ty, "const", IP);
572 DEBUG(dbgs() << "Hoist constant (" << *ConstInfo.BaseConstant << ") to BB "
573 << IP->getParent()->getName() << '\n' << *Base << '\n');
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000574 NumConstantsHoisted++;
575
576 // Emit materialization code for all rebased constants.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000577 for (auto const &RCI : ConstInfo.RebasedConstants) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000578 NumConstantsRebased++;
Juergen Ributzka5429c062014-03-21 06:04:36 +0000579 for (auto const &U : RCI.Uses)
580 emitBaseConstants(Base, RCI.Offset, U);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000581 }
582
583 // Use the same debug location as the last user of the constant.
584 assert(!Base->use_empty() && "The use list is empty!?");
Chandler Carruthcdf47882014-03-09 03:16:01 +0000585 assert(isa<Instruction>(Base->user_back()) &&
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000586 "All uses should be instructions.");
Chandler Carruthcdf47882014-03-09 03:16:01 +0000587 Base->setDebugLoc(cast<Instruction>(Base->user_back())->getDebugLoc());
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000588
589 // Correct for base constant, which we counted above too.
590 NumConstantsRebased--;
591 MadeChange = true;
592 }
593 return MadeChange;
594}
595
Juergen Ributzka5429c062014-03-21 06:04:36 +0000596/// \brief Check all cast instructions we made a copy of and remove them if they
597/// have no more users.
598void ConstantHoisting::deleteDeadCastInst() const {
599 for (auto const &I : ClonedCastMap)
600 if (I.first->use_empty())
Juergen Ributzkae4747522014-03-22 01:49:30 +0000601 I.first->eraseFromParent();
Juergen Ributzka5429c062014-03-21 06:04:36 +0000602}
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000603
Juergen Ributzka5429c062014-03-21 06:04:36 +0000604/// \brief Optimize expensive integer constants in the given function.
605bool ConstantHoisting::optimizeConstants(Function &Fn) {
Juergen Ributzka46357932014-03-20 20:17:13 +0000606 // Collect all constant candidates.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000607 collectConstantCandidates(Fn);
Juergen Ributzka46357932014-03-20 20:17:13 +0000608
Juergen Ributzkaa29a5b82014-03-21 06:04:30 +0000609 // There are no constant candidates to worry about.
610 if (ConstCandVec.empty())
611 return false;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000612
613 // Combine constants that can be easily materialized with an add from a common
614 // base constant.
Juergen Ributzkab8489b32014-03-21 06:04:33 +0000615 findBaseConstants();
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000616
Juergen Ributzka5429c062014-03-21 06:04:36 +0000617 // There are no constants to emit.
618 if (ConstantVec.empty())
619 return false;
620
Juergen Ributzkaf0dff492014-03-21 06:04:45 +0000621 // Finally hoist the base constant and emit materialization code for dependent
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000622 // constants.
Juergen Ributzka5429c062014-03-21 06:04:36 +0000623 bool MadeChange = emitBaseConstants();
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000624
Juergen Ributzka5429c062014-03-21 06:04:36 +0000625 // Cleanup dead instructions.
626 deleteDeadCastInst();
Juergen Ributzkaf26beda2014-01-25 02:02:55 +0000627
628 return MadeChange;
629}