lib/Target/AMDGPU/AMDGPUAlwaysInlinePass.cpp

   1 //===-- AMDGPUAlwaysInlinePass.cpp - Promote Allocas ----------------------===//
   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 /// \file
  11 /// This pass marks all internal functions as always_inline and creates
  12 /// duplicates of all other functions a marks the duplicates as always_inline.
  13 //
  14 //===----------------------------------------------------------------------===//
  15
  16 #include "AMDGPU.h"
  17 #include "llvm/IR/Module.h"
  18 #include "llvm/Transforms/Utils/Cloning.h"
  19
  20 using namespace llvm;
  21
  22 namespace {
  23
  24 class AMDGPUAlwaysInline : public ModulePass {
  25
  26   static char ID;
  27
  28 public:
  29   AMDGPUAlwaysInline() : ModulePass(ID) { }
  30   bool runOnModule(Module &M) override;
  31   const char *getPassName() const override { return "AMDGPU Always Inline Pass"; }
  32 };
  33
  34 } // End anonymous namespace
  35
  36 char AMDGPUAlwaysInline::ID = 0;
  37
  38 bool AMDGPUAlwaysInline::runOnModule(Module &M) {
  39
  40   std::vector<Function*> FuncsToClone;
  41   for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I) {
  42     Function &F = *I;
  43     if (!F.hasLocalLinkage() && !F.isDeclaration() && !F.use_empty() &&
  44         !F.hasFnAttribute(Attribute::NoInline))
  45       FuncsToClone.push_back(&F);
  46   }
  47
  48   for (Function *F : FuncsToClone) {
  49     ValueToValueMapTy VMap;
  50     Function *NewFunc = CloneFunction(F, VMap, false);
  51     NewFunc->setLinkage(GlobalValue::InternalLinkage);
  52     F->getParent()->getFunctionList().push_back(NewFunc);
  53     F->replaceAllUsesWith(NewFunc);
  54   }
  55
  56   for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I) {
  57     Function &F = *I;
  58     if (F.hasLocalLinkage() && !F.hasFnAttribute(Attribute::NoInline)) {
  59       F.addFnAttr(Attribute::AlwaysInline);
  60     }
  61   }
  62   return false;
  63 }
  64
  65 ModulePass *llvm::createAMDGPUAlwaysInlinePass() {
  66   return new AMDGPUAlwaysInline();
  67 }