run.sh

#!/bin/bash -l
#SBATCH --job-name=CPG_WINIT_DIAG
#SBATCH --output=Nest_processing_%A_%a.slurmout
#SBATCH --error=Nest_processing_%A_%a.slurmerr
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --array=0-9
#SBATCH --cpus-per-task=64
#SBATCH --time=03:00:00
#SBATCH --partition=acc

export LANG=${LANG:-C.UTF-8}
export LC_ALL=${LC_ALL:-C.UTF-8}
export PYTHONIOENCODING=utf-8
export PYTHONUNBUFFERED=1

echo "[Slurm] ntasks=$SLURM_NTASKS cpus-per-task=$SLURM_CPUS_PER_TASK array_task=${SLURM_ARRAY_TASK_ID:-NA}"

python3 - <<'PY'
import nest
ks = nest.GetKernelStatus()
mpi = ks.get("mpi_num_processes", ks.get("num_processes", ks.get("total_num_processes", 1)))
thr = ks.get("local_num_threads", ks.get("num_threads", ks.get("threads", 1)))
print("nest", nest.__version__, "mpi_procs", mpi, "local_threads", thr)
PY

# 10 diagnostic (mu,CV) pairs (Option C)
SWEEP_PAIRS="0:0,1:0.8,2:0.4,3:0.2,5:0.1,8:0.05,11:0.2,14:0.4,20:0.8,25:0.05"
OUTDIR="results/"
TAG="winit_diag"
BASE_SEED=12345

srun --cpu-bind=cores --distribution=block:block \
  python3 -u cpg_2legs_fast.py \
    --out cpg_run.h5 \
    --outdir "$OUTDIR" \
    --tag "$TAG" \
    --seed $BASE_SEED \
    --sweep-pairs "$SWEEP_PAIRS" \
    --sweep-run-idx ${SLURM_ARRAY_TASK_ID} \
    --sweep-dist lognormal_cv \
    --sim-ms 60000 \
    --dt-ms 10 \
    --threads $SLURM_CPUS_PER_TASK \
    --nest-verbosity M_ERROR \
    --max-weight-conns 2000 \
    --save-weights snapshots \
    --delay-model length_velocity \
    --species human \
    --delay-jitter-ms 0.2 \
    --long-run