[0-size Tensor Job2 No.97] Add 0-size Tensor support for paddle.Tensor.set_[fluid_ops] (#74200)

* Fix * Fix * Fix

Author: co63oc

paddle/phi/kernels/set_kernel.cc

@@ -13,7 +13,7 @@
 // limitations under the License.
 #include "paddle/phi/kernels/set_kernel.h"
 #include "paddle/phi/core/kernel_registry.h"
-
+#include "paddle/phi/kernels/full_kernel.h"
 namespace phi {
 
 template <typename T, typename Context>
@@ -28,6 +28,17 @@ void SetKernel(const Context& dev_ctx,
  meta.dims = DDim(dims.data(), static_cast<int>(dims.size()));
  meta.strides = DDim(stride.data(), static_cast<int>(stride.size()));
  meta.offset = offset;
+ if (x.numel() == 0 || source.numel() == 0) {
+ if (source.numel() != 0) {
+ out->clear();
+ *out = DenseTensor{source.Holder(), meta};
+ } else if (x.numel() == 0) {
+ phi::Full<T, Context>(
+ dev_ctx, phi::IntArray(common::vectorize(out->dims())), 0, out);
+ }
+ out->ShareInplaceVersionCounterWith(x);
+ return;
+ }
  if (x.IsSharedWith(source)) {
  out->set_meta(meta);
  } else {

test/legacy_test/test_inplace.py

@@ -16,6 +16,7 @@
 import unittest
 
 import numpy as np
+from op_test import get_places
 
 import paddle
 
@@ -2491,5 +2492,16 @@ def test_inplace_api(self):
  self.assertTrue(id(x) == id(inplace_x2))
 
 
+class TestSet_API_ZeroSize(unittest.TestCase):
+ def setUp(self):
+ self.places = get_places()
+
+ def test_set_api(self):
+ for place in self.places:
+ with paddle.base.dygraph.guard(place):
+ out = paddle.randn([20]).set_(paddle.randn([0, 3]), [20], [2])
+ np.testing.assert_allclose(out.shape, [20])
+
+
 if __name__ == '__main__':
  unittest.main()