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 getSubtargetImpl()32 const BPFSubtarget *getSubtargetImpl() const { return &Subtarget; } getSubtargetImpl(const Function &)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 getObjFileLowering()41 TargetLoweringObjectFile *getObjFileLowering() const override { 42 return TLOF.get(); 43 } 44 45 void registerPassBuilderCallbacks(PassBuilder &PB) override; 46 }; 47 } 48 49 #endif 50