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):
|
||||
mem = psutil.virtual_memory()
|
||||
total_mem = mem.total / 1024**3
|
||||
available_mem = mem.available / 1024**3
|
||||
try:
|
||||
mem = psutil.virtual_memory()
|
||||
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:
|
||||
buf_size = int(0.5 * 1024**3)
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user