[llvm-commits] [polly] r141456 - /polly/trunk/lib/RegisterPasses.cpp

[Tobias Grosser]

Author: grosser
Date: Fri Oct  7 19:30:44 2011
New Revision: 141456

URL: http://llvm.org/viewvc/llvm-project?rev=141456&view=rev
Log:
Add flags to run the Polly Printers/Viewers

Modified:
    polly/trunk/lib/RegisterPasses.cpp

Modified: polly/trunk/lib/RegisterPasses.cpp
URL: http://llvm.org/viewvc/llvm-project/polly/trunk/lib/RegisterPasses.cpp?rev=141456&r1=141455&r2=141456&view=diff
==============================================================================
--- polly/trunk/lib/RegisterPasses.cpp (original)
+++ polly/trunk/lib/RegisterPasses.cpp Fri Oct  7 19:30:44 2011
@@ -16,6 +16,7 @@
 #include "llvm/PassRegistry.h"
 #include "llvm/Transforms/Scalar.h"
 #include "llvm/Transforms/IPO/PassManagerBuilder.h"
+#include "llvm/Support/CommandLine.h"
 
 #include "polly/LinkAllPasses.h"
 
@@ -27,6 +28,29 @@
 
 using namespace llvm;
 
+static cl::opt<bool>
+PollyViewer("enable-polly-viewer",
+       cl::desc("Enable the Polly DOT viewer in -O3"), cl::Hidden,
+       cl::value_desc("Run the Polly DOT viewer at -O3"),
+       cl::init(false));
+static cl::opt<bool>
+PollyOnlyViewer("enable-polly-only-viewer",
+       cl::desc("Enable the Polly DOT viewer in -O3 (no BB content)"),
+       cl::Hidden,
+       cl::value_desc("Run the Polly DOT viewer at -O3 (no BB content"),
+       cl::init(false));
+static cl::opt<bool>
+PollyPrinter("enable-polly-printer",
+       cl::desc("Enable the Polly DOT printer in -O3"), cl::Hidden,
+       cl::value_desc("Run the Polly DOT printer at -O3"),
+       cl::init(false));
+static cl::opt<bool>
+PollyOnlyPrinter("enable-polly-only-printer",
+       cl::desc("Enable the Polly DOT printer in -O3 (no BB content)"),
+       cl::Hidden,
+       cl::value_desc("Run the Polly DOT printer at -O3 (no BB content"),
+       cl::init(false));
+
 void initializePollyPasses(PassRegistry &Registry) {
   initializeCloogInfoPass(Registry);
   initializeCodeGenerationPass(Registry);
@@ -78,11 +102,20 @@
 
   PM.add(polly::createCodePreparationPass());
   PM.add(polly::createRegionSimplifyPass());
-
   // FIXME: Needed as RegionSimplifyPass destroys the canonical form of
   //        induction variables (It changes the order of the operands in the
   //        PHI nodes).
   PM.add(llvm::createIndVarSimplifyPass());
+
+  if (PollyViewer)
+    PM.add(polly::createDOTViewerPass());
+  if (PollyOnlyViewer)
+    PM.add(polly::createDOTOnlyViewerPass());
+  if (PollyPrinter)
+    PM.add(polly::createDOTPrinterPass());
+  if (PollyOnlyPrinter)
+    PM.add(polly::createDOTOnlyPrinterPass());
+
   PM.add(polly::createIslScheduleOptimizerPass());
   PM.add(polly::createCodeGenerationPass());
 }