aicpu TASK_INFO log support

pull/441/head
wuweikang 5 years ago
parent 6942dec9d2
commit 173da4b211

@ -150,9 +150,10 @@ namespace ge {
task.op_type_ = op_desc_->GetName();
task.task_info_ = kernel_def_.task_info();
task.dynamic_flag_ = dynamic_flag;
task.kernel_id_ = kernel_id;
auto debug_info = BuildTaskUtils::GetTaskInfo(op_desc_);
GELOGI("[TASK_INFO] %s %s", task.task_info_.c_str(), debug_info.c_str());
GELOGI("[TASK_INFO] %s/%s %s", std::to_string(kernel_id).c_str(), task.op_type_.c_str(), debug_info.c_str());
return SUCCESS;
}
} // namespace ge

@ -401,7 +401,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] is %s", this->task_info_.c_str());
GELOGI("[TASK_INFO] %s/%s", std::to_string(kernel_id_).c_str(), op_type_.c_str());
auto status = OpenDump(stream);
if (status != SUCCESS) {

@ -207,6 +207,7 @@ class AiCpuTask : public AiCpuBaseTask {
void *copy_input_dst_dev_;
vector<void *> out_shape_hbm_;
uint64_t kernel_id_ = 0;
};
class AiCpuCCTask : public AiCpuBaseTask {

Loading…
Cancel
Save