Skip to content

Commit

Permalink
disable pir in all ps mode for Paddle 3.0 (#1004)
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

* [windows] bug fixs trainer divide-zero-bug

* add lightgcn and transformer_conv pglbox

* add centos PDC images for pglbox

* fix pscpu benchmark

* fix pscpu benchmark install paddle

* add pglbox big data benchmark

* disable pir in all ps mode for Paddle 3.0
  • Loading branch information
fightfat authored Sep 12, 2024
1 parent 9fd6283 commit a4725d9
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 0 deletions.
1 change: 1 addition & 0 deletions test_tipc/configs/dselect_k/to_static.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@

import paddle
import os
os.environ['FLAGS_enable_pir_api'] = '0'
import paddle.nn as nn
import time
import logging
Expand Down
1 change: 1 addition & 0 deletions test_tipc/configs/esmm/to_static.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@

import paddle
import os
os.environ['FLAGS_enable_pir_api'] = '0'
import paddle.nn as nn
import time
import logging
Expand Down
1 change: 1 addition & 0 deletions test_tipc/configs/mmoe/to_static.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@

import paddle
import os
os.environ['FLAGS_enable_pir_api'] = '0'
import paddle.nn as nn
import time
import logging
Expand Down
1 change: 1 addition & 0 deletions test_tipc/configs/ple/to_static.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@

import paddle
import os
os.environ['FLAGS_enable_pir_api'] = '0'
import paddle.nn as nn
import time
import logging
Expand Down
1 change: 1 addition & 0 deletions tools/static_gpubox_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import paddle.distributed.fleet.base.role_maker as role_maker
import paddle
import os
os.environ['FLAGS_enable_pir_api'] = '0'
import warnings
import logging

Expand Down

0 comments on commit a4725d9

Please sign in to comment.