Index: include/llvm/Passes/PassBuilder.h =================================================================== --- include/llvm/Passes/PassBuilder.h +++ include/llvm/Passes/PassBuilder.h @@ -29,10 +29,14 @@ /// A struct capturing PGO tunables. struct PGOOptions { - std::string ProfileGenFile = ""; - std::string ProfileUseFile = ""; - std::string SampleProfileFile = ""; - bool RunProfileGen = false; + PGOOptions(std::string ProfileGenFile = "", std::string ProfileUseFile = "", + std::string SampleProfileFile = "", bool RunProfileGen = false) + : ProfileGenFile(ProfileGenFile), ProfileUseFile(ProfileUseFile), + SampleProfileFile(SampleProfileFile), RunProfileGen(RunProfileGen) {} + std::string ProfileGenFile; + std::string ProfileUseFile; + std::string SampleProfileFile; + bool RunProfileGen; }; /// \brief This class provides access to building LLVM's passes. Index: test/Other/Inputs/new-pm-pgo.prof =================================================================== --- /dev/null +++ test/Other/Inputs/new-pm-pgo.prof @@ -0,0 +1 @@ +foo:0:0 Index: test/Other/Inputs/new-pm-pgo.proftext =================================================================== --- /dev/null +++ test/Other/Inputs/new-pm-pgo.proftext @@ -0,0 +1 @@ +:ir Index: test/Other/new-pm-pgo.ll =================================================================== --- /dev/null +++ test/Other/new-pm-pgo.ll @@ -0,0 +1,12 @@ +; RUN: opt -debug-pass-manager -passes='default' -pgo-kind=new-pm-pgo-instr-gen-pipeline -profile-file='temp' %s 2>&1 |FileCheck %s --check-prefixes=GEN +; RUN: llvm-profdata merge %S/Inputs/new-pm-pgo.proftext -o %t.profdata +; RUN: opt -debug-pass-manager -passes='default' -pgo-kind=new-pm-pgo-instr-use-pipeline -profile-file='%t.profdata' %s 2>&1 |FileCheck %s --check-prefixes=USE +; RUN: opt -debug-pass-manager -passes='default' -pgo-kind=new-pm-pgo-sample-use-pipeline -profile-file='%S/Inputs/new-pm-pgo.prof' %s 2>&1 |FileCheck %s --check-prefixes=SAMPLE_USE +; +; GEN: Running pass: PGOInstrumentationGen +; USE: Running pass: PGOInstrumentationUse +; SAMPLE_USE: Running pass: SampleProfileLoaderPass + +define void @foo() { + ret void +} Index: tools/opt/NewPMDriver.cpp =================================================================== --- tools/opt/NewPMDriver.cpp +++ tools/opt/NewPMDriver.cpp @@ -81,6 +81,19 @@ cl::desc("A textual description of the function pass pipeline inserted at " "the VectorizerStart extension point into default pipelines"), cl::Hidden); +enum PGOKind { NoPGO, InstrGen, InstrUse, SampleUse }; +static cl::opt PGOKindFlag( + "pgo-kind", cl::init(NoPGO), cl::Hidden, + cl::desc("The kind of profile guided optimization"), + cl::values(clEnumValN(NoPGO, "nopgo", "Do not use PGO."), + clEnumValN(InstrGen, "new-pm-pgo-instr-gen-pipeline", + "Instrument the IR to generate profile."), + clEnumValN(InstrUse, "new-pm-pgo-instr-use-pipeline", + "Use instrumented profile to guide PGO."), + clEnumValN(SampleUse, "new-pm-pgo-sample-use-pipeline", + "Use sampled profile to guide PGO."))); +static cl::opt ProfileFile( + "profile-file", cl::desc("Path to the profile."), cl::Hidden); /// @}} template @@ -153,7 +166,22 @@ bool ShouldPreserveBitcodeUseListOrder, bool EmitSummaryIndex, bool EmitModuleHash) { bool VerifyEachPass = VK == VK_VerifyEachPass; - PassBuilder PB(TM); + + Optional P; + switch (PGOKindFlag) { + case InstrGen: + P = PGOOptions(ProfileFile, "", "", true); + break; + case InstrUse: + P = PGOOptions("", ProfileFile, "", false); + break; + case SampleUse: + P = PGOOptions("", "", ProfileFile, false); + break; + case NoPGO: + P = None; + } + PassBuilder PB(TM, P); registerEPCallbacks(PB, VerifyEachPass, DebugPM); // Specially handle the alias analysis manager so that we can register