Print SrcValue nodes correctly

llvm-svn: 21803

2 files changed, 11 insertions, 0 deletions

diff --git a/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp b/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
index f4d7adcc4ed..965976cce3d 100644
--- a/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
@@ -1554,6 +1554,7 @@ const char *SDNode::getOperationName() const {
   switch (getOpcode()) {
   default: return "<<Unknown>>";
   case ISD::PCMARKER:      return "PCMarker";
+  case ISD::SRCVALUE:      return "SrcValue";
   case ISD::EntryToken:    return "EntryToken";
   case ISD::TokenFactor:   return "TokenFactor";
   case ISD::Constant:      return "Constant";
@@ -1712,6 +1713,11 @@ void SDNode::dump() const {
     std::cerr << "'" << ES->getSymbol() << "'";
   } else if (const MVTSDNode *M = dyn_cast<MVTSDNode>(this)) {
     std::cerr << " - Ty = " << MVT::getValueTypeString(M->getExtraValueType());
+  } else if (const SrcValueSDNode *M = dyn_cast<SrcValueSDNode>(this)) {
+    if (M->getValue())
+      std::cerr << "<" << M->getValue() << ":" << M->getOffset() << ">";
+    else
+      std::cerr << "<null:" << M->getOffset() << ">";
   }
 }
 
diff --git a/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp b/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp
index 2ed1fe47092..8d8e1fd3662 100644
--- a/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp
@@ -90,6 +90,11 @@ std::string DOTGraphTraits<SelectionDAG*>::getNodeLabel(const SDNode *Node,
     Op += "'" + std::string(ES->getSymbol()) + "'";
   } else if (const MVTSDNode *M = dyn_cast<MVTSDNode>(Node)) {
     Op = Op + " ty=" + MVT::getValueTypeString(M->getExtraValueType());
+  } else if (const SrcValueSDNode *M = dyn_cast<SrcValueSDNode>(Node)) {
+    if (M->getValue())
+      Op += "<" + M->getValue()->getName() + ":" + itostr(M->getOffset()) + ">";
+    else
+      Op += "<null:" + itostr(M->getOffset()) + ">";
   }
   return Op;
 }