From: @zhao_zhixuan Reviewed-by: @xchu42,@ji_chen Signed-off-by: @startzgf168,@ji_chenpull/945/MERGE
| @@ -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_); | 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; | return SUCCESS; | ||||
| } | } | ||||
| } // namespace ge | } // namespace ge | ||||
| @@ -123,7 +123,7 @@ namespace ge { | |||||
| task.kernel_id_ = kernel_id; | task.kernel_id_ = kernel_id; | ||||
| auto debug_info = BuildTaskUtils::GetTaskInfo(op_desc_); | 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; | return SUCCESS; | ||||
| } | } | ||||
| } // namespace ge | } // 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()); | 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] %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); | auto status = OpenDump(stream); | ||||
| if (status != SUCCESS) { | if (status != SUCCESS) { | ||||
| @@ -840,6 +840,7 @@ Status AiCpuCCTask::LaunchKernel(rtStream_t stream) { | |||||
| GELOGE(ret, "Invoke rtCpuKernelLaunch failed. ret = %d", ret); | GELOGE(ret, "Invoke rtCpuKernelLaunch failed. ret = %d", ret); | ||||
| return ret; | return ret; | ||||
| } | } | ||||
| GELOGI("[TASK_INFO] %lu/%s", kernel_id_, op_type_.c_str()); | |||||
| GELOGD("Invoke rtCpuKernelLaunch succeeded"); | GELOGD("Invoke rtCpuKernelLaunch succeeded"); | ||||
| auto status = OpenDump(stream); | auto status = OpenDump(stream); | ||||
| if (status != SUCCESS) { | if (status != SUCCESS) { | ||||
| @@ -227,6 +227,8 @@ private: | |||||
| size_t io_addr_num_ = 0; | size_t io_addr_num_ = 0; | ||||
| bool is_custom_ = false; | bool is_custom_ = false; | ||||
| uint32_t dump_flag_ = RT_KERNEL_DEFAULT; | uint32_t dump_flag_ = RT_KERNEL_DEFAULT; | ||||
| std::string op_type_; | |||||
| uint64_t kernel_id_ = 0; | |||||
| }; | }; | ||||
| } // namespace ge | } // namespace ge | ||||