2 files changed, 12 insertions, 2 deletions
diff --git a/include/llvm/Analysis/DSGraph.h b/include/llvm/Analysis/DSGraph.h
index 9324cb7a3c..2a68365585 100644
--- a/include/llvm/Analysis/DSGraph.h
+++ b/include/llvm/Analysis/DSGraph.h
@@ -473,7 +473,12 @@ class DSGraph {
 public:
   DSGraph() : Func(0) {}           // Create a new, empty, DSGraph.
   DSGraph(Function &F);            // Compute the local DSGraph
-  DSGraph(const DSGraph &DSG);     // Copy ctor
+
+  // Copy ctor - If you want to capture the node mapping between the source and
+  // destination graph, you may optionally do this by specifying a map to record
+  // this into.
+  DSGraph(const DSGraph &DSG);
+  DSGraph(const DSGraph &DSG, std::map<const DSNode*, DSNode*> &BUNodeMapTy);
   ~DSGraph();
 
   bool hasFunction() const { return Func != 0; }
diff --git a/include/llvm/Analysis/DataStructure/DSGraph.h b/include/llvm/Analysis/DataStructure/DSGraph.h
index 9324cb7a3c..2a68365585 100644
--- a/include/llvm/Analysis/DataStructure/DSGraph.h
+++ b/include/llvm/Analysis/DataStructure/DSGraph.h
@@ -473,7 +473,12 @@ class DSGraph {
 public:
   DSGraph() : Func(0) {}           // Create a new, empty, DSGraph.
   DSGraph(Function &F);            // Compute the local DSGraph
-  DSGraph(const DSGraph &DSG);     // Copy ctor
+
+  // Copy ctor - If you want to capture the node mapping between the source and
+  // destination graph, you may optionally do this by specifying a map to record
+  // this into.
+  DSGraph(const DSGraph &DSG);
+  DSGraph(const DSGraph &DSG, std::map<const DSNode*, DSNode*> &BUNodeMapTy);
   ~DSGraph();
 
   bool hasFunction() const { return Func != 0; }