diff options
Diffstat (limited to 'MLP/support/params.h')
-rw-r--r-- | MLP/support/params.h | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/MLP/support/params.h b/MLP/support/params.h new file mode 100644 index 0000000..f9e790e --- /dev/null +++ b/MLP/support/params.h @@ -0,0 +1,56 @@ +#ifndef _PARAMS_H_ +#define _PARAMS_H_ + +#include "common.h" + +typedef struct Params { + unsigned int m_size; + unsigned int n_size; + unsigned int n_warmup; + unsigned int n_reps; +}Params; + +static void usage() { + fprintf(stderr, + "\nUsage: ./program [options]" + "\n" + "\nGeneral options:" + "\n -h help" + "\n -w <W> # of untimed warmup iterations (default=1)" + "\n -e <E> # of timed repetition iterations (default=3)" + "\n" + "\nBenchmark-specific options:" + "\n -m <I> m_size (default=2048 elements)" + "\n -n <I> n_size (default=2048 elements)" + "\n"); +} + +struct Params input_params(int argc, char **argv) { + struct Params p; + p.m_size = 163840; + p.n_size = 4096; + p.n_warmup = 1; + p.n_reps = 3; + + int opt; + while((opt = getopt(argc, argv, "hm:n:w:e:")) >= 0) { + switch(opt) { + case 'h': + usage(); + exit(0); + break; + case 'm': p.m_size = atoi(optarg); break; + case 'n': p.n_size = atoi(optarg); break; + case 'w': p.n_warmup = atoi(optarg); break; + case 'e': p.n_reps = atoi(optarg); break; + default: + fprintf(stderr, "\nUnrecognized option!\n"); + usage(); + exit(0); + } + } + assert(NR_DPUS > 0 && "Invalid # of dpus!"); + + return p; +} +#endif |