#!/bin/bash

# Wrapper API for DNNMark
# 2018 (C) Peter Bryzgalov @ CHITECH Stair Lab

usage=$(cat <<- USAGEBLOCK
Run DNNMark with parameters from CLI.
Usage:
$(basename $0)  [-n <number of images, batch size>]
                [-c <number of channels in input images>]
                [-h <height of input images>]
                [-w <widht of input images>]
                [-k <number of filters, ouput channels>]
                [-s <size of filter kernel>]
                [-u <stride>]
                [-p <padding>]
                [ --algo <cudnnConvolutionBwdFilterAlgo_t> - cuDNN algorithm for backward filter convolution]
                [ --bwd_filter_pref <fastest/no_workspace/specify_workspace_limit> - cuDNN backward filter algorithm selection preference]
                [ --algod <cudnnConvolutionBwdDataAlgo_t> - cuDNN algorithm for backward data convolution]
                [-b <benchmark executable, default=test_bwd_conv>]
                [ --iter <int> - number of FWD+BWD passes to measure time]
                [ --template - benchmark configuration template file]
                [ --debug - debug info ]
                [ --help  - usage info ]
                [ -d <dataset size> - number of samples in dataset, derives number of iterations from batch size and datasetsize]

Configuration saved in temporary file conf_tmp.dnnmark
USAGEBLOCK
)

template="conf_convolution_block.dnntemplate"
config_file="conf_tmp.dnnmark"
conv_bwd_filter_pref="fastest"
# Defaults
N=64
C=3
H=32
W=32
K=128
S=3
U=1
P=1
BENCH="test_composed_model"
ITER=1
debug=0
datasetsize=0


while test $# -gt 0; do
    case "$1" in
        --help)
            echo "$usage"
            exit 0
            ;;
        -n)
            N="$2";shift;
            ;;
        -c)
            C="$2";shift;
            ;;
        -h)
            H="$2";shift;
            ;;
        -w)
            W="$2";shift;
            ;;
        -k)
            K="$2";shift;
            ;;
        -s)
            S="$2";shift;
            ;;
        -u)
            U="$2";shift;
            ;;
        -p)
            P="$2";shift;
            ;;
        -b)
            BENCH="$2";shift;
            ;;
        -d)
            datasetsize="$2";shift;
            ;;
        --algo)
            CBFA="$2";shift;
            ;;
        --bwd_filter_pref)
            conv_bwd_filter_pref="$2";shift;
            ;;
        --algod)
            CBDA="$2";shift;
            ;;
        --iter)
            ITER="$2";shift;
            ;;
        --debug)
            debug=1
            ;;
        --template)
            template="$2";shift;
            ;;
        --)
            shift
            break;;
        -*)
            echo "Unknown option $1";
            echo "$usage"
            exit 1
            ;;
        *)
            break;;
    esac
    shift
done

if [ $CBFA ];then
    CUDNN_CBFA="algo=$CBFA"
fi

if [ $CBDA ];then
    CUDNN_CBDA="algod=$CBDA"
fi

divide_ceil() {
    echo "($1 + $2 - 1)/$2" | bc
}

# Calculate number of iterations from BS ($N) and dataset size
if [ $datasetsize -gt 0 ]; then
    echo "datasetsize=$datasetsize"
    # echo "$datasetsize / $N = "
    # echo "$(divide_ceil $datasetsize $N)"
    ITER=$(divide_ceil $datasetsize $N)
fi

echo "Using template $template"
conf="$(echo EOF;cat $template;echo EOF)"

eval "cat <<$conf" >$config_file
echo "Config: ---"
cat $config_file
echo "-----------"
echo "Benchmark: $BENCH"
echo "Iterations:$ITER"

com="./build/benchmarks/${BENCH}/dnnmark_${BENCH} -config $config_file --warmup 0 --iterations $ITER --debuginfo $debug"
echo $com
$com