diff --git a/python/paddle/fluid/tests/unittests/test_dist_base.py b/python/paddle/fluid/tests/unittests/test_dist_base.py index 618c910b71b2cc6f22e55f7ba1772e71de42e71e..78ed29be569df2fa4290d16c97ee12009733651d 100644 --- a/python/paddle/fluid/tests/unittests/test_dist_base.py +++ b/python/paddle/fluid/tests/unittests/test_dist_base.py @@ -123,6 +123,7 @@ def runtime_main(test_class): ) else fluid.CPUPlace() model.run_trainer(p, endpoints, trainer_id, trainers, is_dist) + import paddle.fluid.compat as cpt