diff --git a/python/paddle/fluid/tests/custom_op/test_custom_tanh_double_grad.py b/python/paddle/fluid/tests/custom_op/test_custom_tanh_double_grad.py index 1127108c361ad884111834e01b0f282f7a9db73f..5664c00d74f8952d6fb76ad4c7a4c72bb1b1d76e 100644 --- a/python/paddle/fluid/tests/custom_op/test_custom_tanh_double_grad.py +++ b/python/paddle/fluid/tests/custom_op/test_custom_tanh_double_grad.py @@ -21,7 +21,8 @@ import paddle.static as static from paddle.utils.cpp_extension import load, get_build_directory from paddle.utils.cpp_extension.extension_utils import run_cmd from utils import paddle_includes, extra_cc_args, extra_nvcc_args -from paddle.fluid.framework import _test_eager_guard +from paddle.fluid.framework import _test_eager_guard, _enable_legacy_dygraph +_enable_legacy_dygraph() # Because Windows don't use docker, the shared lib already exists in the # cache dir, it will not be compiled again unless the shared lib is removed. diff --git a/python/paddle/fluid/tests/unittests/check_nan_inf_base_dygraph.py b/python/paddle/fluid/tests/unittests/check_nan_inf_base_dygraph.py index 08bab306df1b111b0b27814cd27a6f0243adfdf4..f4217d11f2d9be454e209b415708487a2211591f 100644 --- a/python/paddle/fluid/tests/unittests/check_nan_inf_base_dygraph.py +++ b/python/paddle/fluid/tests/unittests/check_nan_inf_base_dygraph.py @@ -25,6 +25,8 @@ os.environ[str("GLOG_vmodule")] = str("nan_inf_utils_detail=10") import paddle import paddle.nn as nn +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() np.random.seed(0) diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet_v2.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet_v2.py index 872d419ff8928df767ac44b502ef354f3c53b43c..ab836b088b09fbeafa11123b9b446b3df0fd9bc5 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet_v2.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet_v2.py @@ -20,6 +20,9 @@ import unittest from simnet_dygraph_model_v2 import BOW, HingeLoss +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() + SEED = 102 random.seed(SEED) diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_spec_names.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_spec_names.py index 361fcbf9c73f5b056389e45c58b83e5e6308d000..bafc4707c4ad996885c0321a22bb44943018577b 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_spec_names.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_spec_names.py @@ -16,6 +16,8 @@ import paddle from paddle.nn import Layer import numpy as np import unittest +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() class Net(Layer): diff --git a/python/paddle/fluid/tests/unittests/test_bfgs.py b/python/paddle/fluid/tests/unittests/test_bfgs.py index c89f7205f0818aab920a5febd51dd82a1f34cd41..4bf6de3eee51099b370cb3a0baa108fc189f7bb7 100644 --- a/python/paddle/fluid/tests/unittests/test_bfgs.py +++ b/python/paddle/fluid/tests/unittests/test_bfgs.py @@ -21,6 +21,9 @@ import paddle.nn.functional as F from paddle.incubate.optimizer.functional.bfgs import minimize_bfgs +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() + np.random.seed(123) diff --git a/python/paddle/fluid/tests/unittests/test_diff_op.py b/python/paddle/fluid/tests/unittests/test_diff_op.py index 4a96827bd7c3c56320a58261abe1824786164d10..99a46bfd9584d9bfb5e5cb42d24482cceced4eb5 100644 --- a/python/paddle/fluid/tests/unittests/test_diff_op.py +++ b/python/paddle/fluid/tests/unittests/test_diff_op.py @@ -19,6 +19,8 @@ import paddle import paddle.fluid as fluid import paddle.fluid.layers as layers import paddle.fluid.core as core +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() class TestDiffOp(unittest.TestCase): diff --git a/python/paddle/fluid/tests/unittests/test_dropout_op.py b/python/paddle/fluid/tests/unittests/test_dropout_op.py index 3aca428ac77af4c9881a3549c201dcb73ba41253..20abeaec7268c7331b2ef3149ab5bbcf05d7c7e8 100644 --- a/python/paddle/fluid/tests/unittests/test_dropout_op.py +++ b/python/paddle/fluid/tests/unittests/test_dropout_op.py @@ -22,7 +22,8 @@ import paddle import paddle.static as static import paddle.fluid as fluid from paddle.fluid import Program, program_guard -from paddle.fluid.framework import _test_eager_guard +from paddle.fluid.framework import _test_eager_guard, _enable_legacy_dygraph +_enable_legacy_dygraph() import os from paddle import _C_ops @@ -951,6 +952,7 @@ class TestDropoutBackward(unittest.TestCase): return mask.astype("float32") def test_backward_downscale_in_infer(self): + _enable_legacy_dygraph() for place in self.places: with fluid.dygraph.guard(place): diff --git a/python/paddle/fluid/tests/unittests/test_eigh_op.py b/python/paddle/fluid/tests/unittests/test_eigh_op.py index 2abbcc98a6b7e33b0e3c99e971978dde6a19fc13..9c9cd883313a2425b11b282ee9039b952a0852f7 100644 --- a/python/paddle/fluid/tests/unittests/test_eigh_op.py +++ b/python/paddle/fluid/tests/unittests/test_eigh_op.py @@ -19,6 +19,8 @@ import numpy as np import paddle from op_test import OpTest from gradient_checker import grad_check +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() def valid_eigh_result(A, eigh_value, eigh_vector, uplo): diff --git a/python/paddle/fluid/tests/unittests/test_faster_tokenizer_op.py b/python/paddle/fluid/tests/unittests/test_faster_tokenizer_op.py index 190345958e0e5e1671b77a70065efe54fcd236f9..87c4656cfa809c3d53463815ca0bd6d8f13c2e52 100755 --- a/python/paddle/fluid/tests/unittests/test_faster_tokenizer_op.py +++ b/python/paddle/fluid/tests/unittests/test_faster_tokenizer_op.py @@ -22,7 +22,8 @@ import numpy as np import paddle import paddle.nn as nn from paddle.dataset.common import DATA_HOME -from paddle.fluid.framework import core, _non_static_mode +from paddle.fluid.framework import core, _non_static_mode, _enable_legacy_dygraph +_enable_legacy_dygraph() from paddle.fluid.layer_helper import LayerHelper from paddle import _C_ops diff --git a/python/paddle/fluid/tests/unittests/test_label_smooth_functional.py b/python/paddle/fluid/tests/unittests/test_label_smooth_functional.py index 54f5e64fda4b6b5b1dc5bc04885fe21992d25f65..83c8ced79b1e8a86ea28cb82349a389547115298 100644 --- a/python/paddle/fluid/tests/unittests/test_label_smooth_functional.py +++ b/python/paddle/fluid/tests/unittests/test_label_smooth_functional.py @@ -19,6 +19,8 @@ import paddle.fluid.dygraph as dg import paddle.nn.functional as F import paddle.fluid.initializer as I import unittest +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() class LabelSmoothTestCase(unittest.TestCase): diff --git a/python/paddle/fluid/tests/unittests/test_lbfgs.py b/python/paddle/fluid/tests/unittests/test_lbfgs.py index bb3818747601fd7a097def4ffafdadabfb671c48..2cad4822b28b14ecc6536829528da156a2ab5f29 100644 --- a/python/paddle/fluid/tests/unittests/test_lbfgs.py +++ b/python/paddle/fluid/tests/unittests/test_lbfgs.py @@ -21,6 +21,9 @@ import paddle.nn.functional as F from paddle.incubate.optimizer.functional.lbfgs import minimize_lbfgs +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() + np.random.seed(123) diff --git a/python/paddle/fluid/tests/unittests/test_nan_inf.py b/python/paddle/fluid/tests/unittests/test_nan_inf.py index 84559048a2b8a919f96ca3cbbd1ea7bec3c56ffb..9b11f6711afc19aaeee7533e978372ae68c781fc 100644 --- a/python/paddle/fluid/tests/unittests/test_nan_inf.py +++ b/python/paddle/fluid/tests/unittests/test_nan_inf.py @@ -20,6 +20,8 @@ import os import sys import subprocess import paddle +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() paddle.enable_static() diff --git a/python/paddle/fluid/tests/unittests/test_nn_functional_embedding_dygraph.py b/python/paddle/fluid/tests/unittests/test_nn_functional_embedding_dygraph.py index acff7daadeb33114646398a0a78250535e73f3aa..e50424126e53e70dc76a2fb929bdd4d228b276d2 100644 --- a/python/paddle/fluid/tests/unittests/test_nn_functional_embedding_dygraph.py +++ b/python/paddle/fluid/tests/unittests/test_nn_functional_embedding_dygraph.py @@ -19,6 +19,8 @@ import unittest import paddle import paddle.nn as nn import numpy as np +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() paddle.disable_static() diff --git a/python/paddle/fluid/tests/unittests/test_tensor_scalar_type_promotion_dynamic.py b/python/paddle/fluid/tests/unittests/test_tensor_scalar_type_promotion_dynamic.py index 5f2dfbdd99e1611c61883b9a34cccc5ac0ec8b71..c5e3cb29e0c208bd1bb89732e1371fcf55c34fb4 100644 --- a/python/paddle/fluid/tests/unittests/test_tensor_scalar_type_promotion_dynamic.py +++ b/python/paddle/fluid/tests/unittests/test_tensor_scalar_type_promotion_dynamic.py @@ -18,6 +18,8 @@ import unittest import numpy as np import paddle +from paddle.fluid.framework import _enable_legacy_dygraph +_enable_legacy_dygraph() # Support types are ref from `paddle.tensor.math` # - Related paddle dtypes: