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

!1033 fix addn error in pynative mode when inputs number over 1000 

Merge pull request !1033 from chujinjin/fix_addn_in_pynative_mode
This commit is contained in:
mindspore-ci-bot 2020-05-11 10:03:20 +08:00 committed by Gitee
parent dd2062bf8d f1d6dd5878
commit ebfdfacd7c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc
View File

@ -240,6 +240,7 @@ void RunOpAscendBackendIRFusionOptimization(const std::shared_ptr<session::Kerne
auto ir_fusion_pm = std::make_shared<PassManager>("ir_fusion_pm");
ir_fusion_pm->AddPass(std::make_shared<BnSplit>());
ir_fusion_pm->AddPass(std::make_shared<TopKSplit>());
ir_fusion_pm->AddPass(std::make_shared<AddnFission>());
optimizer->AddPassManager(ir_fusion_pm);
(void)optimizer->Optimize(kernel_graph);