blob: a57cea8b7e517956a51f7519533ad1717084fc04 [file] [log] [blame]
Misha Brukman373086d2003-05-20 21:01:22 +00001//===- SCCP.cpp - Sparse Conditional Constant Propagation -----------------===//
Misha Brukmanb1c93172005-04-21 23:48:37 +00002//
John Criswell482202a2003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
Misha Brukmanb1c93172005-04-21 23:48:37 +00007//
John Criswell482202a2003-10-20 19:43:21 +00008//===----------------------------------------------------------------------===//
Chris Lattner347389d2001-06-27 23:38:11 +00009//
Misha Brukman373086d2003-05-20 21:01:22 +000010// This file implements sparse conditional constant propagation and merging:
Chris Lattner347389d2001-06-27 23:38:11 +000011//
12// Specifically, this:
13// * Assumes values are constant unless proven otherwise
14// * Assumes BasicBlocks are dead unless proven otherwise
15// * Proves values to be constant, and replaces them with constants
Chris Lattnerdd6522e2002-08-30 23:39:00 +000016// * Proves conditional branches to be unconditional
Chris Lattner347389d2001-06-27 23:38:11 +000017//
18// Notice that:
19// * This pass has a habit of making definitions be dead. It is a good idea
20// to to run a DCE pass sometime after running this pass.
21//
22//===----------------------------------------------------------------------===//
23
Chris Lattner4f031622004-11-15 05:03:30 +000024#define DEBUG_TYPE "sccp"
Chris Lattnerb4cfa7f2002-05-07 20:03:00 +000025#include "llvm/Transforms/Scalar.h"
Chris Lattnerb4394642004-12-10 08:02:06 +000026#include "llvm/Transforms/IPO.h"
Chris Lattner0fe5b322004-01-12 17:43:40 +000027#include "llvm/Constants.h"
Chris Lattner91dbae62004-12-11 05:15:59 +000028#include "llvm/DerivedTypes.h"
Chris Lattnercccc5c72003-04-25 02:50:03 +000029#include "llvm/Instructions.h"
Chris Lattner04805fa2002-02-26 21:46:54 +000030#include "llvm/Pass.h"
Chris Lattner6e560792002-04-18 15:13:15 +000031#include "llvm/Support/InstVisitor.h"
Chris Lattnerff9362a2004-04-13 19:43:54 +000032#include "llvm/Transforms/Utils/Local.h"
Chris Lattnerb4394642004-12-10 08:02:06 +000033#include "llvm/Support/CallSite.h"
Reid Spencer7c16caa2004-09-01 22:55:40 +000034#include "llvm/Support/Debug.h"
35#include "llvm/ADT/hash_map"
36#include "llvm/ADT/Statistic.h"
37#include "llvm/ADT/STLExtras.h"
Chris Lattner347389d2001-06-27 23:38:11 +000038#include <algorithm>
Chris Lattnerc597b8a2006-01-22 23:32:06 +000039#include <iostream>
Chris Lattner347389d2001-06-27 23:38:11 +000040#include <set>
Chris Lattner49525f82004-01-09 06:02:20 +000041using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000042
Chris Lattner4f031622004-11-15 05:03:30 +000043// LatticeVal class - This class represents the different lattice values that an
Chris Lattnerc8e66542002-04-27 06:56:12 +000044// instruction may occupy. It is a simple class with value semantics.
Chris Lattner347389d2001-06-27 23:38:11 +000045//
Chris Lattner7d325382002-04-29 21:26:08 +000046namespace {
Chris Lattnerbf3a0992002-10-01 22:38:41 +000047
Chris Lattner4f031622004-11-15 05:03:30 +000048class LatticeVal {
Misha Brukmanb1c93172005-04-21 23:48:37 +000049 enum {
Chris Lattner3462ae32001-12-03 22:26:30 +000050 undefined, // This instruction has no known value
51 constant, // This instruction has a constant value
Chris Lattner3462ae32001-12-03 22:26:30 +000052 overdefined // This instruction has an unknown value
53 } LatticeValue; // The current lattice position
54 Constant *ConstantVal; // If Constant value, the current value
Chris Lattner347389d2001-06-27 23:38:11 +000055public:
Chris Lattner4f031622004-11-15 05:03:30 +000056 inline LatticeVal() : LatticeValue(undefined), ConstantVal(0) {}
Chris Lattner347389d2001-06-27 23:38:11 +000057
58 // markOverdefined - Return true if this is a new status to be in...
59 inline bool markOverdefined() {
Chris Lattner3462ae32001-12-03 22:26:30 +000060 if (LatticeValue != overdefined) {
61 LatticeValue = overdefined;
Chris Lattner347389d2001-06-27 23:38:11 +000062 return true;
63 }
64 return false;
65 }
66
67 // markConstant - Return true if this is a new status for us...
Chris Lattner3462ae32001-12-03 22:26:30 +000068 inline bool markConstant(Constant *V) {
69 if (LatticeValue != constant) {
70 LatticeValue = constant;
Chris Lattner347389d2001-06-27 23:38:11 +000071 ConstantVal = V;
72 return true;
73 } else {
Chris Lattnerdae05dc2001-09-07 16:43:22 +000074 assert(ConstantVal == V && "Marking constant with different value");
Chris Lattner347389d2001-06-27 23:38:11 +000075 }
76 return false;
77 }
78
Chris Lattner3462ae32001-12-03 22:26:30 +000079 inline bool isUndefined() const { return LatticeValue == undefined; }
80 inline bool isConstant() const { return LatticeValue == constant; }
81 inline bool isOverdefined() const { return LatticeValue == overdefined; }
Chris Lattner347389d2001-06-27 23:38:11 +000082
Chris Lattner05fe6842004-01-12 03:57:30 +000083 inline Constant *getConstant() const {
84 assert(isConstant() && "Cannot get the constant of a non-constant!");
85 return ConstantVal;
86 }
Chris Lattner347389d2001-06-27 23:38:11 +000087};
88
Chris Lattner7d325382002-04-29 21:26:08 +000089} // end anonymous namespace
Chris Lattner347389d2001-06-27 23:38:11 +000090
91
92//===----------------------------------------------------------------------===//
Chris Lattner347389d2001-06-27 23:38:11 +000093//
Chris Lattner074be1f2004-11-15 04:44:20 +000094/// SCCPSolver - This class is a general purpose solver for Sparse Conditional
95/// Constant Propagation.
96///
97class SCCPSolver : public InstVisitor<SCCPSolver> {
Chris Lattner7f74a562002-01-20 22:54:45 +000098 std::set<BasicBlock*> BBExecutable;// The basic blocks that are executable
Chris Lattner4f031622004-11-15 05:03:30 +000099 hash_map<Value*, LatticeVal> ValueState; // The state each value is in...
Chris Lattner347389d2001-06-27 23:38:11 +0000100
Chris Lattner91dbae62004-12-11 05:15:59 +0000101 /// GlobalValue - If we are tracking any values for the contents of a global
102 /// variable, we keep a mapping from the constant accessor to the element of
103 /// the global, to the currently known value. If the value becomes
104 /// overdefined, it's entry is simply removed from this map.
105 hash_map<GlobalVariable*, LatticeVal> TrackedGlobals;
106
Chris Lattnerb4394642004-12-10 08:02:06 +0000107 /// TrackedFunctionRetVals - If we are tracking arguments into and the return
108 /// value out of a function, it will have an entry in this map, indicating
109 /// what the known return value for the function is.
110 hash_map<Function*, LatticeVal> TrackedFunctionRetVals;
111
Chris Lattnerd79334d2004-07-15 23:36:43 +0000112 // The reason for two worklists is that overdefined is the lowest state
113 // on the lattice, and moving things to overdefined as fast as possible
114 // makes SCCP converge much faster.
115 // By having a separate worklist, we accomplish this because everything
116 // possibly overdefined will become overdefined at the soonest possible
117 // point.
Chris Lattnerb4394642004-12-10 08:02:06 +0000118 std::vector<Value*> OverdefinedInstWorkList;
119 std::vector<Value*> InstWorkList;
Chris Lattnerd79334d2004-07-15 23:36:43 +0000120
121
Chris Lattner7f74a562002-01-20 22:54:45 +0000122 std::vector<BasicBlock*> BBWorkList; // The BasicBlock work list
Chris Lattner0bbbe5d2003-10-08 16:55:34 +0000123
Chris Lattner05fe6842004-01-12 03:57:30 +0000124 /// UsersOfOverdefinedPHIs - Keep track of any users of PHI nodes that are not
125 /// overdefined, despite the fact that the PHI node is overdefined.
126 std::multimap<PHINode*, Instruction*> UsersOfOverdefinedPHIs;
127
Chris Lattner0bbbe5d2003-10-08 16:55:34 +0000128 /// KnownFeasibleEdges - Entries in this set are edges which have already had
129 /// PHI nodes retriggered.
130 typedef std::pair<BasicBlock*,BasicBlock*> Edge;
131 std::set<Edge> KnownFeasibleEdges;
Chris Lattner347389d2001-06-27 23:38:11 +0000132public:
133
Chris Lattner074be1f2004-11-15 04:44:20 +0000134 /// MarkBlockExecutable - This method can be used by clients to mark all of
135 /// the blocks that are known to be intrinsically live in the processed unit.
136 void MarkBlockExecutable(BasicBlock *BB) {
137 DEBUG(std::cerr << "Marking Block Executable: " << BB->getName() << "\n");
138 BBExecutable.insert(BB); // Basic block is executable!
139 BBWorkList.push_back(BB); // Add the block to the work list!
Chris Lattner7d325382002-04-29 21:26:08 +0000140 }
141
Chris Lattner91dbae62004-12-11 05:15:59 +0000142 /// TrackValueOfGlobalVariable - Clients can use this method to
Chris Lattnerb4394642004-12-10 08:02:06 +0000143 /// inform the SCCPSolver that it should track loads and stores to the
144 /// specified global variable if it can. This is only legal to call if
145 /// performing Interprocedural SCCP.
Chris Lattner91dbae62004-12-11 05:15:59 +0000146 void TrackValueOfGlobalVariable(GlobalVariable *GV) {
147 const Type *ElTy = GV->getType()->getElementType();
148 if (ElTy->isFirstClassType()) {
149 LatticeVal &IV = TrackedGlobals[GV];
150 if (!isa<UndefValue>(GV->getInitializer()))
151 IV.markConstant(GV->getInitializer());
152 }
153 }
Chris Lattnerb4394642004-12-10 08:02:06 +0000154
155 /// AddTrackedFunction - If the SCCP solver is supposed to track calls into
156 /// and out of the specified function (which cannot have its address taken),
157 /// this method must be called.
158 void AddTrackedFunction(Function *F) {
159 assert(F->hasInternalLinkage() && "Can only track internal functions!");
160 // Add an entry, F -> undef.
161 TrackedFunctionRetVals[F];
162 }
163
Chris Lattner074be1f2004-11-15 04:44:20 +0000164 /// Solve - Solve for constants and executable blocks.
165 ///
166 void Solve();
Chris Lattner347389d2001-06-27 23:38:11 +0000167
Chris Lattner7285f432004-12-10 20:41:50 +0000168 /// ResolveBranchesIn - While solving the dataflow for a function, we assume
169 /// that branches on undef values cannot reach any of their successors.
170 /// However, this is not a safe assumption. After we solve dataflow, this
171 /// method should be use to handle this. If this returns true, the solver
172 /// should be rerun.
173 bool ResolveBranchesIn(Function &F);
174
Chris Lattner074be1f2004-11-15 04:44:20 +0000175 /// getExecutableBlocks - Once we have solved for constants, return the set of
176 /// blocks that is known to be executable.
177 std::set<BasicBlock*> &getExecutableBlocks() {
178 return BBExecutable;
179 }
180
181 /// getValueMapping - Once we have solved for constants, return the mapping of
Chris Lattner4f031622004-11-15 05:03:30 +0000182 /// LLVM values to LatticeVals.
183 hash_map<Value*, LatticeVal> &getValueMapping() {
Chris Lattner074be1f2004-11-15 04:44:20 +0000184 return ValueState;
185 }
186
Chris Lattner99e12952004-12-11 02:53:57 +0000187 /// getTrackedFunctionRetVals - Get the inferred return value map.
188 ///
189 const hash_map<Function*, LatticeVal> &getTrackedFunctionRetVals() {
190 return TrackedFunctionRetVals;
191 }
192
Chris Lattner91dbae62004-12-11 05:15:59 +0000193 /// getTrackedGlobals - Get and return the set of inferred initializers for
194 /// global variables.
195 const hash_map<GlobalVariable*, LatticeVal> &getTrackedGlobals() {
196 return TrackedGlobals;
197 }
198
Chris Lattner99e12952004-12-11 02:53:57 +0000199
Chris Lattner347389d2001-06-27 23:38:11 +0000200private:
Chris Lattnerd79334d2004-07-15 23:36:43 +0000201 // markConstant - Make a value be marked as "constant". If the value
Misha Brukmanb1c93172005-04-21 23:48:37 +0000202 // is not already a constant, add it to the instruction work list so that
Chris Lattner347389d2001-06-27 23:38:11 +0000203 // the users of the instruction are updated later.
204 //
Chris Lattnerb4394642004-12-10 08:02:06 +0000205 inline void markConstant(LatticeVal &IV, Value *V, Constant *C) {
Chris Lattner7324f7c2003-10-08 16:21:03 +0000206 if (IV.markConstant(C)) {
Chris Lattnerb4394642004-12-10 08:02:06 +0000207 DEBUG(std::cerr << "markConstant: " << *C << ": " << *V);
208 InstWorkList.push_back(V);
Chris Lattner347389d2001-06-27 23:38:11 +0000209 }
Chris Lattner7324f7c2003-10-08 16:21:03 +0000210 }
Chris Lattnerb4394642004-12-10 08:02:06 +0000211 inline void markConstant(Value *V, Constant *C) {
212 markConstant(ValueState[V], V, C);
Chris Lattner347389d2001-06-27 23:38:11 +0000213 }
214
Chris Lattnerd79334d2004-07-15 23:36:43 +0000215 // markOverdefined - Make a value be marked as "overdefined". If the
Misha Brukmanb1c93172005-04-21 23:48:37 +0000216 // value is not already overdefined, add it to the overdefined instruction
Chris Lattnerd79334d2004-07-15 23:36:43 +0000217 // work list so that the users of the instruction are updated later.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000218
Chris Lattnerb4394642004-12-10 08:02:06 +0000219 inline void markOverdefined(LatticeVal &IV, Value *V) {
Chris Lattner7324f7c2003-10-08 16:21:03 +0000220 if (IV.markOverdefined()) {
Chris Lattner2f687fd2004-12-11 06:05:53 +0000221 DEBUG(std::cerr << "markOverdefined: ";
222 if (Function *F = dyn_cast<Function>(V))
223 std::cerr << "Function '" << F->getName() << "'\n";
224 else
225 std::cerr << *V);
Chris Lattner074be1f2004-11-15 04:44:20 +0000226 // Only instructions go on the work list
Chris Lattnerb4394642004-12-10 08:02:06 +0000227 OverdefinedInstWorkList.push_back(V);
Chris Lattner347389d2001-06-27 23:38:11 +0000228 }
Chris Lattner7324f7c2003-10-08 16:21:03 +0000229 }
Chris Lattnerb4394642004-12-10 08:02:06 +0000230 inline void markOverdefined(Value *V) {
231 markOverdefined(ValueState[V], V);
232 }
233
234 inline void mergeInValue(LatticeVal &IV, Value *V, LatticeVal &MergeWithV) {
235 if (IV.isOverdefined() || MergeWithV.isUndefined())
236 return; // Noop.
237 if (MergeWithV.isOverdefined())
238 markOverdefined(IV, V);
239 else if (IV.isUndefined())
240 markConstant(IV, V, MergeWithV.getConstant());
241 else if (IV.getConstant() != MergeWithV.getConstant())
242 markOverdefined(IV, V);
Chris Lattner347389d2001-06-27 23:38:11 +0000243 }
Chris Lattner06a0ed12006-02-08 02:38:11 +0000244
245 inline void mergeInValue(Value *V, LatticeVal &MergeWithV) {
246 return mergeInValue(ValueState[V], V, MergeWithV);
247 }
248
Chris Lattner347389d2001-06-27 23:38:11 +0000249
Chris Lattner4f031622004-11-15 05:03:30 +0000250 // getValueState - Return the LatticeVal object that corresponds to the value.
Misha Brukman7eb05a12003-08-18 14:43:39 +0000251 // This function is necessary because not all values should start out in the
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000252 // underdefined state... Argument's should be overdefined, and
Chris Lattner57698e22002-03-26 18:01:55 +0000253 // constants should be marked as constants. If a value is not known to be an
Chris Lattner347389d2001-06-27 23:38:11 +0000254 // Instruction object, then use this accessor to get its value from the map.
255 //
Chris Lattner4f031622004-11-15 05:03:30 +0000256 inline LatticeVal &getValueState(Value *V) {
257 hash_map<Value*, LatticeVal>::iterator I = ValueState.find(V);
Chris Lattner347389d2001-06-27 23:38:11 +0000258 if (I != ValueState.end()) return I->second; // Common case, in the map
Chris Lattner646354b2004-10-16 18:09:41 +0000259
Chris Lattnerd18c16b2004-11-15 05:45:33 +0000260 if (Constant *CPV = dyn_cast<Constant>(V)) {
261 if (isa<UndefValue>(V)) {
262 // Nothing to do, remain undefined.
263 } else {
264 ValueState[CPV].markConstant(CPV); // Constants are constant
265 }
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000266 }
Chris Lattner347389d2001-06-27 23:38:11 +0000267 // All others are underdefined by default...
268 return ValueState[V];
269 }
270
Misha Brukmanb1c93172005-04-21 23:48:37 +0000271 // markEdgeExecutable - Mark a basic block as executable, adding it to the BB
Chris Lattner347389d2001-06-27 23:38:11 +0000272 // work list if it is not already executable...
Misha Brukmanb1c93172005-04-21 23:48:37 +0000273 //
Chris Lattner0bbbe5d2003-10-08 16:55:34 +0000274 void markEdgeExecutable(BasicBlock *Source, BasicBlock *Dest) {
275 if (!KnownFeasibleEdges.insert(Edge(Source, Dest)).second)
276 return; // This edge is already known to be executable!
277
278 if (BBExecutable.count(Dest)) {
279 DEBUG(std::cerr << "Marking Edge Executable: " << Source->getName()
280 << " -> " << Dest->getName() << "\n");
281
282 // The destination is already executable, but we just made an edge
Chris Lattner35e56e72003-10-08 16:56:11 +0000283 // feasible that wasn't before. Revisit the PHI nodes in the block
284 // because they have potentially new operands.
Chris Lattnerb4394642004-12-10 08:02:06 +0000285 for (BasicBlock::iterator I = Dest->begin(); isa<PHINode>(I); ++I)
286 visitPHINode(*cast<PHINode>(I));
Chris Lattnercccc5c72003-04-25 02:50:03 +0000287
288 } else {
Chris Lattner074be1f2004-11-15 04:44:20 +0000289 MarkBlockExecutable(Dest);
Chris Lattnercccc5c72003-04-25 02:50:03 +0000290 }
Chris Lattner347389d2001-06-27 23:38:11 +0000291 }
292
Chris Lattner074be1f2004-11-15 04:44:20 +0000293 // getFeasibleSuccessors - Return a vector of booleans to indicate which
294 // successors are reachable from a given terminator instruction.
295 //
296 void getFeasibleSuccessors(TerminatorInst &TI, std::vector<bool> &Succs);
297
298 // isEdgeFeasible - Return true if the control flow edge from the 'From' basic
299 // block to the 'To' basic block is currently feasible...
300 //
301 bool isEdgeFeasible(BasicBlock *From, BasicBlock *To);
302
303 // OperandChangedState - This method is invoked on all of the users of an
304 // instruction that was just changed state somehow.... Based on this
305 // information, we need to update the specified user of this instruction.
306 //
307 void OperandChangedState(User *U) {
308 // Only instructions use other variable values!
309 Instruction &I = cast<Instruction>(*U);
310 if (BBExecutable.count(I.getParent())) // Inst is executable?
311 visit(I);
312 }
313
314private:
315 friend class InstVisitor<SCCPSolver>;
Chris Lattner347389d2001-06-27 23:38:11 +0000316
Misha Brukmanb1c93172005-04-21 23:48:37 +0000317 // visit implementations - Something changed in this instruction... Either an
Chris Lattner10b250e2001-06-29 23:56:23 +0000318 // operand made a transition, or the instruction is newly executable. Change
319 // the value type of I to reflect these changes if appropriate.
320 //
Chris Lattner113f4f42002-06-25 16:13:24 +0000321 void visitPHINode(PHINode &I);
Chris Lattner6e560792002-04-18 15:13:15 +0000322
323 // Terminators
Chris Lattnerb4394642004-12-10 08:02:06 +0000324 void visitReturnInst(ReturnInst &I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000325 void visitTerminatorInst(TerminatorInst &TI);
Chris Lattner6e560792002-04-18 15:13:15 +0000326
Chris Lattner6e1a1b12002-08-14 17:53:45 +0000327 void visitCastInst(CastInst &I);
Chris Lattner59db22d2004-03-12 05:52:44 +0000328 void visitSelectInst(SelectInst &I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000329 void visitBinaryOperator(Instruction &I);
330 void visitShiftInst(ShiftInst &I) { visitBinaryOperator(I); }
Robert Bocchinobd518d12006-01-10 19:05:05 +0000331 void visitExtractElementInst(ExtractElementInst &I);
Robert Bocchino6dce2502006-01-17 20:06:55 +0000332 void visitInsertElementInst(InsertElementInst &I);
Chris Lattner6e560792002-04-18 15:13:15 +0000333
334 // Instructions that cannot be folded away...
Chris Lattner91dbae62004-12-11 05:15:59 +0000335 void visitStoreInst (Instruction &I);
Chris Lattner49f74522004-01-12 04:29:41 +0000336 void visitLoadInst (LoadInst &I);
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000337 void visitGetElementPtrInst(GetElementPtrInst &I);
Chris Lattnerb4394642004-12-10 08:02:06 +0000338 void visitCallInst (CallInst &I) { visitCallSite(CallSite::get(&I)); }
339 void visitInvokeInst (InvokeInst &II) {
340 visitCallSite(CallSite::get(&II));
341 visitTerminatorInst(II);
Chris Lattnerdf741d62003-08-27 01:08:35 +0000342 }
Chris Lattnerb4394642004-12-10 08:02:06 +0000343 void visitCallSite (CallSite CS);
Chris Lattner9c58cf62003-09-08 18:54:55 +0000344 void visitUnwindInst (TerminatorInst &I) { /*returns void*/ }
Chris Lattner646354b2004-10-16 18:09:41 +0000345 void visitUnreachableInst(TerminatorInst &I) { /*returns void*/ }
Chris Lattner113f4f42002-06-25 16:13:24 +0000346 void visitAllocationInst(Instruction &I) { markOverdefined(&I); }
Chris Lattnerf0fc9be2003-10-18 05:56:52 +0000347 void visitVANextInst (Instruction &I) { markOverdefined(&I); }
348 void visitVAArgInst (Instruction &I) { markOverdefined(&I); }
Chris Lattner113f4f42002-06-25 16:13:24 +0000349 void visitFreeInst (Instruction &I) { /*returns void*/ }
Chris Lattner6e560792002-04-18 15:13:15 +0000350
Chris Lattner113f4f42002-06-25 16:13:24 +0000351 void visitInstruction(Instruction &I) {
Chris Lattner6e560792002-04-18 15:13:15 +0000352 // If a new instruction is added to LLVM that we don't handle...
Chris Lattnercccc5c72003-04-25 02:50:03 +0000353 std::cerr << "SCCP: Don't know how to handle: " << I;
Chris Lattner113f4f42002-06-25 16:13:24 +0000354 markOverdefined(&I); // Just in case
Chris Lattner6e560792002-04-18 15:13:15 +0000355 }
Chris Lattner10b250e2001-06-29 23:56:23 +0000356};
Chris Lattnerb28b6802002-07-23 18:06:35 +0000357
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000358// getFeasibleSuccessors - Return a vector of booleans to indicate which
359// successors are reachable from a given terminator instruction.
360//
Chris Lattner074be1f2004-11-15 04:44:20 +0000361void SCCPSolver::getFeasibleSuccessors(TerminatorInst &TI,
362 std::vector<bool> &Succs) {
Chris Lattnercccc5c72003-04-25 02:50:03 +0000363 Succs.resize(TI.getNumSuccessors());
Chris Lattner113f4f42002-06-25 16:13:24 +0000364 if (BranchInst *BI = dyn_cast<BranchInst>(&TI)) {
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000365 if (BI->isUnconditional()) {
366 Succs[0] = true;
367 } else {
Chris Lattner4f031622004-11-15 05:03:30 +0000368 LatticeVal &BCValue = getValueState(BI->getCondition());
Chris Lattnerfe992d42004-01-12 17:40:36 +0000369 if (BCValue.isOverdefined() ||
370 (BCValue.isConstant() && !isa<ConstantBool>(BCValue.getConstant()))) {
371 // Overdefined condition variables, and branches on unfoldable constant
372 // conditions, mean the branch could go either way.
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000373 Succs[0] = Succs[1] = true;
374 } else if (BCValue.isConstant()) {
375 // Constant condition variables mean the branch can only go a single way
376 Succs[BCValue.getConstant() == ConstantBool::False] = true;
377 }
378 }
Chris Lattner113f4f42002-06-25 16:13:24 +0000379 } else if (InvokeInst *II = dyn_cast<InvokeInst>(&TI)) {
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000380 // Invoke instructions successors are always executable.
381 Succs[0] = Succs[1] = true;
Chris Lattner113f4f42002-06-25 16:13:24 +0000382 } else if (SwitchInst *SI = dyn_cast<SwitchInst>(&TI)) {
Chris Lattner4f031622004-11-15 05:03:30 +0000383 LatticeVal &SCValue = getValueState(SI->getCondition());
Chris Lattnerfe992d42004-01-12 17:40:36 +0000384 if (SCValue.isOverdefined() || // Overdefined condition?
385 (SCValue.isConstant() && !isa<ConstantInt>(SCValue.getConstant()))) {
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000386 // All destinations are executable!
Chris Lattner113f4f42002-06-25 16:13:24 +0000387 Succs.assign(TI.getNumSuccessors(), true);
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000388 } else if (SCValue.isConstant()) {
389 Constant *CPV = SCValue.getConstant();
390 // Make sure to skip the "default value" which isn't a value
391 for (unsigned i = 1, E = SI->getNumSuccessors(); i != E; ++i) {
392 if (SI->getSuccessorValue(i) == CPV) {// Found the right branch...
393 Succs[i] = true;
394 return;
395 }
396 }
397
398 // Constant value not equal to any of the branches... must execute
399 // default branch then...
400 Succs[0] = true;
401 }
402 } else {
Chris Lattnercccc5c72003-04-25 02:50:03 +0000403 std::cerr << "SCCP: Don't know how to handle: " << TI;
Chris Lattner113f4f42002-06-25 16:13:24 +0000404 Succs.assign(TI.getNumSuccessors(), true);
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000405 }
406}
407
408
Chris Lattner13b52e72002-05-02 21:18:01 +0000409// isEdgeFeasible - Return true if the control flow edge from the 'From' basic
410// block to the 'To' basic block is currently feasible...
411//
Chris Lattner074be1f2004-11-15 04:44:20 +0000412bool SCCPSolver::isEdgeFeasible(BasicBlock *From, BasicBlock *To) {
Chris Lattner13b52e72002-05-02 21:18:01 +0000413 assert(BBExecutable.count(To) && "Dest should always be alive!");
414
415 // Make sure the source basic block is executable!!
416 if (!BBExecutable.count(From)) return false;
Misha Brukmanb1c93172005-04-21 23:48:37 +0000417
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000418 // Check to make sure this edge itself is actually feasible now...
Chris Lattner71ac22ff2003-10-08 15:47:41 +0000419 TerminatorInst *TI = From->getTerminator();
420 if (BranchInst *BI = dyn_cast<BranchInst>(TI)) {
421 if (BI->isUnconditional())
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000422 return true;
Chris Lattner71ac22ff2003-10-08 15:47:41 +0000423 else {
Chris Lattner4f031622004-11-15 05:03:30 +0000424 LatticeVal &BCValue = getValueState(BI->getCondition());
Chris Lattner71ac22ff2003-10-08 15:47:41 +0000425 if (BCValue.isOverdefined()) {
426 // Overdefined condition variables mean the branch could go either way.
427 return true;
428 } else if (BCValue.isConstant()) {
Chris Lattnerfe992d42004-01-12 17:40:36 +0000429 // Not branching on an evaluatable constant?
430 if (!isa<ConstantBool>(BCValue.getConstant())) return true;
431
Chris Lattner71ac22ff2003-10-08 15:47:41 +0000432 // Constant condition variables mean the branch can only go a single way
Misha Brukmanb1c93172005-04-21 23:48:37 +0000433 return BI->getSuccessor(BCValue.getConstant() ==
Chris Lattner71ac22ff2003-10-08 15:47:41 +0000434 ConstantBool::False) == To;
435 }
436 return false;
437 }
438 } else if (InvokeInst *II = dyn_cast<InvokeInst>(TI)) {
439 // Invoke instructions successors are always executable.
440 return true;
441 } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
Chris Lattner4f031622004-11-15 05:03:30 +0000442 LatticeVal &SCValue = getValueState(SI->getCondition());
Chris Lattner71ac22ff2003-10-08 15:47:41 +0000443 if (SCValue.isOverdefined()) { // Overdefined condition?
444 // All destinations are executable!
445 return true;
446 } else if (SCValue.isConstant()) {
447 Constant *CPV = SCValue.getConstant();
Chris Lattnerfe992d42004-01-12 17:40:36 +0000448 if (!isa<ConstantInt>(CPV))
449 return true; // not a foldable constant?
450
Chris Lattner71ac22ff2003-10-08 15:47:41 +0000451 // Make sure to skip the "default value" which isn't a value
452 for (unsigned i = 1, E = SI->getNumSuccessors(); i != E; ++i)
453 if (SI->getSuccessorValue(i) == CPV) // Found the taken branch...
454 return SI->getSuccessor(i) == To;
455
456 // Constant value not equal to any of the branches... must execute
457 // default branch then...
458 return SI->getDefaultDest() == To;
459 }
460 return false;
461 } else {
462 std::cerr << "Unknown terminator instruction: " << *TI;
463 abort();
464 }
Chris Lattner13b52e72002-05-02 21:18:01 +0000465}
Chris Lattner347389d2001-06-27 23:38:11 +0000466
Chris Lattner6e560792002-04-18 15:13:15 +0000467// visit Implementations - Something changed in this instruction... Either an
Chris Lattner347389d2001-06-27 23:38:11 +0000468// operand made a transition, or the instruction is newly executable. Change
469// the value type of I to reflect these changes if appropriate. This method
470// makes sure to do the following actions:
471//
472// 1. If a phi node merges two constants in, and has conflicting value coming
473// from different branches, or if the PHI node merges in an overdefined
474// value, then the PHI node becomes overdefined.
475// 2. If a phi node merges only constants in, and they all agree on value, the
476// PHI node becomes a constant value equal to that.
477// 3. If V <- x (op) y && isConstant(x) && isConstant(y) V = Constant
478// 4. If V <- x (op) y && (isOverdefined(x) || isOverdefined(y)) V = Overdefined
479// 5. If V <- MEM or V <- CALL or V <- (unknown) then V = Overdefined
480// 6. If a conditional branch has a value that is constant, make the selected
481// destination executable
482// 7. If a conditional branch has a value that is overdefined, make all
483// successors executable.
484//
Chris Lattner074be1f2004-11-15 04:44:20 +0000485void SCCPSolver::visitPHINode(PHINode &PN) {
Chris Lattner4f031622004-11-15 05:03:30 +0000486 LatticeVal &PNIV = getValueState(&PN);
Chris Lattner05fe6842004-01-12 03:57:30 +0000487 if (PNIV.isOverdefined()) {
488 // There may be instructions using this PHI node that are not overdefined
489 // themselves. If so, make sure that they know that the PHI node operand
490 // changed.
491 std::multimap<PHINode*, Instruction*>::iterator I, E;
492 tie(I, E) = UsersOfOverdefinedPHIs.equal_range(&PN);
493 if (I != E) {
494 std::vector<Instruction*> Users;
495 Users.reserve(std::distance(I, E));
496 for (; I != E; ++I) Users.push_back(I->second);
497 while (!Users.empty()) {
498 visit(Users.back());
499 Users.pop_back();
500 }
501 }
502 return; // Quick exit
503 }
Chris Lattner347389d2001-06-27 23:38:11 +0000504
Chris Lattner7a7b1142004-03-16 19:49:59 +0000505 // Super-extra-high-degree PHI nodes are unlikely to ever be marked constant,
506 // and slow us down a lot. Just mark them overdefined.
507 if (PN.getNumIncomingValues() > 64) {
508 markOverdefined(PNIV, &PN);
509 return;
510 }
511
Chris Lattner6e560792002-04-18 15:13:15 +0000512 // Look at all of the executable operands of the PHI node. If any of them
513 // are overdefined, the PHI becomes overdefined as well. If they are all
514 // constant, and they agree with each other, the PHI becomes the identical
515 // constant. If they are constant and don't agree, the PHI is overdefined.
516 // If there are no executable operands, the PHI remains undefined.
517 //
Chris Lattnercccc5c72003-04-25 02:50:03 +0000518 Constant *OperandVal = 0;
519 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) {
Chris Lattner4f031622004-11-15 05:03:30 +0000520 LatticeVal &IV = getValueState(PN.getIncomingValue(i));
Chris Lattnercccc5c72003-04-25 02:50:03 +0000521 if (IV.isUndefined()) continue; // Doesn't influence PHI node.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000522
Chris Lattner113f4f42002-06-25 16:13:24 +0000523 if (isEdgeFeasible(PN.getIncomingBlock(i), PN.getParent())) {
Chris Lattner7e270582003-06-24 20:29:52 +0000524 if (IV.isOverdefined()) { // PHI node becomes overdefined!
Chris Lattner7324f7c2003-10-08 16:21:03 +0000525 markOverdefined(PNIV, &PN);
Chris Lattner7e270582003-06-24 20:29:52 +0000526 return;
527 }
528
Chris Lattnercccc5c72003-04-25 02:50:03 +0000529 if (OperandVal == 0) { // Grab the first value...
530 OperandVal = IV.getConstant();
Chris Lattner6e560792002-04-18 15:13:15 +0000531 } else { // Another value is being merged in!
532 // There is already a reachable operand. If we conflict with it,
533 // then the PHI node becomes overdefined. If we agree with it, we
534 // can continue on.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000535
Chris Lattner6e560792002-04-18 15:13:15 +0000536 // Check to see if there are two different constants merging...
Chris Lattnercccc5c72003-04-25 02:50:03 +0000537 if (IV.getConstant() != OperandVal) {
Chris Lattner6e560792002-04-18 15:13:15 +0000538 // Yes there is. This means the PHI node is not constant.
539 // You must be overdefined poor PHI.
540 //
Chris Lattner7324f7c2003-10-08 16:21:03 +0000541 markOverdefined(PNIV, &PN); // The PHI node now becomes overdefined
Chris Lattner6e560792002-04-18 15:13:15 +0000542 return; // I'm done analyzing you
Chris Lattnerc4ad64c2001-11-26 18:57:38 +0000543 }
Chris Lattner347389d2001-06-27 23:38:11 +0000544 }
545 }
Chris Lattner347389d2001-06-27 23:38:11 +0000546 }
547
Chris Lattner6e560792002-04-18 15:13:15 +0000548 // If we exited the loop, this means that the PHI node only has constant
Chris Lattnercccc5c72003-04-25 02:50:03 +0000549 // arguments that agree with each other(and OperandVal is the constant) or
550 // OperandVal is null because there are no defined incoming arguments. If
551 // this is the case, the PHI remains undefined.
Chris Lattner347389d2001-06-27 23:38:11 +0000552 //
Chris Lattnercccc5c72003-04-25 02:50:03 +0000553 if (OperandVal)
Misha Brukman8b2bd4e2003-10-10 17:57:28 +0000554 markConstant(PNIV, &PN, OperandVal); // Acquire operand value
Chris Lattner347389d2001-06-27 23:38:11 +0000555}
556
Chris Lattnerb4394642004-12-10 08:02:06 +0000557void SCCPSolver::visitReturnInst(ReturnInst &I) {
558 if (I.getNumOperands() == 0) return; // Ret void
559
560 // If we are tracking the return value of this function, merge it in.
561 Function *F = I.getParent()->getParent();
562 if (F->hasInternalLinkage() && !TrackedFunctionRetVals.empty()) {
563 hash_map<Function*, LatticeVal>::iterator TFRVI =
564 TrackedFunctionRetVals.find(F);
565 if (TFRVI != TrackedFunctionRetVals.end() &&
566 !TFRVI->second.isOverdefined()) {
567 LatticeVal &IV = getValueState(I.getOperand(0));
568 mergeInValue(TFRVI->second, F, IV);
569 }
570 }
571}
572
573
Chris Lattner074be1f2004-11-15 04:44:20 +0000574void SCCPSolver::visitTerminatorInst(TerminatorInst &TI) {
Chris Lattnercccc5c72003-04-25 02:50:03 +0000575 std::vector<bool> SuccFeasible;
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000576 getFeasibleSuccessors(TI, SuccFeasible);
Chris Lattner347389d2001-06-27 23:38:11 +0000577
Chris Lattner0bbbe5d2003-10-08 16:55:34 +0000578 BasicBlock *BB = TI.getParent();
579
Chris Lattnerfe6c9ee2002-05-02 21:44:00 +0000580 // Mark all feasible successors executable...
581 for (unsigned i = 0, e = SuccFeasible.size(); i != e; ++i)
Chris Lattner0bbbe5d2003-10-08 16:55:34 +0000582 if (SuccFeasible[i])
583 markEdgeExecutable(BB, TI.getSuccessor(i));
Chris Lattner6e560792002-04-18 15:13:15 +0000584}
585
Chris Lattner074be1f2004-11-15 04:44:20 +0000586void SCCPSolver::visitCastInst(CastInst &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000587 Value *V = I.getOperand(0);
Chris Lattner4f031622004-11-15 05:03:30 +0000588 LatticeVal &VState = getValueState(V);
Chris Lattner0fe5b322004-01-12 17:43:40 +0000589 if (VState.isOverdefined()) // Inherit overdefinedness of operand
Chris Lattner113f4f42002-06-25 16:13:24 +0000590 markOverdefined(&I);
Chris Lattner0fe5b322004-01-12 17:43:40 +0000591 else if (VState.isConstant()) // Propagate constant value
592 markConstant(&I, ConstantExpr::getCast(VState.getConstant(), I.getType()));
Chris Lattner6e560792002-04-18 15:13:15 +0000593}
594
Chris Lattner074be1f2004-11-15 04:44:20 +0000595void SCCPSolver::visitSelectInst(SelectInst &I) {
Chris Lattner4f031622004-11-15 05:03:30 +0000596 LatticeVal &CondValue = getValueState(I.getCondition());
Chris Lattner06a0ed12006-02-08 02:38:11 +0000597 if (CondValue.isUndefined())
598 return;
599 if (CondValue.isConstant()) {
600 Value *InVal = 0;
Chris Lattner59db22d2004-03-12 05:52:44 +0000601 if (CondValue.getConstant() == ConstantBool::True) {
Chris Lattner06a0ed12006-02-08 02:38:11 +0000602 mergeInValue(&I, getValueState(I.getTrueValue()));
603 return;
Chris Lattner59db22d2004-03-12 05:52:44 +0000604 } else if (CondValue.getConstant() == ConstantBool::False) {
Chris Lattner06a0ed12006-02-08 02:38:11 +0000605 mergeInValue(&I, getValueState(I.getFalseValue()));
606 return;
607 }
608 }
609
610 // Otherwise, the condition is overdefined or a constant we can't evaluate.
611 // See if we can produce something better than overdefined based on the T/F
612 // value.
613 LatticeVal &TVal = getValueState(I.getTrueValue());
614 LatticeVal &FVal = getValueState(I.getFalseValue());
615
616 // select ?, C, C -> C.
617 if (TVal.isConstant() && FVal.isConstant() &&
618 TVal.getConstant() == FVal.getConstant()) {
619 markConstant(&I, FVal.getConstant());
620 return;
621 }
622
623 if (TVal.isUndefined()) { // select ?, undef, X -> X.
624 mergeInValue(&I, FVal);
625 } else if (FVal.isUndefined()) { // select ?, X, undef -> X.
626 mergeInValue(&I, TVal);
627 } else {
628 markOverdefined(&I);
Chris Lattner59db22d2004-03-12 05:52:44 +0000629 }
630}
631
Chris Lattner6e560792002-04-18 15:13:15 +0000632// Handle BinaryOperators and Shift Instructions...
Chris Lattner074be1f2004-11-15 04:44:20 +0000633void SCCPSolver::visitBinaryOperator(Instruction &I) {
Chris Lattner4f031622004-11-15 05:03:30 +0000634 LatticeVal &IV = ValueState[&I];
Chris Lattner05fe6842004-01-12 03:57:30 +0000635 if (IV.isOverdefined()) return;
636
Chris Lattner4f031622004-11-15 05:03:30 +0000637 LatticeVal &V1State = getValueState(I.getOperand(0));
638 LatticeVal &V2State = getValueState(I.getOperand(1));
Chris Lattner05fe6842004-01-12 03:57:30 +0000639
Chris Lattner6e560792002-04-18 15:13:15 +0000640 if (V1State.isOverdefined() || V2State.isOverdefined()) {
Chris Lattnercbc01612004-12-11 23:15:19 +0000641 // If this is an AND or OR with 0 or -1, it doesn't matter that the other
642 // operand is overdefined.
643 if (I.getOpcode() == Instruction::And || I.getOpcode() == Instruction::Or) {
644 LatticeVal *NonOverdefVal = 0;
645 if (!V1State.isOverdefined()) {
646 NonOverdefVal = &V1State;
647 } else if (!V2State.isOverdefined()) {
648 NonOverdefVal = &V2State;
649 }
650
651 if (NonOverdefVal) {
652 if (NonOverdefVal->isUndefined()) {
653 // Could annihilate value.
654 if (I.getOpcode() == Instruction::And)
655 markConstant(IV, &I, Constant::getNullValue(I.getType()));
656 else
657 markConstant(IV, &I, ConstantInt::getAllOnesValue(I.getType()));
658 return;
659 } else {
660 if (I.getOpcode() == Instruction::And) {
661 if (NonOverdefVal->getConstant()->isNullValue()) {
662 markConstant(IV, &I, NonOverdefVal->getConstant());
663 return; // X or 0 = -1
664 }
665 } else {
666 if (ConstantIntegral *CI =
667 dyn_cast<ConstantIntegral>(NonOverdefVal->getConstant()))
668 if (CI->isAllOnesValue()) {
669 markConstant(IV, &I, NonOverdefVal->getConstant());
670 return; // X or -1 = -1
671 }
672 }
673 }
674 }
675 }
676
677
Chris Lattner05fe6842004-01-12 03:57:30 +0000678 // If both operands are PHI nodes, it is possible that this instruction has
679 // a constant value, despite the fact that the PHI node doesn't. Check for
680 // this condition now.
681 if (PHINode *PN1 = dyn_cast<PHINode>(I.getOperand(0)))
682 if (PHINode *PN2 = dyn_cast<PHINode>(I.getOperand(1)))
683 if (PN1->getParent() == PN2->getParent()) {
684 // Since the two PHI nodes are in the same basic block, they must have
685 // entries for the same predecessors. Walk the predecessor list, and
686 // if all of the incoming values are constants, and the result of
687 // evaluating this expression with all incoming value pairs is the
688 // same, then this expression is a constant even though the PHI node
689 // is not a constant!
Chris Lattner4f031622004-11-15 05:03:30 +0000690 LatticeVal Result;
Chris Lattner05fe6842004-01-12 03:57:30 +0000691 for (unsigned i = 0, e = PN1->getNumIncomingValues(); i != e; ++i) {
Chris Lattner4f031622004-11-15 05:03:30 +0000692 LatticeVal &In1 = getValueState(PN1->getIncomingValue(i));
Chris Lattner05fe6842004-01-12 03:57:30 +0000693 BasicBlock *InBlock = PN1->getIncomingBlock(i);
Chris Lattner4f031622004-11-15 05:03:30 +0000694 LatticeVal &In2 =
695 getValueState(PN2->getIncomingValueForBlock(InBlock));
Chris Lattner05fe6842004-01-12 03:57:30 +0000696
697 if (In1.isOverdefined() || In2.isOverdefined()) {
698 Result.markOverdefined();
699 break; // Cannot fold this operation over the PHI nodes!
700 } else if (In1.isConstant() && In2.isConstant()) {
Chris Lattner1b7d4d72004-01-12 19:08:43 +0000701 Constant *V = ConstantExpr::get(I.getOpcode(), In1.getConstant(),
702 In2.getConstant());
Chris Lattner05fe6842004-01-12 03:57:30 +0000703 if (Result.isUndefined())
Chris Lattner1b7d4d72004-01-12 19:08:43 +0000704 Result.markConstant(V);
705 else if (Result.isConstant() && Result.getConstant() != V) {
Chris Lattner05fe6842004-01-12 03:57:30 +0000706 Result.markOverdefined();
707 break;
708 }
709 }
710 }
711
712 // If we found a constant value here, then we know the instruction is
713 // constant despite the fact that the PHI nodes are overdefined.
714 if (Result.isConstant()) {
715 markConstant(IV, &I, Result.getConstant());
716 // Remember that this instruction is virtually using the PHI node
717 // operands.
718 UsersOfOverdefinedPHIs.insert(std::make_pair(PN1, &I));
719 UsersOfOverdefinedPHIs.insert(std::make_pair(PN2, &I));
720 return;
721 } else if (Result.isUndefined()) {
722 return;
723 }
724
725 // Okay, this really is overdefined now. Since we might have
726 // speculatively thought that this was not overdefined before, and
727 // added ourselves to the UsersOfOverdefinedPHIs list for the PHIs,
728 // make sure to clean out any entries that we put there, for
729 // efficiency.
730 std::multimap<PHINode*, Instruction*>::iterator It, E;
731 tie(It, E) = UsersOfOverdefinedPHIs.equal_range(PN1);
732 while (It != E) {
733 if (It->second == &I) {
734 UsersOfOverdefinedPHIs.erase(It++);
735 } else
736 ++It;
737 }
738 tie(It, E) = UsersOfOverdefinedPHIs.equal_range(PN2);
739 while (It != E) {
740 if (It->second == &I) {
741 UsersOfOverdefinedPHIs.erase(It++);
742 } else
743 ++It;
744 }
745 }
746
747 markOverdefined(IV, &I);
Chris Lattner6e560792002-04-18 15:13:15 +0000748 } else if (V1State.isConstant() && V2State.isConstant()) {
Chris Lattner1b7d4d72004-01-12 19:08:43 +0000749 markConstant(IV, &I, ConstantExpr::get(I.getOpcode(), V1State.getConstant(),
750 V2State.getConstant()));
Chris Lattner6e560792002-04-18 15:13:15 +0000751 }
752}
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000753
Robert Bocchinobd518d12006-01-10 19:05:05 +0000754void SCCPSolver::visitExtractElementInst(ExtractElementInst &I) {
755 LatticeVal &ValState = getValueState(I.getOperand(0));
756 LatticeVal &IdxState = getValueState(I.getOperand(1));
757
758 if (ValState.isOverdefined() || IdxState.isOverdefined())
759 markOverdefined(&I);
760 else if(ValState.isConstant() && IdxState.isConstant())
761 markConstant(&I, ConstantExpr::getExtractElement(ValState.getConstant(),
762 IdxState.getConstant()));
763}
764
Robert Bocchino6dce2502006-01-17 20:06:55 +0000765void SCCPSolver::visitInsertElementInst(InsertElementInst &I) {
766 LatticeVal &ValState = getValueState(I.getOperand(0));
767 LatticeVal &EltState = getValueState(I.getOperand(1));
768 LatticeVal &IdxState = getValueState(I.getOperand(2));
769
770 if (ValState.isOverdefined() || EltState.isOverdefined() ||
771 IdxState.isOverdefined())
772 markOverdefined(&I);
773 else if(ValState.isConstant() && EltState.isConstant() &&
774 IdxState.isConstant())
775 markConstant(&I, ConstantExpr::getInsertElement(ValState.getConstant(),
776 EltState.getConstant(),
777 IdxState.getConstant()));
778 else if (ValState.isUndefined() && EltState.isConstant() &&
779 IdxState.isConstant())
780 markConstant(&I, ConstantExpr::getInsertElement(UndefValue::get(I.getType()),
781 EltState.getConstant(),
782 IdxState.getConstant()));
783}
784
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000785// Handle getelementptr instructions... if all operands are constants then we
786// can turn this into a getelementptr ConstantExpr.
787//
Chris Lattner074be1f2004-11-15 04:44:20 +0000788void SCCPSolver::visitGetElementPtrInst(GetElementPtrInst &I) {
Chris Lattner4f031622004-11-15 05:03:30 +0000789 LatticeVal &IV = ValueState[&I];
Chris Lattner49f74522004-01-12 04:29:41 +0000790 if (IV.isOverdefined()) return;
791
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000792 std::vector<Constant*> Operands;
793 Operands.reserve(I.getNumOperands());
794
795 for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) {
Chris Lattner4f031622004-11-15 05:03:30 +0000796 LatticeVal &State = getValueState(I.getOperand(i));
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000797 if (State.isUndefined())
798 return; // Operands are not resolved yet...
799 else if (State.isOverdefined()) {
Chris Lattner49f74522004-01-12 04:29:41 +0000800 markOverdefined(IV, &I);
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000801 return;
802 }
803 assert(State.isConstant() && "Unknown state!");
804 Operands.push_back(State.getConstant());
805 }
806
807 Constant *Ptr = Operands[0];
808 Operands.erase(Operands.begin()); // Erase the pointer from idx list...
809
Misha Brukmanb1c93172005-04-21 23:48:37 +0000810 markConstant(IV, &I, ConstantExpr::getGetElementPtr(Ptr, Operands));
Chris Lattnerdd6522e2002-08-30 23:39:00 +0000811}
Brian Gaeke960707c2003-11-11 22:41:34 +0000812
Chris Lattner91dbae62004-12-11 05:15:59 +0000813void SCCPSolver::visitStoreInst(Instruction &SI) {
814 if (TrackedGlobals.empty() || !isa<GlobalVariable>(SI.getOperand(1)))
815 return;
816 GlobalVariable *GV = cast<GlobalVariable>(SI.getOperand(1));
817 hash_map<GlobalVariable*, LatticeVal>::iterator I = TrackedGlobals.find(GV);
818 if (I == TrackedGlobals.end() || I->second.isOverdefined()) return;
819
820 // Get the value we are storing into the global.
821 LatticeVal &PtrVal = getValueState(SI.getOperand(0));
822
823 mergeInValue(I->second, GV, PtrVal);
824 if (I->second.isOverdefined())
825 TrackedGlobals.erase(I); // No need to keep tracking this!
826}
827
828
Chris Lattner49f74522004-01-12 04:29:41 +0000829// Handle load instructions. If the operand is a constant pointer to a constant
830// global, we can replace the load with the loaded constant value!
Chris Lattner074be1f2004-11-15 04:44:20 +0000831void SCCPSolver::visitLoadInst(LoadInst &I) {
Chris Lattner4f031622004-11-15 05:03:30 +0000832 LatticeVal &IV = ValueState[&I];
Chris Lattner49f74522004-01-12 04:29:41 +0000833 if (IV.isOverdefined()) return;
834
Chris Lattner4f031622004-11-15 05:03:30 +0000835 LatticeVal &PtrVal = getValueState(I.getOperand(0));
Chris Lattner49f74522004-01-12 04:29:41 +0000836 if (PtrVal.isUndefined()) return; // The pointer is not resolved yet!
837 if (PtrVal.isConstant() && !I.isVolatile()) {
838 Value *Ptr = PtrVal.getConstant();
Chris Lattner538fee72004-03-07 22:16:24 +0000839 if (isa<ConstantPointerNull>(Ptr)) {
840 // load null -> null
841 markConstant(IV, &I, Constant::getNullValue(I.getType()));
842 return;
843 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000844
Chris Lattner49f74522004-01-12 04:29:41 +0000845 // Transform load (constant global) into the value loaded.
Chris Lattner91dbae62004-12-11 05:15:59 +0000846 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Ptr)) {
847 if (GV->isConstant()) {
848 if (!GV->isExternal()) {
849 markConstant(IV, &I, GV->getInitializer());
850 return;
851 }
852 } else if (!TrackedGlobals.empty()) {
853 // If we are tracking this global, merge in the known value for it.
854 hash_map<GlobalVariable*, LatticeVal>::iterator It =
855 TrackedGlobals.find(GV);
856 if (It != TrackedGlobals.end()) {
857 mergeInValue(IV, &I, It->second);
858 return;
859 }
Chris Lattner49f74522004-01-12 04:29:41 +0000860 }
Chris Lattner91dbae62004-12-11 05:15:59 +0000861 }
Chris Lattner49f74522004-01-12 04:29:41 +0000862
863 // Transform load (constantexpr_GEP global, 0, ...) into the value loaded.
864 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr))
865 if (CE->getOpcode() == Instruction::GetElementPtr)
Jeff Cohen82639852005-04-23 21:38:35 +0000866 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0)))
867 if (GV->isConstant() && !GV->isExternal())
868 if (Constant *V =
Chris Lattner02ae21e2005-09-26 05:28:52 +0000869 ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE)) {
Jeff Cohen82639852005-04-23 21:38:35 +0000870 markConstant(IV, &I, V);
871 return;
872 }
Chris Lattner49f74522004-01-12 04:29:41 +0000873 }
874
875 // Otherwise we cannot say for certain what value this load will produce.
876 // Bail out.
877 markOverdefined(IV, &I);
878}
Chris Lattnerff9362a2004-04-13 19:43:54 +0000879
Chris Lattnerb4394642004-12-10 08:02:06 +0000880void SCCPSolver::visitCallSite(CallSite CS) {
881 Function *F = CS.getCalledFunction();
882
883 // If we are tracking this function, we must make sure to bind arguments as
884 // appropriate.
885 hash_map<Function*, LatticeVal>::iterator TFRVI =TrackedFunctionRetVals.end();
886 if (F && F->hasInternalLinkage())
887 TFRVI = TrackedFunctionRetVals.find(F);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000888
Chris Lattnerb4394642004-12-10 08:02:06 +0000889 if (TFRVI != TrackedFunctionRetVals.end()) {
890 // If this is the first call to the function hit, mark its entry block
891 // executable.
892 if (!BBExecutable.count(F->begin()))
893 MarkBlockExecutable(F->begin());
894
895 CallSite::arg_iterator CAI = CS.arg_begin();
Chris Lattner531f9e92005-03-15 04:54:21 +0000896 for (Function::arg_iterator AI = F->arg_begin(), E = F->arg_end();
Chris Lattnerb4394642004-12-10 08:02:06 +0000897 AI != E; ++AI, ++CAI) {
898 LatticeVal &IV = ValueState[AI];
899 if (!IV.isOverdefined())
900 mergeInValue(IV, AI, getValueState(*CAI));
901 }
902 }
903 Instruction *I = CS.getInstruction();
904 if (I->getType() == Type::VoidTy) return;
905
906 LatticeVal &IV = ValueState[I];
Chris Lattnerff9362a2004-04-13 19:43:54 +0000907 if (IV.isOverdefined()) return;
908
Chris Lattnerb4394642004-12-10 08:02:06 +0000909 // Propagate the return value of the function to the value of the instruction.
910 if (TFRVI != TrackedFunctionRetVals.end()) {
911 mergeInValue(IV, I, TFRVI->second);
912 return;
913 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000914
Chris Lattnerb4394642004-12-10 08:02:06 +0000915 if (F == 0 || !F->isExternal() || !canConstantFoldCallTo(F)) {
916 markOverdefined(IV, I);
Chris Lattnerff9362a2004-04-13 19:43:54 +0000917 return;
918 }
919
920 std::vector<Constant*> Operands;
Chris Lattnerb4394642004-12-10 08:02:06 +0000921 Operands.reserve(I->getNumOperands()-1);
Chris Lattnerff9362a2004-04-13 19:43:54 +0000922
Chris Lattnerb4394642004-12-10 08:02:06 +0000923 for (CallSite::arg_iterator AI = CS.arg_begin(), E = CS.arg_end();
924 AI != E; ++AI) {
925 LatticeVal &State = getValueState(*AI);
Chris Lattnerff9362a2004-04-13 19:43:54 +0000926 if (State.isUndefined())
927 return; // Operands are not resolved yet...
928 else if (State.isOverdefined()) {
Chris Lattnerb4394642004-12-10 08:02:06 +0000929 markOverdefined(IV, I);
Chris Lattnerff9362a2004-04-13 19:43:54 +0000930 return;
931 }
932 assert(State.isConstant() && "Unknown state!");
933 Operands.push_back(State.getConstant());
934 }
935
936 if (Constant *C = ConstantFoldCall(F, Operands))
Chris Lattnerb4394642004-12-10 08:02:06 +0000937 markConstant(IV, I, C);
Chris Lattnerff9362a2004-04-13 19:43:54 +0000938 else
Chris Lattnerb4394642004-12-10 08:02:06 +0000939 markOverdefined(IV, I);
Chris Lattnerff9362a2004-04-13 19:43:54 +0000940}
Chris Lattner074be1f2004-11-15 04:44:20 +0000941
942
943void SCCPSolver::Solve() {
944 // Process the work lists until they are empty!
Misha Brukmanb1c93172005-04-21 23:48:37 +0000945 while (!BBWorkList.empty() || !InstWorkList.empty() ||
Jeff Cohen82639852005-04-23 21:38:35 +0000946 !OverdefinedInstWorkList.empty()) {
Chris Lattner074be1f2004-11-15 04:44:20 +0000947 // Process the instruction work list...
948 while (!OverdefinedInstWorkList.empty()) {
Chris Lattnerb4394642004-12-10 08:02:06 +0000949 Value *I = OverdefinedInstWorkList.back();
Chris Lattner074be1f2004-11-15 04:44:20 +0000950 OverdefinedInstWorkList.pop_back();
951
Chris Lattnerb4394642004-12-10 08:02:06 +0000952 DEBUG(std::cerr << "\nPopped off OI-WL: " << *I);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000953
Chris Lattner074be1f2004-11-15 04:44:20 +0000954 // "I" got into the work list because it either made the transition from
955 // bottom to constant
956 //
957 // Anything on this worklist that is overdefined need not be visited
958 // since all of its users will have already been marked as overdefined
959 // Update all of the users of this instruction's value...
960 //
961 for (Value::use_iterator UI = I->use_begin(), E = I->use_end();
962 UI != E; ++UI)
963 OperandChangedState(*UI);
964 }
965 // Process the instruction work list...
966 while (!InstWorkList.empty()) {
Chris Lattnerb4394642004-12-10 08:02:06 +0000967 Value *I = InstWorkList.back();
Chris Lattner074be1f2004-11-15 04:44:20 +0000968 InstWorkList.pop_back();
969
970 DEBUG(std::cerr << "\nPopped off I-WL: " << *I);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000971
Chris Lattner074be1f2004-11-15 04:44:20 +0000972 // "I" got into the work list because it either made the transition from
973 // bottom to constant
974 //
975 // Anything on this worklist that is overdefined need not be visited
976 // since all of its users will have already been marked as overdefined.
977 // Update all of the users of this instruction's value...
978 //
979 if (!getValueState(I).isOverdefined())
980 for (Value::use_iterator UI = I->use_begin(), E = I->use_end();
981 UI != E; ++UI)
982 OperandChangedState(*UI);
983 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000984
Chris Lattner074be1f2004-11-15 04:44:20 +0000985 // Process the basic block work list...
986 while (!BBWorkList.empty()) {
987 BasicBlock *BB = BBWorkList.back();
988 BBWorkList.pop_back();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000989
Chris Lattner074be1f2004-11-15 04:44:20 +0000990 DEBUG(std::cerr << "\nPopped off BBWL: " << *BB);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000991
Chris Lattner074be1f2004-11-15 04:44:20 +0000992 // Notify all instructions in this basic block that they are newly
993 // executable.
994 visit(BB);
995 }
996 }
997}
998
Chris Lattner7285f432004-12-10 20:41:50 +0000999/// ResolveBranchesIn - While solving the dataflow for a function, we assume
1000/// that branches on undef values cannot reach any of their successors.
1001/// However, this is not a safe assumption. After we solve dataflow, this
1002/// method should be use to handle this. If this returns true, the solver
1003/// should be rerun.
1004bool SCCPSolver::ResolveBranchesIn(Function &F) {
1005 bool BranchesResolved = false;
Chris Lattner2f687fd2004-12-11 06:05:53 +00001006 for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB)
1007 if (BBExecutable.count(BB)) {
1008 TerminatorInst *TI = BB->getTerminator();
1009 if (BranchInst *BI = dyn_cast<BranchInst>(TI)) {
1010 if (BI->isConditional()) {
1011 LatticeVal &BCValue = getValueState(BI->getCondition());
1012 if (BCValue.isUndefined()) {
1013 BI->setCondition(ConstantBool::True);
1014 BranchesResolved = true;
1015 visit(BI);
1016 }
1017 }
1018 } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
1019 LatticeVal &SCValue = getValueState(SI->getCondition());
1020 if (SCValue.isUndefined()) {
1021 const Type *CondTy = SI->getCondition()->getType();
1022 SI->setCondition(Constant::getNullValue(CondTy));
Chris Lattner7285f432004-12-10 20:41:50 +00001023 BranchesResolved = true;
Chris Lattner2f687fd2004-12-11 06:05:53 +00001024 visit(SI);
Chris Lattner7285f432004-12-10 20:41:50 +00001025 }
1026 }
Chris Lattner7285f432004-12-10 20:41:50 +00001027 }
Chris Lattner2f687fd2004-12-11 06:05:53 +00001028
Chris Lattner7285f432004-12-10 20:41:50 +00001029 return BranchesResolved;
1030}
1031
Chris Lattner074be1f2004-11-15 04:44:20 +00001032
1033namespace {
Chris Lattnerb4394642004-12-10 08:02:06 +00001034 Statistic<> NumInstRemoved("sccp", "Number of instructions removed");
1035 Statistic<> NumDeadBlocks ("sccp", "Number of basic blocks unreachable");
1036
Chris Lattner1890f942004-11-15 07:15:04 +00001037 //===--------------------------------------------------------------------===//
Chris Lattner074be1f2004-11-15 04:44:20 +00001038 //
Chris Lattner1890f942004-11-15 07:15:04 +00001039 /// SCCP Class - This class uses the SCCPSolver to implement a per-function
1040 /// Sparse Conditional COnstant Propagator.
1041 ///
1042 struct SCCP : public FunctionPass {
1043 // runOnFunction - Run the Sparse Conditional Constant Propagation
1044 // algorithm, and return true if the function was modified.
1045 //
1046 bool runOnFunction(Function &F);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001047
Chris Lattner1890f942004-11-15 07:15:04 +00001048 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
1049 AU.setPreservesCFG();
1050 }
1051 };
Chris Lattner074be1f2004-11-15 04:44:20 +00001052
1053 RegisterOpt<SCCP> X("sccp", "Sparse Conditional Constant Propagation");
1054} // end anonymous namespace
1055
1056
1057// createSCCPPass - This is the public interface to this file...
1058FunctionPass *llvm::createSCCPPass() {
1059 return new SCCP();
1060}
1061
1062
Chris Lattner074be1f2004-11-15 04:44:20 +00001063// runOnFunction() - Run the Sparse Conditional Constant Propagation algorithm,
1064// and return true if the function was modified.
1065//
1066bool SCCP::runOnFunction(Function &F) {
Chris Lattnerd18c16b2004-11-15 05:45:33 +00001067 DEBUG(std::cerr << "SCCP on function '" << F.getName() << "'\n");
Chris Lattner074be1f2004-11-15 04:44:20 +00001068 SCCPSolver Solver;
1069
1070 // Mark the first block of the function as being executable.
1071 Solver.MarkBlockExecutable(F.begin());
1072
Chris Lattnerd18c16b2004-11-15 05:45:33 +00001073 // Mark all arguments to the function as being overdefined.
1074 hash_map<Value*, LatticeVal> &Values = Solver.getValueMapping();
Chris Lattner531f9e92005-03-15 04:54:21 +00001075 for (Function::arg_iterator AI = F.arg_begin(), E = F.arg_end(); AI != E; ++AI)
Chris Lattnerd18c16b2004-11-15 05:45:33 +00001076 Values[AI].markOverdefined();
1077
Chris Lattner074be1f2004-11-15 04:44:20 +00001078 // Solve for constants.
Chris Lattner7285f432004-12-10 20:41:50 +00001079 bool ResolvedBranches = true;
1080 while (ResolvedBranches) {
1081 Solver.Solve();
Chris Lattner2f687fd2004-12-11 06:05:53 +00001082 DEBUG(std::cerr << "RESOLVING UNDEF BRANCHES\n");
Chris Lattner7285f432004-12-10 20:41:50 +00001083 ResolvedBranches = Solver.ResolveBranchesIn(F);
1084 }
Chris Lattner074be1f2004-11-15 04:44:20 +00001085
Chris Lattnerd18c16b2004-11-15 05:45:33 +00001086 bool MadeChanges = false;
1087
1088 // If we decided that there are basic blocks that are dead in this function,
1089 // delete their contents now. Note that we cannot actually delete the blocks,
1090 // as we cannot modify the CFG of the function.
1091 //
1092 std::set<BasicBlock*> &ExecutableBBs = Solver.getExecutableBlocks();
1093 for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB)
1094 if (!ExecutableBBs.count(BB)) {
1095 DEBUG(std::cerr << " BasicBlock Dead:" << *BB);
Chris Lattner9a038a32004-11-15 07:02:42 +00001096 ++NumDeadBlocks;
1097
Chris Lattnerd18c16b2004-11-15 05:45:33 +00001098 // Delete the instructions backwards, as it has a reduced likelihood of
1099 // having to update as many def-use and use-def chains.
1100 std::vector<Instruction*> Insts;
1101 for (BasicBlock::iterator I = BB->begin(), E = BB->getTerminator();
1102 I != E; ++I)
1103 Insts.push_back(I);
1104 while (!Insts.empty()) {
1105 Instruction *I = Insts.back();
1106 Insts.pop_back();
1107 if (!I->use_empty())
1108 I->replaceAllUsesWith(UndefValue::get(I->getType()));
1109 BB->getInstList().erase(I);
1110 MadeChanges = true;
Chris Lattner9a038a32004-11-15 07:02:42 +00001111 ++NumInstRemoved;
Chris Lattnerd18c16b2004-11-15 05:45:33 +00001112 }
Chris Lattnerb4394642004-12-10 08:02:06 +00001113 } else {
1114 // Iterate over all of the instructions in a function, replacing them with
1115 // constants if we have found them to be of constant values.
1116 //
1117 for (BasicBlock::iterator BI = BB->begin(), E = BB->end(); BI != E; ) {
1118 Instruction *Inst = BI++;
1119 if (Inst->getType() != Type::VoidTy) {
1120 LatticeVal &IV = Values[Inst];
1121 if (IV.isConstant() || IV.isUndefined() &&
1122 !isa<TerminatorInst>(Inst)) {
1123 Constant *Const = IV.isConstant()
1124 ? IV.getConstant() : UndefValue::get(Inst->getType());
Chris Lattner074be1f2004-11-15 04:44:20 +00001125 DEBUG(std::cerr << " Constant: " << *Const << " = " << *Inst);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001126
Chris Lattnerb4394642004-12-10 08:02:06 +00001127 // Replaces all of the uses of a variable with uses of the constant.
1128 Inst->replaceAllUsesWith(Const);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001129
Chris Lattnerb4394642004-12-10 08:02:06 +00001130 // Delete the instruction.
1131 BB->getInstList().erase(Inst);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001132
Chris Lattnerb4394642004-12-10 08:02:06 +00001133 // Hey, we just changed something!
1134 MadeChanges = true;
1135 ++NumInstRemoved;
Chris Lattner074be1f2004-11-15 04:44:20 +00001136 }
Chris Lattner074be1f2004-11-15 04:44:20 +00001137 }
1138 }
1139 }
1140
1141 return MadeChanges;
1142}
Chris Lattnerb4394642004-12-10 08:02:06 +00001143
1144namespace {
1145 Statistic<> IPNumInstRemoved("ipsccp", "Number of instructions removed");
1146 Statistic<> IPNumDeadBlocks ("ipsccp", "Number of basic blocks unreachable");
1147 Statistic<> IPNumArgsElimed ("ipsccp",
1148 "Number of arguments constant propagated");
Chris Lattner91dbae62004-12-11 05:15:59 +00001149 Statistic<> IPNumGlobalConst("ipsccp",
1150 "Number of globals found to be constant");
Chris Lattnerb4394642004-12-10 08:02:06 +00001151
1152 //===--------------------------------------------------------------------===//
1153 //
1154 /// IPSCCP Class - This class implements interprocedural Sparse Conditional
1155 /// Constant Propagation.
1156 ///
1157 struct IPSCCP : public ModulePass {
1158 bool runOnModule(Module &M);
1159 };
1160
1161 RegisterOpt<IPSCCP>
1162 Y("ipsccp", "Interprocedural Sparse Conditional Constant Propagation");
1163} // end anonymous namespace
1164
1165// createIPSCCPPass - This is the public interface to this file...
1166ModulePass *llvm::createIPSCCPPass() {
1167 return new IPSCCP();
1168}
1169
1170
1171static bool AddressIsTaken(GlobalValue *GV) {
Chris Lattner8cb10a12005-04-19 19:16:19 +00001172 // Delete any dead constantexpr klingons.
1173 GV->removeDeadConstantUsers();
1174
Chris Lattnerb4394642004-12-10 08:02:06 +00001175 for (Value::use_iterator UI = GV->use_begin(), E = GV->use_end();
1176 UI != E; ++UI)
1177 if (StoreInst *SI = dyn_cast<StoreInst>(*UI)) {
Chris Lattner91dbae62004-12-11 05:15:59 +00001178 if (SI->getOperand(0) == GV || SI->isVolatile())
1179 return true; // Storing addr of GV.
Chris Lattnerb4394642004-12-10 08:02:06 +00001180 } else if (isa<InvokeInst>(*UI) || isa<CallInst>(*UI)) {
1181 // Make sure we are calling the function, not passing the address.
1182 CallSite CS = CallSite::get(cast<Instruction>(*UI));
1183 for (CallSite::arg_iterator AI = CS.arg_begin(),
1184 E = CS.arg_end(); AI != E; ++AI)
1185 if (*AI == GV)
1186 return true;
Chris Lattner91dbae62004-12-11 05:15:59 +00001187 } else if (LoadInst *LI = dyn_cast<LoadInst>(*UI)) {
1188 if (LI->isVolatile())
1189 return true;
1190 } else {
Chris Lattnerb4394642004-12-10 08:02:06 +00001191 return true;
1192 }
1193 return false;
1194}
1195
1196bool IPSCCP::runOnModule(Module &M) {
1197 SCCPSolver Solver;
1198
1199 // Loop over all functions, marking arguments to those with their addresses
1200 // taken or that are external as overdefined.
1201 //
1202 hash_map<Value*, LatticeVal> &Values = Solver.getValueMapping();
1203 for (Module::iterator F = M.begin(), E = M.end(); F != E; ++F)
1204 if (!F->hasInternalLinkage() || AddressIsTaken(F)) {
1205 if (!F->isExternal())
1206 Solver.MarkBlockExecutable(F->begin());
Chris Lattner8cb10a12005-04-19 19:16:19 +00001207 for (Function::arg_iterator AI = F->arg_begin(), E = F->arg_end();
1208 AI != E; ++AI)
Chris Lattnerb4394642004-12-10 08:02:06 +00001209 Values[AI].markOverdefined();
1210 } else {
1211 Solver.AddTrackedFunction(F);
1212 }
1213
Chris Lattner91dbae62004-12-11 05:15:59 +00001214 // Loop over global variables. We inform the solver about any internal global
1215 // variables that do not have their 'addresses taken'. If they don't have
1216 // their addresses taken, we can propagate constants through them.
Chris Lattner8cb10a12005-04-19 19:16:19 +00001217 for (Module::global_iterator G = M.global_begin(), E = M.global_end();
1218 G != E; ++G)
Chris Lattner91dbae62004-12-11 05:15:59 +00001219 if (!G->isConstant() && G->hasInternalLinkage() && !AddressIsTaken(G))
1220 Solver.TrackValueOfGlobalVariable(G);
1221
Chris Lattnerb4394642004-12-10 08:02:06 +00001222 // Solve for constants.
Chris Lattner7285f432004-12-10 20:41:50 +00001223 bool ResolvedBranches = true;
1224 while (ResolvedBranches) {
1225 Solver.Solve();
1226
Chris Lattner2f687fd2004-12-11 06:05:53 +00001227 DEBUG(std::cerr << "RESOLVING UNDEF BRANCHES\n");
Chris Lattner7285f432004-12-10 20:41:50 +00001228 ResolvedBranches = false;
1229 for (Module::iterator F = M.begin(), E = M.end(); F != E; ++F)
1230 ResolvedBranches |= Solver.ResolveBranchesIn(*F);
1231 }
Chris Lattnerb4394642004-12-10 08:02:06 +00001232
1233 bool MadeChanges = false;
1234
1235 // Iterate over all of the instructions in the module, replacing them with
1236 // constants if we have found them to be of constant values.
1237 //
1238 std::set<BasicBlock*> &ExecutableBBs = Solver.getExecutableBlocks();
1239 for (Module::iterator F = M.begin(), E = M.end(); F != E; ++F) {
Chris Lattner8cb10a12005-04-19 19:16:19 +00001240 for (Function::arg_iterator AI = F->arg_begin(), E = F->arg_end();
1241 AI != E; ++AI)
Chris Lattnerb4394642004-12-10 08:02:06 +00001242 if (!AI->use_empty()) {
1243 LatticeVal &IV = Values[AI];
1244 if (IV.isConstant() || IV.isUndefined()) {
1245 Constant *CST = IV.isConstant() ?
1246 IV.getConstant() : UndefValue::get(AI->getType());
1247 DEBUG(std::cerr << "*** Arg " << *AI << " = " << *CST <<"\n");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001248
Chris Lattnerb4394642004-12-10 08:02:06 +00001249 // Replaces all of the uses of a variable with uses of the
1250 // constant.
1251 AI->replaceAllUsesWith(CST);
1252 ++IPNumArgsElimed;
1253 }
1254 }
1255
Chris Lattnerbae4b642004-12-10 22:29:08 +00001256 std::vector<BasicBlock*> BlocksToErase;
Chris Lattnerb4394642004-12-10 08:02:06 +00001257 for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB)
1258 if (!ExecutableBBs.count(BB)) {
1259 DEBUG(std::cerr << " BasicBlock Dead:" << *BB);
1260 ++IPNumDeadBlocks;
Chris Lattner7285f432004-12-10 20:41:50 +00001261
Chris Lattnerb4394642004-12-10 08:02:06 +00001262 // Delete the instructions backwards, as it has a reduced likelihood of
1263 // having to update as many def-use and use-def chains.
1264 std::vector<Instruction*> Insts;
Chris Lattnerbae4b642004-12-10 22:29:08 +00001265 TerminatorInst *TI = BB->getTerminator();
1266 for (BasicBlock::iterator I = BB->begin(), E = TI; I != E; ++I)
Chris Lattnerb4394642004-12-10 08:02:06 +00001267 Insts.push_back(I);
Chris Lattnerbae4b642004-12-10 22:29:08 +00001268
Chris Lattnerb4394642004-12-10 08:02:06 +00001269 while (!Insts.empty()) {
1270 Instruction *I = Insts.back();
1271 Insts.pop_back();
1272 if (!I->use_empty())
1273 I->replaceAllUsesWith(UndefValue::get(I->getType()));
1274 BB->getInstList().erase(I);
1275 MadeChanges = true;
1276 ++IPNumInstRemoved;
1277 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001278
Chris Lattnerbae4b642004-12-10 22:29:08 +00001279 for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i) {
1280 BasicBlock *Succ = TI->getSuccessor(i);
1281 if (Succ->begin() != Succ->end() && isa<PHINode>(Succ->begin()))
1282 TI->getSuccessor(i)->removePredecessor(BB);
1283 }
Chris Lattner99e12952004-12-11 02:53:57 +00001284 if (!TI->use_empty())
1285 TI->replaceAllUsesWith(UndefValue::get(TI->getType()));
Chris Lattnerbae4b642004-12-10 22:29:08 +00001286 BB->getInstList().erase(TI);
1287
Chris Lattner8525ebe2004-12-11 05:32:19 +00001288 if (&*BB != &F->front())
1289 BlocksToErase.push_back(BB);
1290 else
1291 new UnreachableInst(BB);
1292
Chris Lattnerb4394642004-12-10 08:02:06 +00001293 } else {
1294 for (BasicBlock::iterator BI = BB->begin(), E = BB->end(); BI != E; ) {
1295 Instruction *Inst = BI++;
1296 if (Inst->getType() != Type::VoidTy) {
1297 LatticeVal &IV = Values[Inst];
1298 if (IV.isConstant() || IV.isUndefined() &&
1299 !isa<TerminatorInst>(Inst)) {
1300 Constant *Const = IV.isConstant()
1301 ? IV.getConstant() : UndefValue::get(Inst->getType());
1302 DEBUG(std::cerr << " Constant: " << *Const << " = " << *Inst);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001303
Chris Lattnerb4394642004-12-10 08:02:06 +00001304 // Replaces all of the uses of a variable with uses of the
1305 // constant.
1306 Inst->replaceAllUsesWith(Const);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001307
Chris Lattnerb4394642004-12-10 08:02:06 +00001308 // Delete the instruction.
1309 if (!isa<TerminatorInst>(Inst) && !isa<CallInst>(Inst))
1310 BB->getInstList().erase(Inst);
1311
1312 // Hey, we just changed something!
1313 MadeChanges = true;
1314 ++IPNumInstRemoved;
1315 }
1316 }
1317 }
1318 }
Chris Lattnerbae4b642004-12-10 22:29:08 +00001319
1320 // Now that all instructions in the function are constant folded, erase dead
1321 // blocks, because we can now use ConstantFoldTerminator to get rid of
1322 // in-edges.
1323 for (unsigned i = 0, e = BlocksToErase.size(); i != e; ++i) {
1324 // If there are any PHI nodes in this successor, drop entries for BB now.
1325 BasicBlock *DeadBB = BlocksToErase[i];
1326 while (!DeadBB->use_empty()) {
1327 Instruction *I = cast<Instruction>(DeadBB->use_back());
1328 bool Folded = ConstantFoldTerminator(I->getParent());
1329 assert(Folded && "Didn't fold away reference to block!");
1330 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001331
Chris Lattnerbae4b642004-12-10 22:29:08 +00001332 // Finally, delete the basic block.
1333 F->getBasicBlockList().erase(DeadBB);
1334 }
Chris Lattnerb4394642004-12-10 08:02:06 +00001335 }
Chris Lattner99e12952004-12-11 02:53:57 +00001336
1337 // If we inferred constant or undef return values for a function, we replaced
1338 // all call uses with the inferred value. This means we don't need to bother
1339 // actually returning anything from the function. Replace all return
1340 // instructions with return undef.
1341 const hash_map<Function*, LatticeVal> &RV =Solver.getTrackedFunctionRetVals();
1342 for (hash_map<Function*, LatticeVal>::const_iterator I = RV.begin(),
1343 E = RV.end(); I != E; ++I)
1344 if (!I->second.isOverdefined() &&
1345 I->first->getReturnType() != Type::VoidTy) {
1346 Function *F = I->first;
1347 for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB)
1348 if (ReturnInst *RI = dyn_cast<ReturnInst>(BB->getTerminator()))
1349 if (!isa<UndefValue>(RI->getOperand(0)))
1350 RI->setOperand(0, UndefValue::get(F->getReturnType()));
1351 }
Chris Lattner91dbae62004-12-11 05:15:59 +00001352
1353 // If we infered constant or undef values for globals variables, we can delete
1354 // the global and any stores that remain to it.
1355 const hash_map<GlobalVariable*, LatticeVal> &TG = Solver.getTrackedGlobals();
1356 for (hash_map<GlobalVariable*, LatticeVal>::const_iterator I = TG.begin(),
1357 E = TG.end(); I != E; ++I) {
1358 GlobalVariable *GV = I->first;
1359 assert(!I->second.isOverdefined() &&
1360 "Overdefined values should have been taken out of the map!");
1361 DEBUG(std::cerr << "Found that GV '" << GV->getName()<< "' is constant!\n");
1362 while (!GV->use_empty()) {
1363 StoreInst *SI = cast<StoreInst>(GV->use_back());
1364 SI->eraseFromParent();
1365 }
1366 M.getGlobalList().erase(GV);
Chris Lattner2f687fd2004-12-11 06:05:53 +00001367 ++IPNumGlobalConst;
Chris Lattner91dbae62004-12-11 05:15:59 +00001368 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001369
Chris Lattnerb4394642004-12-10 08:02:06 +00001370 return MadeChanges;
1371}