llvm/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 #include <optional>
  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, const Triple &TT, StringRef CPU,
  27                    StringRef FS, const TargetOptions &Options,
  28                    std::optional<Reloc::Model> RM,
  29                    std::optional<CodeModel::Model> CM, CodeGenOptLevel OL,
  30                    bool JIT);
  31
  32   const BPFSubtarget *getSubtargetImpl() const { return &Subtarget; }
  33   const BPFSubtarget *getSubtargetImpl(const Function &) const override {
  34     return &Subtarget;
  35   }
  36
  37   TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
  38
  39   TargetTransformInfo getTargetTransformInfo(const Function &F) const override;
  40
  41   TargetLoweringObjectFile *getObjFileLowering() const override {
  42     return TLOF.get();
  43   }
  44
  45   void registerPassBuilderCallbacks(PassBuilder &PB) override;
  46 };
  47 }
  48
  49 #endif