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 cbffb2345f..6e2a53a954 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 GetExpandOps() { prim::kPrimGkDropout, prim::kPrimDropoutGrad, prim::kPrimSoftMax, + prim::kPrimLayerNorm, + prim::kPrimLayerNormGrad, #endif }; return expand_ops;