#!/bin/bash

## parameters
#PBS -N pwr_project_timeit
#PBS -q entry_teachingq
#PBS -l select=1:ncpus=32:mpiprocs=32:mem=512gb
#PBS -o pwr_project_timeit_log.out
#PBS -e pwr_project_timeit_log.err

module load python/gcc
module load openmpi/gcc

## environment 
cd ~/pwr_project 

## delete previous runs
rm pwr_project_timeit_log.*

N=10000

## execute Diag-Vec
echo "Diag-Vec"
mpiexec -n 8 python3 ./src/main_diag_vec_timeit.py $N
mpiexec -n 16 python3 ./src/main_diag_vec_timeit.py $N
mpiexec -n 32 python3 ./src/main_diag_vec_timeit.py $N

## execute Matrix-Vector
echo "Matrix-Vec"
mpiexec -n 8 python3 ./src/main_matrix_vec_timeit.py $N
mpiexec -n 16 python3 ./src/main_matrix_vec_timeit.py $N
mpiexec -n 32 python3 ./src/main_matrix_vec_timeit.py $N

## execute CG
echo "CG"
mpiexec -n 8 python3 ./src/main_cg_timeit.py $(expr $N / 10)
mpiexec -n 16 python3 ./src/main_cg_timeit.py $(expr $N / 10)
mpiexec -n 32 python3 ./src/main_cg_timeit.py $(expr $N / 10)