!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:
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…
Reference in New Issue