diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu index 040c0fe45a..f251279cfa 100644 --- a/backends/iluvatar_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu +++ b/backends/iluvatar_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu @@ -13,8 +13,6 @@ // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/quantize_linear_kernel.cu" //NOLINT -#include "paddle/phi/kernels/impl/quantize_linear_impl.h" #include "paddle/phi/kernels/quantize_linear_kernel.h" PD_CUSTOM_KERNEL_REGISTER(dequantize_linear, diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt index af64a663c2..681d384bad 100755 --- a/backends/metax_gpu/CMakeLists.txt +++ b/backends/metax_gpu/CMakeLists.txt @@ -422,6 +422,7 @@ file( ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/momentum_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/psroi_pool_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/pixel_shuffle_grad_kernel.cu + ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/quantize_linear_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/lu_unpack_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/nll_loss_grad_kernel.cu ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/nanmedian_kernel.cu diff --git a/backends/metax_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu index b2be6f8d0a..728b6d8e28 100644 --- a/backends/metax_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu +++ b/backends/metax_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu @@ -13,8 +13,6 @@ // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/quantize_linear_kernel.cu" //NOLINT -#include "paddle/phi/kernels/impl/quantize_linear_impl.h" #include "paddle/phi/kernels/quantize_linear_kernel.h" PD_CUSTOM_KERNEL_REGISTER(dequantize_linear,