!12006 [GraphKernel][Gpu]enable GraphKernel for layernorm and layernormGrad (sync from r1.1)

From: @hanhuifeng2020
Reviewed-by: @gaoxiong1,@anyrenwei
Signed-off-by: @anyrenwei
pull/12006/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit f9d9bba927

@ -718,6 +718,8 @@ std::unordered_set<PrimitivePtr> GetExpandOps() {
prim::kPrimGkDropout,
prim::kPrimDropoutGrad,
prim::kPrimSoftMax,
prim::kPrimLayerNorm,
prim::kPrimLayerNormGrad,
#endif
};
return expand_ops;

Loading…
Cancel
Save