Browse Source

!940 Add cc_task task_info log.

From: @zhao_zhixuan
Reviewed-by: @xchu42,@ji_chen
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
d14dac3021
4 changed files with 9 additions and 2 deletions
  1. +4
    -0
      ge/single_op/task/aicpu_kernel_task_builder.cc
  2. +1
    -1
      ge/single_op/task/aicpu_task_builder.cc
  3. +2
    -1
      ge/single_op/task/op_task.cc
  4. +2
    -0
      ge/single_op/task/op_task.h

+ 4
- 0
ge/single_op/task/aicpu_kernel_task_builder.cc View File

@@ -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_);
}

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;
}
} // namespace ge

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

@@ -123,7 +123,7 @@ namespace ge {
task.kernel_id_ = kernel_id;

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;
}
} // namespace ge

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

@@ -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());
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);
if (status != SUCCESS) {
@@ -840,6 +840,7 @@ Status AiCpuCCTask::LaunchKernel(rtStream_t stream) {
GELOGE(ret, "Invoke rtCpuKernelLaunch failed. ret = %d", ret);
return ret;
}
GELOGI("[TASK_INFO] %lu/%s", kernel_id_, op_type_.c_str());
GELOGD("Invoke rtCpuKernelLaunch succeeded");
auto status = OpenDump(stream);
if (status != SUCCESS) {


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

@@ -227,6 +227,8 @@ private:
size_t io_addr_num_ = 0;
bool is_custom_ = false;
uint32_t dump_flag_ = RT_KERNEL_DEFAULT;
std::string op_type_;
uint64_t kernel_id_ = 0;
};
} // namespace ge



Loading…
Cancel
Save