From 58faeb90f569478037064a330e80d44a36619021 Mon Sep 17 00:00:00 2001 From: qyh111 Date: Sat, 13 Dec 2025 00:56:18 -0800 Subject: [PATCH] set default local_rank_size to 1 --- ucm/store/pcstore/pcstore_connector.py | 2 +- ucm/store/pcstore/pcstore_connector_v1.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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: