diff --git a/ucm/store/pcstore/pcstore_connector.py b/ucm/store/pcstore/pcstore_connector.py index 00dffa85..7217620e 100644 --- a/ucm/store/pcstore/pcstore_connector.py +++ b/ucm/store/pcstore/pcstore_connector.py @@ -53,7 +53,7 @@ def __init__(self, config: Dict): param.transferIoDirect = config.get("use_direct", False) param.transferStreamNumber = config.get("stream_number", 8) param.transferBufferNumber = config.get("buffer_number", 4096) - param.transferLocalRankSize = config.get("local_rank_size", 8) + param.transferLocalRankSize = config.get("local_rank_size", 1) param.transferScatterGatherEnable = config.get("use_scatter_gatter", False) ret = self.store.Setup(param) if ret != 0: diff --git a/ucm/store/pcstore/pcstore_connector_v1.py b/ucm/store/pcstore/pcstore_connector_v1.py index 3cf964fe..39f0589f 100644 --- a/ucm/store/pcstore/pcstore_connector_v1.py +++ b/ucm/store/pcstore/pcstore_connector_v1.py @@ -54,7 +54,7 @@ def __init__(self, config: Dict): param.transferIoDirect = config.get("use_direct", False) param.transferStreamNumber = config.get("stream_number", 8) param.transferBufferNumber = config.get("buffer_number", 4096) - param.transferLocalRankSize = config.get("local_rank_size", 8) + param.transferLocalRankSize = config.get("local_rank_size", 1) param.transferScatterGatherEnable = config.get("use_scatter_gatter", False) ret = self.store.Setup(param) if ret != 0: