【NewIR】enable dy2st new IR test

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
@@ -102,6 +103,7 @@ def get_static_output(self):
 
         return static_res[0]
 
+    @test_and_compare_with_new_ir(False)
     def test_transformed_static_result(self):
         for func in self.test_funcs:
             self.dygraph_func = func
@@ -259,6 +261,7 @@ def get_static_output(self):
         static_res = exe.run(main_program, fetch_list=static_out)
         return static_res[0]
 
+    @test_and_compare_with_new_ir(False)
     def test_transformed_static_result(self):
         dygraph_res = self.get_dygraph_output()
         static_res = self.get_static_output()
@@ -416,6 +419,7 @@ def get_static_output(self):
         static_res = exe.run(main_program, fetch_list=static_out)
         return static_res[0]
 
+    @test_and_compare_with_new_ir(False)
     def test_transformed_static_result(self):
         dygraph_res = self.get_dygraph_output()
         static_res = self.get_static_output()
@@ -542,6 +546,7 @@ def _get_dygraph_ast_node(self):
     def _get_static_ast_node(self):
         return self.root.body[0].body[2].body[1].value
 
+    @test_and_compare_with_new_ir(False)
     def test_dygraph_api(self):
         self.assertTrue(is_dygraph_api(self._get_dygraph_ast_node()) is True)
         self.assertTrue(is_dygraph_api(self._get_static_ast_node()) is False)

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
@@ -91,6 +95,7 @@ def check_prim(self, net, use_prim):
         self.assertTrue('softmax' not in fwd_ops)
 
     @ast_only_test
+    @test_and_compare_with_new_ir(False)
     def test_cinn_prim_forward(self):
         dy_res = self.train(use_prim=False)
         cinn_res = self.train(use_prim=True)
@@ -158,6 +163,7 @@ def check_prim(self, net, use_prim):
                 self.assertTrue("_grad" not in op)
 
     @ast_only_test
+    @test_and_compare_with_new_ir(False)
     def test_cinn_prim(self):
         dy_res = self.train(use_prim=False)
         cinn_res = self.train(use_prim=True)
@@ -169,6 +175,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_cinn_prim_gelu.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
@@ -107,6 +111,7 @@ def check_prim(self, net, use_prim):
         self.assertTrue('gelu' not in fwd_ops)
 
     @ast_only_test
+    @test_and_compare_with_new_ir(False)
     def test_cinn_prim(self):
         for shape in self.shapes:
             for dtype in self.dtypes:

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
@@ -194,6 +195,7 @@ def init_dygraph_func(self):
             {'func': set('i'), 'test_normal_argument': set('x')},
         ]
 
+    @test_and_compare_with_new_ir(False)
     def test_main(self):
         if self.judge_type == 'push_pop_vars':
             for push_pop_vars, func in zip(
@@ -260,6 +262,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 +274,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 +288,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 +302,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 +316,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]])