Skip to content

Commit

Permalink
Print slurm env vars
Browse files Browse the repository at this point in the history
  • Loading branch information
bartgol committed Oct 25, 2024
1 parent d7145b5 commit b2d7910
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions components/eamxx/scripts/test_all_scream.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand Down

0 comments on commit b2d7910

Please sign in to comment.