| @@ -150,9 +150,10 @@ namespace ge { | |||||
| task.op_type_ = op_desc_->GetName(); | task.op_type_ = op_desc_->GetName(); | ||||
| task.task_info_ = kernel_def_.task_info(); | task.task_info_ = kernel_def_.task_info(); | ||||
| task.dynamic_flag_ = dynamic_flag; | task.dynamic_flag_ = dynamic_flag; | ||||
| task.kernel_id_ = kernel_id; | |||||
| auto debug_info = BuildTaskUtils::GetTaskInfo(op_desc_); | 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; | return SUCCESS; | ||||
| } | } | ||||
| } // namespace ge | } // 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()); | GELOGE(RT_FAILED, "Invoke rtKernelLaunch failed. ret = %d, task = %s", ret, this->op_type_.c_str()); | ||||
| return RT_FAILED; | 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); | auto status = OpenDump(stream); | ||||
| if (status != SUCCESS) { | if (status != SUCCESS) { | ||||
| @@ -207,6 +207,7 @@ class AiCpuTask : public AiCpuBaseTask { | |||||
| void *copy_input_dst_dev_; | void *copy_input_dst_dev_; | ||||
| vector<void *> out_shape_hbm_; | vector<void *> out_shape_hbm_; | ||||
| uint64_t kernel_id_ = 0; | |||||
| }; | }; | ||||
| class AiCpuCCTask : public AiCpuBaseTask { | class AiCpuCCTask : public AiCpuBaseTask { | ||||