【NewIR】enable dy2st new IR test

test/dygraph_to_static/dygraph_to_static_util.py

@@ -153,19 +153,18 @@ def impl(*args, **kwargs):
             ir_outs = test_with_new_ir(func)(*args, **kwargs)
             if not need_check_output:
                 return outs
-            for i in range(len(outs)):
-                np.testing.assert_array_equal(
-                    outs[i],
-                    ir_outs[i],
-                    err_msg='Dy2St Unittest Check ('
-                    + func.__name__
-                    + ') has diff '
-                    + '\nExpect '
-                    + str(outs[i])
-                    + '\n'
-                    + 'But Got'
-                    + str(ir_outs[i]),
-                )
+            np.testing.assert_array_equal(
+                outs,
+                ir_outs,
+                err_msg='Dy2St Unittest Check ('
+                + func.__name__
+                + ') has diff '
+                + '\nExpect '
+                + str(outs)
+                + '\n'
+                + 'But Got'
+                + str(ir_outs),
+            )
             return outs
 
         return impl

test/dygraph_to_static/test_assert.py

@@ -66,6 +66,7 @@ def test_bool_variable(self):
             dyfunc_assert_variable, x=numpy.array([True]), with_exception=False
         )
 
+    @test_and_compare_with_new_ir(False)
     def test_int_variable(self):
         self._run_dy_static(
             dyfunc_assert_variable, x=numpy.array([0]), with_exception=True

test/dygraph_to_static/test_ast_util.py

@@ -17,6 +17,7 @@
 import unittest
 
 import numpy as np
+from dygraph_to_static_util import test_and_compare_with_new_ir
 from ifelse_simple_func import (
     dyfunc_with_if_else,
     dyfunc_with_if_else2,
@@ -60,6 +61,7 @@ def test_ast2func_dygraph(self):
                 test_ret = self._ast2func(func)(x_v).numpy()
                 self.assertTrue((true_ret == test_ret).all())
 
+    @test_and_compare_with_new_ir(False)
     def test_ast2func_static(self):
         paddle.enable_static()
 

test/dygraph_to_static/test_backward_without_params.py

@@ -15,6 +15,7 @@
 import unittest
 
 import numpy as np
+from dygraph_to_static_util import test_and_compare_with_new_ir
 
 import paddle
 
@@ -30,6 +31,7 @@ def forward(self, x):
 
 
 class TestBackwardWithoutParams(unittest.TestCase):
+    @test_and_compare_with_new_ir(False)
     def test_run(self):
         net = Net()
 
@@ -54,6 +56,7 @@ def forward(self, x):
 
 
 class TestZeroSizeNet(unittest.TestCase):
+    @test_and_compare_with_new_ir(False)
     def test_run(self):
         net = ZeroSizeNet()
         x = paddle.ones([2, 2])

test/dygraph_to_static/test_basic_api_transformation.py

@@ -16,6 +16,7 @@
 import unittest
 
 import numpy as np
+from dygraph_to_static_util import test_and_compare_with_new_ir
 
 import paddle
 from paddle import fluid, to_tensor
@@ -91,6 +92,7 @@ def get_dygraph_output(self):
             res = self.dygraph_func(self.input).numpy()
             return res
 
+    @test_and_compare_with_new_ir(True)
     def get_static_output(self):
         main_program = fluid.Program()
         main_program.random_seed = SEED
@@ -245,6 +247,7 @@ def get_dygraph_output(self):
 
             return res
 
+    @test_and_compare_with_new_ir(True)
     def get_static_output(self):
         startup_program = fluid.Program()
         startup_program.random_seed = SEED
@@ -278,6 +281,7 @@ def get_dygraph_output(self):
             res = self.dygraph_func(self.input1, self.input2).numpy()
             return res
 
+    @test_and_compare_with_new_ir(True)
     def get_static_output(self):
         startup_program = fluid.Program()
         startup_program.random_seed = SEED
@@ -403,6 +407,7 @@ def get_dygraph_output(self):
             res = self.dygraph_func().numpy()
             return res
 
+    @test_and_compare_with_new_ir(True)
     def get_static_output(self):
         startup_program = fluid.Program()
         startup_program.random_seed = SEED
@@ -433,6 +438,7 @@ def get_dygraph_output(self):
             res = self.dygraph_func()
             return res
 
+    @test_and_compare_with_new_ir(True)
     def get_static_output(self):
         startup_program = fluid.Program()
         startup_program.random_seed = SEED
@@ -459,6 +465,7 @@ def get_dygraph_output(self):
             res = self.dygraph_func()
             return res
 
+    @test_and_compare_with_new_ir(True)
     def get_static_output(self):
         startup_program = fluid.Program()
         startup_program.random_seed = SEED
@@ -485,6 +492,7 @@ def get_dygraph_output(self):
             res = self.dygraph_func()
             return res
 
+    @test_and_compare_with_new_ir(True)
     def get_static_output(self):
         startup_program = fluid.Program()
         startup_program.random_seed = SEED

test/dygraph_to_static/test_cinn.py

@@ -15,6 +15,7 @@
 import unittest
 
 import numpy as np
+from dygraph_to_static_util import test_and_compare_with_new_ir
 
 import paddle
 
@@ -78,6 +79,7 @@ def train(self, use_cinn):
 
         return res
 
+    @test_and_compare_with_new_ir(False)
     def test_cinn(self):
         dy_res = self.train(use_cinn=False)
         cinn_res = self.train(use_cinn=True)

test/dygraph_to_static/test_cinn_prim.py

@@ -15,7 +15,11 @@
 import unittest
 
 import numpy as np
-from dygraph_to_static_util import ast_only_test, dy2static_unittest
+from dygraph_to_static_util import (
+    ast_only_test,
+    dy2static_unittest,
+    test_and_compare_with_new_ir,
+)
 
 import paddle
 import paddle.nn.functional as F
@@ -169,6 +173,7 @@ def test_cinn_prim(self):
 
 
 class TestBackend(unittest.TestCase):
+    @test_and_compare_with_new_ir(False)
     def test_backend(self):
         x = paddle.randn([2, 4])
         out1 = self.forward(x, 'CINN')

test/dygraph_to_static/test_closure_analysis.py

@@ -16,6 +16,7 @@
 import os
 import unittest
 
+from dygraph_to_static_util import test_and_compare_with_new_ir
 from numpy import append
 
 import paddle
@@ -260,6 +261,7 @@ def init_dygraph_func(self):
 
 
 class TestPushPopTrans(unittest.TestCase):
+    @test_and_compare_with_new_ir(False)
     def test(self):
         def vlist_of_dict(x):
             ma = {'a': []}
@@ -271,6 +273,7 @@ def vlist_of_dict(x):
         print(paddle.jit.to_static(vlist_of_dict).code)
         print(paddle.jit.to_static(vlist_of_dict)(x))
 
+    @test_and_compare_with_new_ir(False)
     def test2(self):
         import numpy as np
 
@@ -284,6 +287,7 @@ def vlist_of_dict(x):
         print(paddle.jit.to_static(vlist_of_dict).code)
         print(paddle.jit.to_static(vlist_of_dict)(x))
 
+    @test_and_compare_with_new_ir(False)
     def test3(self):
         import numpy as np
 
@@ -297,6 +301,7 @@ def vlist_of_dict(x):
         print(paddle.jit.to_static(vlist_of_dict).code)
         print(paddle.jit.to_static(vlist_of_dict)(x))
 
+    @test_and_compare_with_new_ir(False)
     def test4(self):
         import numpy as np
 
@@ -310,6 +315,7 @@ def vlist_of_dict(x):
         print(paddle.jit.to_static(vlist_of_dict).code)
         print(paddle.jit.to_static(vlist_of_dict)(x))
 
+    @test_and_compare_with_new_ir(False)
     def test5(self):
         import numpy as np
 

test/dygraph_to_static/test_convert_call_generator.py

@@ -14,7 +14,7 @@
 
 import unittest
 
-from dygraph_to_static_util import ast_only_test
+from dygraph_to_static_util import ast_only_test, test_and_compare_with_new_ir
 
 import paddle
 from paddle.jit import to_static
@@ -35,6 +35,7 @@ def main_func():
 class TestConvertGenerator(unittest.TestCase):
     # fallback will ok.
     @ast_only_test
+    @test_and_compare_with_new_ir(False)
     def test_raise_error(self):
         translator_logger.verbosity_level = 1
         with self.assertLogs(

test/dygraph_to_static/test_cpu_cuda_to_tensor.py

@@ -19,6 +19,7 @@
     ast_only_test,
     dy2static_unittest,
     sot_only_test,
+    test_and_compare_with_new_ir,
 )
 
 import paddle
@@ -38,6 +39,7 @@ def func(x):
 
 
 class TestToTensor(unittest.TestCase):
+    @test_and_compare_with_new_ir(False)
     def test_to_tensor_with_variable_list(self):
         def func(x):
             ones = paddle.to_tensor(1)
@@ -57,6 +59,7 @@ def func(x):
 @dy2static_unittest
 class TestToTensor1(unittest.TestCase):
     @ast_only_test
+    @test_and_compare_with_new_ir(False)
     def test_to_tensor_with_variable_list(self):
         def func(x):
             ones = paddle.to_tensor([1])
@@ -75,6 +78,7 @@ def func(x):
         )
 
     @sot_only_test
+    @test_and_compare_with_new_ir(False)
     def test_to_tensor_with_variable_list_sot(self):
         def func(x):
             ones = paddle.to_tensor([1])
@@ -96,6 +100,7 @@ def func(x):
 @dy2static_unittest
 class TestToTensor2(unittest.TestCase):
     @ast_only_test
+    @test_and_compare_with_new_ir(False)
     def test_to_tensor_with_variable_list(self):
         def func(x):
             x = paddle.to_tensor([[1], [2], [3], [4]])
@@ -109,6 +114,7 @@ def func(x):
         )
 
     @sot_only_test
+    @test_and_compare_with_new_ir(False)
     def test_to_tensor_with_variable_list_sot(self):
         def func(x):
             x = paddle.to_tensor([[1], [2], [3], [4]])