change mkldnn kernel layout, ALL_LAYOUT->ONEDNN (#46626)

paddle/phi/kernels/onednn/reduce_max_kernel.cc

@@ -34,9 +34,5 @@ void MaxRawKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(max_raw,
-                   OneDNN,
-                   ALL_LAYOUT,
-                   phi::MaxRawKernel,
-                   float,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    max_raw, OneDNN, ONEDNN, phi::MaxRawKernel, float, phi::dtype::bfloat16) {}

paddle/phi/kernels/onednn/reduce_mean_grad_kernel.cc

@@ -55,7 +55,7 @@ void MeanGradKernel(const Context& dev_ctx,
 
 PD_REGISTER_KERNEL(mean_grad,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::MeanGradKernel,
                    float,
                    phi::dtype::bfloat16) {}

paddle/phi/kernels/onednn/reduce_mean_kernel.cc

@@ -34,9 +34,6 @@ void MeanRawKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(mean_raw,
-                   OneDNN,
-                   ALL_LAYOUT,
-                   phi::MeanRawKernel,
-                   float,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    mean_raw, OneDNN, ONEDNN, phi::MeanRawKernel, float, phi::dtype::bfloat16) {
+}

paddle/phi/kernels/onednn/reduce_min_kernel.cc

@@ -35,9 +35,5 @@ void MinRawKernel(const Context& dev_ctx,
 
 }  // namespace phi
 
-PD_REGISTER_KERNEL(min_raw,
-                   OneDNN,
-                   ALL_LAYOUT,
-                   phi::MinRawKernel,
-                   float,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    min_raw, OneDNN, ONEDNN, phi::MinRawKernel, float, phi::dtype::bfloat16) {}

paddle/phi/kernels/onednn/reduce_sum_grad_kernel.cc

@@ -39,9 +39,6 @@ void SumGradKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(sum_grad,
-                   OneDNN,
-                   ALL_LAYOUT,
-                   phi::SumGradKernel,
-                   float,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    sum_grad, OneDNN, ONEDNN, phi::SumGradKernel, float, phi::dtype::bfloat16) {
+}

paddle/phi/kernels/onednn/reduce_sum_kernel.cc

@@ -35,9 +35,5 @@ void SumRawKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(sum_raw,
-                   OneDNN,
-                   ALL_LAYOUT,
-                   phi::SumRawKernel,
-                   float,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    sum_raw, OneDNN, ONEDNN, phi::SumRawKernel, float, phi::dtype::bfloat16) {}

paddle/phi/kernels/reduce_max_kernel.cc

@@ -48,7 +48,7 @@ PD_REGISTER_KERNEL(max, KPS, ALL_LAYOUT, phi::MaxKernel, float) {}
 
 #if defined(PADDLE_WITH_MKLDNN)
 PD_REGISTER_KERNEL(
-    max, OneDNN, ALL_LAYOUT, phi::MaxKernel, float, phi::dtype::bfloat16) {}
+    max, OneDNN, ONEDNN, phi::MaxKernel, float, phi::dtype::bfloat16) {}
 #endif
 
 #if defined(PADDLE_WITH_XPU)

paddle/phi/kernels/reduce_mean_kernel.cc

@@ -53,7 +53,7 @@ PD_REGISTER_KERNEL(mean, KPS, ALL_LAYOUT, phi::MeanKernel, float) {}
 
 #if defined(PADDLE_WITH_MKLDNN)
 PD_REGISTER_KERNEL(
-    mean, OneDNN, ALL_LAYOUT, phi::MeanKernel, float, phi::dtype::bfloat16) {}
+    mean, OneDNN, ONEDNN, phi::MeanKernel, float, phi::dtype::bfloat16) {}
 #endif
 
 #if defined(PADDLE_WITH_XPU)

paddle/phi/kernels/reduce_min_kernel.cc

@@ -48,7 +48,7 @@ PD_REGISTER_KERNEL(min, KPS, ALL_LAYOUT, phi::MinKernel, float) {}
 
 #if defined(PADDLE_WITH_MKLDNN)
 PD_REGISTER_KERNEL(
-    min, OneDNN, ALL_LAYOUT, phi::MinKernel, float, phi::dtype::bfloat16) {}
+    min, OneDNN, ONEDNN, phi::MinKernel, float, phi::dtype::bfloat16) {}
 #endif
 
 #if defined(PADDLE_WITH_XPU)

paddle/phi/kernels/reduce_sum_kernel.cc

@@ -81,7 +81,7 @@ PD_REGISTER_KERNEL(sum, KPS, ALL_LAYOUT, phi::SumKernel, float) {
 
 #if defined(PADDLE_WITH_MKLDNN)
 PD_REGISTER_KERNEL(
-    sum, OneDNN, ALL_LAYOUT, phi::SumKernel, float, phi::dtype::bfloat16) {}
+    sum, OneDNN, ONEDNN, phi::SumKernel, float, phi::dtype::bfloat16) {}
 #endif
 
 #if defined(PADDLE_WITH_XPU)