fix xpu expand

PaddlePaddle · Feb 21, 2023 · 91ab26c · 91ab26c
1 parent 6624842
commit 91ab26c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/paddle/phi/kernels/xpu/expand_as_kernel.cc b/paddle/phi/kernels/xpu/expand_as_kernel.cc
@@ -50,7 +50,6 @@ void ExpandAs(const Context& context,
     }
   }
   if (target_shape.size() == 0) {
-    std::cout << "XYY Debug, debug the 「Tensor holds no memory」" << std::endl;
     phi::DDim out_dims = phi::make_ddim(target_shape);
     out->Resize(out_dims);
     context.template Alloc<T>(out);

diff --git a/paddle/phi/kernels/xpu/expand_kernel.cc b/paddle/phi/kernels/xpu/expand_kernel.cc
@@ -95,6 +95,10 @@ void ExpandKernel(const Context& ctx,
           rank));
 
   if (shape_size == 0) {
+    phi::DDim out_dims = phi::make_ddim(final_expand_shape);
+    out->Resize(out_dims);
+    context.template Alloc<T>(out);
+
     int r = xpu::copy<XPUType>(ctx.x_context(),
                                reinterpret_cast<const XPUType*>(x.data<T>()),
                                reinterpret_cast<XPUType*>(out->data<T>()),