From 37ae8c299e3185814c0f4c8e26f300de4b0e296b Mon Sep 17 00:00:00 2001 From: Le-soleile <3516093767@qq.com> Date: Mon, 29 Sep 2025 13:11:34 +0000 Subject: [PATCH] Change include --- backends/iluvatar_gpu/CMakeLists.txt | 1 + backends/metax_gpu/CMakeLists.txt | 1 + .../kernels/metax_kernel/spectral_norm_kernel_register.cu | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/backends/iluvatar_gpu/CMakeLists.txt b/backends/iluvatar_gpu/CMakeLists.txt index 14856a71562..1e8b384f2f8 100644 --- a/backends/iluvatar_gpu/CMakeLists.txt +++ b/backends/iluvatar_gpu/CMakeLists.txt @@ -362,6 +362,7 @@ file( ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/shuffle_channel_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/shuffle_channel_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/soft_relu_grad_kernel.cu + ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/stack_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/stft_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt index 6048b59e6c1..e4453f8681e 100755 --- a/backends/metax_gpu/CMakeLists.txt +++ b/backends/metax_gpu/CMakeLists.txt @@ -118,6 +118,7 @@ file( ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/math/*.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/eigen/*.cu # kernels/gpu + ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/activation_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/activation_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/adamw_kernel.cu diff --git a/backends/metax_gpu/kernels/metax_kernel/spectral_norm_kernel_register.cu b/backends/metax_gpu/kernels/metax_kernel/spectral_norm_kernel_register.cu index 466937f993b..5d6d9408cfa 100644 --- a/backends/metax_gpu/kernels/metax_kernel/spectral_norm_kernel_register.cu +++ b/backends/metax_gpu/kernels/metax_kernel/spectral_norm_kernel_register.cu @@ -12,7 +12,7 @@ // See the License for the specific language governing permissions and // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/spectral_norm_kernel.cu" // NOLINT +#include "paddle/phi/kernels/spectral_norm_kernel.h" PD_CUSTOM_KERNEL_REGISTER(spectral_norm, metax_gpu,