From 259bb71361f44d54602f43b96fc65c9c7ea92435 Mon Sep 17 00:00:00 2001 From: Luca Bertagna Date: Tue, 22 Oct 2024 13:54:06 -0600 Subject: [PATCH] Print slurm env vars --- components/eamxx/scripts/test_all_scream.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/components/eamxx/scripts/test_all_scream.py b/components/eamxx/scripts/test_all_scream.py index 53759eac53f..8809aeb7f67 100644 --- a/components/eamxx/scripts/test_all_scream.py +++ b/components/eamxx/scripts/test_all_scream.py @@ -402,6 +402,9 @@ def get_taskset_resources(self, test, for_compile=True): ############################################################################### res_name = "compile_res_count" if for_compile else "testing_res_count" + slurm_vars = "\n".join(f"{key}: {value}" for key, value in os.environ.items() if key.startswith('SLURM')) + print(f"SLURM env vars:\n{slurm_vars}") + if not for_compile and self.on_cuda(): # For GPUs, the cpu affinity is irrelevant. Just assume all GPUS are open affinity_cp = list(range(self._ctest_max_jobs))