fix pscpu benchmark (PaddlePaddle#977)

* fix wide_deep gpups error when run in py>3.7 * set FLAGS_dynamic_static_unified_comm=False when psgpu * [windows] bug fixs trainer divide-zero-bug * add lightgcn and transformer_conv pglbox * add centos PDC images for pglbox * fix pscpu benchmark
yusihuu · Jan 30, 2024 · 2967014 · 2967014
1 parent 849925f
commit 2967014
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/test_tipc/configs/dnn/train_infer_python.txt b/test_tipc/configs/dnn/train_infer_python.txt
@@ -1,6 +1,6 @@
 ===========================train_params===========================
 model_name:dnn
-python:python3.7
+python:python
 gpu_list:0
 runner.use_gpu:True|False
 auto_cast:False

diff --git a/test_tipc/configs/wide_deep/train_infer_python.txt b/test_tipc/configs/wide_deep/train_infer_python.txt
@@ -1,6 +1,6 @@
 ===========================train_params===========================
 model_name:wide_deep
-python:python3.7
+python:python
 gpu_list:0
 runner.use_gpu:True|False
 auto_cast:False

diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
@@ -283,6 +283,9 @@ if [ ${MODE} = "benchmark_train" ]; then
         export PADDLE_GLOO_FS_UGI=$gloo_fs_ugi
         export PADDLE_GLOO_FS_PATH=$gloo_fs_path
 
+        cd ../
+        ${python} -m pip install paddlepaddle*.whl -U
+        cd -
         CONFIG=$cpu_config_value
         SC="tools/static_ps_trainer.py -m "
         BATCH="-o runner.train_batch_size="$train_batch_value