xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!1802 [bugfix]remove tuple_get_item combination with make_tuple pattern from graph 

Merge pull request !1802 from zyli2020/add_get_item_pass
This commit is contained in:
mindspore-ci-bot 2020-06-03 17:26:50 +08:00 committed by Gitee
parent 1971e3f993 a62e9fbda2
commit a193d0977a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/session/gpu_session.cc
View File

@ -22,6 +22,7 @@
#include "pre_activate/common/pass_manager.h"
#include "pre_activate/common/helper.h"
#include "pre_activate/pass/communication_op_fusion.h"
#include "pre_activate/pass/getitem_tuple.h"
#include "device/kernel_runtime_manager.h"
#include "predict/predict.h"
#include "common/utils.h"
@ -54,6 +55,7 @@ void GPUSession::Optimize(const std::shared_ptr<KernelGraph> &kernel_graph) {
auto optimizer = std::make_shared<opt::GraphOptimizer>();
auto pm = std::make_shared<opt::PassManager>();
pm->AddPass(std::make_shared<opt::AllReduceFusion>());
pm->AddPass(std::make_shared<opt::GetitemTuple>());
optimizer->AddPassManager(pm);
(void)optimizer->Optimize(kernel_graph);
kernel_graph->SetExecOrderByDefault();