Browse Source

!12395 [auto-monad] Do not insert VirtualDiv after UpdateState.

From: @huangbingjian
Reviewed-by: @hwhewei
Signed-off-by:
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
4a38cbdf4a
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/frontend/parallel/step_parallel.cc

+ 2
- 1
mindspore/ccsrc/frontend/parallel/step_parallel.cc View File

@@ -942,7 +942,8 @@ void InsertVirtualDivOp(const VirtualDivOp &virtual_div_op, const CNodePtr &node
for (size_t index = 1; index < node_size; ++index) {
AnfNodePtr input = node->input(index);
MS_EXCEPTION_IF_NULL(input);
if (!input->isa<CNode>() && !input->isa<Parameter>()) { // if it is not a tensor, continue
// if it is not a tensor, continue
if ((!input->isa<CNode>() && !input->isa<Parameter>()) || HasAbstractMonad(input)) {
MS_LOG(INFO) << "insert div op: the index " << index << " is not tensor, skip";
continue;
}


Loading…
Cancel
Save