Don't fail server startup when psutil can't parse /proc/meminfo (GB300 ShadowCallStack) (#31098)
This commit is contained in:
@@ -222,9 +222,15 @@ def get_zmq_socket_on_host(
|
|||||||
|
|
||||||
|
|
||||||
def config_socket(socket, socket_type: zmq.SocketType):
|
def config_socket(socket, socket_type: zmq.SocketType):
|
||||||
mem = psutil.virtual_memory()
|
try:
|
||||||
total_mem = mem.total / 1024**3
|
mem = psutil.virtual_memory()
|
||||||
available_mem = mem.available / 1024**3
|
total_mem = mem.total / 1024**3
|
||||||
|
available_mem = mem.available / 1024**3
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(
|
||||||
|
"psutil.virtual_memory() failed (%s); using default ZMQ buffer size", e
|
||||||
|
)
|
||||||
|
total_mem = available_mem = 0
|
||||||
if total_mem > 32 and available_mem > 16:
|
if total_mem > 32 and available_mem > 16:
|
||||||
buf_size = int(0.5 * 1024**3)
|
buf_size = int(0.5 * 1024**3)
|
||||||
else:
|
else:
|
||||||
|
|||||||
Reference in New Issue
Block a user