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
24 : public CodeGenPassBuilder<LanaiCodeGenPassBuilder, LanaiTargetMachine> {
25 using Base = CodeGenPassBuilder<LanaiCodeGenPassBuilder, LanaiTargetMachine>;
26
27public:
28 explicit LanaiCodeGenPassBuilder(LanaiTargetMachine &TM,
29 const CGPassBuilderOption &Opts,
30 PassInstrumentationCallbacks *PIC)
31 : CodeGenPassBuilder(TM, Opts, PIC) {}
32
33 void addIRPasses(PassManagerWrapper &PMW) const;
34 Error addInstSelector(PassManagerWrapper &PMW) const;
35 void addPreSched2(PassManagerWrapper &PMW) const;
36 void addPreEmitPass(PassManagerWrapper &PMW) const;
37 void addAsmPrinterBegin(PassManagerWrapper &PMW) const;
38 void addAsmPrinter(PassManagerWrapper &PMW) const;
39 void addAsmPrinterEnd(PassManagerWrapper &PMW) const;
40};
41
42void LanaiCodeGenPassBuilder::addIRPasses(PassManagerWrapper &PMW) const {
43 addFunctionPass(Pass: AtomicExpandPass(TM), PMW);
44
45 Base::addIRPasses(PMW);
46}
47
48Error LanaiCodeGenPassBuilder::addInstSelector(PassManagerWrapper &PMW) const {
49 addMachineFunctionPass(Pass: LanaiISelDAGToDAGPass(TM), PMW);
50 return Error::success();
51}
52
53void LanaiCodeGenPassBuilder::addPreSched2(PassManagerWrapper &PMW) const {
54 addMachineFunctionPass(Pass: LanaiMemAluCombinerPass(), PMW);
55}
56
57void LanaiCodeGenPassBuilder::addPreEmitPass(PassManagerWrapper &PMW) const {
58 addMachineFunctionPass(Pass: LanaiDelaySlotFillerPass(), PMW);
59}
60
61void LanaiCodeGenPassBuilder::addAsmPrinterBegin(
62 PassManagerWrapper &PMW) const {
63 addModulePass(Pass: LanaiAsmPrinterBeginPass(), PMW, /*Force=*/true);
64}
65
66void LanaiCodeGenPassBuilder::addAsmPrinter(PassManagerWrapper &PMW) const {
67 addMachineFunctionPass(Pass: LanaiAsmPrinterPass(), PMW);
68}
69
70void LanaiCodeGenPassBuilder::addAsmPrinterEnd(PassManagerWrapper &PMW) const {
71 addModulePass(Pass: LanaiAsmPrinterEndPass(), PMW);
72}
73
74} // namespace
75
76void LanaiTargetMachine::registerPassBuilderCallbacks(PassBuilder &PB) {
77#define GET_PASS_REGISTRY "LanaiPassRegistry.def"
78#include "llvm/Passes/TargetPassRegistry.inc"
79 // TODO(boomanaiden154): Move this into the base CodeGenPassBuilder once all
80 // targets that currently implement it have a ported asm-printer pass.
81 if (PIC) {
82 PIC->addClassToPassName(ClassName: LanaiAsmPrinterBeginPass::name(),
83 PassName: "lanai-asm-printer-begin");
84 PIC->addClassToPassName(ClassName: LanaiAsmPrinterPass::name(), PassName: "lanai-asmprinter");
85 PIC->addClassToPassName(ClassName: LanaiAsmPrinterEndPass::name(),
86 PassName: "lanai-asm-printer-end");
87 }
88}
89
90Error LanaiTargetMachine::buildCodeGenPipeline(
91 ModulePassManager &MPM, ModuleAnalysisManager &MAM, raw_pwrite_stream &Out,
92 raw_pwrite_stream *DwoOut, CodeGenFileType FileType,
93 const CGPassBuilderOption &Opt, MCContext &Ctx,
94 PassInstrumentationCallbacks *PIC) {
95 auto CGPB = LanaiCodeGenPassBuilder(*this, Opt, PIC);
96 return CGPB.buildPipeline(MPM, MAM, Out, DwoOut, FileType, Ctx);
97}
98