diff --git a/paddle/phi/kernels/gpu/strided_slice_kernel.cu b/paddle/phi/kernels/gpu/strided_slice_kernel.cu index 716150ff47dea9b8f8166ac0fbd918ff2dbc8133..786ccb287c2719c8b6da433b626ca271c7f06710 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, phi::dtype::complex) {} @@ -40,5 +41,6 @@ PD_REGISTER_KERNEL(strided_slice_array, int64_t, float, double, + phi::dtype::float16, phi::dtype::complex, phi::dtype::complex) {} diff --git a/paddle/phi/kernels/strided_slice_kernel.cc b/paddle/phi/kernels/strided_slice_kernel.cc index 547d574cd78d040987b05b382319e3d1c68c0a84..037abf461a306733a87c342057688475af4e65a1 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, phi::dtype::complex) {} #endif