Browse Source

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

From: @hanhuifeng2020
Reviewed-by: @gaoxiong1,@anyrenwei
Signed-off-by: @anyrenwei
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
f9d9bba927
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc

+ 2
- 0
mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc 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;


Loading…
Cancel
Save