Skip to content

Commit f570c06

Browse files
authored
set default local_rank_size to 1 (#515)
1 parent b8c25c4 commit f570c06

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

ucm/store/pcstore/pcstore_connector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def __init__(self, config: Dict):
5353
param.transferIoDirect = config.get("use_direct", False)
5454
param.transferStreamNumber = config.get("stream_number", 8)
5555
param.transferBufferNumber = config.get("buffer_number", 4096)
56-
param.transferLocalRankSize = config.get("local_rank_size", 8)
56+
param.transferLocalRankSize = config.get("local_rank_size", 1)
5757
param.transferScatterGatherEnable = config.get("use_scatter_gatter", False)
5858
ret = self.store.Setup(param)
5959
if ret != 0:

ucm/store/pcstore/pcstore_connector_v1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def __init__(self, config: Dict):
5454
param.transferIoDirect = config.get("use_direct", False)
5555
param.transferStreamNumber = config.get("stream_number", 8)
5656
param.transferBufferNumber = config.get("buffer_number", 4096)
57-
param.transferLocalRankSize = config.get("local_rank_size", 8)
57+
param.transferLocalRankSize = config.get("local_rank_size", 1)
5858
param.transferScatterGatherEnable = config.get("use_scatter_gatter", False)
5959
ret = self.store.Setup(param)
6060
if ret != 0:

0 commit comments

Comments
 (0)