diff --git a/ge/single_op/task/aicpu_task_builder.cc b/ge/single_op/task/aicpu_task_builder.cc index d91bba17..8f28ffda 100755 --- a/ge/single_op/task/aicpu_task_builder.cc +++ b/ge/single_op/task/aicpu_task_builder.cc @@ -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 diff --git a/ge/single_op/task/op_task.cc b/ge/single_op/task/op_task.cc index 1b4b23ff..c3c4e5bb 100755 --- a/ge/single_op/task/op_task.cc +++ b/ge/single_op/task/op_task.cc @@ -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) { diff --git a/ge/single_op/task/op_task.h b/ge/single_op/task/op_task.h index bf8316f6..65c77800 100644 --- a/ge/single_op/task/op_task.h +++ b/ge/single_op/task/op_task.h @@ -207,6 +207,7 @@ class AiCpuTask : public AiCpuBaseTask { void *copy_input_dst_dev_; vector out_shape_hbm_; + uint64_t kernel_id_ = 0; }; class AiCpuCCTask : public AiCpuBaseTask {