[Bugfix] Restore BAGEL CFG KV transfers (#5620)
Signed-off-by: lishunyang12 <lishunyang03@gmail.com> Co-authored-by: lishunyang12 <lishunyang03@gmail.com>
This commit is contained in:
1 parent
185e3d6df6
commit
6f98c807d2
8 files changed
+160
-8
No files matched your search
@@ -403,6 +403,7 @@ class OmniKVTransferManager:
|
||||
self._prefetch_executor: ThreadPoolExecutor | None = (
|
||||
ThreadPoolExecutor(max_workers=1, thread_name_prefix="kv-prefetch") if async_prefetch else None
|
||||
)
|
||||
|
||||
self._prefetch_futures: dict[str, Any] = {}
|
||||
self._bg_copy_stream: current_omni_platform.Stream | None = None
|
||||
|
||||
@@ -415,6 +416,11 @@ class OmniKVTransferManager:
|
||||
except Exception as e:
|
||||
logger.warning("Failed to eagerly initialize sender connector: %s", e)
|
||||
|
||||
@property
|
||||
def tp_topology(self) -> KVTPTopology:
|
||||
"""Return the effective topology after config and runtime detection."""
|
||||
return self._tp_topo
|
||||
|
||||
@property
|
||||
def topo_config(self) -> _TransferTopoConfig:
|
||||
"""Lazily build the parallel topology snapshot on first access."""
|
||||
|
||||
Reference in new issue
Block a user