Skip to content

recognizing ENZYME_RUNPASS for lib/LLVM Enzyme builds #2245

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.

Sign up for GitHub

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

Merged
wsmoses merged 9 commits into from
Feb 21, 2025
Merged

recognizing ENZYME_RUNPASS for lib/LLVM Enzyme builds #2245

Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
844761a
recognizing ENZYME_RUNPASS for lib/LLVM Enzyme builds
ZuseZ4 Feb 15, 2025
0939f0c
Merge branch 'main' into enzyme-runpass
ZuseZ4 Feb 15, 2025
1361fa5
addressing feedback
ZuseZ4 Feb 15, 2025
d0bec48
Add a runtime variant of registerEnzyme for Rust
ZuseZ4 Feb 17, 2025
70e34ed
remove usage examples
ZuseZ4 Feb 19, 2025
5eca4b7
adjust naming
ZuseZ4 Feb 19, 2025
b706c10
fixup
ZuseZ4 Feb 19, 2025
d195b3f
fmt
ZuseZ4 Feb 19, 2025
2a21976
Merge branch 'main' into enzyme-runpass
ZuseZ4 Feb 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 13 additions & 4 deletions enzyme/Enzyme/Enzyme.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3658,10 +3658,11 @@ void augmentPassBuilder(llvm::PassBuilder &PB) {
PB.registerFullLinkTimeOptimizationEarlyEPCallback(loadLTO);
}

extern "C" void registerEnzyme(llvm::PassBuilder &PB) {
#ifdef ENZYME_RUNPASS
augmentPassBuilder(PB);
#endif
extern "C" void registerEnzymeAndPassPipeline(llvm::PassBuilder &PB,
bool augment = false) {
if (augment) {
augmentPassBuilder(PB);
}
PB.registerPipelineParsingCallback(
[](llvm::StringRef Name, llvm::ModulePassManager &MPM,
llvm::ArrayRef<llvm::PassBuilder::PipelineElement>) {
Expand Down Expand Up @@ -3694,6 +3695,14 @@ extern "C" void registerEnzyme(llvm::PassBuilder &PB) {
});
}

extern "C" void registerEnzyme(llvm::PassBuilder &PB) {
#ifdef ENZYME_RUNPASS
registerEnzymeAndPassPipeline(PB, /*augment*/ true);
#else
registerEnzymeAndPassPipeline(PB, /*augment*/ false);
#endif
}

extern "C" ::llvm::PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK
llvmGetPassPluginInfo() {
return {LLVM_PLUGIN_API_VERSION, "EnzymeNewPM", "v0.1", registerEnzyme};
Expand Down
Loading