From 538cc7e21bfad1fe94b1264a3c98131fadd50963 Mon Sep 17 00:00:00 2001
From: seatea <t.cai@huawei.com>
Date: Sun, 26 Apr 2020 19:11:53 +0800
Subject: [PATCH] Fix CSE bug for `DropoutGenMask` operation.

---
 mindspore/ccsrc/optimizer/cse.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/ccsrc/optimizer/cse.cc b/mindspore/ccsrc/optimizer/cse.cc
index 42ebf5a658..ff7291cd86 100644
--- 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;
     }
   }