diff --git a/backends/iluvatar_gpu/CMakeLists.txt b/backends/iluvatar_gpu/CMakeLists.txt index d71fa59857b..38846ab68a1 100644 --- a/backends/iluvatar_gpu/CMakeLists.txt +++ b/backends/iluvatar_gpu/CMakeLists.txt @@ -269,6 +269,7 @@ file( file( GLOB CUDA_SRCS2 + ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/distributed_fused_lamb_init_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/fused_bias_dropout_residual_layer_norm_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/fused_bias_dropout_residual_layer_norm_kernel.cu diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt index bca1ce7aad4..a14ef416c24 100755 --- a/backends/metax_gpu/CMakeLists.txt +++ b/backends/metax_gpu/CMakeLists.txt @@ -120,6 +120,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_grad_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_grad_kernel_register.cu b/backends/metax_gpu/kernels/metax_kernel/spectral_norm_grad_kernel_register.cu index f99621f8ab9..0bf34f29aae 100644 --- a/backends/metax_gpu/kernels/metax_kernel/spectral_norm_grad_kernel_register.cu +++ b/backends/metax_gpu/kernels/metax_kernel/spectral_norm_grad_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_grad_kernel.cu" // NOLINT +#include "paddle/phi/kernels/spectral_norm_grad_kernel.h" PD_CUSTOM_KERNEL_REGISTER(spectral_norm_grad, metax_gpu,