enable GraphKernel for layernorm and layernormGrad (sync from r1.1)

4 years ago · c5f261d894
parent 24ef3efcb8
commit c5f261d894
1 changed files with 2 additions and 0 deletions
--- 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;