diff --git a/lib/CodeGen/SelectionDAG/SelectionDAG.cpp b/lib/CodeGen/SelectionDAG/SelectionDAG.cpp index 33899a411be..ef21a1592bb 100644 --- a/lib/CodeGen/SelectionDAG/SelectionDAG.cpp +++ b/lib/CodeGen/SelectionDAG/SelectionDAG.cpp @@ -6315,7 +6315,11 @@ bool ShuffleVectorSDNode::isSplatMask(const int *Mask, EVT VT) { } static void checkForCyclesHelper(const SDNode *N, - std::set &visited) { + std::set &visited, + std::set &checked) { + if (checked.find(N) != checked.end()) + return; + if (visited.find(N) != visited.end()) { dbgs() << "Offending node:\n"; N->dumprFull(); @@ -6329,16 +6333,18 @@ static void checkForCyclesHelper(const SDNode *N, assert(inserted && "Missed cycle"); for(unsigned i = 0; i < N->getNumOperands(); ++i) { - checkForCyclesHelper(N->getOperand(i).getNode(), visited); + checkForCyclesHelper(N->getOperand(i).getNode(), visited, checked); } visited.erase(i); + checked.insert(N); } void llvm::checkForCycles(const llvm::SDNode *N) { #ifdef XDEBUG assert(N && "Checking nonexistant SDNode"); std::set visited; - checkForCyclesHelper(N, visited); + std::set checked; + checkForCyclesHelper(N, visited, checked); #endif }