ssk015
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

opt addn+zeroslike and opt zeroslike

This commit is contained in:
lianliguang 2021-07-02 09:27:10 +08:00
parent 5f494498a1
commit c659694cf7
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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_,