From c659694cf799f6bf6a8b51fdec43e15c56c2aea3 Mon Sep 17 00:00:00 2001
From: lianliguang <lianliguang@huawei.com>
Date: Fri, 2 Jul 2021 09:27:10 +0800
Subject: [PATCH] opt addn+zeroslike and opt zeroslike

---
 mindspore/ccsrc/pipeline/jit/pass.cc | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/mindspore/ccsrc/pipeline/jit/pass.cc b/mindspore/ccsrc/pipeline/jit/pass.cc
index 3cc4c0b15b6..298a91a60cb 100644
--- a/mindspore/ccsrc/pipeline/jit/pass.cc
+++ b/mindspore/ccsrc/pipeline/jit/pass.cc
@@ -188,9 +188,12 @@ FuncGraphPtr BpropGraphFinalOptPass(const ResourcePtr &res) {
     irpass.reshape_eliminate_,
     irpass.switch_simplify_,
     irpass.addn_zero_filter_,
-    irpass.zero_like_fill_zero_,
   });
-  OptPassGroupMap map({{"ad_final_opt", bg_final_opt}});
+  opt::OptPassConfig fill_zeros_like = opt::OptPassConfig{irpass.zero_like_fill_zero_};
+  OptPassGroupMap map({
+    {"ad_final_opt", bg_final_opt},
+    {"zeros_like", fill_zeros_like},
+  });
   if (pynative::PynativeExecutor::GetInstance()->grad_executor()->need_renormalize()) {
     map.emplace_back(std::make_pair("renormalize", opt::OptPassConfig::Renormalize()));
   }
@@ -256,7 +259,6 @@ opt::OptPassConfig GetOptPassA1(const opt::irpass::OptimizeIRPassLib &irpass) {
     irpass.env_get_set_item_eliminate_,
     irpass.env_get_item_depend_swap_,
 
-    irpass.cast_eliminate_,
     irpass.reshape_eliminate_,
     irpass.reduce_eliminate_,
     irpass.tile_eliminate_,
@@ -283,6 +285,7 @@ OptPassGroupMap GetOptPassesA(const opt::irpass::OptimizeIRPassLib &irpass) {
   opt::OptPassConfig a_1 = GetOptPassA1(irpass);
   opt::OptPassConfig a_2 = opt::OptPassConfig(
     {
+      irpass.cast_eliminate_,
       irpass.specialize_transform_,
       irpass.merge_addn_,
       irpass.float_tuple_getitem_switch_,