From fcae3430808576c6a143562410f2527cc793bc70 Mon Sep 17 00:00:00 2001 From: Yang <3349368+m3ngyang@users.noreply.github.com> Date: Wed, 9 Mar 2022 10:10:55 +0800 Subject: [PATCH] fix take_along_axis cuda op register bug (#40270) * fix take_along_axis cuda op register bug * add comma after float Co-authored-by: Chen Weihang --- paddle/phi/kernels/gpu/take_along_axis_kernel.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/phi/kernels/gpu/take_along_axis_kernel.cu b/paddle/phi/kernels/gpu/take_along_axis_kernel.cu index 63113e3e672..9665a917d9d 100644 --- a/paddle/phi/kernels/gpu/take_along_axis_kernel.cu +++ b/paddle/phi/kernels/gpu/take_along_axis_kernel.cu @@ -53,6 +53,7 @@ PD_REGISTER_KERNEL(take_along_axis, GPU, ALL_LAYOUT, phi::TakeAlongAxisKernel, + float, double, int64_t, int, -- GitLab