diff --git a/test/dygraph_to_static/test_ast_util.py b/test/dygraph_to_static/test_ast_util.py index a6421e4cc60ba..3e98995abdd9b 100644 --- a/test/dygraph_to_static/test_ast_util.py +++ b/test/dygraph_to_static/test_ast_util.py @@ -20,7 +20,7 @@ from dygraph_to_static_utils_new import ( Dy2StTestBase, test_ast_only, - test_legacy_and_pir, + test_legacy_and_pir_api, ) from ifelse_simple_func import ( dyfunc_with_if_else, @@ -48,6 +48,7 @@ def _ast2func(self, func): return transformed_func @test_ast_only + @test_legacy_and_pir_api def test_ast2func(self): def func(x, y): return x + y @@ -56,19 +57,19 @@ def func(x, y): self.assertEqual(func(x, y), self._ast2func(func)(x, y)) @test_ast_only + @test_legacy_and_pir_api def test_ast2func_dygraph(self): paddle.disable_static() funcs = [dyfunc_with_if_else, dyfunc_with_if_else2, nested_if_else] x_data = np.random.random([10, 16]).astype('float32') for func in funcs: - with base.dygraph.guard(): - x_v = base.dygraph.to_variable(x_data) - true_ret = func(x_v).numpy() - test_ret = self._ast2func(func)(x_v).numpy() - self.assertTrue((true_ret == test_ret).all()) + x_v = base.dygraph.to_variable(x_data) + true_ret = func(x_v).numpy() + test_ret = self._ast2func(func)(x_v).numpy() + self.assertTrue((true_ret == test_ret).all()) - @test_legacy_and_pir @test_ast_only + @test_legacy_and_pir_api def test_ast2func_static(self): paddle.enable_static() @@ -83,11 +84,12 @@ def func(x): x_v = paddle.assign(x_data) true_ret = func(x_v) test_ret = self._ast2func(func)(x_v) - exe = base.Executor(base.CPUPlace()) + exe = base.Executor(paddle.CPUPlace()) ret = exe.run(main_program, fetch_list=[true_ret, test_ret]) self.assertTrue((ret[0] == ret[1]).all()) @test_ast_only + @test_legacy_and_pir_api def test_ast2func_error(self): with self.assertRaises(Exception) as e: self.assertRaises(TypeError, ast_to_func("x = a + b", 'foo')) diff --git a/test/dygraph_to_static/test_cast.py b/test/dygraph_to_static/test_cast.py index 48564e2776395..199d8bc1c8230 100644 --- a/test/dygraph_to_static/test_cast.py +++ b/test/dygraph_to_static/test_cast.py @@ -19,29 +19,30 @@ Dy2StTestBase, test_ast_only, test_legacy_and_pir, + test_legacy_and_pir_exe_and_pir_api, ) -from paddle import base -from paddle.jit.api import to_static +import paddle +from paddle.base.dygraph import to_variable SEED = 2020 np.random.seed(SEED) def test_bool_cast(x): - x = base.dygraph.to_variable(x) + x = to_variable(x) x = bool(x) return x def test_int_cast(x): - x = base.dygraph.to_variable(x) + x = to_variable(x) x = int(x) return x def test_float_cast(x): - x = base.dygraph.to_variable(x) + x = to_variable(x) x = float(x) return x @@ -52,7 +53,7 @@ def test_not_var_cast(x): def test_mix_cast(x): - x = base.dygraph.to_variable(x) + x = to_variable(x) x = int(x) x = float(x) x = bool(x) @@ -63,12 +64,11 @@ def test_mix_cast(x): class TestCastBase(Dy2StTestBase): def setUp(self): self.place = ( - base.CUDAPlace(0) - if base.is_compiled_with_cuda() - else base.CPUPlace() + paddle.CUDAPlace(0) + if paddle.is_compiled_with_cuda() + else paddle.CPUPlace() ) self.prepare() - self.set_func() def prepare(self): self.input_shape = (16, 32) @@ -81,16 +81,16 @@ def prepare(self): self.cast_dtype = 'bool' def set_func(self): - self.func = to_static(full_graph=True)(test_bool_cast) + self.func = paddle.jit.to_static(full_graph=True)(test_bool_cast) def do_test(self): - with base.dygraph.guard(): - res = self.func(self.input) - return res + res = self.func(self.input) + return res @test_ast_only # TODO: add new sot only test. - @test_legacy_and_pir + @test_legacy_and_pir_exe_and_pir_api def test_cast_result(self): + self.set_func() res = self.do_test().numpy() self.assertTrue( res.dtype == self.cast_dtype, @@ -119,7 +119,7 @@ def prepare(self): self.cast_dtype = 'int32' def set_func(self): - self.func = to_static(full_graph=True)(test_int_cast) + self.func = paddle.jit.to_static(full_graph=True)(test_int_cast) class TestFloatCast(TestCastBase): @@ -134,7 +134,7 @@ def prepare(self): self.cast_dtype = 'float32' def set_func(self): - self.func = to_static(full_graph=True)(test_float_cast) + self.func = paddle.jit.to_static(full_graph=True)(test_float_cast) class TestMixCast(TestCastBase): @@ -152,11 +152,12 @@ def prepare(self): self.cast_dtype = 'float32' def set_func(self): - self.func = to_static(full_graph=True)(test_mix_cast) + self.func = paddle.jit.to_static(full_graph=True)(test_mix_cast) @test_ast_only # TODO: add new symbolic only test. - @test_legacy_and_pir + @test_legacy_and_pir_exe_and_pir_api def test_cast_result(self): + self.set_func() res = self.do_test().numpy() self.assertTrue( res.dtype == self.cast_dtype, @@ -184,11 +185,12 @@ def prepare(self): self.cast_dtype = 'int' def set_func(self): - self.func = to_static(full_graph=True)(test_not_var_cast) + self.func = paddle.jit.to_static(full_graph=True)(test_not_var_cast) @test_ast_only @test_legacy_and_pir def test_cast_result(self): + self.set_func() # breakpoint() # print("run once!!!") res = self.do_test() diff --git a/test/dygraph_to_static/test_cinn_prim.py b/test/dygraph_to_static/test_cinn_prim.py index 95df5d498c6fb..ce9e0afcc52f0 100644 --- a/test/dygraph_to_static/test_cinn_prim.py +++ b/test/dygraph_to_static/test_cinn_prim.py @@ -18,7 +18,7 @@ from dygraph_to_static_utils_new import ( Dy2StTestBase, test_ast_only, - test_legacy_and_pir, + test_legacy_and_pir_exe_and_pir_api, ) import paddle @@ -171,7 +171,7 @@ def test_cinn_prim(self): class TestBackend(Dy2StTestBase): - @test_legacy_and_pir + @test_legacy_and_pir_exe_and_pir_api def test_backend(self): x = paddle.randn([2, 4]) out1 = self.forward(x, 'CINN') diff --git a/test/dygraph_to_static/test_tensor_hook.py b/test/dygraph_to_static/test_tensor_hook.py index 3a4174f0febcf..5fad08c189e44 100644 --- a/test/dygraph_to_static/test_tensor_hook.py +++ b/test/dygraph_to_static/test_tensor_hook.py @@ -15,7 +15,10 @@ import unittest import numpy as np -from dygraph_to_static_utils_new import Dy2StTestBase +from dygraph_to_static_utils_new import ( + Dy2StTestBase, + test_legacy_and_pir_exe_and_pir_api, +) import paddle from paddle import nn @@ -94,6 +97,7 @@ def h(g): loss.backward() np.testing.assert_allclose(x.grad.numpy(), x_jit.grad.numpy()) + @test_legacy_and_pir_exe_and_pir_api def test_hook_in_init_for_layer(self): def hook(grad): return grad * 2 diff --git a/test/dygraph_to_static/test_variable_trans_func.py b/test/dygraph_to_static/test_variable_trans_func.py index cbfe76ec2824c..2880692c32b59 100644 --- a/test/dygraph_to_static/test_variable_trans_func.py +++ b/test/dygraph_to_static/test_variable_trans_func.py @@ -14,13 +14,14 @@ import unittest -from dygraph_to_static_utils_new import Dy2StTestBase +from dygraph_to_static_utils_new import Dy2StTestBase, test_legacy_and_pir_api from paddle.jit.dy2static.utils import ast_to_source_code from paddle.jit.dy2static.variable_trans_func import create_fill_constant_node class TestVariableTransFunc(Dy2StTestBase): + @test_legacy_and_pir_api def test_create_fill_constant_node(self): node = create_fill_constant_node("a", 1.0) source = "a = paddle.full(shape=[1], dtype='float64', fill_value=1.0, name='a')"