matrix_multiply.c

Go to the documentation of this file.

#include <stdio.h>

#define NUM_RUNS 3

#define MATRIX_SIZE 512
static double a[MATRIX_SIZE][MATRIX_SIZE];
static double b[MATRIX_SIZE][MATRIX_SIZE];
static double c[MATRIX_SIZE][MATRIX_SIZE];

long long naive_matrix_multiply_estimated_flops(int quiet) {

    long long muls,divs,adds;

    /* setup */
    muls=MATRIX_SIZE*MATRIX_SIZE;
    divs=MATRIX_SIZE*MATRIX_SIZE;
    adds=MATRIX_SIZE*MATRIX_SIZE;

    /* multiply */
    muls+=MATRIX_SIZE*MATRIX_SIZE*MATRIX_SIZE;
    adds+=MATRIX_SIZE*MATRIX_SIZE*MATRIX_SIZE;

    /* sum */
    adds+=MATRIX_SIZE*MATRIX_SIZE;

    if (!quiet) {
        printf("Estimated flops: adds: %lld muls: %lld divs: %lld\n",
            adds,muls,divs);
    }

    return adds+muls+divs;
}


long long naive_matrix_multiply_estimated_loads(int quiet) {

    long long loads=0;

    /* setup */
    loads+=0;

    /* multiply */
    loads+=MATRIX_SIZE*MATRIX_SIZE*MATRIX_SIZE*2;

    /* sum */
    loads+=MATRIX_SIZE*MATRIX_SIZE;

    if (!quiet) {
        printf("Estimated loads: %lld\n",loads);
    }

    return loads;
}

long long naive_matrix_multiply_estimated_stores(int quiet) {

    long long stores=0;

    /* setup */
    stores+=MATRIX_SIZE*MATRIX_SIZE*2;

    /* multiply */
    stores+=MATRIX_SIZE*MATRIX_SIZE;

    /* sum */
    stores+=1;

    if (!quiet) {
        printf("Estimated stores: %lld\n",stores);
    }

    return stores;
}


double naive_matrix_multiply(int quiet) {

    double s;
    int i,j,k;

    for(i=0;i<MATRIX_SIZE;i++) {
        for(j=0;j<MATRIX_SIZE;j++) {
            a[i][j]=(double)i*(double)j;
            b[i][j]=(double)i/(double)(j+5);
        }
    }

    for(j=0;j<MATRIX_SIZE;j++) {
        for(i=0;i<MATRIX_SIZE;i++) {
            s=0;
            for(k=0;k<MATRIX_SIZE;k++) {
                s+=a[i][k]*b[k][j];
            }
            c[i][j] = s;
        }
    }

    s=0.0;

    for(i=0;i<MATRIX_SIZE;i++) {
        for(j=0;j<MATRIX_SIZE;j++) {
            s+=c[i][j];
        }
    }

    if (!quiet) printf("Matrix multiply sum: s=%lf\n",s);

    return s;
}