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   TargetTransformInfo getTargetTransformInfo(const Function &F) override;
38 
39   TargetLoweringObjectFile *getObjFileLowering() const override {
40     return TLOF.get();
41   }
42 
43   void adjustPassManager(PassManagerBuilder &) override;
44   void registerPassBuilderCallbacks(PassBuilder &PB,
45                                     bool DebugPassManager) override;
46 };
47 }
48 
49 #endif
50