p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

do not merge cast to receive & send op

This commit is contained in:
LianLiguang 2020-12-28 10:11:44 +08:00
parent 8f2b70261a
commit da09b8e515
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/backend/optimizer/ascend/format_type/merge_cast_to_op.cc
View File

@ -154,7 +154,10 @@ AnfNodePtr MergeCastToNextOp(const FuncGraphPtr &graph, const CNodePtr &node, co
if (AnfAlgo::IsGraphKernel(next_node)) { if (AnfAlgo::IsGraphKernel(next_node)) {
return nullptr; return nullptr;
} }
auto next_op_name = AnfAlgo::GetCNodeName(next_node); auto next_op_name = AnfAlgo::GetCNodeName(next_cnode);
if (next_op_name == prim::kPrimSend->name()) {
return nullptr;
}
std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list;
kernel_query->Query(next_cnode, &kernel_info_list); kernel_query->Query(next_cnode, &kernel_info_list);
@ -231,6 +234,9 @@ AnfNodePtr MergeCastToPriorOp(const FuncGraphPtr &graph, const CNodePtr &cur_nod
} }
std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list;
if (AnfAlgo::GetCNodeName(prior_op) == prim::kPrimReceive->name()) {
return nullptr;
}
kernel_query->Query(prior_op, &kernel_info_list); kernel_query->Query(prior_op, &kernel_info_list);
auto kernel_info_it = std::find_if( auto kernel_info_it = std::find_if(
kernel_info_list.begin(), kernel_info_list.end(), kernel_info_list.begin(), kernel_info_list.end(),