From: @zhao_zhixuan Reviewed-by: @xchu42,@ji_chen Signed-off-by: @ji_chentags/v1.2.0
@@ -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 | ||||