From de3209523a16c3c2c298455d2e1f42eecf9eadf1 Mon Sep 17 00:00:00 2001 From: Daniel Maier <daniel.maier@tu-berlin.de> Date: Mon, 29 Oct 2018 12:02:20 +0100 Subject: [PATCH] gemm application --- codes/gemm/gemm.c | 146 ++++++++++++++++++++++++++++++++++++++++++++++ codes/gemm/gemm.h | 86 +++++++++++++++++++++++++++ 2 files changed, 232 insertions(+) create mode 100644 codes/gemm/gemm.c create mode 100644 codes/gemm/gemm.h diff --git a/codes/gemm/gemm.c b/codes/gemm/gemm.c new file mode 100644 index 0000000..bdbd33a --- /dev/null +++ b/codes/gemm/gemm.c @@ -0,0 +1,146 @@ +/** + * This version is stamped on May 10, 2016 + * + * Contact: + * Louis-Noel Pouchet <pouchet.ohio-state.edu> + * Tomofumi Yuki <tomofumi.yuki.fr> + * + * Web address: http://polybench.sourceforge.net + */ +/* gemm.c: this file is part of PolyBench/C */ + +#include <stdio.h> +#include <unistd.h> +#include <string.h> +#include <math.h> + +/* Include polybench common header. */ +#include <polybench.h> + +/* Include benchmark-specific header. */ +#include "gemm.h" + + +/* Array initialization. */ +static +void init_array(int ni, int nj, int nk, + DATA_TYPE *alpha, + DATA_TYPE *beta, + DATA_TYPE POLYBENCH_2D(C,NI,NJ,ni,nj), + DATA_TYPE POLYBENCH_2D(A,NI,NK,ni,nk), + DATA_TYPE POLYBENCH_2D(B,NK,NJ,nk,nj)) +{ + int i, j; + + *alpha = 1.5; + *beta = 1.2; + for (i = 0; i < ni; i++) + for (j = 0; j < nj; j++) + C[i][j] = (DATA_TYPE) ((i*j+1) % ni) / ni; + for (i = 0; i < ni; i++) + for (j = 0; j < nk; j++) + A[i][j] = (DATA_TYPE) (i*(j+1) % nk) / nk; + for (i = 0; i < nk; i++) + for (j = 0; j < nj; j++) + B[i][j] = (DATA_TYPE) (i*(j+2) % nj) / nj; +} + + +/* DCE code. Must scan the entire live-out data. + Can be used also to check the correctness of the output. */ +static +void print_array(int ni, int nj, + DATA_TYPE POLYBENCH_2D(C,NI,NJ,ni,nj)) +{ + int i, j; + + POLYBENCH_DUMP_START; + POLYBENCH_DUMP_BEGIN("C"); + for (i = 0; i < ni; i++) + for (j = 0; j < nj; j++) { + if ((i * ni + j) % 20 == 0) fprintf (POLYBENCH_DUMP_TARGET, "\n"); + fprintf (POLYBENCH_DUMP_TARGET, DATA_PRINTF_MODIFIER, C[i][j]); + } + POLYBENCH_DUMP_END("C"); + POLYBENCH_DUMP_FINISH; +} + + +/* Main computational kernel. The whole function will be timed, + including the call and return. */ +static +void kernel_gemm(int ni, int nj, int nk, + DATA_TYPE alpha, + DATA_TYPE beta, + DATA_TYPE POLYBENCH_2D(C,NI,NJ,ni,nj), + DATA_TYPE POLYBENCH_2D(A,NI,NK,ni,nk), + DATA_TYPE POLYBENCH_2D(B,NK,NJ,nk,nj)) +{ + int i, j, k; + +//BLAS PARAMS +//TRANSA = 'N' +//TRANSB = 'N' +// => Form C := alpha*A*B + beta*C, +//A is NIxNK +//B is NKxNJ +//C is NIxNJ +#pragma scop + for (i = 0; i < _PB_NI; i++) { + for (j = 0; j < _PB_NJ; j++) + C[i][j] *= beta; + for (k = 0; k < _PB_NK; k++) { + for (j = 0; j < _PB_NJ; j++) + C[i][j] += alpha * A[i][k] * B[k][j]; + } + } +#pragma endscop + +} + + +int main(int argc, char** argv) +{ + /* Retrieve problem size. */ + int ni = NI; + int nj = NJ; + int nk = NK; + + /* Variable declaration/allocation. */ + DATA_TYPE alpha; + DATA_TYPE beta; + POLYBENCH_2D_ARRAY_DECL(C,DATA_TYPE,NI,NJ,ni,nj); + POLYBENCH_2D_ARRAY_DECL(A,DATA_TYPE,NI,NK,ni,nk); + POLYBENCH_2D_ARRAY_DECL(B,DATA_TYPE,NK,NJ,nk,nj); + + /* Initialize array(s). */ + init_array (ni, nj, nk, &alpha, &beta, + POLYBENCH_ARRAY(C), + POLYBENCH_ARRAY(A), + POLYBENCH_ARRAY(B)); + + /* Start timer. */ + polybench_start_instruments; + + /* Run kernel. */ + kernel_gemm (ni, nj, nk, + alpha, beta, + POLYBENCH_ARRAY(C), + POLYBENCH_ARRAY(A), + POLYBENCH_ARRAY(B)); + + /* Stop and print timer. */ + polybench_stop_instruments; + polybench_print_instruments; + + /* Prevent dead-code elimination. All live-out data must be printed + by the function call in argument. */ + polybench_prevent_dce(print_array(ni, nj, POLYBENCH_ARRAY(C))); + + /* Be clean. */ + POLYBENCH_FREE_ARRAY(C); + POLYBENCH_FREE_ARRAY(A); + POLYBENCH_FREE_ARRAY(B); + + return 0; +} diff --git a/codes/gemm/gemm.h b/codes/gemm/gemm.h new file mode 100644 index 0000000..0d18a0b --- /dev/null +++ b/codes/gemm/gemm.h @@ -0,0 +1,86 @@ +/** + * This version is stamped on May 10, 2016 + * + * Contact: + * Louis-Noel Pouchet <pouchet.ohio-state.edu> + * Tomofumi Yuki <tomofumi.yuki.fr> + * + * Web address: http://polybench.sourceforge.net + */ +#ifndef _GEMM_H +# define _GEMM_H + +/* Default to LARGE_DATASET. */ +# if !defined(MINI_DATASET) && !defined(SMALL_DATASET) && !defined(MEDIUM_DATASET) && !defined(LARGE_DATASET) && !defined(EXTRALARGE_DATASET) +# define LARGE_DATASET +# endif + +# if !defined(NI) && !defined(NJ) && !defined(NK) +/* Define sample dataset sizes. */ +# ifdef MINI_DATASET +# define NI 20 +# define NJ 25 +# define NK 30 +# endif + +# ifdef SMALL_DATASET +# define NI 60 +# define NJ 70 +# define NK 80 +# endif + +# ifdef MEDIUM_DATASET +# define NI 200 +# define NJ 220 +# define NK 240 +# endif + +# ifdef LARGE_DATASET +# define NI 1000 +# define NJ 1100 +# define NK 1200 +# endif + +# ifdef EXTRALARGE_DATASET +# define NI 2000 +# define NJ 2300 +# define NK 2600 +# endif + + +#endif /* !(NI NJ NK) */ + +# define _PB_NI POLYBENCH_LOOP_BOUND(NI,ni) +# define _PB_NJ POLYBENCH_LOOP_BOUND(NJ,nj) +# define _PB_NK POLYBENCH_LOOP_BOUND(NK,nk) + + +/* Default data type */ +# if !defined(DATA_TYPE_IS_INT) && !defined(DATA_TYPE_IS_FLOAT) && !defined(DATA_TYPE_IS_DOUBLE) +# define DATA_TYPE_IS_DOUBLE +# endif + +#ifdef DATA_TYPE_IS_INT +# define DATA_TYPE int +# define DATA_PRINTF_MODIFIER "%d " +#endif + +#ifdef DATA_TYPE_IS_FLOAT +# define DATA_TYPE float +# define DATA_PRINTF_MODIFIER "%0.2f " +# define SCALAR_VAL(x) x##f +# define SQRT_FUN(x) sqrtf(x) +# define EXP_FUN(x) expf(x) +# define POW_FUN(x,y) powf(x,y) +# endif + +#ifdef DATA_TYPE_IS_DOUBLE +# define DATA_TYPE double +# define DATA_PRINTF_MODIFIER "%0.2lf " +# define SCALAR_VAL(x) x +# define SQRT_FUN(x) sqrt(x) +# define EXP_FUN(x) exp(x) +# define POW_FUN(x,y) pow(x,y) +# endif + +#endif /* !_GEMM_H */ -- GitLab