Skip to content

Commit 44ef7b6

Browse files
authored
Set MC_MAX_MR_SIZE to avoid register hang (#7162)
1 parent 242ed02 commit 44ef7b6

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

fastdeploy/cache_manager/transfer_factory/mooncake_store/mooncake_store.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,9 @@ def __init__(self, tp_rank=None):
106106
host_ip = get_host_ip()
107107
os.environ["MC_TCP_BIND_ADDRESS"] = host_ip
108108
logger.info(f"Set MC_TCP_BIND_ADDRESS to {host_ip}")
109+
if os.environ.get("MC_MAX_MR_SIZE") is None:
110+
os.environ["MC_MAX_MR_SIZE"] = "4294967296" # 4GB
111+
logger.info("MC_MAX_MR_SIZE is not set, default to 4GB.")
109112

110113
try:
111114
from mooncake.store import MooncakeDistributedStore

0 commit comments

Comments
 (0)