[XPU] zero size support for gaussian op (#71062)

houj04 · web-flow · commit f3e14feb3692 · 2025-02-11T09:57:05.000+08:00
* [XPU] zero size support for gaussian op

* fix ut list
diff --git a/paddle/phi/kernels/xpu/gaussian_kernel.cc b/paddle/phi/kernels/xpu/gaussian_kernel.cc
@@ -30,6 +30,11 @@ void GaussianKernel(const Context& ctx,
                     DenseTensor* out) {
   out->Resize(common::make_ddim(shape.GetData()));
   T* data = ctx.template Alloc<T>(out);
+
+  if (out->numel() == 0) {
+    return;
+  }
+
   using XPUType = typename XPUTypeTrait<T>::Type;
   int64_t real_seed = seed != 0 ? seed : ctx.GetGenerator()->Random64();
 
diff --git a/test/xpu/test_gaussian_random_op_xpu.py b/test/xpu/test_gaussian_random_op_xpu.py
@@ -342,6 +342,13 @@ def test_default_fp64():
         paddle.enable_static()
 
 
+class TestZeroSizeRandN(unittest.TestCase):
+    def test_zero_size_randn(self):
+        paddle.disable_static()
+        x = paddle.randn((0,))
+        paddle.enable_static()
+
+
 support_types = get_xpu_op_support_types('gaussian_random')
 for stype in support_types:
     create_test_class(globals(), XPUTestGaussianRandomOp, stype)
diff --git a/tools/xpu/disable_ut_xpu_kl3.local b/tools/xpu/disable_ut_xpu_kl3.local
@@ -41,7 +41,6 @@ test_auto_checkpoint_dist_basic
 test_auto_checkpoint_multiple
 test_auto_recompute_dy2static
 test_autograd_functional_dynamic
-test_backward_without_params
 test_c_comm_init_op
 test_c_concat
 test_c_split
