[Dy2St][PIR] Value.place should be property (#63404)

SigureMo · web-flow · commit b6301664e60c · 2024-04-11T18:23:54.000+08:00
diff --git a/python/paddle/pir/math_op_patch.py b/python/paddle/pir/math_op_patch.py
@@ -141,6 +141,7 @@ def cuda(self, device_id=None, blocking=True):
         # 1 means cuda place, see paddle/phi/kernels/memcpy_kernel.cc
         return _C_ops.memcpy(self, 1)
 
+    @property
     def place(self):
         """
         Value don't have 'place' interface in static graph mode
diff --git a/test/dygraph_to_static/test_place.py b/test/dygraph_to_static/test_place.py
@@ -17,23 +17,33 @@
 
 from dygraph_to_static_utils import (
     Dy2StTestBase,
-    test_legacy_and_pt_and_pir,
+    test_legacy_and_pt,
+    test_pir_only,
 )
 
 import paddle
 
 
 class TestPlace(Dy2StTestBase):
-    @test_legacy_and_pt_and_pir
-    def test_place(self):
+    @test_legacy_and_pt
+    def test_place_legacy(self):
+        # TODO(cleanup-legacy-ir): remove this test case
         paddle.enable_static()
         x = paddle.to_tensor([1, 2, 3, 4])
         with warnings.catch_warnings(record=True) as w:
             warnings.simplefilter("always")
             self.assertIsNone(x.place())
             self.assertTrue(len(w) == 1)
-            if paddle.framework.use_pir_api():
-                self.assertIn("Value do not have 'place'", str(w[-1].message))
+
+    @test_pir_only
+    def test_place(self):
+        paddle.enable_static()
+        x = paddle.to_tensor([1, 2, 3, 4])
+        with warnings.catch_warnings(record=True) as w:
+            warnings.simplefilter("always")
+            self.assertIsNone(x.place)
+            self.assertTrue(len(w) == 1)
+            self.assertIn("Value do not have 'place'", str(w[-1].message))
 
 
 if __name__ == '__main__':
diff --git a/test/legacy_test/test_math_op_patch_pir.py b/test/legacy_test/test_math_op_patch_pir.py
@@ -408,7 +408,7 @@ def test_place(self):
             warnings.simplefilter("always")
             with paddle.pir_utils.IrGuard():
                 x = paddle.static.data(name='x', shape=[3, 2, 1])
-                x.place()
+                _ = x.place
                 self.assertTrue(len(w) == 1)
                 self.assertTrue("place" in str(w[-1].message))
 
