Browse Source

aicpu TASK_INFO log support

tags/v1.1.0
wuweikang 3 years ago
parent
commit
173da4b211
3 changed files with 4 additions and 2 deletions
  1. +2
    -1
      ge/single_op/task/aicpu_task_builder.cc
  2. +1
    -1
      ge/single_op/task/op_task.cc
  3. +1
    -0
      ge/single_op/task/op_task.h

+ 2
- 1
ge/single_op/task/aicpu_task_builder.cc View File

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

+ 1
- 1
ge/single_op/task/op_task.cc View File

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


+ 1
- 0
ge/single_op/task/op_task.h View File

@@ -207,6 +207,7 @@ class AiCpuTask : public AiCpuBaseTask {
void *copy_input_dst_dev_;

vector<void *> out_shape_hbm_;
uint64_t kernel_id_ = 0;
};

class AiCpuCCTask : public AiCpuBaseTask {


Loading…
Cancel
Save