diff --git a/mindspore/ccsrc/backend/session/kernel_graph.cc b/mindspore/ccsrc/backend/session/kernel_graph.cc index fff7c2db4b..b1a3574ea8 100644 --- a/mindspore/ccsrc/backend/session/kernel_graph.cc +++ b/mindspore/ccsrc/backend/session/kernel_graph.cc @@ -354,11 +354,11 @@ void KernelGraph::CheckLoop() { if (node_input_it == node_input_edges_.end()) { MS_LOG(EXCEPTION) << "Can't find node [" << it.first->DebugString() << "]"; } - for (const auto &input_edge : node_input_edges_[it.first]) { - MS_EXCEPTION_IF_NULL(input_edge.first); - str = str.append(input_edge.first->DebugString()).append("|"); - } if (it.second != 0) { + for (const auto &input_edge : node_input_edges_[it.first]) { + MS_EXCEPTION_IF_NULL(input_edge.first); + str = str.append(input_edge.first->DebugString()).append("|"); + } MS_LOG(WARNING) << "Node:" << it.first->DebugString() << ",inputs:" << str << ",input num:" << it.second; none_zero_nodes[it.first] = it.second; } diff --git a/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc b/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc index 3108471cdf..5eefa96532 100644 --- a/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc +++ b/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc @@ -172,9 +172,9 @@ void UpdateCurMatchCounts(const kernel::KernelBuildInfo &kernel_build_info, cons } } -void PrintRaiseOrReducePrecisionSelectedInfo(const CNodePtr &cnode, - const std::shared_ptr &selected_kernel_build_info, - bool precision_reduce) { +std::string PrintRaiseOrReducePrecisionSelectedInfo( + const CNodePtr &cnode, const std::shared_ptr &selected_kernel_build_info, + bool precision_reduce) { MS_EXCEPTION_IF_NULL(selected_kernel_build_info); MS_EXCEPTION_IF_NULL(cnode); std::ostringstream buffer; @@ -186,7 +186,7 @@ void PrintRaiseOrReducePrecisionSelectedInfo(const CNodePtr &cnode, } PrintInputAndOutputInferType(buffer, cnode); buffer << ", select kernel:" << selected_kernel_build_info->ToString(); - MS_LOG(INFO) << buffer.str(); + return buffer.str(); } std::shared_ptr ChooseMatchedKernelInfo( @@ -381,7 +381,7 @@ KernelSelectStatus SetMatchedKernelInfo(const CNodePtr &kernel_node, if (selected_kernel_info == nullptr) { return select_status; } else { - PrintRaiseOrReducePrecisionSelectedInfo(kernel_node, selected_kernel_info, precision_reduce); + MS_LOG(INFO) << PrintRaiseOrReducePrecisionSelectedInfo(kernel_node, selected_kernel_info, precision_reduce); select_status = precision_reduce ? kStatusReducePrecision : kStatusRaisePrecision; } }