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

#conda activate py39-pyvkfft-test
#source ~/dev/pynx-py38-power9-env/bin/activate
#export CUDA_VISIBLE_DEVICES=1
module load mamba
conda activate /home/esrf/favre/miniconda3/envs/pynx-py311-cu11.7
#pyvkfft-test-suite --backend pyopencl --gpu a40 --gpumem 48 --nproc 32 &> test.log

pyvkfft-test --systematic --backend pyopencl --gpu a40 --max-nb-tests 0 --nproc 8 --ndim 1 --range 2 4096 --dct 2 --radix --inplace --lut --norm 1 --range-mb 0 4100
pyvkfft-test --systematic --backend pyopencl --gpu a40 --max-nb-tests 0 --nproc 8 --ndim 2 --range 2 4500 --r2c --bluestein --double --norm 1 --range-mb 0 4100
pyvkfft-test --systematic --backend pyopencl --gpu a40 --max-nb-tests 0 --nproc 4 --ndim 3 --range 2 550 --r2c --bluestein --double --norm 1 --range-mb 0 4100
pyvkfft-test --systematic --backend pyopencl --gpu a40 --max-nb-tests 0 --nproc 8 --ndim 2 --range 2 4500 --bluestein --inplace --norm 1 --range-mb 0 4100
pyvkfft-test --systematic --backend pyopencl --gpu a40 --max-nb-tests 0 --nproc 8 --ndim 2 --range 2 4500 --r2c --bluestein --inplace --double --norm 1 --range-mb 0 4100
pyvkfft-test --systematic --backend pyopencl --gpu a40 --max-nb-tests 0 --nproc 8 --ndim 2 --range 2 4500 --r2c --bluestein --double --norm 0 --range-mb 0 4100
pyvkfft-test --systematic --backend pyopencl --gpu a40 --max-nb-tests 0 --nproc 8 --ndim 2 --range 2 4500 --r2c --bluestein --inplace --double --norm 0 --range-mb 0 4100

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