Skip to content

Commit 97ea150

Browse files
authored
disable cp parallel (#205)
1 parent 256fb47 commit 97ea150

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

videosys/core/parallel_mgr.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def initialize(
8585
init_method=None,
8686
seed: Optional[int] = None,
8787
sp_size: Optional[int] = None,
88-
enable_cp: bool = True,
88+
enable_cp: bool = False,
8989
):
9090
if not dist.is_initialized():
9191
try:
@@ -107,7 +107,8 @@ def initialize(
107107
dp_size = dist.get_world_size() // sp_size
108108

109109
# update cfg parallel
110-
if enable_cp and sp_size % 2 == 0:
110+
# NOTE: enable cp parallel will be slower. disable it for now.
111+
if False and enable_cp and sp_size % 2 == 0:
111112
sp_size = sp_size // 2
112113
cp_size = 2
113114
else:

0 commit comments

Comments
 (0)