Merge pull request #26698 from gottesmm/pr-251ac47676a7789236a438744e9fe9b38c45cbde

[loop-region-printer] A few small improvements

Author: gottesmm

lib/SILOptimizer/UtilityPasses/LoopRegionPrinter.cpp

@@ -20,37 +20,66 @@
 #include "swift/SILOptimizer/Analysis/LoopRegionAnalysis.h"
 #include "swift/SILOptimizer/PassManager/Passes.h"
 #include "swift/SILOptimizer/PassManager/Transforms.h"
+#include "llvm/Support/CommandLine.h"
 
 using namespace swift;
 
+static llvm::cl::opt<std::string>
+    SILViewCFGOnlyFun("sil-loop-region-view-cfg-only-function",
+                      llvm::cl::init(""),
+                      llvm::cl::desc("Only produce a graphviz file for the "
+                                     "loop region info of this function"));
+
+static llvm::cl::opt<std::string>
+    SILViewCFGOnlyFuns("sil-loop-region-view-cfg-only-functions",
+                       llvm::cl::init(""),
+                       llvm::cl::desc("Only produce a graphviz file for the "
+                                      "loop region info for the functions "
+                                      "whose name contains this substring"));
+
 namespace {
 
 class LoopRegionViewText : public SILModuleTransform {
   void run() override {
     invalidateAll();
-    LoopRegionAnalysis *LRA = PM->getAnalysis<LoopRegionAnalysis>();
-    for (auto &Fn : *getModule()) {
-      if (Fn.isExternalDeclaration()) continue;
+    auto *lra = PM->getAnalysis<LoopRegionAnalysis>();
+
+    for (auto &fn : *getModule()) {
+      if (fn.isExternalDeclaration())
+        continue;
+      if (!SILViewCFGOnlyFun.empty() && fn.getName() != SILViewCFGOnlyFun)
+        continue;
+      if (!SILViewCFGOnlyFuns.empty() &&
+          fn.getName().find(SILViewCFGOnlyFuns, 0) == StringRef::npos)
+        continue;
 
-      llvm::outs() << "@" << Fn.getName() << "@\n";
-      LRA->get(&Fn)->dump();
-      llvm::outs() << "\n";
+      // Ok, we are going to analyze this function. Invalidate all state
+      // associated with it so we recompute the loop regions.
+      llvm::outs() << "Start @" << fn.getName() << "@\n";
+      lra->get(&fn)->dump();
+      llvm::outs() << "End @" << fn.getName() << "@\n";
       llvm::outs().flush();
     }
   }
-
 };
 
 class LoopRegionViewCFG : public SILModuleTransform {
   void run() override {
+    invalidateAll();
+    auto *lra = PM->getAnalysis<LoopRegionAnalysis>();
 
-    LoopRegionAnalysis *LRA = PM->getAnalysis<LoopRegionAnalysis>();
-
-    auto *M = getModule();
-    for (auto &Fn : M->getFunctions()) {
-      if (Fn.isExternalDeclaration())
+    for (auto &fn : *getModule()) {
+      if (fn.isExternalDeclaration())
         continue;
-      LRA->get(&Fn)->viewLoopRegions();
+      if (!SILViewCFGOnlyFun.empty() && fn.getName() != SILViewCFGOnlyFun)
+        continue;
+      if (!SILViewCFGOnlyFuns.empty() &&
+          fn.getName().find(SILViewCFGOnlyFuns, 0) == StringRef::npos)
+        continue;
+
+      // Ok, we are going to analyze this function. Invalidate all state
+      // associated with it so we recompute the loop regions.
+      lra->get(&fn)->viewLoopRegions();
     }
   }
 };