From 88de0cffa94789bb11994c56ef9e2cd1bdfdfec3 Mon Sep 17 00:00:00 2001
From: r1chardf1d0 <6567503+r1chardf1d0@user.noreply.gitee.com>
Date: Mon, 14 Sep 2020 11:37:13 +0800
Subject: [PATCH] open graph kernel expander opt for gpu

---
 .../backend/optimizer/graph_kernel/graph_kernel_helper.cc      | 3 ---
 mindspore/ccsrc/backend/session/gpu_session.cc                 | 1 +
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc b/mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
index 132147efa0..89894fdc84 100644
--- a/mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
+++ b/mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
@@ -648,9 +648,6 @@ bool JsonDescToAnf(const std::string &json_desc, const std::map<std::string, Anf
 std::unordered_set<PrimitivePtr> GetExpandOps() {
   std::unordered_set<PrimitivePtr> expand_ops = {
     prim::kPrimSquare,
-    prim::kPrimGelu,
-    prim::kPrimSoftmax,
-    prim::kPrimLayerNorm,
   };
   return expand_ops;
 }
diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc
index 6e72cd81ea..c103aee7cc 100644
--- a/mindspore/ccsrc/backend/session/gpu_session.cc
+++ b/mindspore/ccsrc/backend/session/gpu_session.cc
@@ -116,6 +116,7 @@ void GPUSession::GraphKernelOptimize(const std::shared_ptr<KernelGraph> &kernel_
   }
   auto optimizer = std::make_shared<opt::GraphOptimizer>();
   auto pm = std::make_shared<opt::PassManager>("graph_kernel_pm");
+  pm->AddPass(std::make_shared<opt::GraphKernelExpander>());
   pm->AddPass(std::make_shared<opt::BasicOpsFusion>());
   pm->AddPass(std::make_shared<opt::CompositeOpsFusion>());
   pm->AddPass(std::make_shared<opt::GraphKernelSplitter>());