[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:
SYLARandlishunyang12 authored and GitHub committed 2026-08-01 15:06:05 +00:00
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."""