contrib/llvm/lib/Target/NVPTX/NVPTXAllocaHoisting.cpp

   1 //===-- AllocaHoisting.cpp - Hoist allocas to the entry block --*- C++ -*-===//
   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 // Hoist the alloca instructions in the non-entry blocks to the entry blocks.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "NVPTXAllocaHoisting.h"
  15 #include "llvm/CodeGen/StackProtector.h"
  16 #include "llvm/IR/Constants.h"
  17 #include "llvm/IR/Function.h"
  18 #include "llvm/IR/Instructions.h"
  19 using namespace llvm;
  20
  21 namespace {
  22 // Hoisting the alloca instructions in the non-entry blocks to the entry
  23 // block.
  24 class NVPTXAllocaHoisting : public FunctionPass {
  25 public:
  26   static char ID; // Pass ID
  27   NVPTXAllocaHoisting() : FunctionPass(ID) {}
  28
  29   void getAnalysisUsage(AnalysisUsage &AU) const override {
  30     AU.addPreserved<StackProtector>();
  31   }
  32
  33   StringRef getPassName() const override {
  34     return "NVPTX specific alloca hoisting";
  35   }
  36
  37   bool runOnFunction(Function &function) override;
  38 };
  39 } // namespace
  40
  41 bool NVPTXAllocaHoisting::runOnFunction(Function &function) {
  42   bool functionModified = false;
  43   Function::iterator I = function.begin();
  44   TerminatorInst *firstTerminatorInst = (I++)->getTerminator();
  45
  46   for (Function::iterator E = function.end(); I != E; ++I) {
  47     for (BasicBlock::iterator BI = I->begin(), BE = I->end(); BI != BE;) {
  48       AllocaInst *allocaInst = dyn_cast<AllocaInst>(BI++);
  49       if (allocaInst && isa<ConstantInt>(allocaInst->getArraySize())) {
  50         allocaInst->moveBefore(firstTerminatorInst);
  51         functionModified = true;
  52       }
  53     }
  54   }
  55
  56   return functionModified;
  57 }
  58
  59 char NVPTXAllocaHoisting::ID = 0;
  60
  61 namespace llvm {
  62 void initializeNVPTXAllocaHoistingPass(PassRegistry &);
  63 }
  64
  65 INITIALIZE_PASS(
  66     NVPTXAllocaHoisting, "alloca-hoisting",
  67     "Hoisting alloca instructions in non-entry blocks to the entry block",
  68     false, false)
  69
  70 FunctionPass *llvm::createAllocaHoisting() { return new NVPTXAllocaHoisting; }