Browse Source

!15386 Add NonTask info in task_info.ir

From: @laiyongqiang
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
pull/15386/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
ba5b751418
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/runtime/device/ascend/tasksink/task_generator.cc

+ 4
- 0
mindspore/ccsrc/runtime/device/ascend/tasksink/task_generator.cc View File

@@ -151,6 +151,10 @@ bool TaskGenerator::LaunchKernel(const CNodePtr &anf_node_ptr, uint32_t stream_i
auto op_name = AnfAlgo::GetCNodeName(anf_node_ptr);
if ((op_name == kSplitOpName || op_name == kSplitVOpName) && AnfAlgo::HasNodeAttr(kAttrNonTask, anf_node_ptr)) {
MS_LOG(INFO) << "Skip task generation for NonTask op " << anf_node_ptr->fullname_with_scope();
auto debug_info = std::make_shared<TaskDebugInfo>();
debug_info->op_name_ = anf_node_ptr->fullname_with_scope() + "-NonTask";
debug_info->task_num_ = 0;
task_debug_info_list_.push_back(debug_info);
return true;
}



Loading…
Cancel
Save