Set ALL_BACKEND for shape cpu kernel (#44884)

5e74ba33 · WangZhen · GitHub · d5de7886 · 5e74ba33
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

paddle/phi/kernels/shape_kernel.cc paddle/phi/kernels/shape_kernel.cc +3 -1

未找到文件。
--- a/paddle/phi/kernels/shape_kernel.cc
+++ b/paddle/phi/kernels/shape_kernel.cc
@@ -48,7 +48,9 @@ PD_REGISTER_KERNEL(shape,
                   float,
                   double,
                   phi::dtype::complex<float>,
-                   phi::dtype::complex<double>) {}
+                   phi::dtype::complex<double>) {
+  kernel->InputAt(0).SetBackend(phi::Backend::ALL_BACKEND);
+}

 #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 PD_REGISTER_KERNEL(shape,