#!/bin/bash -l
#SBATCH --partition=low
#SBATCH --gres=gpu:1
#SBATCH -C a40
# SBATCH --nodelist=gpu3-01
#SBATCH --mem=128g
#SBATCH --cpus-per-gpu=32
#SBATCH --time=48:00:00
#SBATCH --output=log-%j.txt

#export CUDA_VISIBLE_DEVICES=1
module purge
module load mamba
conda activate pyvkfft-test
module list
#rm -f *.html *.svg *.log *.txt
#pyvkfft-test-suite --backend pycuda --gpu a40 --gpumem 48 --nproc 32 --skip &> test.log
timeout 47h pyvkfft-test-suite --backend cupy --gpu a40 --gpumem 48 --nproc 16 --skip

if [[ $? -eq 124 ]]; then
  sbatch go.slurm
fi

cat pyvkfft-test1*.html > pyvkfft-test.html