From 38574482c2bd30fce7e7a886a83fb1890c6104cd Mon Sep 17 00:00:00 2001 From: algorithm1832 <737634857@qq.com> Date: Tue, 14 Oct 2025 05:19:32 +0000 Subject: [PATCH] include .h file for c_scatter_kernel --- .../kernels/cuda_kernels/c_scatter_kernel_register.cu | 2 +- backends/metax_gpu/CMakeLists.txt | 1 + .../metax_gpu/kernels/cuda_kernels/c_scatter_kernel_register.cu | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) 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 4a5f59af478..fe6c5eba253 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 e357a5e5912..e51d775608e 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 3c8565069c0..b599c17a69f 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,