From 3b90a7f3587c251a29ffabc39655eb81fd1e8f5d Mon Sep 17 00:00:00 2001
From: ZZK <359521840@qq.com>
Date: Fri, 23 Dec 2022 11:13:58 +0800
Subject: [PATCH] Register half datatype for Roll Kernel (#49192)

* register half datatype

* register roll grad fp16 kernel
---
 paddle/phi/kernels/gpu/roll_grad_kernel.cu | 1 +
 paddle/phi/kernels/gpu/roll_kernel.cu      | 1 +
 2 files changed, 2 insertions(+)

diff --git a/paddle/phi/kernels/gpu/roll_grad_kernel.cu b/paddle/phi/kernels/gpu/roll_grad_kernel.cu
index ff3ac29117..bce5997ec6 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 60db02d846..6e627374b0 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,
-- 
GitLab