【Paddle Tensor 规范化第二期】paddle.copysign support 0-size tensor (#72692)

fangfangssj · web-flow · commit d8cb30c9bca1 · 2025-05-16T16:48:06.000+08:00
* add copysign 0size

* add test

* fix test

* fix test

* fix coverage

* fix

* fix
diff --git a/paddle/fluid/eager/auto_code_generator/generator/eager_gen.py b/paddle/fluid/eager/auto_code_generator/generator/eager_gen.py
@@ -119,6 +119,7 @@
     "huber_loss": ["input", "label"],
     "nextafter": ["x", "y"],
     "atan2": ["x", "y"],
+    "copysign": ["x", "y"],
 }
 
 type_promote_inplace_white_list = {
@@ -137,6 +138,7 @@
     "logical_or_": ["x", "y"],
     "logical_xor_": ["x", "y"],
     "remainder_": ["x", "y"],
+    "copysign_": ["x", "y"],
 }
 
 # ops support casting int tensor into float32 to do forward calculation
diff --git a/paddle/fluid/pir/dialect/op_generator/api_gen.py b/paddle/fluid/pir/dialect/op_generator/api_gen.py
@@ -52,6 +52,7 @@
     "nextafter": ["x", "y"],
     "atan2": ["x", "y"],
     "multiply": ["x", "y"],
+    "copysign": ["x", "y"],
 }
 
 type_promote_inplace_white_list = {
@@ -70,6 +71,7 @@
     "logical_or_": ["x", "y"],
     "logical_xor_": ["x", "y"],
     "remainder_": ["x", "y"],
+    "copysign_": ["x", "y"],
 }
 
 # ops support casting int tensor into float32 to do forward calculation
