From 60d890c998556a1ada09a7d32848cadca8a1d0ff Mon Sep 17 00:00:00 2001
From: Miki Bonacci <46074008+mikibonacci@users.noreply.github.com>
Date: Tue, 11 Feb 2025 11:57:55 +0100
Subject: [PATCH] Update thor-setup.yml

This automatically sets the threads to be consistend with requested cpus_per_task.

useful in case we want to use thread-parallelized codes (e.g. yambo)
---
 thor.psi.ch/default/thor-setup.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/thor.psi.ch/default/thor-setup.yml b/thor.psi.ch/default/thor-setup.yml
index 6bfcb4e..7c45c4b 100644
--- a/thor.psi.ch/default/thor-setup.yml
+++ b/thor.psi.ch/default/thor-setup.yml
@@ -6,7 +6,7 @@ hostname: lms-login.psi.ch
 label: thor
 mpiprocs_per_machine: 48
 mpirun_command: mpirun -np {tot_num_mpiprocs}
-prepend_text: ''
+prepend_text: 'export OMP_NUM_THREADS=$SLURM_CPUS_PER_TASK'
 scheduler: core.slurm
 shebang: "#!/bin/bash"
 transport: core.ssh