diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu index 972127d6c3..d4a2b9d763 100644 --- a/backends/iluvatar_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu +++ b/backends/iluvatar_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu @@ -13,7 +13,7 @@ // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/box_clip_kernel.cu" //NOLINT +#include "paddle/phi/kernels/gpu/box_clip_kernel.h" PD_CUSTOM_KERNEL_REGISTER( box_clip, iluvatar_gpu, ALL_LAYOUT, phi::GPUBoxClipKernel, float, double) {} diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt index c80982449a..e42722f611 100755 --- a/backends/metax_gpu/CMakeLists.txt +++ b/backends/metax_gpu/CMakeLists.txt @@ -146,6 +146,7 @@ file( ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/angle_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/adamax_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/bincount_kernel.cu + ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/box_clip_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cast_kernel.cu diff --git a/backends/metax_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu index 6c16891a84..efa1405776 100644 --- a/backends/metax_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu +++ b/backends/metax_gpu/kernels/cuda_kernels/box_clip_kernel_register.cu @@ -13,7 +13,7 @@ // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/box_clip_kernel.cu" //NOLINT +#include "paddle/phi/kernels/gpu/box_clip_kernel.h" PD_CUSTOM_KERNEL_REGISTER( box_clip, metax_gpu, ALL_LAYOUT, phi::GPUBoxClipKernel, float, double) {}