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/IR/Constants.h"
16#include "llvm/IR/Function.h"
17#include "llvm/IR/Instructions.h"
18
19namespace llvm {
20
21bool NVPTXAllocaHoisting::runOnFunction(Function &function) {
22  bool functionModified = false;
23  Function::iterator I = function.begin();
24  TerminatorInst *firstTerminatorInst = (I++)->getTerminator();
25
26  for (Function::iterator E = function.end(); I != E; ++I) {
27    for (BasicBlock::iterator BI = I->begin(), BE = I->end(); BI != BE;) {
28      AllocaInst *allocaInst = dyn_cast<AllocaInst>(BI++);
29      if (allocaInst && isa<ConstantInt>(allocaInst->getArraySize())) {
30        allocaInst->moveBefore(firstTerminatorInst);
31        functionModified = true;
32      }
33    }
34  }
35
36  return functionModified;
37}
38
39char NVPTXAllocaHoisting::ID = 1;
40static RegisterPass<NVPTXAllocaHoisting>
41X("alloca-hoisting", "Hoisting alloca instructions in non-entry "
42                     "blocks to the entry block");
43
44FunctionPass *createAllocaHoisting() { return new NVPTXAllocaHoisting(); }
45
46} // end namespace llvm
47