From 84d595faecddb9a3e53b007f63bde5b4aac6c390 Mon Sep 17 00:00:00 2001
From: ming1753 <61511741+ming1753@users.noreply.github.com>
Date: Wed, 27 Jul 2022 17:03:45 +0800
Subject: [PATCH] Strided slice fp16 (#44653)

---
 paddle/phi/kernels/gpu/strided_slice_kernel.cu | 2 ++
 paddle/phi/kernels/strided_slice_kernel.cc     | 1 +
 2 files changed, 3 insertions(+)

diff --git a/paddle/phi/kernels/gpu/strided_slice_kernel.cu b/paddle/phi/kernels/gpu/strided_slice_kernel.cu
index 716150ff47d..786ccb287c2 100644
--- a/paddle/phi/kernels/gpu/strided_slice_kernel.cu
+++ b/paddle/phi/kernels/gpu/strided_slice_kernel.cu
@@ -28,6 +28,7 @@ PD_REGISTER_KERNEL(strided_slice_raw,
                    int64_t,
                    float,
                    double,
+                   phi::dtype::float16,
                    phi::dtype::complex<float>,
                    phi::dtype::complex<double>) {}
 
@@ -40,5 +41,6 @@ PD_REGISTER_KERNEL(strided_slice_array,
                    int64_t,
                    float,
                    double,
+                   phi::dtype::float16,
                    phi::dtype::complex<float>,
                    phi::dtype::complex<double>) {}
diff --git a/paddle/phi/kernels/strided_slice_kernel.cc b/paddle/phi/kernels/strided_slice_kernel.cc
index 547d574cd78..037abf461a3 100644
--- a/paddle/phi/kernels/strided_slice_kernel.cc
+++ b/paddle/phi/kernels/strided_slice_kernel.cc
@@ -55,6 +55,7 @@ PD_REGISTER_KERNEL(strided_slice,
                    int64_t,
                    float,
                    double,
+                   phi::dtype::float16,
                    phi::dtype::complex<float>,
                    phi::dtype::complex<double>) {}
 #endif
-- 
GitLab