#!/bin/bash
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --partition=gpu
#SBATCH --cpus-per-task=18
#SBATCH --time=10:00:00
#SBATCH --gpus-per-node=1

REPLICAS=100

# activate conda environment
ENVNAME=nnpdf_dev
source ~/.bashrc
anaconda
conda activate $ENVNAME

# Verify GPU usage:
ngpus=$(python3 -c "import tensorflow as tf; print(len(tf.config.list_physical_devices('GPU')))")
if [ "${ngpus}" -eq 0 ]; then
    echo "GPUs not being used!"
else
    echo "Using GPUs! Num GPUs Available: ${ngpus}"
fi

runcard="nnpdf40-like.yml"

CMD_0="vp-setupfit $runcard"
echo "Running vp-setup"
echo $CMD_0
eval $CMD_0

CMD_1="n3fit $runcard 1 -r 120"
echo "Running n3fit"
echo $CMD_1
eval $CMD_1

CMD_2="evolven3fit evolve nnpdf40-like"
echo "Running evolven3fit"
echo $CMD_2
eval $CMD_2

CMD_3="postfit $REPLICAS nnpdf40-like"
echo "Running postfit"
echo $CMD_3
eval $CMD_3