!706 Fix CSE bug for `DropoutGenMask` operation

Merge pull request !706 from seatea/fix-cse-bug-for-dropoutgenmask
5 years ago · 81ad297f43
parent b2dccfd393 538cc7e21b
commit 81ad297f43
1 changed files with 3 additions and 0 deletions
--- a/mindspore/ccsrc/optimizer/cse.cc
+++ b/mindspore/ccsrc/optimizer/cse.cc
@ -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;
    }
  }