1 files changed, 5 insertions, 2 deletions
diff --git a/include/llvm/Transforms/Utils/UnifyFunctionExitNodes.h b/include/llvm/Transforms/Utils/UnifyFunctionExitNodes.h
index d31e55b3a8..01f055e801 100644
--- a/include/llvm/Transforms/Utils/UnifyFunctionExitNodes.h
+++ b/include/llvm/Transforms/Utils/UnifyFunctionExitNodes.h
@@ -7,8 +7,8 @@
 //
 //===----------------------------------------------------------------------===//
 
-#ifndef LLVM_XFORMS_UNIFY_FUNCTION_EXIT_NODES_H
-#define LLVM_XFORMS_UNIFY_FUNCTION_EXIT_NODES_H
+#ifndef LLVM_TRANSFORMS_UNIFYFUNCTIONEXITNODES_H
+#define LLVM_TRANSFORMS_UNIFYFUNCTIONEXITNODES_H
 
 #include "llvm/Pass.h"
 
@@ -17,6 +17,9 @@ struct UnifyFunctionExitNodes : public FunctionPass {
 public:
   UnifyFunctionExitNodes() : ExitNode(0) {}
 
+  // We can preserve non-critical-edgeness when we unify function exit nodes
+  virtual void getAnalysisUsage(AnalysisUsage &AU) const;
+
   // getExitNode - Return the new single (or nonexistant) exit node of the CFG.
   //
   BasicBlock *getExitNode() const { return ExitNode; }