From 40ab6faf5b2c243ec89509bb9536bf1eb0605c80 Mon Sep 17 00:00:00 2001
From: ming1753 <61511741+ming1753@users.noreply.github.com>
Date: Thu, 29 Sep 2022 10:58:41 +0800
Subject: [PATCH] add register for strided_slice_grad (#46549)

---
 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 38dd360ea66..8551fc6e845 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<float>,
                    phi::dtype::complex<double>) {}
 #endif
-- 
GitLab