#!/bin/sh
#< 1 node with 7 GPU
#SBATCH --partition=gpu
#SBATCH --qos=gpu
#SBATCH --nodes=1
#SBATCH --gres=gpu:p100:7
#SBATCH --time=0-00:30:00
#SBATCH --mem=3G
#< Charge resources to account 
##SBATCH --account=<account> 

module load cuda

echo "# SLURM_JOB_NODELIST  : $SLURM_JOB_NODELIST"
echo "# CUDA_VISIBLE_DEVICES: $CUDA_VISIBLE_DEVICES"

/hpc/share/samples/cuda/nbody/nbody -benchmark -numbodies=1024000 -numdevices=7