diff --git a/paddle/phi/common/type_promotion.h b/paddle/phi/common/type_promotion.h
@@ -92,6 +92,7 @@ static std::unordered_set<std::string> support_promotion_ops = {
     "divide",    "elementwise_div", "truediv",         "floor_divide",
     "pow",       "elementwise_pow", "equal",           "not_equal",
     "less_than", "less_equal",      "greater_than",    "greater_equal",
+    "copysign",
 };
 
 static std::unordered_set<std::string> support_autocast_ops = {
diff --git a/paddle/phi/kernels/cpu/elementwise_kernel.cc b/paddle/phi/kernels/cpu/elementwise_kernel.cc
@@ -83,6 +83,10 @@ void CopySignKernel(const Context& dev_ctx,
                     const DenseTensor& x,
                     const DenseTensor& y,
                     DenseTensor* out) {
+  if (out->numel() == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
   dev_ctx.template Alloc<T>(out);
   auto x_dims = x.dims();
   auto y_dims = y.dims();
diff --git a/paddle/phi/kernels/kps/elementwise_kernel.cu b/paddle/phi/kernels/kps/elementwise_kernel.cu
@@ -189,6 +189,10 @@ void CopySignKernel(const Context& dev_ctx,
                     const DenseTensor& x,
                     const DenseTensor& y,
                     DenseTensor* out) {
+  if (out->numel() == 0) {
+    dev_ctx.template Alloc<T>(out);
+    return;
+  }
   std::vector<const DenseTensor*> inputs = {&x, &y};
   std::vector<DenseTensor*> outputs = {out};
   dev_ctx.template Alloc<T>(out);
diff --git a/test/legacy_test/test_copysign_op.py b/test/legacy_test/test_copysign_op.py
@@ -132,7 +132,7 @@ def test_static_api(self):
                 y = self.y
             else:
                 y = paddle.static.data(
-                    name='y', shape=self.y.shape, dtype=self.x.dtype
+                    name='y', shape=self.y.shape, dtype=self.y.dtype
                 )
             out = paddle.copysign(x, y)
             exe = paddle.static.Executor(self.place)
@@ -300,6 +300,114 @@ def input_init(self):
         self.y = (np.random.randn(3, 4, 5) * 10).astype(dtype)
 
 
+class TestCopySignZeroSize1(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.random.randn(0, 5)
+        self.y = np.random.randn(0, 5)
+
+    def place_init(self):
+        self.place = paddle.CPUPlace()
+
+
+class TestCopySignZeroSize2(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.random.randn(0, 5)
+        self.y = np.random.randn(3, 0, 5)
+
+    def place_init(self):
+        self.place = paddle.CPUPlace()
+
+
+class TestCopySignZeroSize3(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.random.randn(3, 0, 5)
+        self.y = np.random.randn(0, 5)
+
+
+class TestCopySignZeroSize4(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.random.randn(5, 0)
+        self.y = np.random.randn(3, 5, 0)
+
+
+class TestCopySignZeroSize5(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.random.randn(2, 5)
+        self.y = np.random.randn(0, 2, 5)
+
+
+class TestCopySignZeroSize6(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.random.randn(0)
+        self.y = np.random.randn(0)
+
+
+class TestCopySignTypePromotion(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.random.randn(2, 5).astype(np.float64)
+        self.y = np.random.randn(2, 5).astype(np.float32)
+
+
+class TestCopySignNan1(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[1.0, 2.0], [3.0, 4.0]], dtype=np.float64)
+        self.y = np.array(
+            [[np.nan, np.nan], [np.nan, np.nan]], dtype=np.float64
+        )
+        self.y.view('uint64')[0, 0] |= np.uint64(0x8000000000000000)
+
+
+class TestCopySignNan2(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[1.0, 2.0], [3.0, 4.0]], dtype=np.float64)
+        self.y = np.array(
+            [[np.nan, np.nan], [np.nan, np.nan]], dtype=np.float64
+        )
+        self.y.view('uint64')[0, 0] &= ~np.uint64(0x8000000000000000)
+
+
+class TestCopySignNan3(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[np.nan, 2.0], [3.0, 4.0]])
+        self.y = np.array([[-1.0, np.nan], [np.nan, np.nan]])
+
+
+class TestCopySignNan4(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[np.nan, 2.0], [3.0, 4.0]])
+        self.y = np.array([[-0.0, np.nan], [np.nan, np.nan]])
+
+
+class TestCopySignNan5(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[np.nan, 2.0], [3.0, 4.0]])
+        self.y = np.array([[0.0, np.nan], [np.nan, np.nan]])
+
+
+class TestCopySignNan6(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[np.nan, 2.0], [3.0, 4.0]])
+        self.y = np.array([[1.0, np.nan], [np.nan, np.nan]])
+
+
+class TestCopySignNan7(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[np.nan, 2.0], [3.0, 4.0]], dtype=np.float64)
+        self.y = np.array(
+            [[np.nan, np.nan], [np.nan, np.nan]], dtype=np.float64
+        )
+        self.y.view('uint64')[0, 0] |= np.uint64(0x8000000000000000)
+
+
+class TestCopySignNan8(TestCopySignAPI):
+    def input_init(self):
+        self.x = np.array([[np.nan, 2.0], [3.0, 4.0]], dtype=np.float64)
+        self.y = np.array(
+            [[np.nan, np.nan], [np.nan, np.nan]], dtype=np.float64
+        )
+        self.y.view('uint64')[0, 0] &= ~np.uint64(0x8000000000000000)
+
+
 if __name__ == "__main__":
     paddle.enable_static()
     unittest.main()

Original file line number	Diff line number	Diff line change
`@@ -119,6 +119,7 @@`
`119`	`119`	`"huber_loss": ["input", "label"],`
`120`	`120`	`"nextafter": ["x", "y"],`
`121`	`121`	`"atan2": ["x", "y"],`
	`122`	`+ "copysign": ["x", "y"],`
`122`	`123`	`}`
`123`	`124`
`124`	`125`	`type_promote_inplace_white_list = {`
`@@ -137,6 +138,7 @@`
`137`	`138`	`"logical_or_": ["x", "y"],`
`138`	`139`	`"logical_xor_": ["x", "y"],`
`139`	`140`	`"remainder_": ["x", "y"],`
	`141`	`+ "copysign_": ["x", "y"],`
`140`	`142`	`}`
`141`	`143`
`142`	`144`	`# ops support casting int tensor into float32 to do forward calculation`
Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,7 @@`
`52`	`52`	`"nextafter": ["x", "y"],`
`53`	`53`	`"atan2": ["x", "y"],`
`54`	`54`	`"multiply": ["x", "y"],`
	`55`	`+ "copysign": ["x", "y"],`
`55`	`56`	`}`
`56`	`57`
`57`	`58`	`type_promote_inplace_white_list = {`
`@@ -70,6 +71,7 @@`
`70`	`71`	`"logical_or_": ["x", "y"],`
`71`	`72`	`"logical_xor_": ["x", "y"],`
`72`	`73`	`"remainder_": ["x", "y"],`
	`74`	`+ "copysign_": ["x", "y"],`
`73`	`75`	`}`
`74`	`76`
`75`	`77`	`# ops support casting int tensor into float32 to do forward calculation`