Browse Source

Support NopNode as input of hccl operator

tags/v0.3.0-alpha
caifubi 5 years ago
parent
commit
f25f338008
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc

+ 1
- 1
mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc View File

@@ -25,7 +25,7 @@ namespace opt {
namespace { namespace {
bool InputIsParameterOrValueNode(const AnfNodePtr &node) { bool InputIsParameterOrValueNode(const AnfNodePtr &node) {
MS_EXCEPTION_IF_NULL(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>(); return kernel_with_index.first->isa<Parameter>() || kernel_with_index.first->isa<ValueNode>();
} }




Loading…
Cancel
Save