From b5cdac829690f7c7911cb3535c134b4536840034 Mon Sep 17 00:00:00 2001 From: Dmitry Meyer Date: Thu, 9 Jul 2026 09:06:36 +0000 Subject: [PATCH] Slurm: fix memory request `memory: ..4GB` should be translated to `--mem=4096M`, not `--mem=8192M` (where 8GB is taken from `DEFAULT_MEMORY_SIZE.min`) --- src/dstack/_internal/core/backends/slurm/resources.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/dstack/_internal/core/backends/slurm/resources.py b/src/dstack/_internal/core/backends/slurm/resources.py index 9a481f6c6..d93a7b9c6 100644 --- a/src/dstack/_internal/core/backends/slurm/resources.py +++ b/src/dstack/_internal/core/backends/slurm/resources.py @@ -128,6 +128,8 @@ def get_requested_resources_from_resources_spec(spec: ResourcesSpec) -> Requeste # The easiest/safest option is to use some sane default memory = spec.memory.min or DEFAULT_MEMORY_SIZE.min assert memory + if spec.memory.max: + memory = min(memory, spec.memory.max) memory_mib = round(memory * 1024) gpu_count: int = 0