#!/bin/bash --login
#SBATCH --job-name=mdrun_mpi_omp_gpu
#SBATCH --output=%x.o%j
#SBATCH --error=%x.e%j
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=8
#SBATCH --gres=gpu:2
#SBATCH --time=0-24:00:00
#SBATCH --mem=16G
#SBATCH --partition=gpu
#SBATCH --qos=gpu
#SBATCH --account=<account>

shopt -q login_shell || exit 1
test -n "$SLURM_NODELIST" || exit 1

module load gnu8
module load openmpi3
module load gromacs/2021.7-gpu

OMP_NUM_THREADS=2

export OMP_NUM_THREADS

mpirun -np 4 gmx mdrun -deffnm meta -dlb auto -plumed plumed.dat