1//===----------------------------------------------------------------------===//
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#include "Lanai.h"
10#include "LanaiAsmPrinter.h"
11#include "LanaiTargetMachine.h"
12#include "llvm/CodeGen/AtomicExpand.h"
13#include "llvm/IR/PassInstrumentation.h"
14#include "llvm/MC/MCStreamer.h"
15#include "llvm/Passes/CodeGenPassBuilder.h"
16#include "llvm/Passes/PassBuilder.h"
17#include "llvm/Target/CGPassBuilderOption.h"
18
19using namespace llvm;
20
21namespace {
22
23class LanaiCodeGenPassBuilder : public CodeGenPassBuilder {
24 using Base = CodeGenPassBuilder;
25
26 LanaiTargetMachine &getTM() const {
27 return static_cast<LanaiTargetMachine &>(TM);
28 }
29
30public:
31 explicit LanaiCodeGenPassBuilder(LanaiTargetMachine &TM,
32 const CGPassBuilderOption &Opts,
33 PassInstrumentationCallbacks *PIC)
34 : CodeGenPassBuilder(TM, Opts, PIC) {}
35
36 void addIRPasses(PassManagerWrapper &PMW) override;
37 Error addInstSelector(PassManagerWrapper &PMW) override;
38 void addPreSched2(PassManagerWrapper &PMW) override;
39 void addPreEmitPass(PassManagerWrapper &PMW) override;
40 void addAsmPrinterBegin(PassManagerWrapper &PMW) override;
41 void addAsmPrinter(PassManagerWrapper &PMW) override;
42 void addAsmPrinterEnd(PassManagerWrapper &PMW) override;
43};
44
45void LanaiCodeGenPassBuilder::addIRPasses(PassManagerWrapper &PMW) {
46 addFunctionPass(Pass: AtomicExpandPass(TM), PMW);
47
48 Base::addIRPasses(PMW);
49}
50
51Error LanaiCodeGenPassBuilder::addInstSelector(PassManagerWrapper &PMW) {
52 addMachineFunctionPass(Pass: LanaiISelDAGToDAGPass(getTM()), PMW);
53 return Error::success();
54}
55
56void LanaiCodeGenPassBuilder::addPreSched2(PassManagerWrapper &PMW) {
57 addMachineFunctionPass(Pass: LanaiMemAluCombinerPass(), PMW);
58}
59
60void LanaiCodeGenPassBuilder::addPreEmitPass(PassManagerWrapper &PMW) {
61 addMachineFunctionPass(Pass: LanaiDelaySlotFillerPass(), PMW);
62}
63
64void LanaiCodeGenPassBuilder::addAsmPrinterBegin(PassManagerWrapper &PMW) {
65 addModulePass(Pass: LanaiAsmPrinterBeginPass(), PMW, /*Force=*/true);
66}
67
68void LanaiCodeGenPassBuilder::addAsmPrinter(PassManagerWrapper &PMW) {
69 addMachineFunctionPass(Pass: LanaiAsmPrinterPass(), PMW);
70}
71
72void LanaiCodeGenPassBuilder::addAsmPrinterEnd(PassManagerWrapper &PMW) {
73 addModulePass(Pass: LanaiAsmPrinterEndPass(), PMW, /*Force=*/true);
74}
75
76} // namespace
77
78void LanaiTargetMachine::registerPassBuilderCallbacks(PassBuilder &PB){
79#define GET_PASS_REGISTRY "LanaiPassRegistry.def"
80#include "llvm/Passes/TargetPassRegistry.inc"
81}
82
83Error LanaiTargetMachine::buildCodeGenPipeline(
84 ModulePassManager &MPM, ModuleAnalysisManager &MAM, raw_pwrite_stream &Out,
85 raw_pwrite_stream *DwoOut, CodeGenFileType FileType,
86 const CGPassBuilderOption &Opt, MCContext &Ctx,
87 PassInstrumentationCallbacks *PIC) {
88 auto CGPB = LanaiCodeGenPassBuilder(*this, Opt, PIC);
89 return CGPB.buildPipeline(MPM, MAM, Out, DwoOut, FileType, Ctx);
90}
91