[PIR]Open uts for scatter_nd_add (#60698)

0x45f · web-flow · commit 78eec98207e1 · 2024-01-11T10:50:12.000+08:00
* [PIR]Open uts for scatter_nd_add

* Fix ut
diff --git a/test/legacy_test/test_scatter_nd_op.py b/test/legacy_test/test_scatter_nd_op.py
@@ -350,6 +350,7 @@ class TestScatterNdOpAPI(unittest.TestCase):
     test scatter_nd_add api and scatter_nd api
     """
 
+    @test_with_pir_api
     def testcase1(self):
         with static_guard():
             ref1 = paddle.static.data(
@@ -369,6 +370,7 @@ def testcase1(self):
             )
             output1 = paddle.scatter_nd_add(ref1, index1, updates1)
 
+    @test_with_pir_api
     def testcase2(self):
         with static_guard():
             ref2 = paddle.static.data(
@@ -390,6 +392,7 @@ def testcase2(self):
                 ref2, index2, updates2, name="scatter_nd_add"
             )
 
+    @test_with_pir_api
     def testcase3(self):
         with static_guard():
             shape3 = [10, 9, 8, 1, 3]
@@ -405,6 +408,7 @@ def testcase3(self):
             )
             output3 = paddle.scatter_nd(index3, updates3, shape3)
 
+    @test_with_pir_api
     def testcase4(self):
         with static_guard():
             shape4 = [10, 9, 8, 1, 3]
@@ -486,6 +490,7 @@ def test_static_graph():
 
 # Test Raise Error
 class TestScatterNdOpRaise(unittest.TestCase):
+    @test_with_pir_api
     def test_check_raise(self):
         def check_raise_is_test():
             with static_guard():
@@ -527,6 +532,7 @@ def test_check_raise2(self):
                 )
                 output6 = paddle.scatter_nd_add(ref6, index6, updates6)
 
+    @test_with_pir_api
     def test_check_raise3(self):
         def check_raise_is_test():
             with static_guard():
@@ -546,7 +552,7 @@ def check_raise_is_test():
                     if t in str(e):
                         raise ValueError
 
-            self.assertRaises(ValueError, check_raise_is_test)
+        self.assertRaises(ValueError, check_raise_is_test)
 
 
 class TestDygraph(unittest.TestCase):
