Commit 192ec7d6 authored by Jakub Klinkovský's avatar Jakub Klinkovský
Browse files

Simplified run_benchmark script

parent 77a99abd
Loading
Loading
Loading
Loading
+15 −17
Original line number Original line Diff line number Diff line
@@ -2,16 +2,12 @@


set -e
set -e


NCPUS="$SLURM_CPUS_PER_TASK"
# maximum number of CPU cores available on the system
# set fallback when not running under Slurm
# (this respects even $SLURM_CPUS_PER_TASK when executed under Slurm)
if [[ "$NCPUS" == "" ]]; then
NCPUS=$(nproc)
    NCPUS=12
fi


function run_benchmark()
function run_benchmark()
{
{
    echo "mesh = $(basename $mesh_file)"

    # set number of OpenMP threads
    # set number of OpenMP threads
    omp_threads=${omp_threads:=1}
    omp_threads=${omp_threads:=1}
    if [[ "$omp_threads" == "1" ]]; then
    if [[ "$omp_threads" == "1" ]]; then
@@ -48,6 +44,10 @@ function run_benchmark()


function run_set()
function run_set()
{
{
    if (( $omp_threads > $NCPUS )); then
        return
    fi

    for mesh_type in mrizka cube1m; do
    for mesh_type in mrizka cube1m; do
    for id in 1 2 3 4 5; do
    for id in 1 2 3 4 5; do
        [[ "$mesh_type" == "mrizka" ]] && id="$id.rcm"
        [[ "$mesh_type" == "mrizka" ]] && id="$id.rcm"
@@ -61,13 +61,11 @@ function run_set()
#omp_threads=1
#omp_threads=1
#run_set
#run_set


if [[ "$CUDA_VISIBLE_DEVICES" == "" ]]; then
solver="./tnl-benchmark-mesh --devices host"
solver="./tnl-benchmark-mesh --devices host"
    for omp_threads in 1 2 4 6 8 $NCPUS; do
for omp_threads in 1 2 4 6 8 12; do
   run_set
   run_set
done
done
else

solver="./tnl-benchmark-mesh-cuda --devices cuda"
solver="./tnl-benchmark-mesh-cuda --devices cuda"
omp_threads=1
omp_threads=1
run_set
run_set
fi