!945 Add cc_task task_info log.

From: @zhao_zhixuan
Reviewed-by: @xchu42,@ji_chen
Signed-off-by: @startzgf168,@ji_chen
pull/945/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit cb019693b1

@ -109,6 +109,10 @@ Status AiCpuCCTaskBuilder::BuildTask(AiCpuCCTask &task, uint64_t kernel_id, cons
aicpu_param_head->extInfoAddr = reinterpret_cast<uintptr_t>(task.ext_info_addr_dev_);
}
task.op_type_ = op_desc_->GetName();
task.kernel_id_ = kernel_id;
auto debug_info = BuildTaskUtils::GetTaskInfo(op_desc_);
GELOGI("[TASK_INFO] %lu/%s %s", kernel_id, task.op_type_.c_str(), debug_info.c_str());
return SUCCESS;
}
} // namespace ge

@ -123,7 +123,7 @@ namespace ge {
task.kernel_id_ = kernel_id;
auto debug_info = BuildTaskUtils::GetTaskInfo(op_desc_);
GELOGI("[TASK_INFO] %s/%s %s", std::to_string(kernel_id).c_str(), task.op_type_.c_str(), debug_info.c_str());
GELOGI("[TASK_INFO] %lu/%s %s", kernel_id, task.op_type_.c_str(), debug_info.c_str());
return SUCCESS;
}
} // namespace ge

@ -567,7 +567,7 @@ Status AiCpuTask::LaunchKernel(rtStream_t stream) {
GELOGE(RT_FAILED, "Invoke rtKernelLaunch failed. ret = %d, task = %s", ret, this->op_type_.c_str());
return RT_FAILED;
}
GELOGI("[TASK_INFO] %s/%s", std::to_string(kernel_id_).c_str(), op_type_.c_str());
GELOGI("[TASK_INFO] %lu/%s", kernel_id_, op_type_.c_str());
auto status = OpenDump(stream);
if (status != SUCCESS) {
@ -840,6 +840,7 @@ Status AiCpuCCTask::LaunchKernel(rtStream_t stream) {
GELOGE(ret, "Invoke rtCpuKernelLaunch failed. ret = %d", ret);
return ret;
}
GELOGI("[TASK_INFO] %lu/%s", kernel_id_, op_type_.c_str());
GELOGD("Invoke rtCpuKernelLaunch succeeded");
auto status = OpenDump(stream);
if (status != SUCCESS) {

@ -227,6 +227,8 @@ private:
size_t io_addr_num_ = 0;
bool is_custom_ = false;
uint32_t dump_flag_ = RT_KERNEL_DEFAULT;
std::string op_type_;
uint64_t kernel_id_ = 0;
};
} // namespace ge

Loading…
Cancel
Save