forked from mindspore-Ecosystem/mindspore
log level
This commit is contained in:
parent
b87e6c542e
commit
4851580446
|
@ -104,7 +104,7 @@ bool HcomUtil::GetHcclOpSize(const HcclDataType &data_type, const vector<size_t>
|
||||||
|
|
||||||
*size = SizetMulWithOverflowCheck(tmp_size, type_size);
|
*size = SizetMulWithOverflowCheck(tmp_size, type_size);
|
||||||
|
|
||||||
MS_LOG(INFO) << "size[" << *size << "]";
|
MS_LOG(DEBUG) << "size[" << *size << "]";
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -119,7 +119,7 @@ std::vector<TaskInfoPtr> TbeKernelMod::GenTask(const std::vector<AddressPtr> &in
|
||||||
|
|
||||||
std::string stub_func = KernelManager::GetStubFuncName(kernel_pack_);
|
std::string stub_func = KernelManager::GetStubFuncName(kernel_pack_);
|
||||||
|
|
||||||
MS_LOG(INFO) << "block_dim is:" << block_dim_;
|
MS_LOG(DEBUG) << "block_dim is:" << block_dim_;
|
||||||
|
|
||||||
TbeTaskInfoPtr task_info_ptr = std::make_shared<mindspore::ge::model_runner::TbeTaskInfo>(
|
TbeTaskInfoPtr task_info_ptr = std::make_shared<mindspore::ge::model_runner::TbeTaskInfo>(
|
||||||
unique_name_, stream_id, stub_func, block_dim_, args, 0, sm_desc, nullptr, 0, meta_data, input_data_addrs,
|
unique_name_, stream_id, stub_func, block_dim_, args, 0, sm_desc, nullptr, 0, meta_data, input_data_addrs,
|
||||||
|
|
|
@ -278,18 +278,16 @@ void AscendStreamAssign::AssignStream(const NotNull<KernelGraphPtr> &graph_ptr)
|
||||||
|
|
||||||
GetNeedActiveStreams(graph_ptr);
|
GetNeedActiveStreams(graph_ptr);
|
||||||
|
|
||||||
MS_LOG(INFO) << "Before check resource assign";
|
MS_LOG(INFO) << "After finish stream assign and before check resource assign:";
|
||||||
graph_ptr->PrintGraphExecuteOrder();
|
graph_ptr->PrintGraphExecuteOrder();
|
||||||
|
|
||||||
CheckResourceAssign(graph_ptr);
|
CheckResourceAssign(graph_ptr);
|
||||||
MS_LOG(INFO) << "After finish stream assign";
|
|
||||||
#ifdef ENABLE_DUMP_IR
|
#ifdef ENABLE_DUMP_IR
|
||||||
SubModuleId module = SubModuleId::SM_SESSION;
|
SubModuleId module = SubModuleId::SM_SESSION;
|
||||||
std::string name = "assign_stream." + std::to_string(graph_ptr->graph_id());
|
std::string name = "assign_stream." + std::to_string(graph_ptr->graph_id());
|
||||||
const std::vector<CNodePtr> &exec_order = graph_ptr->execution_order();
|
const std::vector<CNodePtr> &exec_order = graph_ptr->execution_order();
|
||||||
(void)mindspore::RDR::RecordStreamExecOrder(module, name, exec_order);
|
(void)mindspore::RDR::RecordStreamExecOrder(module, name, exec_order);
|
||||||
#endif
|
#endif
|
||||||
graph_ptr->PrintGraphExecuteOrder();
|
|
||||||
SetNodeStreamIDAttr(graph_ptr);
|
SetNodeStreamIDAttr(graph_ptr);
|
||||||
FindStreamRelations(graph_ptr);
|
FindStreamRelations(graph_ptr);
|
||||||
PrintStreamRelations();
|
PrintStreamRelations();
|
||||||
|
|
|
@ -167,7 +167,7 @@ bool IsAtomicNode(const CNodePtr &kernel_node) {
|
||||||
auto param_output = parameters_indexes.at(input_num + i);
|
auto param_output = parameters_indexes.at(input_num + i);
|
||||||
if (param_output == 1) {
|
if (param_output == 1) {
|
||||||
output_indexes.emplace_back(i);
|
output_indexes.emplace_back(i);
|
||||||
MS_LOG(INFO) << "Atomic clear output index: " << i;
|
MS_LOG(DEBUG) << "Atomic clear output index: " << i;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -182,7 +182,7 @@ bool IsAtomicNode(const CNodePtr &kernel_node) {
|
||||||
auto param_workspace = parameters_indexes.at(input_num + output_num + k);
|
auto param_workspace = parameters_indexes.at(input_num + output_num + k);
|
||||||
if (param_workspace == 1) {
|
if (param_workspace == 1) {
|
||||||
workspace_indexes.emplace_back(k);
|
workspace_indexes.emplace_back(k);
|
||||||
MS_LOG(INFO) << "Atomic clear workspace index: " << k;
|
MS_LOG(DEBUG) << "Atomic clear workspace index: " << k;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (!workspace_indexes.empty()) {
|
if (!workspace_indexes.empty()) {
|
||||||
|
|
|
@ -263,7 +263,7 @@ bool TaskGenerator::LaunchAllKernel(const std::vector<CNodePtr> &anf_node_list,
|
||||||
size_t old_size = task_info_list->size();
|
size_t old_size = task_info_list->size();
|
||||||
uint32_t stream_id = AnfAlgo::GetStreamId(anf_node_ptr);
|
uint32_t stream_id = AnfAlgo::GetStreamId(anf_node_ptr);
|
||||||
MS_EXCEPTION_IF_NULL(anf_node_ptr);
|
MS_EXCEPTION_IF_NULL(anf_node_ptr);
|
||||||
MS_LOG(INFO) << "Task gen launch begin, current_op_idx:" << current_op_index
|
MS_LOG(DEBUG) << "Task gen launch begin, current_op_idx:" << current_op_index
|
||||||
<< " name:" << anf_node_ptr->fullname_with_scope() << ", stream id:" << stream_id;
|
<< " name:" << anf_node_ptr->fullname_with_scope() << ", stream id:" << stream_id;
|
||||||
if (!LaunchKernel(anf_node_ptr, stream_id, task_info_list)) {
|
if (!LaunchKernel(anf_node_ptr, stream_id, task_info_list)) {
|
||||||
MS_LOG(ERROR) << "LaunchKernel failed.";
|
MS_LOG(ERROR) << "LaunchKernel failed.";
|
||||||
|
|
|
@ -892,7 +892,7 @@ void KernelRuntime::AssignNodeOutputMem(MemType type, const AnfNodePtr &node, in
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
if (NodeOutputDeviceAddressExist(node, i)) {
|
if (NodeOutputDeviceAddressExist(node, i)) {
|
||||||
MS_LOG(INFO) << "Already malloc index:" << i;
|
MS_LOG(DEBUG) << "Already malloc index:" << i;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
MS_LOG(DEBUG) << "Assign Node:" << node->fullname_with_scope() << " output memory size:" << output_sizes[i];
|
MS_LOG(DEBUG) << "Assign Node:" << node->fullname_with_scope() << " output memory size:" << output_sizes[i];
|
||||||
|
|
Loading…
Reference in New Issue