fix 2 pir uts (#67399)

ooooo-create · web-flow · commit d0c89b47ad7d · 2024-08-14T09:54:29.000+08:00
diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py
@@ -7808,7 +7808,7 @@ def copysign(x: Tensor, y: Tensor | float, name: str | None = None) -> Tensor:
     if isinstance(y, (float, int)):
         y = paddle.to_tensor(y, dtype=x.dtype)
     out_shape = broadcast_shape(x.shape, y.shape)
-    if out_shape != x.shape:
+    if out_shape != list(x.shape):
         warnings.warn(
             f"The shape of broadcast output {out_shape} is different from the input tensor x with shape: {x.shape}, please make sure you are using copysign api correctly."
         )
diff --git a/test/legacy_test/op_test.py b/test/legacy_test/op_test.py
@@ -2064,8 +2064,13 @@ def check_inplace_output_with_place(
         if getattr(self, "no_need_check_inplace", False):
             return
 
-        if os.getenv("FLAGS_enable_pir_in_executor") or os.getenv(
-            "FLAGS_enable_pir_api"
+        if (
+            os.getenv("FLAGS_enable_pir_in_executor")
+            or os.getenv("FLAGS_enable_pir_api")
+            or get_flags("FLAGS_enable_pir_in_executor")[
+                "FLAGS_enable_pir_in_executor"
+            ]
+            or get_flags("FLAGS_enable_pir_api")["FLAGS_enable_pir_api"]
         ):
             return
 
diff --git a/test/legacy_test/test_copysign_op.py b/test/legacy_test/test_copysign_op.py
@@ -45,16 +45,16 @@ def setUp(self):
         self.outputs = {'out': self.target}
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_pir=True)
 
     def test_check_grad(self):
-        self.check_grad(['x', 'y'], ['out'])
+        self.check_grad(['x', 'y'], ['out'], check_pir=True)
 
     def test_check_grad_ignore_x(self):
-        self.check_grad(['y'], ['out'])
+        self.check_grad(['y'], ['out'], check_pir=True)
 
     def test_check_grad_ignore_y(self):
-        self.check_grad(['x'], ['out'])
+        self.check_grad(['x'], ['out'], check_pir=True)
 
     def init_config(self):
         self.x = np.random.randn(20, 6).astype('float64')
@@ -87,19 +87,21 @@ def init_dtype(self):
 
     def test_check_output(self):
         place = core.CUDAPlace(0)
-        self.check_output_with_place(place)
+        self.check_output_with_place(place, check_pir=True)
 
     def test_check_grad(self):
-        self.check_grad_with_place(self.place, ['x', 'y'], ['out'])
+        self.check_grad_with_place(
+            self.place, ['x', 'y'], ['out'], check_pir=True
+        )
 
     def test_check_grad_ignore_x(self):
         self.check_grad_with_place(
-            self.place, ['y'], ['out'], no_grad_set=set('x')
+            self.place, ['y'], ['out'], no_grad_set=set('x'), check_pir=True
         )
 
     def test_check_grad_ignore_y(self):
         self.check_grad_with_place(
-            self.place, ['x'], ['out'], no_grad_set=set('y')
+            self.place, ['x'], ['out'], no_grad_set=set('y'), check_pir=True
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -7808,7 +7808,7 @@ def copysign(x: Tensor, y: Tensor \| float, name: str \| None = None) -> Tensor:`
`7808`	`7808`	`if isinstance(y, (float, int)):`
`7809`	`7809`	`y = paddle.to_tensor(y, dtype=x.dtype)`
`7810`	`7810`	`out_shape = broadcast_shape(x.shape, y.shape)`
`7811`		`- if out_shape != x.shape:`
	`7811`	`+ if out_shape != list(x.shape):`
`7812`	`7812`	`warnings.warn(`
`7813`	`7813`	`f"The shape of broadcast output {out_shape} is different from the input tensor x with shape: {x.shape}, please make sure you are using copysign api correctly."`
`7814`	`7814`	`)`