Browse Source

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

tags/v1.2.0-rc1
hanhuifeng2020 4 years ago
parent
commit
c5f261d894
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