@@ -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 { | ||||