Apply feedback from Chris.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@19432 91177308-0d34-0410-b5e6-96231b3b80d8

2 files changed, 5 insertions, 4 deletions

diff --git a/include/llvm/Transforms/IPO.h b/include/llvm/Transforms/IPO.h
index d2371de517..a0515a7f5e 100644
--- a/include/llvm/Transforms/IPO.h
+++ b/include/llvm/Transforms/IPO.h
@@ -19,6 +19,7 @@
 
 namespace llvm {
 
+class FunctionPass;
 class ModulePass;
 class Function;
 class BasicBlock;
@@ -150,12 +151,12 @@ ModulePass *createIPSCCPPass();
 /// createLoopExtractorPass - This pass extracts all natural loops from the
 /// program into a function if it can.
 ///
-ModulePass *createLoopExtractorPass();
+FunctionPass *createLoopExtractorPass();
 
 /// createSingleLoopExtractorPass - This pass extracts one natural loop from the
 /// program into a function if it can.  This is used by bugpoint.
 ///
-ModulePass *createSingleLoopExtractorPass();
+FunctionPass *createSingleLoopExtractorPass();
 
 // createBlockExtractorPass - This pass extracts all blocks (except those
 // specified in the argument list) from the functions in the module.
diff --git a/lib/Transforms/IPO/LoopExtractor.cpp b/lib/Transforms/IPO/LoopExtractor.cpp
index 638820e398..4fd2335c28 100644
--- a/lib/Transforms/IPO/LoopExtractor.cpp
+++ b/lib/Transforms/IPO/LoopExtractor.cpp
@@ -61,7 +61,7 @@ namespace {
 // createLoopExtractorPass - This pass extracts all natural loops from the
 // program into a function if it can.
 //
-ModulePass *llvm::createLoopExtractorPass() { return new LoopExtractor(); }
+FunctionPass *llvm::createLoopExtractorPass() { return new LoopExtractor(); }
 
 bool LoopExtractor::runOnFunction(Function &F) {
   LoopInfo &LI = getAnalysis<LoopInfo>();
@@ -131,7 +131,7 @@ bool LoopExtractor::runOnFunction(Function &F) {
 // createSingleLoopExtractorPass - This pass extracts one natural loop from the
 // program into a function if it can.  This is used by bugpoint.
 //
-ModulePass *llvm::createSingleLoopExtractorPass() {
+FunctionPass *llvm::createSingleLoopExtractorPass() {
   return new SingleLoopExtractor();
 }