From 41902dda8abfa1c6b986c998b4dbfa2786ce0779 Mon Sep 17 00:00:00 2001
From: wangzhen38 <41941775+wangzhen38@users.noreply.github.com>
Date: Wed, 15 Feb 2023 15:37:47 +0800
Subject: [PATCH] [gpups update] add gpups ci log print (#50522)

---
 python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py     | 5 +++--
 .../fleet/parameter_server/distribute_transpiler/__init__.py | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py b/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py
index 7c7fedd812c..4562561fce4 100755
--- a/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py
+++ b/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py
@@ -27,8 +27,9 @@ class GpuBoxTest(unittest.TestCase):
         exitcode = os.system('bash gpubox_run.sh')
         if os.path.exists('./train_data'):
             os.system('rm -rf train_data')
-        if os.path.exists('./log'):
-            os.system('rm -rf log')
+        if exitcode:
+            os.system('cat ./log/worker.0.log')
+
         assert exitcode == 0
 
 
diff --git a/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py b/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
index 99e5dd2219e..fa1da4bcf0b 100644
--- a/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
+++ b/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
@@ -57,8 +57,9 @@ from paddle.incubate.fleet.parameter_server.distribute_transpiler.distributed_st
     StrategyFactory,
 )
 
-from paddle.fluid.transpiler.details.checkport import wait_server_ready
-
+from paddle.distributed.fleet.base.private_helper_function import (
+    wait_server_ready,
+)
 from paddle.fluid.incubate.fleet.parameter_server.mode import PSMode
 from paddle.fluid.incubate.fleet.base.fleet_base import DistributedOptimizer
 
-- 
GitLab