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

!1252 Support NopNode as hccl operator input 

Merge pull request !1252 from caifubi/fix-hccl-nopnode-input
This commit is contained in:
mindspore-ci-bot 2020-05-19 19:57:27 +08:00 committed by Gitee
parent 2a1aad0f55 f25f338008
commit 2aa78e7c10
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc
View File

@ -25,7 +25,7 @@ namespace opt {
namespace {
bool InputIsParameterOrValueNode(const AnfNodePtr &node) {
MS_EXCEPTION_IF_NULL(node);
auto kernel_with_index = AnfAlgo::VisitKernel(node, 0);
auto kernel_with_index = AnfAlgo::VisitKernelWithReturnType(node, 0, true);
return kernel_with_index.first->isa<Parameter>() || kernel_with_index.first->isa<ValueNode>();
}