Skip to content

Commit

Permalink
Move the pipeline to the new partition
Browse files Browse the repository at this point in the history
  • Loading branch information
nefrathenrici committed Feb 28, 2024
1 parent 6e2c27c commit 80c3236
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 4 deletions.
8 changes: 6 additions & 2 deletions experiments/pipeline.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#!/bin/bash

# Configure the environment
source experiments/common_env.sh
export MODULEPATH=/groups/esm/modules:$MODULEPATH
module load climacommon/2024_02_27

# Parse command line
experiment_id=${1?Error: no experiment ID given}
Expand All @@ -11,6 +11,7 @@ echo "Running experiment $experiment_id with $tasks_per_model_run tasks per mode
echo 'Initializing ensemble for calibration.'
init_id=$(sbatch --parsable \
--output=init_$experiment_id.out \
--partition=expansion \
experiments/initialize.sbatch $experiment_id)

# Get ensemble size and the number of iterations from configuration file
Expand All @@ -30,6 +31,7 @@ do
--output=/dev/null \
--ntasks=$tasks_per_model_run \
--array=1-$ensemble_size \
--partition=expansion \
experiments/model_run.sbatch $experiment_id $i)

dependency=afterany:$ensemble_array_id
Expand All @@ -39,8 +41,10 @@ do
sbatch --dependency=$dependency --kill-on-invalid-dep=yes --parsable \
--job=update-$i \
--output=output/$experiment_id/$format_i/update_log.out \
--partition=expansion \
experiments/update.sbatch $experiment_id $i)

dependency=afterany:$update_id
echo "Update $i job id: $update_id"
echo ""
done
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,15 @@
#SBATCH --ntasks=32
#SBATCH --cpus-per-task=8
#SBATCH --output="experiments/sphere_held_suarez_rhoe_equilmoist/truth_simulation/model_log.out"
#SBATCH --partition=expansion

# Configure the environment
source experiments/common_env.sh
export MODULEPATH=/groups/esm/modules:$MODULEPATH
module load climacommon/2024_02_20

echo "Generating truth observations."

srun --open-mode=append julia --project=experiments -e '
srun --open-mode=append julia --project=experiments -e '
using ClimaComms
ClimaComms.init(ClimaComms.context())
import ClimaAtmos as CA
Expand Down

0 comments on commit 80c3236

Please sign in to comment.