summaryrefslogtreecommitdiff
path: root/lib/Target/PTX/PTXRegAlloc.cpp
blob: 2d2d5c30c8ca989e72533d1ab68b5514c96857e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
//===-- PTXRegAlloc.cpp - PTX Register Allocator --------------------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file contains a register allocator for PTX code.
//
//===----------------------------------------------------------------------===//

#define DEBUG_TYPE "ptx-reg-alloc"

#include "PTX.h"
#include "llvm/CodeGen/MachineFunctionPass.h"
#include "llvm/CodeGen/RegAllocRegistry.h"

using namespace llvm;

namespace {
  // Special register allocator for PTX.
  class PTXRegAlloc : public MachineFunctionPass {
  public:
    static char ID;
    PTXRegAlloc() : MachineFunctionPass(ID) {
      initializePHIEliminationPass(*PassRegistry::getPassRegistry());
      initializeTwoAddressInstructionPassPass(*PassRegistry::getPassRegistry());
    }

    virtual const char* getPassName() const {
      return "PTX Register Allocator";
    }

    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
      AU.setPreservesCFG();
      AU.addRequiredID(PHIEliminationID);
      AU.addRequiredID(TwoAddressInstructionPassID);
      MachineFunctionPass::getAnalysisUsage(AU);
    }

    virtual bool runOnMachineFunction(MachineFunction &MF) {
      // We do not actually do anything (at least not yet).
      return false;
    }
  };

  char PTXRegAlloc::ID = 0;

  static RegisterRegAlloc
    ptxRegAlloc("ptx", "PTX register allocator", createPTXRegisterAllocator);
}

FunctionPass *llvm::createPTXRegisterAllocator() {
  return new PTXRegAlloc();
}