Skip to content

Commit 1cc0cf2

Browse files
authored
[BugFix] Set MC_MAX_MR_SIZE to avoid register hang in default (#7161)
* Set MC_MAX_MR_SIZE to avoid register hang * Set MC_MAX_MR_SIZE to avoid register hang
1 parent 2632e6c commit 1cc0cf2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

fastdeploy/cache_manager/transfer_factory/mooncake_store/mooncake_store.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,9 @@ def __init__(self, tp_rank=None):
111111
host_ip = get_host_ip()
112112
os.environ["MC_TCP_BIND_ADDRESS"] = host_ip
113113
logger.info(f"Set MC_TCP_BIND_ADDRESS to {host_ip}")
114+
if os.environ.get("MC_MAX_MR_SIZE") is None:
115+
os.environ["MC_MAX_MR_SIZE"] = "4294967296" # 4GB
116+
logger.info("MC_MAX_MR_SIZE is not set, default to 4GB.")
114117

115118
try:
116119
from mooncake.store import MooncakeDistributedStore

0 commit comments

Comments
 (0)