https://github.com/optimisan created
https://github.com/llvm/llvm-project/pull/116166
None
>From 197b28c684fcf3ba751a1283fd124bd2d090dfc7 Mon Sep 17 00:00:00 2001
From: Akshat Oke <akshat....@amd.com>
Date: Thu, 14 Nov 2024 05:57:01 +0000
Subject: [PATCH] [NewPM] Introduce MFAnalysisGetter for a common analysis
getter
---
.../include/llvm/CodeGen/MachinePassManager.h | 80 +++++++++++++++++++
1 file changed, 80 insertions(+)
diff --git a/llvm/include/llvm/CodeGen/MachinePassManager.h
b/llvm/include/llvm/CodeGen/MachinePassManager.h
index 69b5f6e92940c4..37669faff96c33 100644
--- a/llvm/include/llvm/CodeGen/MachinePassManager.h
+++ b/llvm/include/llvm/CodeGen/MachinePassManager.h
@@ -21,12 +21,19 @@
#ifndef LLVM_CODEGEN_MACHINEPASSMANAGER_H
#define LLVM_CODEGEN_MACHINEPASSMANAGER_H
+#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/FunctionExtras.h"
+#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/CodeGen/MachineFunction.h"
+#include "llvm/IR/Function.h"
#include "llvm/IR/PassManager.h"
#include "llvm/IR/PassManagerInternal.h"
+#include "llvm/Pass.h"
#include "llvm/Support/Error.h"
+#include <memory>
+#include <type_traits>
+#include <utility>
namespace llvm {
class Module;
@@ -236,6 +243,79 @@ using MachineFunctionPassManager =
PassManager<MachineFunction>;
/// preserve.
PreservedAnalyses getMachineFunctionPassPreservedAnalyses();
+/// For migrating to new pass manager
+/// Provides a common interface to fetch analyses instead of doing it twice in
+/// the *LegacyPass::runOnMachineFunction and NPM Pass::run NPM analyses must
+/// have the LegacyWrapper type to indicate which legacy analysis to run.
+///
+/// Outer analyses passes(Module or Function) can also be requested through
+/// `getAnalysis` or `getCachedAnalysis`.
+class MFAnalysisGetter {
+private:
+ Pass *LegacyPass;
+ MachineFunctionAnalysisManager *MFAM;
+
+ template <typename T>
+ using type_of_run =
+ typename function_traits<decltype(&T::run)>::template arg_t<0>;
+
+ template <typename T>
+ static constexpr bool IsFunctionAnalysis =
+ std::is_same_v<Function, type_of_run<T>>;
+
+ template <typename T>
+ static constexpr bool IsModuleAnalysis =
+ std::is_same_v<Module, type_of_run<T>>;
+
+public:
+ MFAnalysisGetter(Pass *LegacyPass) : LegacyPass(LegacyPass) {}
+ MFAnalysisGetter(MachineFunctionAnalysisManager *MFAM) : MFAM(MFAM) {}
+
+ /// Outer analyses requested from NPM will be cached results and can be null
+ template <typename AnalysisT>
+ typename AnalysisT::Result *getAnalysis(MachineFunction &MF) {
+ if (MFAM) {
+ // need a proxy to get the result for outer analyses
+ // this can return null
+ if constexpr (IsModuleAnalysis<AnalysisT>)
+ return MFAM->getResult<ModuleAnalysisManagerMachineFunctionProxy>(MF)
+ .getCachedResult<AnalysisT>(*MF.getFunction().getParent());
+ else if constexpr (IsFunctionAnalysis<AnalysisT>) {
+ return
&MFAM->getResult<FunctionAnalysisManagerMachineFunctionProxy>(MF)
+ .getManager()
+ .getResult<AnalysisT>(MF.getFunction());
+ }
+ return &MFAM->getResult<AnalysisT>(MF);
+ }
+ return &LegacyPass->getAnalysis<typename AnalysisT::LegacyWrapper>()
+ .getResult();
+ }
+
+ template <typename AnalysisT>
+ typename AnalysisT::Result *getCachedAnalysis(MachineFunction &MF) {
+ if (MFAM) {
+ if constexpr (IsFunctionAnalysis<AnalysisT>) {
+ return MFAM->getResult<FunctionAnalysisManagerMachineFunctionProxy>(MF)
+ .getManager()
+ .getCachedResult<AnalysisT>(MF.getFunction());
+ } else if constexpr (IsModuleAnalysis<AnalysisT>)
+ return MFAM->getResult<ModuleAnalysisManagerMachineFunctionProxy>(MF)
+ .getCachedResult<AnalysisT>(*MF.getFunction().getParent());
+
+ return &MFAM->getCachedResult<AnalysisT>(MF);
+ }
+
+ if (auto *P =
+ LegacyPass->getAnalysisIfAvailable<AnalysisT::LegacyWrapper>())
+ return &P->getResult();
+ return nullptr;
+ }
+
+ /// This is not intended to be used to invoke getAnalysis()
+ Pass *getLegacyPass() const { return LegacyPass; }
+ MachineFunctionAnalysisManager *getMFAM() const { return MFAM; }
+};
+
} // end namespace llvm
#endif // LLVM_CODEGEN_MACHINEPASSMANAGER_H
_______________________________________________
llvm-branch-commits mailing list
llvm-branch-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits
