lib/Target/BPF/BPFTargetMachine.h

   1 //===-- BPFTargetMachine.h - Define TargetMachine for BPF --- C++ ---===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file declares the BPF specific subclass of TargetMachine.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef LLVM_LIB_TARGET_BPF_BPFTARGETMACHINE_H
  14 #define LLVM_LIB_TARGET_BPF_BPFTARGETMACHINE_H
  15
  16 #include "BPFSubtarget.h"
  17 #include "llvm/Target/TargetMachine.h"
  18
  19 namespace llvm {
  20 class BPFTargetMachine : public LLVMTargetMachine {
  21   std::unique_ptr<TargetLoweringObjectFile> TLOF;
  22   BPFSubtarget Subtarget;
  23
  24 public:
  25   BPFTargetMachine(const Target &T, const Triple &TT, StringRef CPU,
  26                    StringRef FS, const TargetOptions &Options,
  27                    Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM,
  28                    CodeGenOpt::Level OL, bool JIT);
  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