We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 256fb47 commit 97ea150Copy full SHA for 97ea150
videosys/core/parallel_mgr.py
@@ -85,7 +85,7 @@ def initialize(
85
init_method=None,
86
seed: Optional[int] = None,
87
sp_size: Optional[int] = None,
88
- enable_cp: bool = True,
+ enable_cp: bool = False,
89
):
90
if not dist.is_initialized():
91
try:
@@ -107,7 +107,8 @@ def initialize(
107
dp_size = dist.get_world_size() // sp_size
108
109
# update cfg parallel
110
- if enable_cp and sp_size % 2 == 0:
+ # NOTE: enable cp parallel will be slower. disable it for now.
111
+ if False and enable_cp and sp_size % 2 == 0:
112
sp_size = sp_size // 2
113
cp_size = 2
114
else:
0 commit comments