|
|
|
@@ -302,20 +302,20 @@ void KernelGraph::GetLoopNodesByDFS(AnfNodePtr node, uint32_t *loop_num) { |
|
|
|
} else { |
|
|
|
AnfNodePtr node_iter = node; |
|
|
|
MS_EXCEPTION_IF_NULL(node_iter); |
|
|
|
MS_LOG(DEBUG) << "Print loop nodes start:"; |
|
|
|
MS_LOG(INFO) << "Print loop nodes start:"; |
|
|
|
for (; node_iter != input_edge.first && node_iter != nullptr; node_iter = edge_to_[node_iter]) { |
|
|
|
loop_nodes_.push(node_iter); |
|
|
|
node_input_num_[node_iter]--; |
|
|
|
MS_LOG(DEBUG) << "Get loop node:" << node_iter->DebugString(); |
|
|
|
MS_LOG(INFO) << "Get loop node:" << node_iter->DebugString(); |
|
|
|
} |
|
|
|
if (node_iter != nullptr) { |
|
|
|
loop_nodes_.push(node_iter); |
|
|
|
loop_nodes_.push(node); |
|
|
|
(*loop_num)++; |
|
|
|
node_input_num_[node_iter]--; |
|
|
|
MS_LOG(DEBUG) << "Get loop node:" << node_iter->DebugString(); |
|
|
|
MS_LOG(DEBUG) << "Get loop node:" << node->DebugString(); |
|
|
|
MS_LOG(DEBUG) << "Print loop nodes end, Loop num:" << *loop_num; |
|
|
|
MS_LOG(INFO) << "Get loop node:" << node_iter->DebugString(); |
|
|
|
MS_LOG(INFO) << "Get loop node:" << node->DebugString(); |
|
|
|
MS_LOG(INFO) << "Print loop nodes end, Loop num:" << *loop_num; |
|
|
|
while (!loop_nodes_.empty()) { |
|
|
|
loop_nodes_.pop(); |
|
|
|
} |
|
|
|
|