From 23f2a4eab715c83b8b68fcbf8114ea8f2f6ed235 Mon Sep 17 00:00:00 2001 From: Zhang Ting Date: Wed, 19 Oct 2022 20:20:42 +0800 Subject: [PATCH] [cherry-pick] strided_slice grad add fp16 support (#47159) * strided_slice grad add fp16 support --- paddle/phi/kernels/strided_slice_grad_kernel.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/phi/kernels/strided_slice_grad_kernel.cc b/paddle/phi/kernels/strided_slice_grad_kernel.cc index 38dd360ea6..8551fc6e84 100644 --- 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, phi::dtype::complex) {} #endif -- GitLab