From c5f261d89430ddc6a1f224d13225e422dbef203c Mon Sep 17 00:00:00 2001
From: hanhuifeng2020 <hanhuifeng1@huawei.com>
Date: Tue, 19 Jan 2021 20:30:13 +0800
Subject: [PATCH] enable GraphKernel for layernorm and layernormGrad (sync from
 r1.1)

---
 .../ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc | 2 ++
 1 file changed, 2 insertions(+)

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 cbffb2345f4..6e2a53a9544 100644
--- a/mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
+++ b/mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
@@ -718,6 +718,8 @@ std::unordered_set<PrimitivePtr> GetExpandOps() {
     prim::kPrimGkDropout,
     prim::kPrimDropoutGrad,
     prim::kPrimSoftMax,
+    prim::kPrimLayerNorm,
+    prim::kPrimLayerNormGrad,
 #endif
   };
   return expand_ops;