diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu index 4a5f59af47..fe6c5eba25 100644 --- a/backends/iluvatar_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu +++ b/backends/iluvatar_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu @@ -13,7 +13,7 @@ // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/c_scatter_kernel.cu" //NOLINT +#include "paddle/phi/kernels/gpu/c_scatter_kernel.h" PD_CUSTOM_KERNEL_REGISTER(c_scatter, iluvatar_gpu, diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt index e357a5e591..e51d775608 100755 --- a/backends/metax_gpu/CMakeLists.txt +++ b/backends/metax_gpu/CMakeLists.txt @@ -148,6 +148,7 @@ file( ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/bincount_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/c_scatter_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cast_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/clip_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/clip_grad_kernel.cu diff --git a/backends/metax_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu index 3c8565069c..b599c17a69 100644 --- a/backends/metax_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu +++ b/backends/metax_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu @@ -13,7 +13,7 @@ // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/c_scatter_kernel.cu" //NOLINT +#include "paddle/phi/kernels/gpu/c_scatter_kernel.h" PD_CUSTOM_KERNEL_REGISTER(c_scatter, metax_gpu,