Fix warning about || and && without explicit grouping.
[oota-llvm.git] / lib / Transforms / Scalar / CodeGenPrepare.cpp
1 //===- CodeGenPrepare.cpp - Prepare a function for code generation --------===//
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 munges the code in the input function to better prepare it for
11 // SelectionDAG-based code generation. This works around limitations in it's
12 // basic-block-at-a-time approach. It should eventually be removed.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #define DEBUG_TYPE "codegenprepare"
17 #include "llvm/Transforms/Scalar.h"
18 #include "llvm/Constants.h"
19 #include "llvm/DerivedTypes.h"
20 #include "llvm/Function.h"
21 #include "llvm/InlineAsm.h"
22 #include "llvm/Instructions.h"
23 #include "llvm/IntrinsicInst.h"
24 #include "llvm/Pass.h"
25 #include "llvm/Analysis/Dominators.h"
26 #include "llvm/Analysis/InstructionSimplify.h"
27 #include "llvm/Analysis/ProfileInfo.h"
28 #include "llvm/Target/TargetData.h"
29 #include "llvm/Target/TargetLowering.h"
30 #include "llvm/Transforms/Utils/AddrModeMatcher.h"
31 #include "llvm/Transforms/Utils/BasicBlockUtils.h"
32 #include "llvm/Transforms/Utils/Local.h"
33 #include "llvm/Transforms/Utils/BuildLibCalls.h"
34 #include "llvm/ADT/DenseMap.h"
35 #include "llvm/ADT/SmallSet.h"
36 #include "llvm/ADT/Statistic.h"
37 #include "llvm/Assembly/Writer.h"
38 #include "llvm/Support/CallSite.h"
39 #include "llvm/Support/CommandLine.h"
40 #include "llvm/Support/Debug.h"
41 #include "llvm/Support/GetElementPtrTypeIterator.h"
42 #include "llvm/Support/PatternMatch.h"
43 #include "llvm/Support/raw_ostream.h"
44 #include "llvm/Support/IRBuilder.h"
45 #include "llvm/Support/ValueHandle.h"
46 using namespace llvm;
47 using namespace llvm::PatternMatch;
48
49 STATISTIC(NumBlocksElim, "Number of blocks eliminated");
50 STATISTIC(NumPHIsElim,   "Number of trivial PHIs eliminated");
51 STATISTIC(NumGEPsElim,   "Number of GEPs converted to casts");
52 STATISTIC(NumCmpUses, "Number of uses of Cmp expressions replaced with uses of "
53                       "sunken Cmps");
54 STATISTIC(NumCastUses, "Number of uses of Cast expressions replaced with uses "
55                        "of sunken Casts");
56 STATISTIC(NumMemoryInsts, "Number of memory instructions whose address "
57                           "computations were sunk");
58 STATISTIC(NumExtsMoved,  "Number of [s|z]ext instructions combined with loads");
59 STATISTIC(NumExtUses,    "Number of uses of [s|z]ext instructions optimized");
60 STATISTIC(NumRetsDup,    "Number of return instructions duplicated");
61
62 static cl::opt<bool> DisableBranchOpts(
63   "disable-cgp-branch-opts", cl::Hidden, cl::init(false),
64   cl::desc("Disable branch optimizations in CodeGenPrepare"));
65
66 namespace {
67   class CodeGenPrepare : public FunctionPass {
68     /// TLI - Keep a pointer of a TargetLowering to consult for determining
69     /// transformation profitability.
70     const TargetLowering *TLI;
71     DominatorTree *DT;
72     ProfileInfo *PFI;
73     
74     /// CurInstIterator - As we scan instructions optimizing them, this is the
75     /// next instruction to optimize.  Xforms that can invalidate this should
76     /// update it.
77     BasicBlock::iterator CurInstIterator;
78
79     /// Keeps track of non-local addresses that have been sunk into a block.
80     /// This allows us to avoid inserting duplicate code for blocks with
81     /// multiple load/stores of the same address.
82     DenseMap<Value*, Value*> SunkAddrs;
83
84     /// ModifiedDT - If CFG is modified in anyway, dominator tree may need to
85     /// be updated.
86     bool ModifiedDT;
87
88   public:
89     static char ID; // Pass identification, replacement for typeid
90     explicit CodeGenPrepare(const TargetLowering *tli = 0)
91       : FunctionPass(ID), TLI(tli) {
92         initializeCodeGenPreparePass(*PassRegistry::getPassRegistry());
93       }
94     bool runOnFunction(Function &F);
95
96     virtual void getAnalysisUsage(AnalysisUsage &AU) const {
97       AU.addPreserved<DominatorTree>();
98       AU.addPreserved<ProfileInfo>();
99     }
100
101   private:
102     bool EliminateMostlyEmptyBlocks(Function &F);
103     bool CanMergeBlocks(const BasicBlock *BB, const BasicBlock *DestBB) const;
104     void EliminateMostlyEmptyBlock(BasicBlock *BB);
105     bool OptimizeBlock(BasicBlock &BB);
106     bool OptimizeInst(Instruction *I);
107     bool OptimizeMemoryInst(Instruction *I, Value *Addr, const Type *AccessTy);
108     bool OptimizeInlineAsmInst(CallInst *CS);
109     bool OptimizeCallInst(CallInst *CI);
110     bool MoveExtToFormExtLoad(Instruction *I);
111     bool OptimizeExtUses(Instruction *I);
112     bool DupRetToEnableTailCallOpts(ReturnInst *RI);
113   };
114 }
115
116 char CodeGenPrepare::ID = 0;
117 INITIALIZE_PASS(CodeGenPrepare, "codegenprepare",
118                 "Optimize for code generation", false, false)
119
120 FunctionPass *llvm::createCodeGenPreparePass(const TargetLowering *TLI) {
121   return new CodeGenPrepare(TLI);
122 }
123
124 bool CodeGenPrepare::runOnFunction(Function &F) {
125   bool EverMadeChange = false;
126
127   ModifiedDT = false;
128   DT = getAnalysisIfAvailable<DominatorTree>();
129   PFI = getAnalysisIfAvailable<ProfileInfo>();
130
131   // First pass, eliminate blocks that contain only PHI nodes and an
132   // unconditional branch.
133   EverMadeChange |= EliminateMostlyEmptyBlocks(F);
134
135   bool MadeChange = true;
136   while (MadeChange) {
137     MadeChange = false;
138     for (Function::iterator I = F.begin(), E = F.end(); I != E; ) {
139       BasicBlock *BB = I++;
140       MadeChange |= OptimizeBlock(*BB);
141     }
142     EverMadeChange |= MadeChange;
143   }
144
145   SunkAddrs.clear();
146
147   if (!DisableBranchOpts) {
148     MadeChange = false;
149     for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB)
150       MadeChange |= ConstantFoldTerminator(BB, true);
151
152     if (MadeChange)
153       ModifiedDT = true;
154     EverMadeChange |= MadeChange;
155   }
156
157   if (ModifiedDT && DT)
158     DT->DT->recalculate(F);
159
160   return EverMadeChange;
161 }
162
163 /// EliminateMostlyEmptyBlocks - eliminate blocks that contain only PHI nodes,
164 /// debug info directives, and an unconditional branch.  Passes before isel
165 /// (e.g. LSR/loopsimplify) often split edges in ways that are non-optimal for
166 /// isel.  Start by eliminating these blocks so we can split them the way we
167 /// want them.
168 bool CodeGenPrepare::EliminateMostlyEmptyBlocks(Function &F) {
169   bool MadeChange = false;
170   // Note that this intentionally skips the entry block.
171   for (Function::iterator I = ++F.begin(), E = F.end(); I != E; ) {
172     BasicBlock *BB = I++;
173
174     // If this block doesn't end with an uncond branch, ignore it.
175     BranchInst *BI = dyn_cast<BranchInst>(BB->getTerminator());
176     if (!BI || !BI->isUnconditional())
177       continue;
178
179     // If the instruction before the branch (skipping debug info) isn't a phi
180     // node, then other stuff is happening here.
181     BasicBlock::iterator BBI = BI;
182     if (BBI != BB->begin()) {
183       --BBI;
184       while (isa<DbgInfoIntrinsic>(BBI)) {
185         if (BBI == BB->begin())
186           break;
187         --BBI;
188       }
189       if (!isa<DbgInfoIntrinsic>(BBI) && !isa<PHINode>(BBI))
190         continue;
191     }
192
193     // Do not break infinite loops.
194     BasicBlock *DestBB = BI->getSuccessor(0);
195     if (DestBB == BB)
196       continue;
197
198     if (!CanMergeBlocks(BB, DestBB))
199       continue;
200
201     EliminateMostlyEmptyBlock(BB);
202     MadeChange = true;
203   }
204   return MadeChange;
205 }
206
207 /// CanMergeBlocks - Return true if we can merge BB into DestBB if there is a
208 /// single uncond branch between them, and BB contains no other non-phi
209 /// instructions.
210 bool CodeGenPrepare::CanMergeBlocks(const BasicBlock *BB,
211                                     const BasicBlock *DestBB) const {
212   // We only want to eliminate blocks whose phi nodes are used by phi nodes in
213   // the successor.  If there are more complex condition (e.g. preheaders),
214   // don't mess around with them.
215   BasicBlock::const_iterator BBI = BB->begin();
216   while (const PHINode *PN = dyn_cast<PHINode>(BBI++)) {
217     for (Value::const_use_iterator UI = PN->use_begin(), E = PN->use_end();
218          UI != E; ++UI) {
219       const Instruction *User = cast<Instruction>(*UI);
220       if (User->getParent() != DestBB || !isa<PHINode>(User))
221         return false;
222       // If User is inside DestBB block and it is a PHINode then check
223       // incoming value. If incoming value is not from BB then this is
224       // a complex condition (e.g. preheaders) we want to avoid here.
225       if (User->getParent() == DestBB) {
226         if (const PHINode *UPN = dyn_cast<PHINode>(User))
227           for (unsigned I = 0, E = UPN->getNumIncomingValues(); I != E; ++I) {
228             Instruction *Insn = dyn_cast<Instruction>(UPN->getIncomingValue(I));
229             if (Insn && Insn->getParent() == BB &&
230                 Insn->getParent() != UPN->getIncomingBlock(I))
231               return false;
232           }
233       }
234     }
235   }
236
237   // If BB and DestBB contain any common predecessors, then the phi nodes in BB
238   // and DestBB may have conflicting incoming values for the block.  If so, we
239   // can't merge the block.
240   const PHINode *DestBBPN = dyn_cast<PHINode>(DestBB->begin());
241   if (!DestBBPN) return true;  // no conflict.
242
243   // Collect the preds of BB.
244   SmallPtrSet<const BasicBlock*, 16> BBPreds;
245   if (const PHINode *BBPN = dyn_cast<PHINode>(BB->begin())) {
246     // It is faster to get preds from a PHI than with pred_iterator.
247     for (unsigned i = 0, e = BBPN->getNumIncomingValues(); i != e; ++i)
248       BBPreds.insert(BBPN->getIncomingBlock(i));
249   } else {
250     BBPreds.insert(pred_begin(BB), pred_end(BB));
251   }
252
253   // Walk the preds of DestBB.
254   for (unsigned i = 0, e = DestBBPN->getNumIncomingValues(); i != e; ++i) {
255     BasicBlock *Pred = DestBBPN->getIncomingBlock(i);
256     if (BBPreds.count(Pred)) {   // Common predecessor?
257       BBI = DestBB->begin();
258       while (const PHINode *PN = dyn_cast<PHINode>(BBI++)) {
259         const Value *V1 = PN->getIncomingValueForBlock(Pred);
260         const Value *V2 = PN->getIncomingValueForBlock(BB);
261
262         // If V2 is a phi node in BB, look up what the mapped value will be.
263         if (const PHINode *V2PN = dyn_cast<PHINode>(V2))
264           if (V2PN->getParent() == BB)
265             V2 = V2PN->getIncomingValueForBlock(Pred);
266
267         // If there is a conflict, bail out.
268         if (V1 != V2) return false;
269       }
270     }
271   }
272
273   return true;
274 }
275
276
277 /// EliminateMostlyEmptyBlock - Eliminate a basic block that have only phi's and
278 /// an unconditional branch in it.
279 void CodeGenPrepare::EliminateMostlyEmptyBlock(BasicBlock *BB) {
280   BranchInst *BI = cast<BranchInst>(BB->getTerminator());
281   BasicBlock *DestBB = BI->getSuccessor(0);
282
283   DEBUG(dbgs() << "MERGING MOSTLY EMPTY BLOCKS - BEFORE:\n" << *BB << *DestBB);
284
285   // If the destination block has a single pred, then this is a trivial edge,
286   // just collapse it.
287   if (BasicBlock *SinglePred = DestBB->getSinglePredecessor()) {
288     if (SinglePred != DestBB) {
289       // Remember if SinglePred was the entry block of the function.  If so, we
290       // will need to move BB back to the entry position.
291       bool isEntry = SinglePred == &SinglePred->getParent()->getEntryBlock();
292       MergeBasicBlockIntoOnlyPred(DestBB, this);
293
294       if (isEntry && BB != &BB->getParent()->getEntryBlock())
295         BB->moveBefore(&BB->getParent()->getEntryBlock());
296       
297       DEBUG(dbgs() << "AFTER:\n" << *DestBB << "\n\n\n");
298       return;
299     }
300   }
301
302   // Otherwise, we have multiple predecessors of BB.  Update the PHIs in DestBB
303   // to handle the new incoming edges it is about to have.
304   PHINode *PN;
305   for (BasicBlock::iterator BBI = DestBB->begin();
306        (PN = dyn_cast<PHINode>(BBI)); ++BBI) {
307     // Remove the incoming value for BB, and remember it.
308     Value *InVal = PN->removeIncomingValue(BB, false);
309
310     // Two options: either the InVal is a phi node defined in BB or it is some
311     // value that dominates BB.
312     PHINode *InValPhi = dyn_cast<PHINode>(InVal);
313     if (InValPhi && InValPhi->getParent() == BB) {
314       // Add all of the input values of the input PHI as inputs of this phi.
315       for (unsigned i = 0, e = InValPhi->getNumIncomingValues(); i != e; ++i)
316         PN->addIncoming(InValPhi->getIncomingValue(i),
317                         InValPhi->getIncomingBlock(i));
318     } else {
319       // Otherwise, add one instance of the dominating value for each edge that
320       // we will be adding.
321       if (PHINode *BBPN = dyn_cast<PHINode>(BB->begin())) {
322         for (unsigned i = 0, e = BBPN->getNumIncomingValues(); i != e; ++i)
323           PN->addIncoming(InVal, BBPN->getIncomingBlock(i));
324       } else {
325         for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI)
326           PN->addIncoming(InVal, *PI);
327       }
328     }
329   }
330
331   // The PHIs are now updated, change everything that refers to BB to use
332   // DestBB and remove BB.
333   BB->replaceAllUsesWith(DestBB);
334   if (DT && !ModifiedDT) {
335     BasicBlock *BBIDom  = DT->getNode(BB)->getIDom()->getBlock();
336     BasicBlock *DestBBIDom = DT->getNode(DestBB)->getIDom()->getBlock();
337     BasicBlock *NewIDom = DT->findNearestCommonDominator(BBIDom, DestBBIDom);
338     DT->changeImmediateDominator(DestBB, NewIDom);
339     DT->eraseNode(BB);
340   }
341   if (PFI) {
342     PFI->replaceAllUses(BB, DestBB);
343     PFI->removeEdge(ProfileInfo::getEdge(BB, DestBB));
344   }
345   BB->eraseFromParent();
346   ++NumBlocksElim;
347
348   DEBUG(dbgs() << "AFTER:\n" << *DestBB << "\n\n\n");
349 }
350
351 /// OptimizeNoopCopyExpression - If the specified cast instruction is a noop
352 /// copy (e.g. it's casting from one pointer type to another, i32->i8 on PPC),
353 /// sink it into user blocks to reduce the number of virtual
354 /// registers that must be created and coalesced.
355 ///
356 /// Return true if any changes are made.
357 ///
358 static bool OptimizeNoopCopyExpression(CastInst *CI, const TargetLowering &TLI){
359   // If this is a noop copy,
360   EVT SrcVT = TLI.getValueType(CI->getOperand(0)->getType());
361   EVT DstVT = TLI.getValueType(CI->getType());
362
363   // This is an fp<->int conversion?
364   if (SrcVT.isInteger() != DstVT.isInteger())
365     return false;
366
367   // If this is an extension, it will be a zero or sign extension, which
368   // isn't a noop.
369   if (SrcVT.bitsLT(DstVT)) return false;
370
371   // If these values will be promoted, find out what they will be promoted
372   // to.  This helps us consider truncates on PPC as noop copies when they
373   // are.
374   if (TLI.getTypeAction(SrcVT) == TargetLowering::Promote)
375     SrcVT = TLI.getTypeToTransformTo(CI->getContext(), SrcVT);
376   if (TLI.getTypeAction(DstVT) == TargetLowering::Promote)
377     DstVT = TLI.getTypeToTransformTo(CI->getContext(), DstVT);
378
379   // If, after promotion, these are the same types, this is a noop copy.
380   if (SrcVT != DstVT)
381     return false;
382
383   BasicBlock *DefBB = CI->getParent();
384
385   /// InsertedCasts - Only insert a cast in each block once.
386   DenseMap<BasicBlock*, CastInst*> InsertedCasts;
387
388   bool MadeChange = false;
389   for (Value::use_iterator UI = CI->use_begin(), E = CI->use_end();
390        UI != E; ) {
391     Use &TheUse = UI.getUse();
392     Instruction *User = cast<Instruction>(*UI);
393
394     // Figure out which BB this cast is used in.  For PHI's this is the
395     // appropriate predecessor block.
396     BasicBlock *UserBB = User->getParent();
397     if (PHINode *PN = dyn_cast<PHINode>(User)) {
398       UserBB = PN->getIncomingBlock(UI);
399     }
400
401     // Preincrement use iterator so we don't invalidate it.
402     ++UI;
403
404     // If this user is in the same block as the cast, don't change the cast.
405     if (UserBB == DefBB) continue;
406
407     // If we have already inserted a cast into this block, use it.
408     CastInst *&InsertedCast = InsertedCasts[UserBB];
409
410     if (!InsertedCast) {
411       BasicBlock::iterator InsertPt = UserBB->getFirstNonPHI();
412
413       InsertedCast =
414         CastInst::Create(CI->getOpcode(), CI->getOperand(0), CI->getType(), "",
415                          InsertPt);
416       MadeChange = true;
417     }
418
419     // Replace a use of the cast with a use of the new cast.
420     TheUse = InsertedCast;
421     ++NumCastUses;
422   }
423
424   // If we removed all uses, nuke the cast.
425   if (CI->use_empty()) {
426     CI->eraseFromParent();
427     MadeChange = true;
428   }
429
430   return MadeChange;
431 }
432
433 /// OptimizeCmpExpression - sink the given CmpInst into user blocks to reduce
434 /// the number of virtual registers that must be created and coalesced.  This is
435 /// a clear win except on targets with multiple condition code registers
436 ///  (PowerPC), where it might lose; some adjustment may be wanted there.
437 ///
438 /// Return true if any changes are made.
439 static bool OptimizeCmpExpression(CmpInst *CI) {
440   BasicBlock *DefBB = CI->getParent();
441
442   /// InsertedCmp - Only insert a cmp in each block once.
443   DenseMap<BasicBlock*, CmpInst*> InsertedCmps;
444
445   bool MadeChange = false;
446   for (Value::use_iterator UI = CI->use_begin(), E = CI->use_end();
447        UI != E; ) {
448     Use &TheUse = UI.getUse();
449     Instruction *User = cast<Instruction>(*UI);
450
451     // Preincrement use iterator so we don't invalidate it.
452     ++UI;
453
454     // Don't bother for PHI nodes.
455     if (isa<PHINode>(User))
456       continue;
457
458     // Figure out which BB this cmp is used in.
459     BasicBlock *UserBB = User->getParent();
460
461     // If this user is in the same block as the cmp, don't change the cmp.
462     if (UserBB == DefBB) continue;
463
464     // If we have already inserted a cmp into this block, use it.
465     CmpInst *&InsertedCmp = InsertedCmps[UserBB];
466
467     if (!InsertedCmp) {
468       BasicBlock::iterator InsertPt = UserBB->getFirstNonPHI();
469
470       InsertedCmp =
471         CmpInst::Create(CI->getOpcode(),
472                         CI->getPredicate(),  CI->getOperand(0),
473                         CI->getOperand(1), "", InsertPt);
474       MadeChange = true;
475     }
476
477     // Replace a use of the cmp with a use of the new cmp.
478     TheUse = InsertedCmp;
479     ++NumCmpUses;
480   }
481
482   // If we removed all uses, nuke the cmp.
483   if (CI->use_empty())
484     CI->eraseFromParent();
485
486   return MadeChange;
487 }
488
489 namespace {
490 class CodeGenPrepareFortifiedLibCalls : public SimplifyFortifiedLibCalls {
491 protected:
492   void replaceCall(Value *With) {
493     CI->replaceAllUsesWith(With);
494     CI->eraseFromParent();
495   }
496   bool isFoldable(unsigned SizeCIOp, unsigned, bool) const {
497       if (ConstantInt *SizeCI =
498                              dyn_cast<ConstantInt>(CI->getArgOperand(SizeCIOp)))
499         return SizeCI->isAllOnesValue();
500     return false;
501   }
502 };
503 } // end anonymous namespace
504
505 bool CodeGenPrepare::OptimizeCallInst(CallInst *CI) {
506   BasicBlock *BB = CI->getParent();
507   
508   // Lower inline assembly if we can.
509   // If we found an inline asm expession, and if the target knows how to
510   // lower it to normal LLVM code, do so now.
511   if (TLI && isa<InlineAsm>(CI->getCalledValue())) {
512     if (TLI->ExpandInlineAsm(CI)) {
513       // Avoid invalidating the iterator.
514       CurInstIterator = BB->begin();
515       // Avoid processing instructions out of order, which could cause
516       // reuse before a value is defined.
517       SunkAddrs.clear();
518       return true;
519     }
520     // Sink address computing for memory operands into the block.
521     if (OptimizeInlineAsmInst(CI))
522       return true;
523   }
524   
525   // Lower all uses of llvm.objectsize.*
526   IntrinsicInst *II = dyn_cast<IntrinsicInst>(CI);
527   if (II && II->getIntrinsicID() == Intrinsic::objectsize) {
528     bool Min = (cast<ConstantInt>(II->getArgOperand(1))->getZExtValue() == 1);
529     const Type *ReturnTy = CI->getType();
530     Constant *RetVal = ConstantInt::get(ReturnTy, Min ? 0 : -1ULL);    
531     
532     // Substituting this can cause recursive simplifications, which can
533     // invalidate our iterator.  Use a WeakVH to hold onto it in case this
534     // happens.
535     WeakVH IterHandle(CurInstIterator);
536     
537     ReplaceAndSimplifyAllUses(CI, RetVal, TLI ? TLI->getTargetData() : 0,
538                               ModifiedDT ? 0 : DT);
539
540     // If the iterator instruction was recursively deleted, start over at the
541     // start of the block.
542     if (IterHandle != CurInstIterator) {
543       CurInstIterator = BB->begin();
544       SunkAddrs.clear();
545     }
546     return true;
547   }
548
549   // From here on out we're working with named functions.
550   if (CI->getCalledFunction() == 0) return false;
551
552   // llvm.dbg.value is far away from the value then iSel may not be able
553   // handle it properly. iSel will drop llvm.dbg.value if it can not 
554   // find a node corresponding to the value.
555   if (DbgValueInst *DVI = dyn_cast<DbgValueInst>(CI))
556     if (Instruction *VI = dyn_cast_or_null<Instruction>(DVI->getValue()))
557       if (!VI->isTerminator() &&
558           (DVI->getParent() != VI->getParent() || DT->dominates(DVI, VI))) {
559         DEBUG(dbgs() << "Moving Debug Value before :\n" << *DVI << ' ' << *VI);
560         DVI->removeFromParent();
561         if (isa<PHINode>(VI))
562           DVI->insertBefore(VI->getParent()->getFirstNonPHI());
563         else
564           DVI->insertAfter(VI);
565         return true;
566       }
567
568   // We'll need TargetData from here on out.
569   const TargetData *TD = TLI ? TLI->getTargetData() : 0;
570   if (!TD) return false;
571   
572   // Lower all default uses of _chk calls.  This is very similar
573   // to what InstCombineCalls does, but here we are only lowering calls
574   // that have the default "don't know" as the objectsize.  Anything else
575   // should be left alone.
576   CodeGenPrepareFortifiedLibCalls Simplifier;
577   return Simplifier.fold(CI, TD);
578 }
579
580 /// DupRetToEnableTailCallOpts - Look for opportunities to duplicate return
581 /// instructions to the predecessor to enable tail call optimizations. The
582 /// case it is currently looking for is:
583 /// bb0:
584 ///   %tmp0 = tail call i32 @f0()
585 ///   br label %return
586 /// bb1:
587 ///   %tmp1 = tail call i32 @f1()
588 ///   br label %return
589 /// bb2:
590 ///   %tmp2 = tail call i32 @f2()
591 ///   br label %return
592 /// return:
593 ///   %retval = phi i32 [ %tmp0, %bb0 ], [ %tmp1, %bb1 ], [ %tmp2, %bb2 ]
594 ///   ret i32 %retval
595 ///
596 /// =>
597 ///
598 /// bb0:
599 ///   %tmp0 = tail call i32 @f0()
600 ///   ret i32 %tmp0
601 /// bb1:
602 ///   %tmp1 = tail call i32 @f1()
603 ///   ret i32 %tmp1
604 /// bb2:
605 ///   %tmp2 = tail call i32 @f2()
606 ///   ret i32 %tmp2
607 ///
608 bool CodeGenPrepare::DupRetToEnableTailCallOpts(ReturnInst *RI) {
609   if (!TLI)
610     return false;
611
612   Value *V = RI->getReturnValue();
613   PHINode *PN = V ? dyn_cast<PHINode>(V) : NULL;
614   if (V && !PN)
615     return false;
616
617   BasicBlock *BB = RI->getParent();
618   if (PN && PN->getParent() != BB)
619     return false;
620
621   // It's not safe to eliminate the sign / zero extension of the return value.
622   // See llvm::isInTailCallPosition().
623   const Function *F = BB->getParent();
624   unsigned CallerRetAttr = F->getAttributes().getRetAttributes();
625   if ((CallerRetAttr & Attribute::ZExt) || (CallerRetAttr & Attribute::SExt))
626     return false;
627
628   // Make sure there are no instructions between the PHI and return, or that the
629   // return is the first instruction in the block.
630   if (PN) {
631     BasicBlock::iterator BI = BB->begin();
632     do { ++BI; } while (isa<DbgInfoIntrinsic>(BI));
633     if (&*BI != RI)
634       return false;
635   } else {
636     BasicBlock::iterator BI = BB->begin();
637     while (isa<DbgInfoIntrinsic>(BI)) ++BI;
638     if (&*BI != RI)
639       return false;
640   }
641
642   /// Only dup the ReturnInst if the CallInst is likely to be emitted as a tail
643   /// call.
644   SmallVector<CallInst*, 4> TailCalls;
645   if (PN) {
646     for (unsigned I = 0, E = PN->getNumIncomingValues(); I != E; ++I) {
647       CallInst *CI = dyn_cast<CallInst>(PN->getIncomingValue(I));
648       // Make sure the phi value is indeed produced by the tail call.
649       if (CI && CI->hasOneUse() && CI->getParent() == PN->getIncomingBlock(I) &&
650           TLI->mayBeEmittedAsTailCall(CI))
651         TailCalls.push_back(CI);
652     }
653   } else {
654     SmallPtrSet<BasicBlock*, 4> VisitedBBs;
655     for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
656       if (!VisitedBBs.insert(*PI))
657         continue;
658
659       BasicBlock::InstListType &InstList = (*PI)->getInstList();
660       BasicBlock::InstListType::reverse_iterator RI = InstList.rbegin();
661       BasicBlock::InstListType::reverse_iterator RE = InstList.rend();
662       do { ++RI; } while (RI != RE && isa<DbgInfoIntrinsic>(&*RI));
663       if (RI == RE)
664         continue;
665
666       CallInst *CI = dyn_cast<CallInst>(&*RI);
667       if (CI && CI->use_empty() && TLI->mayBeEmittedAsTailCall(CI))
668         TailCalls.push_back(CI);
669     }
670   }
671
672   bool Changed = false;
673   for (unsigned i = 0, e = TailCalls.size(); i != e; ++i) {
674     CallInst *CI = TailCalls[i];
675     CallSite CS(CI);
676
677     // Conservatively require the attributes of the call to match those of the
678     // return. Ignore noalias because it doesn't affect the call sequence.
679     unsigned CalleeRetAttr = CS.getAttributes().getRetAttributes();
680     if ((CalleeRetAttr ^ CallerRetAttr) & ~Attribute::NoAlias)
681       continue;
682
683     // Make sure the call instruction is followed by an unconditional branch to
684     // the return block.
685     BasicBlock *CallBB = CI->getParent();
686     BranchInst *BI = dyn_cast<BranchInst>(CallBB->getTerminator());
687     if (!BI || !BI->isUnconditional() || BI->getSuccessor(0) != BB)
688       continue;
689
690     // Duplicate the return into CallBB.
691     (void)FoldReturnIntoUncondBranch(RI, BB, CallBB);
692     ModifiedDT = Changed = true;
693     ++NumRetsDup;
694   }
695
696   // If we eliminated all predecessors of the block, delete the block now.
697   if (Changed && pred_begin(BB) == pred_end(BB))
698     BB->eraseFromParent();
699
700   return Changed;
701 }
702
703 //===----------------------------------------------------------------------===//
704 // Memory Optimization
705 //===----------------------------------------------------------------------===//
706
707 /// IsNonLocalValue - Return true if the specified values are defined in a
708 /// different basic block than BB.
709 static bool IsNonLocalValue(Value *V, BasicBlock *BB) {
710   if (Instruction *I = dyn_cast<Instruction>(V))
711     return I->getParent() != BB;
712   return false;
713 }
714
715 /// OptimizeMemoryInst - Load and Store Instructions often have
716 /// addressing modes that can do significant amounts of computation.  As such,
717 /// instruction selection will try to get the load or store to do as much
718 /// computation as possible for the program.  The problem is that isel can only
719 /// see within a single block.  As such, we sink as much legal addressing mode
720 /// stuff into the block as possible.
721 ///
722 /// This method is used to optimize both load/store and inline asms with memory
723 /// operands.
724 bool CodeGenPrepare::OptimizeMemoryInst(Instruction *MemoryInst, Value *Addr,
725                                         const Type *AccessTy) {
726   Value *Repl = Addr;
727   
728   // Try to collapse single-value PHI nodes.  This is necessary to undo 
729   // unprofitable PRE transformations.
730   SmallVector<Value*, 8> worklist;
731   SmallPtrSet<Value*, 16> Visited;
732   worklist.push_back(Addr);
733   
734   // Use a worklist to iteratively look through PHI nodes, and ensure that
735   // the addressing mode obtained from the non-PHI roots of the graph
736   // are equivalent.
737   Value *Consensus = 0;
738   unsigned NumUsesConsensus = 0;
739   bool IsNumUsesConsensusValid = false;
740   SmallVector<Instruction*, 16> AddrModeInsts;
741   ExtAddrMode AddrMode;
742   while (!worklist.empty()) {
743     Value *V = worklist.back();
744     worklist.pop_back();
745     
746     // Break use-def graph loops.
747     if (Visited.count(V)) {
748       Consensus = 0;
749       break;
750     }
751     
752     Visited.insert(V);
753     
754     // For a PHI node, push all of its incoming values.
755     if (PHINode *P = dyn_cast<PHINode>(V)) {
756       for (unsigned i = 0, e = P->getNumIncomingValues(); i != e; ++i)
757         worklist.push_back(P->getIncomingValue(i));
758       continue;
759     }
760     
761     // For non-PHIs, determine the addressing mode being computed.
762     SmallVector<Instruction*, 16> NewAddrModeInsts;
763     ExtAddrMode NewAddrMode =
764       AddressingModeMatcher::Match(V, AccessTy,MemoryInst,
765                                    NewAddrModeInsts, *TLI);
766
767     // This check is broken into two cases with very similar code to avoid using
768     // getNumUses() as much as possible. Some values have a lot of uses, so
769     // calling getNumUses() unconditionally caused a significant compile-time
770     // regression.
771     if (!Consensus) {
772       Consensus = V;
773       AddrMode = NewAddrMode;
774       AddrModeInsts = NewAddrModeInsts;
775       continue;
776     } else if (NewAddrMode == AddrMode) {
777       if (!IsNumUsesConsensusValid) {
778         NumUsesConsensus = Consensus->getNumUses();
779         IsNumUsesConsensusValid = true;
780       }
781
782       // Ensure that the obtained addressing mode is equivalent to that obtained
783       // for all other roots of the PHI traversal.  Also, when choosing one
784       // such root as representative, select the one with the most uses in order
785       // to keep the cost modeling heuristics in AddressingModeMatcher
786       // applicable.
787       unsigned NumUses = V->getNumUses();
788       if (NumUses > NumUsesConsensus) {
789         Consensus = V;
790         NumUsesConsensus = NumUses;
791         AddrModeInsts = NewAddrModeInsts;
792       }
793       continue;
794     }
795     
796     Consensus = 0;
797     break;
798   }
799   
800   // If the addressing mode couldn't be determined, or if multiple different
801   // ones were determined, bail out now.
802   if (!Consensus) return false;
803   
804   // Check to see if any of the instructions supersumed by this addr mode are
805   // non-local to I's BB.
806   bool AnyNonLocal = false;
807   for (unsigned i = 0, e = AddrModeInsts.size(); i != e; ++i) {
808     if (IsNonLocalValue(AddrModeInsts[i], MemoryInst->getParent())) {
809       AnyNonLocal = true;
810       break;
811     }
812   }
813
814   // If all the instructions matched are already in this BB, don't do anything.
815   if (!AnyNonLocal) {
816     DEBUG(dbgs() << "CGP: Found      local addrmode: " << AddrMode << "\n");
817     return false;
818   }
819
820   // Insert this computation right after this user.  Since our caller is
821   // scanning from the top of the BB to the bottom, reuse of the expr are
822   // guaranteed to happen later.
823   BasicBlock::iterator InsertPt = MemoryInst;
824
825   // Now that we determined the addressing expression we want to use and know
826   // that we have to sink it into this block.  Check to see if we have already
827   // done this for some other load/store instr in this block.  If so, reuse the
828   // computation.
829   Value *&SunkAddr = SunkAddrs[Addr];
830   if (SunkAddr) {
831     DEBUG(dbgs() << "CGP: Reusing nonlocal addrmode: " << AddrMode << " for "
832                  << *MemoryInst);
833     if (SunkAddr->getType() != Addr->getType())
834       SunkAddr = new BitCastInst(SunkAddr, Addr->getType(), "tmp", InsertPt);
835   } else {
836     DEBUG(dbgs() << "CGP: SINKING nonlocal addrmode: " << AddrMode << " for "
837                  << *MemoryInst);
838     const Type *IntPtrTy =
839           TLI->getTargetData()->getIntPtrType(AccessTy->getContext());
840
841     Value *Result = 0;
842
843     // Start with the base register. Do this first so that subsequent address
844     // matching finds it last, which will prevent it from trying to match it
845     // as the scaled value in case it happens to be a mul. That would be
846     // problematic if we've sunk a different mul for the scale, because then
847     // we'd end up sinking both muls.
848     if (AddrMode.BaseReg) {
849       Value *V = AddrMode.BaseReg;
850       if (V->getType()->isPointerTy())
851         V = new PtrToIntInst(V, IntPtrTy, "sunkaddr", InsertPt);
852       if (V->getType() != IntPtrTy)
853         V = CastInst::CreateIntegerCast(V, IntPtrTy, /*isSigned=*/true,
854                                         "sunkaddr", InsertPt);
855       Result = V;
856     }
857
858     // Add the scale value.
859     if (AddrMode.Scale) {
860       Value *V = AddrMode.ScaledReg;
861       if (V->getType() == IntPtrTy) {
862         // done.
863       } else if (V->getType()->isPointerTy()) {
864         V = new PtrToIntInst(V, IntPtrTy, "sunkaddr", InsertPt);
865       } else if (cast<IntegerType>(IntPtrTy)->getBitWidth() <
866                  cast<IntegerType>(V->getType())->getBitWidth()) {
867         V = new TruncInst(V, IntPtrTy, "sunkaddr", InsertPt);
868       } else {
869         V = new SExtInst(V, IntPtrTy, "sunkaddr", InsertPt);
870       }
871       if (AddrMode.Scale != 1)
872         V = BinaryOperator::CreateMul(V, ConstantInt::get(IntPtrTy,
873                                                                 AddrMode.Scale),
874                                       "sunkaddr", InsertPt);
875       if (Result)
876         Result = BinaryOperator::CreateAdd(Result, V, "sunkaddr", InsertPt);
877       else
878         Result = V;
879     }
880
881     // Add in the BaseGV if present.
882     if (AddrMode.BaseGV) {
883       Value *V = new PtrToIntInst(AddrMode.BaseGV, IntPtrTy, "sunkaddr",
884                                   InsertPt);
885       if (Result)
886         Result = BinaryOperator::CreateAdd(Result, V, "sunkaddr", InsertPt);
887       else
888         Result = V;
889     }
890
891     // Add in the Base Offset if present.
892     if (AddrMode.BaseOffs) {
893       Value *V = ConstantInt::get(IntPtrTy, AddrMode.BaseOffs);
894       if (Result)
895         Result = BinaryOperator::CreateAdd(Result, V, "sunkaddr", InsertPt);
896       else
897         Result = V;
898     }
899
900     if (Result == 0)
901       SunkAddr = Constant::getNullValue(Addr->getType());
902     else
903       SunkAddr = new IntToPtrInst(Result, Addr->getType(), "sunkaddr",InsertPt);
904   }
905
906   MemoryInst->replaceUsesOfWith(Repl, SunkAddr);
907
908   // If we have no uses, recursively delete the value and all dead instructions
909   // using it.
910   if (Repl->use_empty()) {
911     // This can cause recursive deletion, which can invalidate our iterator.
912     // Use a WeakVH to hold onto it in case this happens.
913     WeakVH IterHandle(CurInstIterator);
914     BasicBlock *BB = CurInstIterator->getParent();
915     
916     RecursivelyDeleteTriviallyDeadInstructions(Repl);
917
918     if (IterHandle != CurInstIterator) {
919       // If the iterator instruction was recursively deleted, start over at the
920       // start of the block.
921       CurInstIterator = BB->begin();
922       SunkAddrs.clear();
923     } else {
924       // This address is now available for reassignment, so erase the table
925       // entry; we don't want to match some completely different instruction.
926       SunkAddrs[Addr] = 0;
927     }    
928   }
929   ++NumMemoryInsts;
930   return true;
931 }
932
933 /// OptimizeInlineAsmInst - If there are any memory operands, use
934 /// OptimizeMemoryInst to sink their address computing into the block when
935 /// possible / profitable.
936 bool CodeGenPrepare::OptimizeInlineAsmInst(CallInst *CS) {
937   bool MadeChange = false;
938
939   TargetLowering::AsmOperandInfoVector 
940     TargetConstraints = TLI->ParseConstraints(CS);
941   unsigned ArgNo = 0;
942   for (unsigned i = 0, e = TargetConstraints.size(); i != e; ++i) {
943     TargetLowering::AsmOperandInfo &OpInfo = TargetConstraints[i];
944     
945     // Compute the constraint code and ConstraintType to use.
946     TLI->ComputeConstraintToUse(OpInfo, SDValue());
947
948     if (OpInfo.ConstraintType == TargetLowering::C_Memory &&
949         OpInfo.isIndirect) {
950       Value *OpVal = CS->getArgOperand(ArgNo++);
951       MadeChange |= OptimizeMemoryInst(CS, OpVal, OpVal->getType());
952     } else if (OpInfo.Type == InlineAsm::isInput)
953       ArgNo++;
954   }
955
956   return MadeChange;
957 }
958
959 /// MoveExtToFormExtLoad - Move a zext or sext fed by a load into the same
960 /// basic block as the load, unless conditions are unfavorable. This allows
961 /// SelectionDAG to fold the extend into the load.
962 ///
963 bool CodeGenPrepare::MoveExtToFormExtLoad(Instruction *I) {
964   // Look for a load being extended.
965   LoadInst *LI = dyn_cast<LoadInst>(I->getOperand(0));
966   if (!LI) return false;
967
968   // If they're already in the same block, there's nothing to do.
969   if (LI->getParent() == I->getParent())
970     return false;
971
972   // If the load has other users and the truncate is not free, this probably
973   // isn't worthwhile.
974   if (!LI->hasOneUse() &&
975       TLI && (TLI->isTypeLegal(TLI->getValueType(LI->getType())) ||
976               !TLI->isTypeLegal(TLI->getValueType(I->getType()))) &&
977       !TLI->isTruncateFree(I->getType(), LI->getType()))
978     return false;
979
980   // Check whether the target supports casts folded into loads.
981   unsigned LType;
982   if (isa<ZExtInst>(I))
983     LType = ISD::ZEXTLOAD;
984   else {
985     assert(isa<SExtInst>(I) && "Unexpected ext type!");
986     LType = ISD::SEXTLOAD;
987   }
988   if (TLI && !TLI->isLoadExtLegal(LType, TLI->getValueType(LI->getType())))
989     return false;
990
991   // Move the extend into the same block as the load, so that SelectionDAG
992   // can fold it.
993   I->removeFromParent();
994   I->insertAfter(LI);
995   ++NumExtsMoved;
996   return true;
997 }
998
999 bool CodeGenPrepare::OptimizeExtUses(Instruction *I) {
1000   BasicBlock *DefBB = I->getParent();
1001
1002   // If the result of a {s|z}ext and its source are both live out, rewrite all
1003   // other uses of the source with result of extension.
1004   Value *Src = I->getOperand(0);
1005   if (Src->hasOneUse())
1006     return false;
1007
1008   // Only do this xform if truncating is free.
1009   if (TLI && !TLI->isTruncateFree(I->getType(), Src->getType()))
1010     return false;
1011
1012   // Only safe to perform the optimization if the source is also defined in
1013   // this block.
1014   if (!isa<Instruction>(Src) || DefBB != cast<Instruction>(Src)->getParent())
1015     return false;
1016
1017   bool DefIsLiveOut = false;
1018   for (Value::use_iterator UI = I->use_begin(), E = I->use_end();
1019        UI != E; ++UI) {
1020     Instruction *User = cast<Instruction>(*UI);
1021
1022     // Figure out which BB this ext is used in.
1023     BasicBlock *UserBB = User->getParent();
1024     if (UserBB == DefBB) continue;
1025     DefIsLiveOut = true;
1026     break;
1027   }
1028   if (!DefIsLiveOut)
1029     return false;
1030
1031   // Make sure non of the uses are PHI nodes.
1032   for (Value::use_iterator UI = Src->use_begin(), E = Src->use_end();
1033        UI != E; ++UI) {
1034     Instruction *User = cast<Instruction>(*UI);
1035     BasicBlock *UserBB = User->getParent();
1036     if (UserBB == DefBB) continue;
1037     // Be conservative. We don't want this xform to end up introducing
1038     // reloads just before load / store instructions.
1039     if (isa<PHINode>(User) || isa<LoadInst>(User) || isa<StoreInst>(User))
1040       return false;
1041   }
1042
1043   // InsertedTruncs - Only insert one trunc in each block once.
1044   DenseMap<BasicBlock*, Instruction*> InsertedTruncs;
1045
1046   bool MadeChange = false;
1047   for (Value::use_iterator UI = Src->use_begin(), E = Src->use_end();
1048        UI != E; ++UI) {
1049     Use &TheUse = UI.getUse();
1050     Instruction *User = cast<Instruction>(*UI);
1051
1052     // Figure out which BB this ext is used in.
1053     BasicBlock *UserBB = User->getParent();
1054     if (UserBB == DefBB) continue;
1055
1056     // Both src and def are live in this block. Rewrite the use.
1057     Instruction *&InsertedTrunc = InsertedTruncs[UserBB];
1058
1059     if (!InsertedTrunc) {
1060       BasicBlock::iterator InsertPt = UserBB->getFirstNonPHI();
1061
1062       InsertedTrunc = new TruncInst(I, Src->getType(), "", InsertPt);
1063     }
1064
1065     // Replace a use of the {s|z}ext source with a use of the result.
1066     TheUse = InsertedTrunc;
1067     ++NumExtUses;
1068     MadeChange = true;
1069   }
1070
1071   return MadeChange;
1072 }
1073
1074 bool CodeGenPrepare::OptimizeInst(Instruction *I) {
1075   if (PHINode *P = dyn_cast<PHINode>(I)) {
1076     // It is possible for very late stage optimizations (such as SimplifyCFG)
1077     // to introduce PHI nodes too late to be cleaned up.  If we detect such a
1078     // trivial PHI, go ahead and zap it here.
1079     if (Value *V = SimplifyInstruction(P)) {
1080       P->replaceAllUsesWith(V);
1081       P->eraseFromParent();
1082       ++NumPHIsElim;
1083       return true;
1084     }
1085     return false;
1086   }
1087   
1088   if (CastInst *CI = dyn_cast<CastInst>(I)) {
1089     // If the source of the cast is a constant, then this should have
1090     // already been constant folded.  The only reason NOT to constant fold
1091     // it is if something (e.g. LSR) was careful to place the constant
1092     // evaluation in a block other than then one that uses it (e.g. to hoist
1093     // the address of globals out of a loop).  If this is the case, we don't
1094     // want to forward-subst the cast.
1095     if (isa<Constant>(CI->getOperand(0)))
1096       return false;
1097
1098     if (TLI && OptimizeNoopCopyExpression(CI, *TLI))
1099       return true;
1100
1101     if (isa<ZExtInst>(I) || isa<SExtInst>(I)) {
1102       bool MadeChange = MoveExtToFormExtLoad(I);
1103       return MadeChange | OptimizeExtUses(I);
1104     }
1105     return false;
1106   }
1107   
1108   if (CmpInst *CI = dyn_cast<CmpInst>(I))
1109     return OptimizeCmpExpression(CI);
1110   
1111   if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1112     if (TLI)
1113       return OptimizeMemoryInst(I, I->getOperand(0), LI->getType());
1114     return false;
1115   }
1116   
1117   if (StoreInst *SI = dyn_cast<StoreInst>(I)) {
1118     if (TLI)
1119       return OptimizeMemoryInst(I, SI->getOperand(1),
1120                                 SI->getOperand(0)->getType());
1121     return false;
1122   }
1123   
1124   if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(I)) {
1125     if (GEPI->hasAllZeroIndices()) {
1126       /// The GEP operand must be a pointer, so must its result -> BitCast
1127       Instruction *NC = new BitCastInst(GEPI->getOperand(0), GEPI->getType(),
1128                                         GEPI->getName(), GEPI);
1129       GEPI->replaceAllUsesWith(NC);
1130       GEPI->eraseFromParent();
1131       ++NumGEPsElim;
1132       OptimizeInst(NC);
1133       return true;
1134     }
1135     return false;
1136   }
1137   
1138   if (CallInst *CI = dyn_cast<CallInst>(I))
1139     return OptimizeCallInst(CI);
1140
1141   if (ReturnInst *RI = dyn_cast<ReturnInst>(I))
1142     return DupRetToEnableTailCallOpts(RI);
1143
1144   return false;
1145 }
1146
1147 // In this pass we look for GEP and cast instructions that are used
1148 // across basic blocks and rewrite them to improve basic-block-at-a-time
1149 // selection.
1150 bool CodeGenPrepare::OptimizeBlock(BasicBlock &BB) {
1151   SunkAddrs.clear();
1152   bool MadeChange = false;
1153
1154   CurInstIterator = BB.begin();
1155   for (BasicBlock::iterator E = BB.end(); CurInstIterator != E; )
1156     MadeChange |= OptimizeInst(CurInstIterator++);
1157
1158   return MadeChange;
1159 }