Chris Lattner | 237ef56 | 2003-08-31 19:10:30 +0000 | [diff] [blame] | 1 | //===- InlineSimple.cpp - Code to perform simple function inlining --------===// |
Misha Brukman | fd93908 | 2005-04-21 23:48:37 +0000 | [diff] [blame] | 2 | // |
John Criswell | b576c94 | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 4ee451d | 2007-12-29 20:36:04 +0000 | [diff] [blame^] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | fd93908 | 2005-04-21 23:48:37 +0000 | [diff] [blame] | 7 | // |
John Criswell | b576c94 | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
Chris Lattner | 0095054 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 9 | // |
Chris Lattner | ca398dc | 2003-05-29 15:11:31 +0000 | [diff] [blame] | 10 | // This file implements bottom-up inlining of functions into callees. |
Chris Lattner | 0154505 | 2002-04-18 18:52:03 +0000 | [diff] [blame] | 11 | // |
Chris Lattner | 0095054 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Tanya Lattner | 6f7426e | 2007-06-19 22:29:50 +0000 | [diff] [blame] | 14 | #define DEBUG_TYPE "inline" |
Chris Lattner | 49fbff4 | 2005-05-18 04:30:33 +0000 | [diff] [blame] | 15 | #include "llvm/CallingConv.h" |
Chris Lattner | 869adc2 | 2003-11-21 21:46:09 +0000 | [diff] [blame] | 16 | #include "llvm/Instructions.h" |
Chris Lattner | 8db93f1 | 2004-11-22 17:21:44 +0000 | [diff] [blame] | 17 | #include "llvm/IntrinsicInst.h" |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 18 | #include "llvm/Module.h" |
Chris Lattner | 619d354 | 2004-03-13 23:15:45 +0000 | [diff] [blame] | 19 | #include "llvm/Type.h" |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 20 | #include "llvm/Analysis/CallGraph.h" |
Chris Lattner | e544533 | 2003-08-24 06:59:28 +0000 | [diff] [blame] | 21 | #include "llvm/Support/CallSite.h" |
Reid Spencer | 9133fe2 | 2007-02-05 23:32:05 +0000 | [diff] [blame] | 22 | #include "llvm/Support/Compiler.h" |
Chris Lattner | 237ef56 | 2003-08-31 19:10:30 +0000 | [diff] [blame] | 23 | #include "llvm/Transforms/IPO.h" |
Tanya Lattner | 6f7426e | 2007-06-19 22:29:50 +0000 | [diff] [blame] | 24 | #include "llvm/Transforms/IPO/InlinerPass.h" |
Devang Patel | 6899b31 | 2007-07-25 18:00:25 +0000 | [diff] [blame] | 25 | #include "llvm/Transforms/Utils/InlineCost.h" |
Devang Patel | 29381fb | 2007-07-27 18:34:27 +0000 | [diff] [blame] | 26 | #include "llvm/ADT/SmallPtrSet.h" |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 27 | |
Chris Lattner | 869adc2 | 2003-11-21 21:46:09 +0000 | [diff] [blame] | 28 | using namespace llvm; |
Brian Gaeke | d0fde30 | 2003-11-11 22:41:34 +0000 | [diff] [blame] | 29 | |
Chris Lattner | bd0ef77 | 2002-02-26 21:46:54 +0000 | [diff] [blame] | 30 | namespace { |
Chris Lattner | 884d6c4 | 2003-10-06 15:52:43 +0000 | [diff] [blame] | 31 | |
Reid Spencer | 9133fe2 | 2007-02-05 23:32:05 +0000 | [diff] [blame] | 32 | class VISIBILITY_HIDDEN SimpleInliner : public Inliner { |
Devang Patel | 29381fb | 2007-07-27 18:34:27 +0000 | [diff] [blame] | 33 | // Functions that are never inlined |
| 34 | SmallPtrSet<const Function*, 16> NeverInline; |
Devang Patel | 6899b31 | 2007-07-25 18:00:25 +0000 | [diff] [blame] | 35 | InlineCostAnalyzer CA; |
Chris Lattner | 884d6c4 | 2003-10-06 15:52:43 +0000 | [diff] [blame] | 36 | public: |
Chris Lattner | 1ce6f8d | 2007-05-06 23:13:56 +0000 | [diff] [blame] | 37 | SimpleInliner() : Inliner(&ID) {} |
Nick Lewycky | ecd94c8 | 2007-05-06 13:37:16 +0000 | [diff] [blame] | 38 | static char ID; // Pass identification, replacement for typeid |
Devang Patel | 6899b31 | 2007-07-25 18:00:25 +0000 | [diff] [blame] | 39 | int getInlineCost(CallSite CS) { |
| 40 | return CA.getInlineCost(CS, NeverInline); |
| 41 | } |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 42 | virtual bool doInitialization(CallGraph &CG); |
Chris Lattner | bd0ef77 | 2002-02-26 21:46:54 +0000 | [diff] [blame] | 43 | }; |
Devang Patel | 1997473 | 2007-05-03 01:11:54 +0000 | [diff] [blame] | 44 | char SimpleInliner::ID = 0; |
Chris Lattner | 7f8897f | 2006-08-27 22:42:52 +0000 | [diff] [blame] | 45 | RegisterPass<SimpleInliner> X("inline", "Function Integration/Inlining"); |
Chris Lattner | bd0ef77 | 2002-02-26 21:46:54 +0000 | [diff] [blame] | 46 | } |
| 47 | |
Devang Patel | c71ca3c | 2007-01-26 00:47:38 +0000 | [diff] [blame] | 48 | Pass *llvm::createFunctionInliningPass() { return new SimpleInliner(); } |
Chris Lattner | 869adc2 | 2003-11-21 21:46:09 +0000 | [diff] [blame] | 49 | |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 50 | // doInitialization - Initializes the vector of functions that have been |
| 51 | // annotated with the noinline attribute. |
| 52 | bool SimpleInliner::doInitialization(CallGraph &CG) { |
| 53 | |
| 54 | Module &M = CG.getModule(); |
| 55 | |
| 56 | // Get llvm.noinline |
| 57 | GlobalVariable *GV = M.getNamedGlobal("llvm.noinline"); |
| 58 | |
Tanya Lattner | b62fa8a | 2007-06-07 17:12:16 +0000 | [diff] [blame] | 59 | if (GV == 0) |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 60 | return false; |
| 61 | |
Anton Korobeynikov | 311c4b6 | 2007-11-22 22:30:10 +0000 | [diff] [blame] | 62 | // Don't crash on invalid code |
| 63 | if (!GV->hasInitializer()) |
| 64 | return false; |
| 65 | |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 66 | const ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); |
| 67 | |
Tanya Lattner | b62fa8a | 2007-06-07 17:12:16 +0000 | [diff] [blame] | 68 | if (InitList == 0) |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 69 | return false; |
| 70 | |
| 71 | // Iterate over each element and add to the NeverInline set |
| 72 | for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) { |
| 73 | |
| 74 | // Get Source |
| 75 | const Constant *Elt = InitList->getOperand(i); |
| 76 | |
| 77 | if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(Elt)) |
| 78 | if (CE->getOpcode() == Instruction::BitCast) |
| 79 | Elt = CE->getOperand(0); |
| 80 | |
| 81 | // Insert into set of functions to never inline |
Tanya Lattner | b62fa8a | 2007-06-07 17:12:16 +0000 | [diff] [blame] | 82 | if (const Function *F = dyn_cast<Function>(Elt)) |
| 83 | NeverInline.insert(F); |
Tanya Lattner | 682f683 | 2007-06-06 21:59:26 +0000 | [diff] [blame] | 84 | } |
| 85 | |
| 86 | return false; |
| 87 | } |
Devang Patel | 6899b31 | 2007-07-25 18:00:25 +0000 | [diff] [blame] | 88 | |