Browse Source

!10676 do not merge cast to send and receive node

From: @lianliguang
Reviewed-by: @zhoufeng54,@kisnwang
Signed-off-by: @kisnwang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
b35435cc02
1 changed files with 7 additions and 1 deletions
  1. +7
    -1
      mindspore/ccsrc/backend/optimizer/ascend/format_type/merge_cast_to_op.cc

+ 7
- 1
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(),


Loading…
Cancel
Save