未验证 提交 fc5a85b0 编写于 作者: T tianshuo78520a 提交者: GitHub

Add gpups test in PR-CI-GpuPS (#43592)

* test=gpups
上级 61591afe
......@@ -2024,6 +2024,26 @@ function get_failedUts_precise_map_file {
fi
}
function parallel_test_base_gpups() {
mkdir -p ${PADDLE_ROOT}/build
cd ${PADDLE_ROOT}/build
if [ ${WITH_TESTING:-ON} == "ON" ] ; then
cat <<EOF
========================================
Running unit GpuPS tests ...
========================================
EOF
ut_startTime_s=`date +%s`
ctest -L "RUN_TYPE=GPUPS" --timeout 120
ut_endTime_s=`date +%s`
echo "GPUPS testCase Time: $[ $ut_endTime_s - $ut_startTime_s ]s"
if [[ "$EXIT_CODE" != "0" ]]; then
exit 8;
fi
fi
}
function parallel_test_base_xpu() {
mkdir -p ${PADDLE_ROOT}/build
cd ${PADDLE_ROOT}/build
......@@ -2720,6 +2740,8 @@ function parallel_test() {
ut_total_startTime_s=`date +%s`
if [ "$WITH_CINN" == "ON" ];then
parallel_test_base_cinn
elif [ "$WITH_GPU" == "ON" ] && [ "$WITH_HETERPS" == "ON" ];then
parallel_test_base_gpups
elif [ "$WITH_GPU" == "ON" ] || [ "$WITH_ROCM" == "ON" ];then
parallel_test_base_gpu_test
elif [ "$WITH_XPU" == "ON" ];then
......
......@@ -1179,6 +1179,11 @@ if(WITH_XPU_BKCL)
SRCS "test_collective_allreduce_api.py")
endif()
if(WITH_HETERPS)
set_tests_properties(test_dist_fleet_ps11 PROPERTIES LABELS "RUN_TYPE=GPUPS")
set_tests_properties(test_dist_fleet_ps12 PROPERTIES LABELS "RUN_TYPE=GPUPS")
endif()
if(WIN32)
cc_test(
cc_imp_py_test
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册