From 8065a35450d0f174db2f9dbb92445294e928035f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Tue, 10 Mar 2026 12:54:07 +0000 Subject: [PATCH] fix: Add GROUP_RANK MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- nemo_run/core/execution/templates/dgxc.sh.j2 | 1 + 1 file changed, 1 insertion(+) diff --git a/nemo_run/core/execution/templates/dgxc.sh.j2 b/nemo_run/core/execution/templates/dgxc.sh.j2 index 75bdede2..df7ab9f9 100644 --- a/nemo_run/core/execution/templates/dgxc.sh.j2 +++ b/nemo_run/core/execution/templates/dgxc.sh.j2 @@ -8,6 +8,7 @@ export TORCHX_MAX_RETRIES={{max_retries}} {%- for env_var in env_vars %} {{env_var}} {%- endfor %} +export GROUP_RANK=$(echo $HOSTNAME | grep -oE '[0-9]+$') {%- if ft_enabled %} {{ fault_tolerance.ft_launcher_setup(fault_tol_cfg_path, fault_tol_finished_flag_file, fault_tol_job_results_file) }}