Changeset 5000 for BOL/LMDZ_Setup_amaury/lmdz_env.sh

Timestamp:

Jul 1, 2024, 10:09:23 AM (3 months ago)

Author:

abarral

Message:

update lmdz_env.sh to separate bash run and srun

File:

• BOL/LMDZ_Setup_amaury/lmdz_env.sh (modified) (4 diffs)

BOL/LMDZ_Setup_amaury/lmdz_env.sh

@@ -44 +44 @@
       NB_MPI_MAX=2000
       NB_OMP_MAX=20
-      RUNCMD="srun --label -n"
+      MPICMD="srun -n"
+      RUNBASHCMD="srun -A $account@cpu --label -n 1 -c"
       SUBMITCMD="sbatch -A $account@cpu"
       ;;
@@ -54 +55 @@
       module load cdo/2.3.0
 
-      root_dir=~/"LMDZ_SETUP_ROOT/LMDZ_Setup";
+      root_dir="/data/abarral/LMDZ_SETUP_ROOT/LMDZ_Setup";
       ARCH="X64_MESOIPSL-GNU"
       SIMRUNBASEDIR="$SCRATCH/"
       LMDZD="$root_dir/LMDZD"
       LMDZ_INIT="$root_dir/LMDZ_Init"
-      NB_MPI_MAX=20
+      NB_MPI_MAX=5
       NB_OMP_MAX=1
-      RUNCMD="mpirun -n"  # on spirit, we can't run MPI using srun from within sbatch
+      MPICMD="mpirun -np"  # on spirit, we can't run MPI using srun from within sbatch
+      RUNBASHCMD="bash"
       SUBMITCMD="sbatch"
       ;;
@@ -94 +96 @@
       NB_MPI_MAX=1  # TODO 2000 in JZ
       NB_OMP_MAX=1  # TODO 20 in JZ
-      RUNCMD="srun --label --account=$account --constraint=GENOA -n"
+      MPICMD="srun -n"
+      RUNBASHCMD="srun --label --account=$account --constraint=GENOA --ntasks-per-node=1 -n 1 --time=00:10:00 -c"
       SUBMITCMD="sbatch --constraint=GENOA --account=$account"
       ;;
@@ -105 +108 @@
       NB_MPI_MAX=2  # Max number of MPI cores (only for running simulations)
       NB_OMP_MAX=1  # Max number of OMP threads (only for running simulations)
-      RUNCMD="mpirun -np" # command to run a job, as $runcmd <nprocs> <script>
+      MPICMD="mpirun -np" # command to run an mpi executable, as $MPICMD <nprocs> <script>
+      RUNBASHCMD="bash" # command to run a bash job, as $runbashcmd (nthreads) <script> [nthreads only supplied if =/="bash"]
       SUBMITCMD="."  # command to sumbit a job, as $submitcmd <script>
       ;;