ScopDetection: Allow to limit the scop detection to a single function

-polly-detect-only=<functionname> allows to limit the scop detection to
a single function.

llvm-svn: 142750

1 files changed, 12 insertions, 0 deletions

diff --git a/polly/lib/Analysis/ScopDetection.cpp b/polly/lib/Analysis/ScopDetection.cpp
index 083ad439f84..ab9720c6741 100644
--- a/polly/lib/Analysis/ScopDetection.cpp
+++ b/polly/lib/Analysis/ScopDetection.cpp
@@ -63,6 +63,13 @@
 using namespace llvm;
 using namespace polly;
 
+static cl::opt<std::string>
+OnlyFunction("polly-detect-only",
+             cl::desc("Only detect scops in function"), cl::Hidden,
+             cl::value_desc("The function name to detect scops in"),
+             cl::ValueRequired, cl::init(""));
+
+
 //===----------------------------------------------------------------------===//
 // Statistics.
 
@@ -553,6 +560,11 @@ bool ScopDetection::runOnFunction(llvm::Function &F) {
   RI = &getAnalysis<RegionInfo>();
   Region *TopRegion = RI->getTopLevelRegion();
 
+  releaseMemory();
+
+  if (OnlyFunction != "" && F.getNameStr() != OnlyFunction)
+    return false;
+
   if(!isValidFunction(F))
     return false;