[XShape]Fix optimzier kernel param == param_out (#67572)

Aurelius84 · web-flow · commit bbb53d396850 · 2024-08-20T18:49:28.000+08:00
diff --git a/paddle/phi/kernels/impl/adagrad_kernel_impl.h b/paddle/phi/kernels/impl/adagrad_kernel_impl.h
@@ -113,8 +113,8 @@ void AdagradSparseKernel(const Context& ctx,
                         "the input tensor not equal with output tensor"));
 
   auto* moment_tensor = &moment_t;
-  PADDLE_ENFORCE_EQ(moment_tensor,
-                    moment_out_tensor,
+  PADDLE_ENFORCE_EQ(moment_tensor->IsSharedBufferWith(*moment_out_tensor),
+                    true,
                     common::errors::InvalidArgument(
                         "the input moment not equal with output moment"));
 
