Register half datatype for Roll Kernel (#49192)

* register half datatype * register roll grad fp16 kernel

Register half datatype for Roll Kernel (#49192)
* register half datatype * register roll grad fp16 kernel
3b90a7f3 · MarDino · GitHub · eaf90003 · 3b90a7f3 · 3b90a7f3
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/phi/kernels/gpu/roll_grad_kernel.cu paddle/phi/kernels/gpu/roll_grad_kernel.cu +1 -0

paddle/phi/kernels/gpu/roll_kernel.cu paddle/phi/kernels/gpu/roll_kernel.cu +1 -0

未找到文件。
--- 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,

--- 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,