From a62e9fbda2d41f986f285f09c0ce0c2565c4fd18 Mon Sep 17 00:00:00 2001
From: lizhenyu <lizhenyu13@huawei.com>
Date: Tue, 2 Jun 2020 20:31:45 +0800
Subject: [PATCH] remove tuple_get_item combination with make_tuple pattern
 from graph

---
 mindspore/ccsrc/session/gpu_session.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/ccsrc/session/gpu_session.cc b/mindspore/ccsrc/session/gpu_session.cc
index b843514793..aa0cf3ca42 100644
--- a/mindspore/ccsrc/session/gpu_session.cc
+++ b/mindspore/ccsrc/session/gpu_session.cc
@@ -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();