Skip to content

Commit

Permalink
set FLAGS_dynamic_static_unified_comm=False when psgpu (PaddlePaddle#955
Browse files Browse the repository at this point in the history
)

* fix wide_deep gpups error when run in py>3.7

* set FLAGS_dynamic_static_unified_comm=False when psgpu
  • Loading branch information
fightfat authored Oct 23, 2023
1 parent eb869a1 commit 22d28cf
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions test_tipc/test_train_inference_python.sh
Original file line number Diff line number Diff line change
Expand Up @@ -225,6 +225,7 @@ if [ ${MODE} = "benchmark_train" ]; then
echo "Create log floder for store running log"
fi
if [ ${run_mode_value} = "PSGPU" ]; then
export FLAGS_dynamic_static_unified_comm=False #PSGPU不支持新通信库
export FLAGS_LAUNCH_BARRIER=0
export PADDLE_TRAINER_ID=0
export PADDLE_PSERVER_NUMS=1
Expand Down

0 comments on commit 22d28cf

Please sign in to comment.