未验证 提交 0536b526 编写于 作者: G gongweibao 提交者: GitHub

fix test_launch_ps test=develop (#23333)

上级 99d30bfc
......@@ -302,7 +302,7 @@ if(WITH_DISTRIBUTE)
# NOTE. test_launch only work in gpu collective mode
bash_test_modules(test_launch MODULES test_launch.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
endif()
bash_test_modules(test_launch_ps MODULES test_launch_ps.sh)
bash_test_modules(test_launch_ps MODULES test_launch_ps.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
set(dist_ut_port 1000)
foreach(TEST_OP ${DIST_TEST_OPS})
......
#!/bin/bash
set -e
# use default values
python -m paddle.distributed.launch_ps fleet_ps_training.py 2> ut.elog
launch_py=${PADDLE_BINARY_DIR}/python/paddle/distributed/launch_ps.py
python ${launch_py} fleet_ps_training.py 2> ut.elog
if grep -q "server are killed" ut.elog; then
echo "succeed"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册