Strided slice fp16 (#44653)

84d595fa · ming1753 · GitHub · a6d05771 · 84d595fa · 84d595fa
Showing with 3 addition and 0 deletion

paddle/phi/kernels/gpu/strided_slice_kernel.cu paddle/phi/kernels/gpu/strided_slice_kernel.cu +2 -0

paddle/phi/kernels/strided_slice_kernel.cc paddle/phi/kernels/strided_slice_kernel.cc +1 -0

未找到文件。
--- 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>) {}
--- 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