From 8e49b05bd0e20cd8d0afd2341765716992648e78 Mon Sep 17 00:00:00 2001 From: Wang Huan Date: Sun, 18 Feb 2024 03:03:45 +0000 Subject: [PATCH] fix view grad kernel name error --- paddle/phi/kernels/stride/view_grad_kernel.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/phi/kernels/stride/view_grad_kernel.cc b/paddle/phi/kernels/stride/view_grad_kernel.cc index d04998c95622e5..19674670b27077 100644 --- a/paddle/phi/kernels/stride/view_grad_kernel.cc +++ b/paddle/phi/kernels/stride/view_grad_kernel.cc @@ -39,7 +39,7 @@ void ViewDtypeGradKernel(const Context& dev_ctx, } // namespace phi PD_REGISTER_KERNEL_FOR_ALL_BACKEND_DTYPE_EXCEPT_CUSTOM( - view_grad_shape, STRIDED, phi::ViewShapeGradKernel) {} + view_shape_grad, STRIDED, phi::ViewShapeGradKernel) {} PD_REGISTER_KERNEL_FOR_ALL_BACKEND_DTYPE_EXCEPT_CUSTOM( - view_grad_dtype, STRIDED, phi::ViewDtypeGradKernel) {} + view_dtype_grad, STRIDED, phi::ViewDtypeGradKernel) {}