From f72d197ec5a5f1a3314a52bbcd4106e575137ac6 Mon Sep 17 00:00:00 2001 From: gongweibao Date: Tue, 23 Mar 2021 11:30:50 +0800 Subject: [PATCH] fix launch ps ut test=develop (#31771) fix launch ps ut test=develop --- .../tests/unittests/test_fleet_launch_ps.sh | 26 ++++++++++++------- 1 file changed, 16 insertions(+), 10 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_fleet_launch_ps.sh b/python/paddle/fluid/tests/unittests/test_fleet_launch_ps.sh index 67a8d7e575..0f28be614c 100644 --- a/python/paddle/fluid/tests/unittests/test_fleet_launch_ps.sh +++ b/python/paddle/fluid/tests/unittests/test_fleet_launch_ps.sh @@ -16,18 +16,24 @@ set -e -server_port_0=${PADDLE_DIST_UT_PORT} -server_port_1=$(( PADDLE_DIST_UT_PORT + 1 )) -worker_port_0=$(( PADDLE_DIST_UT_PORT + 2 )) -worker_port_1=$(( PADDLE_DIST_UT_PORT + 3 )) -heter_worker_port_0=$(( PADDLE_DIST_UT_PORT + 4 )) -heter_worker_port_1=$(( PADDLE_DIST_UT_PORT + 5 )) +server_port_00=${PADDLE_DIST_UT_PORT} +server_port_10=$(( PADDLE_DIST_UT_PORT + 1 )) +worker_port_00=$(( PADDLE_DIST_UT_PORT + 2 )) +worker_port_10=$(( PADDLE_DIST_UT_PORT + 3 )) + +server_port_01=$(( PADDLE_DIST_UT_PORT + 4 )) +server_port_11=$(( PADDLE_DIST_UT_PORT + 5 )) +worker_port_01=$(( PADDLE_DIST_UT_PORT + 6 )) +worker_port_11=$(( PADDLE_DIST_UT_PORT + 7 )) + +heter_worker_port_0=$(( PADDLE_DIST_UT_PORT + 8 )) +heter_worker_port_1=$(( PADDLE_DIST_UT_PORT + 9 )) function test_launch_ps(){ python -m paddle.distributed.fleet.launch \ - --servers="127.0.0.1:${server_port_0},127.0.0.1:${server_port_1}" \ - --workers="127.0.0.1:${worker_port_0},127.0.0.1:${worker_port_1}" \ + --servers="127.0.0.1:${server_port_00},127.0.0.1:${server_port_10}" \ + --workers="127.0.0.1:${worker_port_00},127.0.0.1:${worker_port_10}" \ fleet_ps_training.py 2> ut.elog if grep -q "server are killed" ut.elog; then echo "test pserver launch succeed" @@ -39,8 +45,8 @@ function test_launch_ps(){ function test_launch_ps_heter(){ python -m paddle.distributed.fleet.launch \ - --servers="127.0.0.1:${server_port_0},127.0.0.1:${server_port_1}" \ - --workers="127.0.0.1:${worker_port_0},127.0.0.1:${worker_port_1}" \ + --servers="127.0.0.1:${server_port_01},127.0.0.1:${server_port_11}" \ + --workers="127.0.0.1:${worker_port_01},127.0.0.1:${worker_port_11}" \ --heter_workers="127.0.0.1:${heter_worker_port_0},127.0.0.1:${heter_worker_port_1}" \ fleet_ps_training.py 2> ut.elog if grep -q "server are killed" ut.elog; then