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

From: @hanhuifeng2020
Reviewed-by: @gaoxiong1,@anyrenwei
Signed-off-by: @anyrenwei
This commit is contained in:
mindspore-ci-bot 2021-02-03 17:58:37 +08:00 committed by Gitee
commit f9d9bba927
1 changed files with 2 additions and 0 deletions

View File

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