mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!15386 Add NonTask info in task_info.ir 

From: @laiyongqiang
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
This commit is contained in:
mindspore-ci-bot 2021-04-20 10:19:35 +08:00 committed by Gitee
parent 23973aa830 59fd6358c5
commit ba5b751418
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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;
}