diff --git a/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp b/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp index 8ff50c1a3ad..66ada59acdb 100644 --- a/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp +++ b/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp @@ -446,17 +446,24 @@ std::string DOTGraphTraits::getNodeLabel(const SUnit *SU, const ScheduleDAG *G) { std::string Op; - if (!SU->getNode()) - Op = ""; - else { - SmallVector FlaggedNodes; - for (SDNode *N = SU->getNode(); N; N = N->getFlaggedNode()) - FlaggedNodes.push_back(N); - while (!FlaggedNodes.empty()) { - Op += DOTGraphTraits::getNodeLabel(FlaggedNodes.back(), - G->DAG) + "\n"; - FlaggedNodes.pop_back(); + if (G->DAG) { + if (!SU->getNode()) + Op = ""; + else { + SmallVector FlaggedNodes; + for (SDNode *N = SU->getNode(); N; N = N->getFlaggedNode()) + FlaggedNodes.push_back(N); + while (!FlaggedNodes.empty()) { + Op += DOTGraphTraits::getNodeLabel(FlaggedNodes.back(), + G->DAG) + "\n"; + FlaggedNodes.pop_back(); + } } + } else { + std::string s; + raw_string_ostream oss(s); + SU->MI->print(oss); + Op += oss.str(); } return Op;