Reverting unnecessary changes

Signed-off-by: Alexey Kudinkin <ak@anyscale.com>
ray-project · Sep 24, 2024 · 695ee1c · 695ee1c
1 parent 4dffbf8
commit 695ee1c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/ray/autoscaler/_private/command_runner.py b/python/ray/autoscaler/_private/command_runner.py
@@ -11,6 +11,7 @@
 
 import click
 
+from ray._private.ray_constants import DEFAULT_OBJECT_STORE_MAX_MEMORY_BYTES
 from ray.autoscaler._private.cli_logger import cf, cli_logger
 from ray.autoscaler._private.constants import (
     AUTOSCALER_NODE_SSH_INTERVAL_S,
@@ -903,8 +904,9 @@ def _auto_configure_shm(self, run_options: List[str]) -> List[str]:
             )
             available_memory_bytes = available_memory * 1024
             # Overestimate SHM size by 10%
-            shm_size = (
-                available_memory_bytes * DEFAULT_OBJECT_STORE_MEMORY_PROPORTION * 1.1
+            shm_size = min(
+                (available_memory_bytes * DEFAULT_OBJECT_STORE_MEMORY_PROPORTION * 1.1),
+                DEFAULT_OBJECT_STORE_MAX_MEMORY_BYTES,
             )
             return run_options + [f"--shm-size='{shm_size}b'"]
         except Exception as e: