Skip to content

[Polly] Introduce PhaseManager and remove LPM support #125442

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 13 commits into
base: users/meinersbur/polly_CodePreparation
Choose a base branch
from

Conversation

Meinersbur
Copy link
Member

@Meinersbur Meinersbur commented Feb 3, 2025

Instead of relying on any pass manager to schedule Polly's passes, add Polly's own pipeline manager which is seen as a monolithic pass in LLVM's pass manager. Polly's former passes are now phases of the new PhaseManager component.

Relying on LLVM's pass manager (the legacy as well as the New Pass Manager1) to manage Polly's phases never was a good fit that the PhaseManager resolves:

  • Polly passes were modifying analysis results, in particular RegionInfo and ScopInfo. This means that there was not just one unique and "definite" analysis result, the actual result depended on which analyses ran prior, and the pass manager was not allowed to throw away cached analyses or prior SCoP optimizations would have been forgotten. The LLVM pass manger's persistance of analysis results is not contractual but designed for caching.

  • Polly depends on a particular execution order of passes and regions (e.g. regression tests, invalidation of consecutive SCoPs). LLVM's pass manager does not guarantee any excecution order.

  • Polly does not completely preserve DominatorTree, RegionInfo, LoopInfo, or ScalarEvolution, but only as-needed for Polly's own uses. Because the ScopDetection object stores references to those analyses, it still had to lie to the pass manager that they would be preserved, or the pass manager would have released and recomputed the invalidated analysis objects that ScopDetection/ScopInfo was still referencing. To ensure that no non-Polly pass would see these not-completely-preserved analyses, all analyses still had to be thrown away after the ScopPassManager, respectively with a BarrierNoopPass in case of the LPM.

  • The NPM's PassInstrumentation wraps the IR unit into an llvm::Any object, but implementatons such as PrintIRInstrumentation call llvm_unreachable on encountering an unknown IR unit, such as SCoPs, with no extension points to add support. Hence LLVM crashes when dumping IR between SCoP passes (such as -print-before-changed with Polly being active).

The new PhaseManager uses some command line options that previously belonged to Polly's legacy passes, such as -polly-print-detect (so the option will continue to work). Hence the LPM support is incompatible with the new approach and support for it is removed.

PRs from this patch series:

Footnotes

  1. When Chandler Carruth announced working on the NPM, I talked to him about supporting Polly's use cases. His response was "passes must adhere to the constraints of the pass manager".

@kartcq
Copy link
Contributor

kartcq commented Feb 3, 2025 via email

@Meinersbur
Copy link
Member Author

Meinersbur commented Feb 3, 2025

FYI, @rahulana-quic @tobiasgrosser

For some reason I cannot add you as reviewers, but feel free to start a review.

@Meinersbur Meinersbur marked this pull request as ready for review February 3, 2025 15:01
@kartcq
Copy link
Contributor

kartcq commented Feb 12, 2025

Hi @Meinersbur
As the change is huge, please give us some time to try it out downstream before merging.
Thanks.

@Meinersbur
Copy link
Member Author

OK, please signal your LGTM when you are ready

@kartcq
Copy link
Contributor

kartcq commented Apr 9, 2025

Thanks for your patience @Meinersbur.
I am posting some of my observations as review comments.

// Phase: scops
auto &AC = FAM.getResult<AssumptionAnalysis>(F);
const DataLayout &DL = F.getParent()->getDataLayout();
ScopInfo Info(DL, SD, SE, LI, AA, DT, AC, ORE);
Copy link
Contributor

@kartcq kartcq Apr 9, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This way of computing ScopInfo processes the regions in same order, as opposed to reverse order, followed by FunctionToScopPassAdaptor earlier.
To pertain to old behaviour, can we either invoke ScopBuilder in reverse order here or change the order in ScopInfo::recompute.

Copy link
Member Author

@Meinersbur Meinersbur May 18, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you clarify? There are two places we iterate over scops:

  1. For -print-scops: This PR's code reuses the LPM did:

    addRegionIntoQueue(*TLR, Regions);
    vs
    addRegionIntoQueue(*RI->getTopLevelRegion(), RQ);

  2. For the phase pipeline itself it resuses the code from the NPM/FunctionToScopPassAdaptor:

    SmallPriorityWorklist<Region *, 4> Worklist;
    for (auto &[R, S] : Info)
    if (S)
    Worklist.insert(R);
    auto &&TTI = FAM.getResult<TargetIRAnalysis>(F);
    while (!Worklist.empty()) {
    Region *R = Worklist.pop_back_val();
    if (!SD.isMaxRegionInScop(*R, /*Verify=*/false))
    continue;
    Scop *S = Info.getScop(R);
    vs
    SmallPriorityWorklist<Region *, 4> Worklist;
    for (auto &S : SI)
    if (S.second)
    Worklist.insert(S.first);
    ScopStandardAnalysisResults AR = {AM.getResult<DominatorTreeAnalysis>(F),
    AM.getResult<ScopInfoAnalysis>(F),
    AM.getResult<ScalarEvolutionAnalysis>(F),
    AM.getResult<LoopAnalysis>(F),
    AM.getResult<RegionInfoAnalysis>(F),
    AM.getResult<TargetIRAnalysis>(F)};
    ScopAnalysisManager &SAM =
    AM.getResult<ScopAnalysisManagerFunctionProxy>(F).getManager();
    SPMUpdater Updater{Worklist, SAM};
    while (!Worklist.empty()) {
    Region *R = Worklist.pop_back_val();
    if (!SD.isMaxRegionInScop(*R, /*Verify=*/false))
    continue;
    Scop *scop = SI.getScop(R);

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It's in second case for pass pipeline, the order is changed.

@Meinersbur Meinersbur marked this pull request as draft May 17, 2025 22:27
@Meinersbur Meinersbur changed the base branch from users/meinersbur/polly_NPM-ScopInliner to users/meinersbur/polly_CodePreparation May 17, 2025 23:53
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants