浏览代码

Support NopNode as input of hccl operator

tags/v0.3.0-alpha
caifubi 5 年前
父节点
当前提交
f25f338008
共有 1 个文件被更改,包括 1 次插入1 次删除
  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 查看文件

@@ -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>();
}



正在加载...
取消
保存