Use OneDNN's LayerNorm kernel (#40418)

d1811010 · Tomasz Socha · GitHub · f70f5e4f · d1811010
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/operators/mkldnn/layer_norm_mkldnn_op.cc paddle/fluid/operators/mkldnn/layer_norm_mkldnn_op.cc +1 -0

未找到文件。
--- a/paddle/fluid/operators/mkldnn/layer_norm_mkldnn_op.cc
+++ b/paddle/fluid/operators/mkldnn/layer_norm_mkldnn_op.cc
@@ -150,4 +150,5 @@ class LayerNormMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
 // TODO(jczaja): Enable FP32 when performance is good
 namespace ops = paddle::operators;
 REGISTER_OP_KERNEL(layer_norm, MKLDNN, ::paddle::platform::CPUPlace,
+                   ops::LayerNormMKLDNNOpKernel<float>,
                   ops::LayerNormMKLDNNOpKernel<paddle::platform::bfloat16>);