p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22833 Add env eliminate pass for run control flow with ms function 

Merge pull request !22833 from JoyLvliang/add_env_eliminate_pass_for_run_control_flow_with_ms_function
This commit is contained in:
i-robot 2021-09-03 01:37:05 +00:00 committed by Gitee
parent 60d583190b d3b856f19a
commit 6b3d787655
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/pipeline/jit/pass.cc
View File

@ -191,8 +191,15 @@ FuncGraphPtr BpropGraphFinalOptPass(const ResourcePtr &res) {
{"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()));
opt::OptPassConfig env_eliminate = opt::OptPassConfig({
irpass.incorporate_call_,
irpass.incorporate_call_switch_,
irpass.incorporate_getitem_set_,
});
map.emplace_back(std::make_pair("env_eliminate", env_eliminate));
}
auto bprop_graph_final_opt = opt::Optimizer::MakeOptimizer("bprop_graph_final_opt", res, map);