[PATCH] D102463: Making Instrumentation aware of LoopNest Pass

[Raghesh Aloor via Phabricator via llvm-commits]

raghesh updated this revision to Diff 347285.
raghesh added a comment.

Addressed the comments. Changed functions to static and made testcase changes


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D102463/new/

https://reviews.llvm.org/D102463

Files:
  llvm/include/llvm/Transforms/Scalar/LoopPassManager.h
  llvm/test/Other/loopnest-callback.ll


Index: llvm/test/Other/loopnest-callback.ll
===================================================================
--- /dev/null
+++ llvm/test/Other/loopnest-callback.ll
@@ -0,0 +1,9 @@
+;RUN: opt -disable-output -passes=loop-interchange -print-after-all < %s 2>&1 | FileCheck %s
+
+; CHECK: IR Dump After LoopInterchangePass
+define void @foo() {
+entry:
+  br label %for.cond
+for.cond:
+  br label %for.cond
+}
Index: llvm/include/llvm/Transforms/Scalar/LoopPassManager.h
===================================================================
--- llvm/include/llvm/Transforms/Scalar/LoopPassManager.h
+++ llvm/include/llvm/Transforms/Scalar/LoopPassManager.h
@@ -183,6 +183,12 @@
   PreservedAnalyses runWithoutLoopNestPasses(Loop &L, LoopAnalysisManager &AM,
                                              LoopStandardAnalysisResults &AR,
                                              LPMUpdater &U);
+
+private:
+  static const Loop &getLoopFromIR(Loop &L) { return L; }
+  static const Loop &getLoopFromIR(LoopNest &LN) {
+    return LN.getOutermostLoop();
+  }
 };
 
 /// The Loop pass manager.
@@ -358,9 +364,12 @@
 Optional<PreservedAnalyses> LoopPassManager::runSinglePass(
     IRUnitT &IR, PassT &Pass, LoopAnalysisManager &AM,
     LoopStandardAnalysisResults &AR, LPMUpdater &U, PassInstrumentation &PI) {
+  // Get the loop in case of Loop pass and outermost loop in case of LoopNest
+  // pass which is to be passed to BeforePass and AfterPass call backs.
+  const Loop &L = getLoopFromIR(IR);
   // Check the PassInstrumentation's BeforePass callbacks before running the
   // pass, skip its execution completely if asked to (callback returns false).
-  if (!PI.runBeforePass<IRUnitT>(*Pass, IR))
+  if (!PI.runBeforePass<Loop>(*Pass, L))
     return None;
 
   PreservedAnalyses PA;
@@ -373,7 +382,7 @@
   if (U.skipCurrentLoop())
     PI.runAfterPassInvalidated<IRUnitT>(*Pass, PA);
   else
-    PI.runAfterPass<IRUnitT>(*Pass, IR, PA);
+    PI.runAfterPass<Loop>(*Pass, L, PA);
   return PA;
 }
 


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D102463.347285.patch
Type: text/x-patch
Size: 2010 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20210524/2748195a/attachment.bin>