| 1 | #!/bin/bash |
|---|
| 2 | ### Partition to use |
|---|
| 3 | #SBATCH --account=cin0391 |
|---|
| 4 | #SBATCH --constraint=GENOA |
|---|
| 5 | ### Number of nodes/cores to use |
|---|
| 6 | #SBATCH --nodes=1 |
|---|
| 7 | #SBATCH --ntasks-per-node=24 |
|---|
| 8 | #SBATCH --cpus-per-task=8 |
|---|
| 9 | #SBATCH --threads-per-core=1 # --hint=nomultithread |
|---|
| 10 | ### Job information |
|---|
| 11 | #SBATCH --job-name=jobPCM0 |
|---|
| 12 | #SBATCH --output=jobPCM_%j.out |
|---|
| 13 | #SBATCH --time=4:00:00 |
|---|
| 14 | |
|---|
| 15 | ######################################################################## |
|---|
| 16 | # Modify here the parameters depending on your setup |
|---|
| 17 | #################################################### |
|---|
| 18 | # Path to the arch.env to source: |
|---|
| 19 | source ../trunk/LMDZ.COMMON/arch.env |
|---|
| 20 | |
|---|
| 21 | # Number of threads to use (must be the same as "#SBATCH --cpus-per-task=" above) |
|---|
| 22 | export OMP_NUM_THREADS=$SLURM_CPUS_PER_TASK |
|---|
| 23 | export OMP_STACKSIZE=400M |
|---|
| 24 | |
|---|
| 25 | # Name of executable for the PCM: |
|---|
| 26 | pcm_exe="gcm_64x48x32_phymars_para.e" |
|---|
| 27 | |
|---|
| 28 | # Execution command: |
|---|
| 29 | exec_cmd="srun --ntasks-per-node=${SLURM_NTASKS_PER_NODE} --cpu-bind=none --mem-bind=none --label -- ./adastra_cpu_binding.sh ./$pcm_exe" |
|---|
| 30 | ######################################################################## |
|---|
| 31 | |
|---|
| 32 | |
|---|
| 33 | ulimit -s unlimited |
|---|
| 34 | |
|---|
| 35 | # Running the PCM |
|---|
| 36 | read n_yr_sim ntot_yr_sim r_plnt2earth_yr i_pcm_run i_pem_run n_pcm_runs n_pcm_runs_ini < pem_workflow.sts |
|---|
| 37 | echo "Run \"PCM $i_pcm_run\" is starting." |
|---|
| 38 | cp run_pcm.def run.def |
|---|
| 39 | eval "$exec_cmd > run.log 2>&1" |
|---|
| 40 | if [ ! -f "restartfi.nc" ] || ! (tail -n 100 run.log | grep -iq "everything is cool!"); then # Check if it ended abnormally |
|---|
| 41 | echo "Error: the run \"PCM $i_pcm_run\" crashed!" |
|---|
| 42 | if [ $exec_mode -ne 0 ]; then |
|---|
| 43 | echo "Be careful: there may be dependent jobs remaining in the queue! You can cancel them by executing the script \"kill_pem_workflow.sh\"." |
|---|
| 44 | fi |
|---|
| 45 | exit 1 |
|---|
| 46 | fi |
|---|
| 47 | |
|---|
| 48 | # Copy data files and prepare the next run |
|---|
| 49 | mv run.log logs/run_pcm${i_pcm_run}.log |
|---|
| 50 | if [ -f "diagfi.nc" ]; then |
|---|
| 51 | mv diagfi.nc diags/diagfi${i_pcm_run}.nc |
|---|
| 52 | fi |
|---|
| 53 | if [ -f "diagsoil.nc" ]; then |
|---|
| 54 | mv diagsoil.nc diags/diagsoil${i_pcm_run}.nc |
|---|
| 55 | fi |
|---|
| 56 | if [ -f "stats.nc" ]; then |
|---|
| 57 | mv stats.nc diags/stats${i_pcm_run}.nc |
|---|
| 58 | fi |
|---|
| 59 | k=0 |
|---|
| 60 | if [ $(echo "$k > 0" | bc) -eq 1 ]; then # Only the last 2 years are taken for the PEM |
|---|
| 61 | cp Xoutdaily4pem.nc Xoutdaily4pem_Y${k}.nc |
|---|
| 62 | cp Xoutyearly4pem.nc Xoutyearly4pem_Y${k}.nc |
|---|
| 63 | fi |
|---|
| 64 | mv Xoutdaily4pem.nc diags/Xoutdaily4pem${i_pcm_run}.nc |
|---|
| 65 | mv Xoutyearly4pem.nc diags/Xoutyearly4pem${i_pcm_run}.nc |
|---|
| 66 | cp restartfi.nc starts/restartfi${i_pcm_run}.nc |
|---|
| 67 | mv restartfi.nc startfi.nc |
|---|
| 68 | if [ -f "restart.nc" ]; then |
|---|
| 69 | cp restart.nc starts/restart${i_pcm_run}.nc |
|---|
| 70 | mv restart.nc start.nc |
|---|
| 71 | elif [ -f "restart1D.txt" ]; then |
|---|
| 72 | cp restart1D.txt starts/restart1D${i_pcm_run}.txt |
|---|
| 73 | mv restart1D.txt start1D.txt |
|---|
| 74 | fi |
|---|
| 75 | ((i_pcm_run++)) |
|---|
| 76 | sed -i "1s/.*/$n_yr_sim $ntot_yr_sim $r_plnt2earth_yr $i_pcm_run $i_pem_run $n_pcm_runs $n_pcm_runs_ini/" pem_workflow.sts |
|---|
