Browse Source

!706 Fix CSE bug for `DropoutGenMask` operation

Merge pull request !706 from seatea/fix-cse-bug-for-dropoutgenmask
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
81ad297f43
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/optimizer/cse.cc

+ 3
- 0
mindspore/ccsrc/optimizer/cse.cc View File

@@ -122,6 +122,9 @@ bool CSE::CheckReplace(const AnfNodePtr &main, const AnfNodePtr &node) const {
break;
}
}
if (IsPrimitiveCNode(c_main, prim::kPrimDropoutGenMask)) {
appsame = false;
}
replace = appsame;
}
}


Loading…
Cancel
Save