diff --git a/paddle/phi/kernels/cpu/concat_kernel.cc b/paddle/phi/kernels/cpu/concat_kernel.cc index a80c9db43c8b46ebc5d5a5bacf8b99bed475cd29..625d7d7436a4687bea71e9dfc108f213afa345b0 100644 --- a/paddle/phi/kernels/cpu/concat_kernel.cc +++ b/paddle/phi/kernels/cpu/concat_kernel.cc @@ -15,9 +15,9 @@ #include "paddle/phi/kernels/concat_kernel.h" #include "paddle/fluid/operators/strided_memcpy.h" -#include "paddle/fluid/platform/bfloat16.h" #include "paddle/fluid/platform/complex.h" #include "paddle/phi/backends/cpu/cpu_context.h" +#include "paddle/phi/common/bfloat16.h" #include "paddle/phi/common/scalar.h" #include "paddle/phi/core/dense_tensor.h" #include "paddle/phi/core/kernel_registry.h" diff --git a/paddle/phi/kernels/cpu/fill_grad_kernel.cc b/paddle/phi/kernels/cpu/fill_grad_kernel.cc index fda723ee2ba801480c6f83ee07626cdc69b20741..04cbb18e938ba28b489b6d3b553cf50347697199 100644 --- a/paddle/phi/kernels/cpu/fill_grad_kernel.cc +++ b/paddle/phi/kernels/cpu/fill_grad_kernel.cc @@ -26,5 +26,5 @@ PD_REGISTER_KERNEL(fill_grad, int64_t, int, phi::dtype::float16, - paddle::platform::bfloat16, + phi::dtype::bfloat16, bool) {} diff --git a/paddle/phi/kernels/cpu/fill_kernel.cc b/paddle/phi/kernels/cpu/fill_kernel.cc index 7e8534a5fbeb86f845f4b949a45ca197213f71af..937e0cf3e43085f3728710c32e85cc1bf34d83e3 100644 --- a/paddle/phi/kernels/cpu/fill_kernel.cc +++ b/paddle/phi/kernels/cpu/fill_kernel.cc @@ -26,5 +26,5 @@ PD_REGISTER_KERNEL(fill, int64_t, int, phi::dtype::float16, - paddle::platform::bfloat16, + phi::dtype::bfloat16, bool) {} diff --git a/paddle/phi/kernels/funcs/math_function.cc b/paddle/phi/kernels/funcs/math_function.cc index 8c32f8bcd61f4fb69c458a55454cce4e294d65b4..77d8a49a7eab2ee616fd9e286ea909f96a1e7ca1 100644 --- a/paddle/phi/kernels/funcs/math_function.cc +++ b/paddle/phi/kernels/funcs/math_function.cc @@ -26,8 +26,8 @@ limitations under the License. */ #include #include -#include "paddle/fluid/platform/bfloat16.h" #include "paddle/phi/backends/cpu/cpu_context.h" +#include "paddle/phi/common/bfloat16.h" #include "paddle/phi/common/data_type.h" #include "paddle/phi/common/float16.h" #include "paddle/phi/kernels/funcs/eigen/common.h" diff --git a/paddle/phi/kernels/funcs/math_function.cu b/paddle/phi/kernels/funcs/math_function.cu index decd668e3a358901aa6ad90483f5cdc25fd3c076..db4cdc57e2f04d54e4ac3122910be492fe511c33 100644 --- a/paddle/phi/kernels/funcs/math_function.cu +++ b/paddle/phi/kernels/funcs/math_function.cu @@ -16,8 +16,8 @@ limitations under the License. */ #include "paddle/fluid/memory/malloc.h" #include "paddle/fluid/memory/memcpy.h" -#include "paddle/fluid/platform/bfloat16.h" #include "paddle/phi/backends/gpu/gpu_context.h" +#include "paddle/phi/common/bfloat16.h" #include "paddle/phi/common/data_type.h" #include "paddle/phi/common/float16.h" #include "paddle/phi/kernels/funcs/blas/blas.h" diff --git a/paddle/phi/kernels/gpu/concat_kernel.cu b/paddle/phi/kernels/gpu/concat_kernel.cu index 6d32205b0bb643491039cc8c43e55b59893de5d9..7635c94cde789fdac4ea19d4a612308c6d386500 100644 --- a/paddle/phi/kernels/gpu/concat_kernel.cu +++ b/paddle/phi/kernels/gpu/concat_kernel.cu @@ -15,9 +15,9 @@ #include "paddle/phi/kernels/concat_kernel.h" #include "paddle/fluid/operators/strided_memcpy.h" -#include "paddle/fluid/platform/bfloat16.h" #include "paddle/fluid/platform/complex.h" #include "paddle/phi/backends/gpu/gpu_context.h" +#include "paddle/phi/common/bfloat16.h" #include "paddle/phi/common/scalar.h" #include "paddle/phi/core/dense_tensor.h" #include "paddle/phi/core/kernel_registry.h" diff --git a/paddle/phi/kernels/gpu/fill_grad_kernel.cu b/paddle/phi/kernels/gpu/fill_grad_kernel.cu index 512aef8c1c9841713af598e9c613cbd874bf1cfc..be1cb3fe2223d945da5352945ee8a9fde93eea8b 100644 --- a/paddle/phi/kernels/gpu/fill_grad_kernel.cu +++ b/paddle/phi/kernels/gpu/fill_grad_kernel.cu @@ -27,5 +27,5 @@ PD_REGISTER_KERNEL(fill_grad, int64_t, int, phi::dtype::float16, - paddle::platform::bfloat16, + phi::dtype::bfloat16, bool) {} diff --git a/paddle/phi/kernels/gpu/fill_kernel.cu b/paddle/phi/kernels/gpu/fill_kernel.cu index de46e678782f2d6d68a6cf78afdb8f0ea0368338..9d17f8af71808ece142f5c9c327bd328790f3237 100644 --- a/paddle/phi/kernels/gpu/fill_kernel.cu +++ b/paddle/phi/kernels/gpu/fill_kernel.cu @@ -27,5 +27,5 @@ PD_REGISTER_KERNEL(fill, int64_t, int, phi::dtype::float16, - paddle::platform::bfloat16, + phi::dtype::bfloat16, bool) {}