xref: /freebsd-src/contrib/llvm-project/llvm/tools/opt/NewPMDriver.cpp (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
10b57cec5SDimitry Andric //===- NewPMDriver.cpp - Driver for opt with new PM -----------------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric /// \file
90b57cec5SDimitry Andric ///
100b57cec5SDimitry Andric /// This file is just a split of the code that logically belongs in opt.cpp but
110b57cec5SDimitry Andric /// that includes the new pass manager headers.
120b57cec5SDimitry Andric ///
130b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
140b57cec5SDimitry Andric 
150b57cec5SDimitry Andric #include "NewPMDriver.h"
165ffd83dbSDimitry Andric #include "llvm/ADT/SmallVector.h"
170b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h"
180b57cec5SDimitry Andric #include "llvm/Analysis/AliasAnalysis.h"
190b57cec5SDimitry Andric #include "llvm/Analysis/CGSCCPassManager.h"
20e8d8bef9SDimitry Andric #include "llvm/Analysis/TargetLibraryInfo.h"
210b57cec5SDimitry Andric #include "llvm/Bitcode/BitcodeWriterPass.h"
220b57cec5SDimitry Andric #include "llvm/Config/llvm-config.h"
230b57cec5SDimitry Andric #include "llvm/IR/Dominators.h"
240b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h"
250b57cec5SDimitry Andric #include "llvm/IR/Module.h"
260b57cec5SDimitry Andric #include "llvm/IR/PassManager.h"
270b57cec5SDimitry Andric #include "llvm/IR/Verifier.h"
28bdd1243dSDimitry Andric #include "llvm/IRPrinter/IRPrintingPasses.h"
290b57cec5SDimitry Andric #include "llvm/Passes/PassBuilder.h"
300b57cec5SDimitry Andric #include "llvm/Passes/PassPlugin.h"
310b57cec5SDimitry Andric #include "llvm/Passes/StandardInstrumentations.h"
320b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
330b57cec5SDimitry Andric #include "llvm/Support/ToolOutputFile.h"
3406c3fb27SDimitry Andric #include "llvm/Support/VirtualFileSystem.h"
35bdd1243dSDimitry Andric #include "llvm/Support/raw_ostream.h"
360b57cec5SDimitry Andric #include "llvm/Target/TargetMachine.h"
370b57cec5SDimitry Andric #include "llvm/Transforms/IPO/ThinLTOBitcodeWriter.h"
38e8d8bef9SDimitry Andric #include "llvm/Transforms/Instrumentation/AddressSanitizer.h"
390b57cec5SDimitry Andric #include "llvm/Transforms/Scalar/LoopPassManager.h"
40480093f4SDimitry Andric #include "llvm/Transforms/Utils/Debugify.h"
410b57cec5SDimitry Andric 
420b57cec5SDimitry Andric using namespace llvm;
430b57cec5SDimitry Andric using namespace opt_tool;
440b57cec5SDimitry Andric 
45e8d8bef9SDimitry Andric namespace llvm {
46e8d8bef9SDimitry Andric cl::opt<bool> DebugifyEach(
47e8d8bef9SDimitry Andric     "debugify-each",
48e8d8bef9SDimitry Andric     cl::desc("Start each pass with debugify and end it with check-debugify"));
49e8d8bef9SDimitry Andric 
50e8d8bef9SDimitry Andric cl::opt<std::string>
51e8d8bef9SDimitry Andric     DebugifyExport("debugify-export",
52e8d8bef9SDimitry Andric                    cl::desc("Export per-pass debugify statistics to this file"),
53e8d8bef9SDimitry Andric                    cl::value_desc("filename"));
54753f127fSDimitry Andric 
55753f127fSDimitry Andric cl::opt<bool> VerifyEachDebugInfoPreserve(
56753f127fSDimitry Andric     "verify-each-debuginfo-preserve",
57753f127fSDimitry Andric     cl::desc("Start each pass with collecting and end it with checking of "
58753f127fSDimitry Andric              "debug info preservation."));
59753f127fSDimitry Andric 
60753f127fSDimitry Andric cl::opt<std::string>
61753f127fSDimitry Andric     VerifyDIPreserveExport("verify-di-preserve-export",
62753f127fSDimitry Andric                    cl::desc("Export debug info preservation failures into "
63753f127fSDimitry Andric                             "specified (JSON) file (should be abs path as we use"
64753f127fSDimitry Andric                             " append mode to insert new JSON objects)"),
65753f127fSDimitry Andric                    cl::value_desc("filename"), cl::init(""));
66753f127fSDimitry Andric 
67e8d8bef9SDimitry Andric } // namespace llvm
68e8d8bef9SDimitry Andric 
69fe6060f1SDimitry Andric enum class DebugLogging { None, Normal, Verbose, Quiet };
70fe6060f1SDimitry Andric 
71fe6060f1SDimitry Andric static cl::opt<DebugLogging> DebugPM(
72fe6060f1SDimitry Andric     "debug-pass-manager", cl::Hidden, cl::ValueOptional,
73fe6060f1SDimitry Andric     cl::desc("Print pass management debugging information"),
74fe6060f1SDimitry Andric     cl::init(DebugLogging::None),
75fe6060f1SDimitry Andric     cl::values(
76fe6060f1SDimitry Andric         clEnumValN(DebugLogging::Normal, "", ""),
77fe6060f1SDimitry Andric         clEnumValN(DebugLogging::Quiet, "quiet",
78fe6060f1SDimitry Andric                    "Skip printing info about analyses"),
79fe6060f1SDimitry Andric         clEnumValN(
80fe6060f1SDimitry Andric             DebugLogging::Verbose, "verbose",
81fe6060f1SDimitry Andric             "Print extra information about adaptors and pass managers")));
820b57cec5SDimitry Andric 
830b57cec5SDimitry Andric // This flag specifies a textual description of the alias analysis pipeline to
840b57cec5SDimitry Andric // use when querying for aliasing information. It only works in concert with
850b57cec5SDimitry Andric // the "passes" flag above.
860b57cec5SDimitry Andric static cl::opt<std::string>
870b57cec5SDimitry Andric     AAPipeline("aa-pipeline",
880b57cec5SDimitry Andric                cl::desc("A textual description of the alias analysis "
890b57cec5SDimitry Andric                         "pipeline for handling managed aliasing queries"),
90e8d8bef9SDimitry Andric                cl::Hidden, cl::init("default"));
910b57cec5SDimitry Andric 
920b57cec5SDimitry Andric /// {{@ These options accept textual pipeline descriptions which will be
930b57cec5SDimitry Andric /// inserted into default pipelines at the respective extension points
940b57cec5SDimitry Andric static cl::opt<std::string> PeepholeEPPipeline(
950b57cec5SDimitry Andric     "passes-ep-peephole",
960b57cec5SDimitry Andric     cl::desc("A textual description of the function pass pipeline inserted at "
970b57cec5SDimitry Andric              "the Peephole extension points into default pipelines"),
980b57cec5SDimitry Andric     cl::Hidden);
990b57cec5SDimitry Andric static cl::opt<std::string> LateLoopOptimizationsEPPipeline(
1000b57cec5SDimitry Andric     "passes-ep-late-loop-optimizations",
1010b57cec5SDimitry Andric     cl::desc(
1020b57cec5SDimitry Andric         "A textual description of the loop pass pipeline inserted at "
1030b57cec5SDimitry Andric         "the LateLoopOptimizations extension point into default pipelines"),
1040b57cec5SDimitry Andric     cl::Hidden);
1050b57cec5SDimitry Andric static cl::opt<std::string> LoopOptimizerEndEPPipeline(
1060b57cec5SDimitry Andric     "passes-ep-loop-optimizer-end",
1070b57cec5SDimitry Andric     cl::desc("A textual description of the loop pass pipeline inserted at "
1080b57cec5SDimitry Andric              "the LoopOptimizerEnd extension point into default pipelines"),
1090b57cec5SDimitry Andric     cl::Hidden);
1100b57cec5SDimitry Andric static cl::opt<std::string> ScalarOptimizerLateEPPipeline(
1110b57cec5SDimitry Andric     "passes-ep-scalar-optimizer-late",
1120b57cec5SDimitry Andric     cl::desc("A textual description of the function pass pipeline inserted at "
1130b57cec5SDimitry Andric              "the ScalarOptimizerLate extension point into default pipelines"),
1140b57cec5SDimitry Andric     cl::Hidden);
1150b57cec5SDimitry Andric static cl::opt<std::string> CGSCCOptimizerLateEPPipeline(
1160b57cec5SDimitry Andric     "passes-ep-cgscc-optimizer-late",
1170b57cec5SDimitry Andric     cl::desc("A textual description of the cgscc pass pipeline inserted at "
1180b57cec5SDimitry Andric              "the CGSCCOptimizerLate extension point into default pipelines"),
1190b57cec5SDimitry Andric     cl::Hidden);
1200b57cec5SDimitry Andric static cl::opt<std::string> VectorizerStartEPPipeline(
1210b57cec5SDimitry Andric     "passes-ep-vectorizer-start",
1220b57cec5SDimitry Andric     cl::desc("A textual description of the function pass pipeline inserted at "
1230b57cec5SDimitry Andric              "the VectorizerStart extension point into default pipelines"),
1240b57cec5SDimitry Andric     cl::Hidden);
1250b57cec5SDimitry Andric static cl::opt<std::string> PipelineStartEPPipeline(
1260b57cec5SDimitry Andric     "passes-ep-pipeline-start",
127e8d8bef9SDimitry Andric     cl::desc("A textual description of the module pass pipeline inserted at "
1280b57cec5SDimitry Andric              "the PipelineStart extension point into default pipelines"),
1290b57cec5SDimitry Andric     cl::Hidden);
130e8d8bef9SDimitry Andric static cl::opt<std::string> PipelineEarlySimplificationEPPipeline(
131e8d8bef9SDimitry Andric     "passes-ep-pipeline-early-simplification",
132e8d8bef9SDimitry Andric     cl::desc("A textual description of the module pass pipeline inserted at "
133e8d8bef9SDimitry Andric              "the EarlySimplification extension point into default pipelines"),
134e8d8bef9SDimitry Andric     cl::Hidden);
13581ad6265SDimitry Andric static cl::opt<std::string> OptimizerEarlyEPPipeline(
13681ad6265SDimitry Andric     "passes-ep-optimizer-early",
13781ad6265SDimitry Andric     cl::desc("A textual description of the module pass pipeline inserted at "
13881ad6265SDimitry Andric              "the OptimizerEarly extension point into default pipelines"),
13981ad6265SDimitry Andric     cl::Hidden);
1400b57cec5SDimitry Andric static cl::opt<std::string> OptimizerLastEPPipeline(
1410b57cec5SDimitry Andric     "passes-ep-optimizer-last",
142e8d8bef9SDimitry Andric     cl::desc("A textual description of the module pass pipeline inserted at "
1430b57cec5SDimitry Andric              "the OptimizerLast extension point into default pipelines"),
1440b57cec5SDimitry Andric     cl::Hidden);
14581ad6265SDimitry Andric static cl::opt<std::string> FullLinkTimeOptimizationEarlyEPPipeline(
14681ad6265SDimitry Andric     "passes-ep-full-link-time-optimization-early",
14781ad6265SDimitry Andric     cl::desc("A textual description of the module pass pipeline inserted at "
14881ad6265SDimitry Andric              "the FullLinkTimeOptimizationEarly extension point into default "
14981ad6265SDimitry Andric              "pipelines"),
15081ad6265SDimitry Andric     cl::Hidden);
15181ad6265SDimitry Andric static cl::opt<std::string> FullLinkTimeOptimizationLastEPPipeline(
15281ad6265SDimitry Andric     "passes-ep-full-link-time-optimization-last",
15381ad6265SDimitry Andric     cl::desc("A textual description of the module pass pipeline inserted at "
15481ad6265SDimitry Andric              "the FullLinkTimeOptimizationLast extension point into default "
15581ad6265SDimitry Andric              "pipelines"),
15681ad6265SDimitry Andric     cl::Hidden);
157bdd1243dSDimitry Andric /// @}}
1580b57cec5SDimitry Andric 
159bdd1243dSDimitry Andric static cl::opt<bool> DisablePipelineVerification(
160bdd1243dSDimitry Andric     "disable-pipeline-verification",
161bdd1243dSDimitry Andric     cl::desc("Only has an effect when specified with -print-pipeline-passes. "
162bdd1243dSDimitry Andric              "Disables verifying that the textual pipeline generated by "
163bdd1243dSDimitry Andric              "-print-pipeline-passes can be used to create a pipeline."),
164bdd1243dSDimitry Andric     cl::Hidden);
1655ffd83dbSDimitry Andric 
166bdd1243dSDimitry Andric 
167bdd1243dSDimitry Andric static cl::opt<PGOKind>
168bdd1243dSDimitry Andric     PGOKindFlag("pgo-kind", cl::init(NoPGO), cl::Hidden,
169bdd1243dSDimitry Andric                 cl::desc("The kind of profile guided optimization"),
170bdd1243dSDimitry Andric                 cl::values(clEnumValN(NoPGO, "nopgo", "Do not use PGO."),
171bdd1243dSDimitry Andric                            clEnumValN(InstrGen, "pgo-instr-gen-pipeline",
172bdd1243dSDimitry Andric                                       "Instrument the IR to generate profile."),
173bdd1243dSDimitry Andric                            clEnumValN(InstrUse, "pgo-instr-use-pipeline",
174bdd1243dSDimitry Andric                                       "Use instrumented profile to guide PGO."),
175bdd1243dSDimitry Andric                            clEnumValN(SampleUse, "pgo-sample-use-pipeline",
176bdd1243dSDimitry Andric                                       "Use sampled profile to guide PGO.")));
177bdd1243dSDimitry Andric static cl::opt<std::string> ProfileFile("profile-file",
178bdd1243dSDimitry Andric                                  cl::desc("Path to the profile."), cl::Hidden);
17906c3fb27SDimitry Andric static cl::opt<std::string>
18006c3fb27SDimitry Andric     MemoryProfileFile("memory-profile-file",
18106c3fb27SDimitry Andric                       cl::desc("Path to the memory profile."), cl::Hidden);
182bdd1243dSDimitry Andric 
183bdd1243dSDimitry Andric static cl::opt<CSPGOKind> CSPGOKindFlag(
184bdd1243dSDimitry Andric     "cspgo-kind", cl::init(NoCSPGO), cl::Hidden,
185bdd1243dSDimitry Andric     cl::desc("The kind of context sensitive profile guided optimization"),
186bdd1243dSDimitry Andric     cl::values(
187bdd1243dSDimitry Andric         clEnumValN(NoCSPGO, "nocspgo", "Do not use CSPGO."),
188bdd1243dSDimitry Andric         clEnumValN(
189bdd1243dSDimitry Andric             CSInstrGen, "cspgo-instr-gen-pipeline",
190bdd1243dSDimitry Andric             "Instrument (context sensitive) the IR to generate profile."),
191bdd1243dSDimitry Andric         clEnumValN(
192bdd1243dSDimitry Andric             CSInstrUse, "cspgo-instr-use-pipeline",
193bdd1243dSDimitry Andric             "Use instrumented (context sensitive) profile to guide PGO.")));
194bdd1243dSDimitry Andric 
195bdd1243dSDimitry Andric static cl::opt<std::string> CSProfileGenFile(
196bdd1243dSDimitry Andric     "cs-profilegen-file",
197bdd1243dSDimitry Andric     cl::desc("Path to the instrumented context sensitive profile."),
198bdd1243dSDimitry Andric     cl::Hidden);
1990b57cec5SDimitry Andric 
2000b57cec5SDimitry Andric static cl::opt<std::string>
2010b57cec5SDimitry Andric     ProfileRemappingFile("profile-remapping-file",
2020b57cec5SDimitry Andric                          cl::desc("Path to the profile remapping file."),
2030b57cec5SDimitry Andric                          cl::Hidden);
204bdd1243dSDimitry Andric 
205*0fca6ea1SDimitry Andric static cl::opt<PGOOptions::ColdFuncOpt> PGOColdFuncAttr(
206*0fca6ea1SDimitry Andric     "pgo-cold-func-opt", cl::init(PGOOptions::ColdFuncOpt::Default), cl::Hidden,
207*0fca6ea1SDimitry Andric     cl::desc(
208*0fca6ea1SDimitry Andric         "Function attribute to apply to cold functions as determined by PGO"),
209*0fca6ea1SDimitry Andric     cl::values(clEnumValN(PGOOptions::ColdFuncOpt::Default, "default",
210*0fca6ea1SDimitry Andric                           "Default (no attribute)"),
211*0fca6ea1SDimitry Andric                clEnumValN(PGOOptions::ColdFuncOpt::OptSize, "optsize",
212*0fca6ea1SDimitry Andric                           "Mark cold functions with optsize."),
213*0fca6ea1SDimitry Andric                clEnumValN(PGOOptions::ColdFuncOpt::MinSize, "minsize",
214*0fca6ea1SDimitry Andric                           "Mark cold functions with minsize."),
215*0fca6ea1SDimitry Andric                clEnumValN(PGOOptions::ColdFuncOpt::OptNone, "optnone",
216*0fca6ea1SDimitry Andric                           "Mark cold functions with optnone.")));
217*0fca6ea1SDimitry Andric 
2180b57cec5SDimitry Andric static cl::opt<bool> DebugInfoForProfiling(
219bdd1243dSDimitry Andric     "debug-info-for-profiling", cl::init(false), cl::Hidden,
2200b57cec5SDimitry Andric     cl::desc("Emit special debug info to enable PGO profile generation."));
221bdd1243dSDimitry Andric 
222e8d8bef9SDimitry Andric static cl::opt<bool> PseudoProbeForProfiling(
223bdd1243dSDimitry Andric     "pseudo-probe-for-profiling", cl::init(false), cl::Hidden,
224e8d8bef9SDimitry Andric     cl::desc("Emit pseudo probes to enable PGO profile generation."));
225bdd1243dSDimitry Andric 
226bdd1243dSDimitry Andric static cl::opt<bool> DisableLoopUnrolling(
227bdd1243dSDimitry Andric     "disable-loop-unrolling",
228bdd1243dSDimitry Andric     cl::desc("Disable loop unrolling in all relevant passes"), cl::init(false));
229bdd1243dSDimitry Andric 
230bdd1243dSDimitry Andric namespace llvm {
231bdd1243dSDimitry Andric extern cl::opt<bool> PrintPipelinePasses;
232bdd1243dSDimitry Andric } // namespace llvm
2330b57cec5SDimitry Andric 
2340b57cec5SDimitry Andric template <typename PassManagerT>
2350b57cec5SDimitry Andric bool tryParsePipelineText(PassBuilder &PB,
2360b57cec5SDimitry Andric                           const cl::opt<std::string> &PipelineOpt) {
2370b57cec5SDimitry Andric   if (PipelineOpt.empty())
2380b57cec5SDimitry Andric     return false;
2390b57cec5SDimitry Andric 
2400b57cec5SDimitry Andric   // Verify the pipeline is parseable:
2410b57cec5SDimitry Andric   PassManagerT PM;
2420b57cec5SDimitry Andric   if (auto Err = PB.parsePassPipeline(PM, PipelineOpt)) {
2430b57cec5SDimitry Andric     errs() << "Could not parse -" << PipelineOpt.ArgStr
2440b57cec5SDimitry Andric            << " pipeline: " << toString(std::move(Err))
2450b57cec5SDimitry Andric            << "... I'm going to ignore it.\n";
2460b57cec5SDimitry Andric     return false;
2470b57cec5SDimitry Andric   }
2480b57cec5SDimitry Andric   return true;
2490b57cec5SDimitry Andric }
2500b57cec5SDimitry Andric 
2510b57cec5SDimitry Andric /// If one of the EPPipeline command line options was given, register callbacks
2520b57cec5SDimitry Andric /// for parsing and inserting the given pipeline
253e8d8bef9SDimitry Andric static void registerEPCallbacks(PassBuilder &PB) {
2540b57cec5SDimitry Andric   if (tryParsePipelineText<FunctionPassManager>(PB, PeepholeEPPipeline))
2550b57cec5SDimitry Andric     PB.registerPeepholeEPCallback(
256349cc55cSDimitry Andric         [&PB](FunctionPassManager &PM, OptimizationLevel Level) {
2570b57cec5SDimitry Andric           ExitOnError Err("Unable to parse PeepholeEP pipeline: ");
258e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, PeepholeEPPipeline));
2590b57cec5SDimitry Andric         });
2600b57cec5SDimitry Andric   if (tryParsePipelineText<LoopPassManager>(PB,
2610b57cec5SDimitry Andric                                             LateLoopOptimizationsEPPipeline))
2620b57cec5SDimitry Andric     PB.registerLateLoopOptimizationsEPCallback(
263349cc55cSDimitry Andric         [&PB](LoopPassManager &PM, OptimizationLevel Level) {
2640b57cec5SDimitry Andric           ExitOnError Err("Unable to parse LateLoopOptimizationsEP pipeline: ");
265e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, LateLoopOptimizationsEPPipeline));
2660b57cec5SDimitry Andric         });
2670b57cec5SDimitry Andric   if (tryParsePipelineText<LoopPassManager>(PB, LoopOptimizerEndEPPipeline))
2680b57cec5SDimitry Andric     PB.registerLoopOptimizerEndEPCallback(
269349cc55cSDimitry Andric         [&PB](LoopPassManager &PM, OptimizationLevel Level) {
2700b57cec5SDimitry Andric           ExitOnError Err("Unable to parse LoopOptimizerEndEP pipeline: ");
271e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, LoopOptimizerEndEPPipeline));
2720b57cec5SDimitry Andric         });
2730b57cec5SDimitry Andric   if (tryParsePipelineText<FunctionPassManager>(PB,
2740b57cec5SDimitry Andric                                                 ScalarOptimizerLateEPPipeline))
2750b57cec5SDimitry Andric     PB.registerScalarOptimizerLateEPCallback(
276349cc55cSDimitry Andric         [&PB](FunctionPassManager &PM, OptimizationLevel Level) {
2770b57cec5SDimitry Andric           ExitOnError Err("Unable to parse ScalarOptimizerLateEP pipeline: ");
278e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, ScalarOptimizerLateEPPipeline));
2790b57cec5SDimitry Andric         });
2800b57cec5SDimitry Andric   if (tryParsePipelineText<CGSCCPassManager>(PB, CGSCCOptimizerLateEPPipeline))
2810b57cec5SDimitry Andric     PB.registerCGSCCOptimizerLateEPCallback(
282349cc55cSDimitry Andric         [&PB](CGSCCPassManager &PM, OptimizationLevel Level) {
2830b57cec5SDimitry Andric           ExitOnError Err("Unable to parse CGSCCOptimizerLateEP pipeline: ");
284e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, CGSCCOptimizerLateEPPipeline));
2850b57cec5SDimitry Andric         });
2860b57cec5SDimitry Andric   if (tryParsePipelineText<FunctionPassManager>(PB, VectorizerStartEPPipeline))
2870b57cec5SDimitry Andric     PB.registerVectorizerStartEPCallback(
288349cc55cSDimitry Andric         [&PB](FunctionPassManager &PM, OptimizationLevel Level) {
2890b57cec5SDimitry Andric           ExitOnError Err("Unable to parse VectorizerStartEP pipeline: ");
290e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, VectorizerStartEPPipeline));
2910b57cec5SDimitry Andric         });
2920b57cec5SDimitry Andric   if (tryParsePipelineText<ModulePassManager>(PB, PipelineStartEPPipeline))
2930b57cec5SDimitry Andric     PB.registerPipelineStartEPCallback(
294349cc55cSDimitry Andric         [&PB](ModulePassManager &PM, OptimizationLevel) {
2950b57cec5SDimitry Andric           ExitOnError Err("Unable to parse PipelineStartEP pipeline: ");
296e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, PipelineStartEPPipeline));
297e8d8bef9SDimitry Andric         });
298e8d8bef9SDimitry Andric   if (tryParsePipelineText<ModulePassManager>(
299e8d8bef9SDimitry Andric           PB, PipelineEarlySimplificationEPPipeline))
300e8d8bef9SDimitry Andric     PB.registerPipelineEarlySimplificationEPCallback(
301349cc55cSDimitry Andric         [&PB](ModulePassManager &PM, OptimizationLevel) {
302e8d8bef9SDimitry Andric           ExitOnError Err("Unable to parse EarlySimplification pipeline: ");
303e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, PipelineEarlySimplificationEPPipeline));
3040b57cec5SDimitry Andric         });
30581ad6265SDimitry Andric   if (tryParsePipelineText<ModulePassManager>(PB, OptimizerEarlyEPPipeline))
30681ad6265SDimitry Andric     PB.registerOptimizerEarlyEPCallback(
30781ad6265SDimitry Andric         [&PB](ModulePassManager &PM, OptimizationLevel) {
30881ad6265SDimitry Andric           ExitOnError Err("Unable to parse OptimizerEarlyEP pipeline: ");
30981ad6265SDimitry Andric           Err(PB.parsePassPipeline(PM, OptimizerEarlyEPPipeline));
31081ad6265SDimitry Andric         });
31181ad6265SDimitry Andric   if (tryParsePipelineText<ModulePassManager>(PB, OptimizerLastEPPipeline))
3120b57cec5SDimitry Andric     PB.registerOptimizerLastEPCallback(
313349cc55cSDimitry Andric         [&PB](ModulePassManager &PM, OptimizationLevel) {
3140b57cec5SDimitry Andric           ExitOnError Err("Unable to parse OptimizerLastEP pipeline: ");
315e8d8bef9SDimitry Andric           Err(PB.parsePassPipeline(PM, OptimizerLastEPPipeline));
3160b57cec5SDimitry Andric         });
31781ad6265SDimitry Andric   if (tryParsePipelineText<ModulePassManager>(
31881ad6265SDimitry Andric           PB, FullLinkTimeOptimizationEarlyEPPipeline))
31981ad6265SDimitry Andric     PB.registerFullLinkTimeOptimizationEarlyEPCallback(
32081ad6265SDimitry Andric         [&PB](ModulePassManager &PM, OptimizationLevel) {
32181ad6265SDimitry Andric           ExitOnError Err(
32281ad6265SDimitry Andric               "Unable to parse FullLinkTimeOptimizationEarlyEP pipeline: ");
32381ad6265SDimitry Andric           Err(PB.parsePassPipeline(PM,
32481ad6265SDimitry Andric                                    FullLinkTimeOptimizationEarlyEPPipeline));
32581ad6265SDimitry Andric         });
32681ad6265SDimitry Andric   if (tryParsePipelineText<ModulePassManager>(
32781ad6265SDimitry Andric           PB, FullLinkTimeOptimizationLastEPPipeline))
32881ad6265SDimitry Andric     PB.registerFullLinkTimeOptimizationLastEPCallback(
32981ad6265SDimitry Andric         [&PB](ModulePassManager &PM, OptimizationLevel) {
33081ad6265SDimitry Andric           ExitOnError Err(
33181ad6265SDimitry Andric               "Unable to parse FullLinkTimeOptimizationLastEP pipeline: ");
33281ad6265SDimitry Andric           Err(PB.parsePassPipeline(PM, FullLinkTimeOptimizationLastEPPipeline));
33381ad6265SDimitry Andric         });
3340b57cec5SDimitry Andric }
3350b57cec5SDimitry Andric 
336480093f4SDimitry Andric #define HANDLE_EXTENSION(Ext)                                                  \
337480093f4SDimitry Andric   llvm::PassPluginLibraryInfo get##Ext##PluginInfo();
338480093f4SDimitry Andric #include "llvm/Support/Extension.def"
3390b57cec5SDimitry Andric 
34006c3fb27SDimitry Andric bool llvm::runPassPipeline(
34106c3fb27SDimitry Andric     StringRef Arg0, Module &M, TargetMachine *TM, TargetLibraryInfoImpl *TLII,
34206c3fb27SDimitry Andric     ToolOutputFile *Out, ToolOutputFile *ThinLTOLinkOut,
34306c3fb27SDimitry Andric     ToolOutputFile *OptRemarkFile, StringRef PassPipeline,
344*0fca6ea1SDimitry Andric     ArrayRef<PassPlugin> PassPlugins,
345*0fca6ea1SDimitry Andric     ArrayRef<std::function<void(llvm::PassBuilder &)>> PassBuilderCallbacks,
346*0fca6ea1SDimitry Andric     OutputKind OK, VerifierKind VK, bool ShouldPreserveAssemblyUseListOrder,
34706c3fb27SDimitry Andric     bool ShouldPreserveBitcodeUseListOrder, bool EmitSummaryIndex,
34806c3fb27SDimitry Andric     bool EmitModuleHash, bool EnableDebugify, bool VerifyDIPreserve,
34906c3fb27SDimitry Andric     bool UnifiedLTO) {
3500b57cec5SDimitry Andric   bool VerifyEachPass = VK == VK_VerifyEachPass;
3510b57cec5SDimitry Andric 
35206c3fb27SDimitry Andric   auto FS = vfs::getRealFileSystem();
353bdd1243dSDimitry Andric   std::optional<PGOOptions> P;
3540b57cec5SDimitry Andric   switch (PGOKindFlag) {
3550b57cec5SDimitry Andric   case InstrGen:
35606c3fb27SDimitry Andric     P = PGOOptions(ProfileFile, "", "", MemoryProfileFile, FS,
357*0fca6ea1SDimitry Andric                    PGOOptions::IRInstr, PGOOptions::NoCSAction,
358*0fca6ea1SDimitry Andric                    PGOColdFuncAttr);
3590b57cec5SDimitry Andric     break;
3600b57cec5SDimitry Andric   case InstrUse:
36106c3fb27SDimitry Andric     P = PGOOptions(ProfileFile, "", ProfileRemappingFile, MemoryProfileFile, FS,
362*0fca6ea1SDimitry Andric                    PGOOptions::IRUse, PGOOptions::NoCSAction, PGOColdFuncAttr);
3630b57cec5SDimitry Andric     break;
3640b57cec5SDimitry Andric   case SampleUse:
36506c3fb27SDimitry Andric     P = PGOOptions(ProfileFile, "", ProfileRemappingFile, MemoryProfileFile, FS,
366*0fca6ea1SDimitry Andric                    PGOOptions::SampleUse, PGOOptions::NoCSAction,
367*0fca6ea1SDimitry Andric                    PGOColdFuncAttr);
3680b57cec5SDimitry Andric     break;
3690b57cec5SDimitry Andric   case NoPGO:
37006c3fb27SDimitry Andric     if (DebugInfoForProfiling || PseudoProbeForProfiling ||
37106c3fb27SDimitry Andric         !MemoryProfileFile.empty())
37206c3fb27SDimitry Andric       P = PGOOptions("", "", "", MemoryProfileFile, FS, PGOOptions::NoAction,
373*0fca6ea1SDimitry Andric                      PGOOptions::NoCSAction, PGOColdFuncAttr,
374*0fca6ea1SDimitry Andric                      DebugInfoForProfiling, PseudoProbeForProfiling);
3750b57cec5SDimitry Andric     else
376bdd1243dSDimitry Andric       P = std::nullopt;
3770b57cec5SDimitry Andric   }
3780b57cec5SDimitry Andric   if (CSPGOKindFlag != NoCSPGO) {
3790b57cec5SDimitry Andric     if (P && (P->Action == PGOOptions::IRInstr ||
380bdd1243dSDimitry Andric               P->Action == PGOOptions::SampleUse)) {
3810b57cec5SDimitry Andric       errs() << "CSPGOKind cannot be used with IRInstr or SampleUse";
382bdd1243dSDimitry Andric       return false;
383bdd1243dSDimitry Andric     }
3840b57cec5SDimitry Andric     if (CSPGOKindFlag == CSInstrGen) {
385bdd1243dSDimitry Andric       if (CSProfileGenFile.empty()) {
3860b57cec5SDimitry Andric         errs() << "CSInstrGen needs to specify CSProfileGenFile";
387bdd1243dSDimitry Andric         return false;
388bdd1243dSDimitry Andric       }
3890b57cec5SDimitry Andric       if (P) {
3900b57cec5SDimitry Andric         P->CSAction = PGOOptions::CSIRInstr;
3910b57cec5SDimitry Andric         P->CSProfileGenFile = CSProfileGenFile;
3920b57cec5SDimitry Andric       } else
3930b57cec5SDimitry Andric         P = PGOOptions("", CSProfileGenFile, ProfileRemappingFile,
39406c3fb27SDimitry Andric                        /*MemoryProfile=*/"", FS, PGOOptions::NoAction,
39506c3fb27SDimitry Andric                        PGOOptions::CSIRInstr);
3960b57cec5SDimitry Andric     } else /* CSPGOKindFlag == CSInstrUse */ {
397bdd1243dSDimitry Andric       if (!P) {
3980b57cec5SDimitry Andric         errs() << "CSInstrUse needs to be together with InstrUse";
399bdd1243dSDimitry Andric         return false;
400bdd1243dSDimitry Andric       }
4010b57cec5SDimitry Andric       P->CSAction = PGOOptions::CSIRUse;
4020b57cec5SDimitry Andric     }
4030b57cec5SDimitry Andric   }
404349cc55cSDimitry Andric   if (TM)
405349cc55cSDimitry Andric     TM->setPGOOption(P);
406349cc55cSDimitry Andric 
407fe6060f1SDimitry Andric   LoopAnalysisManager LAM;
408fe6060f1SDimitry Andric   FunctionAnalysisManager FAM;
409fe6060f1SDimitry Andric   CGSCCAnalysisManager CGAM;
410fe6060f1SDimitry Andric   ModuleAnalysisManager MAM;
411fe6060f1SDimitry Andric 
4120b57cec5SDimitry Andric   PassInstrumentationCallbacks PIC;
413fe6060f1SDimitry Andric   PrintPassOptions PrintPassOpts;
414fe6060f1SDimitry Andric   PrintPassOpts.Verbose = DebugPM == DebugLogging::Verbose;
415fe6060f1SDimitry Andric   PrintPassOpts.SkipAnalyses = DebugPM == DebugLogging::Quiet;
416bdd1243dSDimitry Andric   StandardInstrumentations SI(M.getContext(), DebugPM != DebugLogging::None,
417bdd1243dSDimitry Andric                               VerifyEachPass, PrintPassOpts);
41806c3fb27SDimitry Andric   SI.registerCallbacks(PIC, &MAM);
419e8d8bef9SDimitry Andric   DebugifyEachInstrumentation Debugify;
420753f127fSDimitry Andric   DebugifyStatsMap DIStatsMap;
421753f127fSDimitry Andric   DebugInfoPerPass DebugInfoBeforePass;
422753f127fSDimitry Andric   if (DebugifyEach) {
423753f127fSDimitry Andric     Debugify.setDIStatsMap(DIStatsMap);
424753f127fSDimitry Andric     Debugify.setDebugifyMode(DebugifyMode::SyntheticDebugInfo);
42506c3fb27SDimitry Andric     Debugify.registerCallbacks(PIC, MAM);
426753f127fSDimitry Andric   } else if (VerifyEachDebugInfoPreserve) {
427753f127fSDimitry Andric     Debugify.setDebugInfoBeforePass(DebugInfoBeforePass);
428753f127fSDimitry Andric     Debugify.setDebugifyMode(DebugifyMode::OriginalDebugInfo);
429753f127fSDimitry Andric     Debugify.setOrigDIVerifyBugsReportFilePath(
430753f127fSDimitry Andric       VerifyDIPreserveExport);
43106c3fb27SDimitry Andric     Debugify.registerCallbacks(PIC, MAM);
432753f127fSDimitry Andric   }
4330b57cec5SDimitry Andric 
4345ffd83dbSDimitry Andric   PipelineTuningOptions PTO;
4355ffd83dbSDimitry Andric   // LoopUnrolling defaults on to true and DisableLoopUnrolling is initialized
4365ffd83dbSDimitry Andric   // to false above so we shouldn't necessarily need to check whether or not the
4375ffd83dbSDimitry Andric   // option has been enabled.
4385ffd83dbSDimitry Andric   PTO.LoopUnrolling = !DisableLoopUnrolling;
43906c3fb27SDimitry Andric   PTO.UnifiedLTO = UnifiedLTO;
440fe6060f1SDimitry Andric   PassBuilder PB(TM, PTO, P, &PIC);
441e8d8bef9SDimitry Andric   registerEPCallbacks(PB);
4420b57cec5SDimitry Andric 
44381ad6265SDimitry Andric   // For any loaded plugins, let them register pass builder callbacks.
44481ad6265SDimitry Andric   for (auto &PassPlugin : PassPlugins)
44581ad6265SDimitry Andric     PassPlugin.registerPassBuilderCallbacks(PB);
4460b57cec5SDimitry Andric 
447*0fca6ea1SDimitry Andric   // Load any explicitly specified plugins.
448*0fca6ea1SDimitry Andric   for (auto &PassCallback : PassBuilderCallbacks)
449*0fca6ea1SDimitry Andric     PassCallback(PB);
450*0fca6ea1SDimitry Andric 
451480093f4SDimitry Andric #define HANDLE_EXTENSION(Ext)                                                  \
452480093f4SDimitry Andric   get##Ext##PluginInfo().RegisterPassBuilderCallbacks(PB);
453480093f4SDimitry Andric #include "llvm/Support/Extension.def"
4540b57cec5SDimitry Andric 
4550b57cec5SDimitry Andric   // Specially handle the alias analysis manager so that we can register
4560b57cec5SDimitry Andric   // a custom pipeline of AA passes with it.
4570b57cec5SDimitry Andric   AAManager AA;
4580b57cec5SDimitry Andric   if (auto Err = PB.parseAAPipeline(AA, AAPipeline)) {
4590b57cec5SDimitry Andric     errs() << Arg0 << ": " << toString(std::move(Err)) << "\n";
4600b57cec5SDimitry Andric     return false;
4610b57cec5SDimitry Andric   }
4620b57cec5SDimitry Andric 
4630b57cec5SDimitry Andric   // Register the AA manager first so that our version is the one used.
4640b57cec5SDimitry Andric   FAM.registerPass([&] { return std::move(AA); });
465e8d8bef9SDimitry Andric   // Register our TargetLibraryInfoImpl.
466e8d8bef9SDimitry Andric   FAM.registerPass([&] { return TargetLibraryAnalysis(*TLII); });
4670b57cec5SDimitry Andric 
4680b57cec5SDimitry Andric   // Register all the basic analyses with the managers.
4690b57cec5SDimitry Andric   PB.registerModuleAnalyses(MAM);
4700b57cec5SDimitry Andric   PB.registerCGSCCAnalyses(CGAM);
4710b57cec5SDimitry Andric   PB.registerFunctionAnalyses(FAM);
4720b57cec5SDimitry Andric   PB.registerLoopAnalyses(LAM);
4730b57cec5SDimitry Andric   PB.crossRegisterProxies(LAM, FAM, CGAM, MAM);
4740b57cec5SDimitry Andric 
475fe6060f1SDimitry Andric   ModulePassManager MPM;
4760b57cec5SDimitry Andric   if (EnableDebugify)
4770b57cec5SDimitry Andric     MPM.addPass(NewPMDebugifyPass());
478753f127fSDimitry Andric   if (VerifyDIPreserve)
479753f127fSDimitry Andric     MPM.addPass(NewPMDebugifyPass(DebugifyMode::OriginalDebugInfo, "",
480753f127fSDimitry Andric                                   &DebugInfoBeforePass));
4810b57cec5SDimitry Andric 
482349cc55cSDimitry Andric   // Add passes according to the -passes options.
4835ffd83dbSDimitry Andric   if (!PassPipeline.empty()) {
484e8d8bef9SDimitry Andric     if (auto Err = PB.parsePassPipeline(MPM, PassPipeline)) {
4850b57cec5SDimitry Andric       errs() << Arg0 << ": " << toString(std::move(Err)) << "\n";
4860b57cec5SDimitry Andric       return false;
4870b57cec5SDimitry Andric     }
4885ffd83dbSDimitry Andric   }
4890b57cec5SDimitry Andric 
4900b57cec5SDimitry Andric   if (VK > VK_NoVerifier)
4910b57cec5SDimitry Andric     MPM.addPass(VerifierPass());
4920b57cec5SDimitry Andric   if (EnableDebugify)
493753f127fSDimitry Andric     MPM.addPass(NewPMCheckDebugifyPass(false, "", &DIStatsMap));
494753f127fSDimitry Andric   if (VerifyDIPreserve)
495753f127fSDimitry Andric     MPM.addPass(NewPMCheckDebugifyPass(
496bdd1243dSDimitry Andric         false, "", nullptr, DebugifyMode::OriginalDebugInfo,
497bdd1243dSDimitry Andric         &DebugInfoBeforePass, VerifyDIPreserveExport));
4980b57cec5SDimitry Andric 
4990b57cec5SDimitry Andric   // Add any relevant output pass at the end of the pipeline.
5000b57cec5SDimitry Andric   switch (OK) {
5010b57cec5SDimitry Andric   case OK_NoOutput:
5020b57cec5SDimitry Andric     break; // No output pass needed.
5030b57cec5SDimitry Andric   case OK_OutputAssembly:
504bdd1243dSDimitry Andric     MPM.addPass(PrintModulePass(
505bdd1243dSDimitry Andric         Out->os(), "", ShouldPreserveAssemblyUseListOrder, EmitSummaryIndex));
5060b57cec5SDimitry Andric     break;
5070b57cec5SDimitry Andric   case OK_OutputBitcode:
5080b57cec5SDimitry Andric     MPM.addPass(BitcodeWriterPass(Out->os(), ShouldPreserveBitcodeUseListOrder,
5090b57cec5SDimitry Andric                                   EmitSummaryIndex, EmitModuleHash));
5100b57cec5SDimitry Andric     break;
5110b57cec5SDimitry Andric   case OK_OutputThinLTOBitcode:
5120b57cec5SDimitry Andric     MPM.addPass(ThinLTOBitcodeWriterPass(
5130b57cec5SDimitry Andric         Out->os(), ThinLTOLinkOut ? &ThinLTOLinkOut->os() : nullptr));
5140b57cec5SDimitry Andric     break;
5150b57cec5SDimitry Andric   }
5160b57cec5SDimitry Andric 
5170b57cec5SDimitry Andric   // Before executing passes, print the final values of the LLVM options.
5180b57cec5SDimitry Andric   cl::PrintOptionValues();
5190b57cec5SDimitry Andric 
520349cc55cSDimitry Andric   // Print a textual, '-passes=' compatible, representation of pipeline if
521349cc55cSDimitry Andric   // requested.
522349cc55cSDimitry Andric   if (PrintPipelinePasses) {
523bdd1243dSDimitry Andric     std::string Pipeline;
524bdd1243dSDimitry Andric     raw_string_ostream SOS(Pipeline);
525bdd1243dSDimitry Andric     MPM.printPipeline(SOS, [&PIC](StringRef ClassName) {
526349cc55cSDimitry Andric       auto PassName = PIC.getPassNameForClassName(ClassName);
527349cc55cSDimitry Andric       return PassName.empty() ? ClassName : PassName;
528349cc55cSDimitry Andric     });
529bdd1243dSDimitry Andric     outs() << Pipeline;
530349cc55cSDimitry Andric     outs() << "\n";
531bdd1243dSDimitry Andric 
532bdd1243dSDimitry Andric     if (!DisablePipelineVerification) {
533bdd1243dSDimitry Andric       // Check that we can parse the returned pipeline string as an actual
534bdd1243dSDimitry Andric       // pipeline.
535bdd1243dSDimitry Andric       ModulePassManager TempPM;
536bdd1243dSDimitry Andric       if (auto Err = PB.parsePassPipeline(TempPM, Pipeline)) {
537bdd1243dSDimitry Andric         errs() << "Could not parse dumped pass pipeline: "
538bdd1243dSDimitry Andric                << toString(std::move(Err)) << "\n";
539bdd1243dSDimitry Andric         return false;
540bdd1243dSDimitry Andric       }
541bdd1243dSDimitry Andric     }
542bdd1243dSDimitry Andric 
543349cc55cSDimitry Andric     return true;
544349cc55cSDimitry Andric   }
545349cc55cSDimitry Andric 
5460b57cec5SDimitry Andric   // Now that we have all of the passes ready, run them.
5470b57cec5SDimitry Andric   MPM.run(M, MAM);
5480b57cec5SDimitry Andric 
5490b57cec5SDimitry Andric   // Declare success.
5500b57cec5SDimitry Andric   if (OK != OK_NoOutput) {
5510b57cec5SDimitry Andric     Out->keep();
5520b57cec5SDimitry Andric     if (OK == OK_OutputThinLTOBitcode && ThinLTOLinkOut)
5530b57cec5SDimitry Andric       ThinLTOLinkOut->keep();
5540b57cec5SDimitry Andric   }
5550b57cec5SDimitry Andric 
5560b57cec5SDimitry Andric   if (OptRemarkFile)
5570b57cec5SDimitry Andric     OptRemarkFile->keep();
5580b57cec5SDimitry Andric 
559e8d8bef9SDimitry Andric   if (DebugifyEach && !DebugifyExport.empty())
560753f127fSDimitry Andric     exportDebugifyStats(DebugifyExport, Debugify.getDebugifyStatsMap());
561e8d8bef9SDimitry Andric 
5620b57cec5SDimitry Andric   return true;
5630b57cec5SDimitry Andric }
564fe6060f1SDimitry Andric 
565fe6060f1SDimitry Andric void llvm::printPasses(raw_ostream &OS) {
566fe6060f1SDimitry Andric   PassBuilder PB;
567fe6060f1SDimitry Andric   PB.printPassNames(OS);
568fe6060f1SDimitry Andric }
569