diff --git a/paddle/phi/kernels/gpu/roll_grad_kernel.cu b/paddle/phi/kernels/gpu/roll_grad_kernel.cu index ff3ac29117e413d6cce7ed041b2c60b1a8eb4ac6..bce5997ec6cd2ec8cc163d3cceb59c549e3b36c1 100644 --- a/paddle/phi/kernels/gpu/roll_grad_kernel.cu +++ b/paddle/phi/kernels/gpu/roll_grad_kernel.cu @@ -80,6 +80,7 @@ PD_REGISTER_KERNEL(roll_grad, GPU, ALL_LAYOUT, phi::RollGradKernel, + phi::dtype::float16, float, double, int, diff --git a/paddle/phi/kernels/gpu/roll_kernel.cu b/paddle/phi/kernels/gpu/roll_kernel.cu index 60db02d8463174eb032ed1ad6813bc8882d3854b..6e627374b08646724c98fa42968eab35b6d951a6 100644 --- a/paddle/phi/kernels/gpu/roll_kernel.cu +++ b/paddle/phi/kernels/gpu/roll_kernel.cu @@ -82,6 +82,7 @@ PD_REGISTER_KERNEL(roll, GPU, ALL_LAYOUT, phi::RollKernel, + phi::dtype::float16, float, double, int,