add register for strided_slice_grad (#46549)

40ab6faf · ming1753 · GitHub · f6039929 · 40ab6faf
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

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

未找到文件。
--- a/paddle/phi/kernels/strided_slice_grad_kernel.cc
+++ b/paddle/phi/kernels/strided_slice_grad_kernel.cc
@@ -64,6 +64,7 @@ PD_REGISTER_KERNEL(strided_slice_grad,
                   int64_t,
                   float,
                   double,
+                   phi::dtype::float16,
                   phi::dtype::complex<float>,
                   phi::dtype::complex<double>) {}
 #endif