[bpf] fix build
[oota-llvm.git] / lib / Target / BPF / BPFTargetMachine.h
1 //===-- BPFTargetMachine.h - Define TargetMachine for BPF --- 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 // This file declares the BPF specific subclass of TargetMachine.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #ifndef LLVM_LIB_TARGET_BPF_BPFTARGETMACHINE_H
15 #define LLVM_LIB_TARGET_BPF_BPFTARGETMACHINE_H
16
17 #include "BPFSubtarget.h"
18 #include "llvm/Target/TargetMachine.h"
19
20 namespace llvm {
21 class BPFTargetMachine : public LLVMTargetMachine {
22   std::unique_ptr<TargetLoweringObjectFile> TLOF;
23   BPFSubtarget Subtarget;
24
25 public:
26   BPFTargetMachine(const Target &T, StringRef TT, StringRef CPU, StringRef FS,
27                    const TargetOptions &Options, Reloc::Model RM,
28                    CodeModel::Model CM, CodeGenOpt::Level OL);
29
30   const BPFSubtarget *getSubtargetImpl() const { return &Subtarget; }
31   const BPFSubtarget *getSubtargetImpl(const Function &) const override {
32     return &Subtarget;
33   }
34
35   TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
36
37   TargetLoweringObjectFile *getObjFileLowering() const override {
38     return TLOF.get();
39   }
40 };
41 }
42
43 #endif