Skip to content

Commit 5ea50a9

Browse files
authored
fix check_recommended_shm_size (#1021)
Co-authored-by: none <none>
1 parent 63ba5bd commit 5ea50a9

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

lightllm/utils/shm_size_check.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,18 @@
1313

1414

1515
def check_recommended_shm_size(args):
16-
shm_size, recommended_shm_size, is_shm_sufficient = _check_shm_size(args)
17-
if not is_shm_sufficient:
18-
_start_shm_size_warning_thread(shm_size, recommended_shm_size)
19-
else:
20-
logger.info(
21-
f"SHM check: Available={shm_size:.2f} GB,"
22-
f"Recommended={recommended_shm_size:.2f} GB."
23-
f"Sufficient: {is_shm_sufficient}",
24-
)
16+
try:
17+
shm_size, recommended_shm_size, is_shm_sufficient = _check_shm_size(args)
18+
if not is_shm_sufficient:
19+
_start_shm_size_warning_thread(shm_size, recommended_shm_size)
20+
else:
21+
logger.info(
22+
f"SHM check: Available={shm_size:.2f} GB,"
23+
f"Recommended={recommended_shm_size:.2f} GB."
24+
f"Sufficient: {is_shm_sufficient}",
25+
)
26+
except BaseException as e:
27+
logger.error(f"check_recommended_shm_size error: {str(e)}")
2528

2629

2730
def _check_shm_size(args):

0 commit comments

Comments
 (0)