mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
[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
This commit is contained in:
@@ -111,6 +111,9 @@ class MooncakeStore(KVCacheStorage):
|
||||
host_ip = get_host_ip()
|
||||
os.environ["MC_TCP_BIND_ADDRESS"] = host_ip
|
||||
logger.info(f"Set MC_TCP_BIND_ADDRESS to {host_ip}")
|
||||
if os.environ.get("MC_MAX_MR_SIZE") is None:
|
||||
os.environ["MC_MAX_MR_SIZE"] = "4294967296" # 4GB
|
||||
logger.info("MC_MAX_MR_SIZE is not set, default to 4GB.")
|
||||
|
||||
try:
|
||||
from mooncake.store import MooncakeDistributedStore
|
||||
|
||||
Reference in New Issue
Block a user