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
19namespace llvm {
20class BPFTargetMachine : public LLVMTargetMachine {
21  std::unique_ptr<TargetLoweringObjectFile> TLOF;
22  BPFSubtarget Subtarget;
23
24public:
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
44