Home Browse Files 
      1
      2
      3
      4
      5
      6
      7
      8
      9
     10
     11
     12
     13
     14
     15
     16
     17
     18
     19
     20
     21
     22
     23
     24
     25
     26
     27
     28
     29
     30
     31
     32
     33
     34
     35
     36
     37
     38
     39
     40
     41
     42
     43
     44
     45
     46
     47
     48
     49
     50
     51
     52
     53
     54
     55
     56
     57
     58
     59
     60

#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>

void
initGeMatrix(size_t m, size_t n,
             double *A,
             ptrdiff_t incRowA, ptrdiff_t incColA)
{
    for (size_t i=0; i<m; ++i) {
        for (size_t j=0; j<n; ++j) {
            A[i*incRowA+j*incColA] = i*n + j + 1;
        }
    }
}

void
printGeMatrix(size_t m, size_t n,
              const double *A,
              ptrdiff_t incRowA, ptrdiff_t incColA)
{
    for (size_t i=0; i<m; ++i) {
        for (size_t j=0; j<n; ++j) {
            printf("%10.4lf ", A[i*incRowA+j*incColA]);
        }
        printf("\n");
    }
    printf("\n");
}

#ifndef DGEMM_MC
#define DGEMM_MC    8
#endif

#ifndef DGEMM_KC
#define DGEMM_KC    9
#endif

#ifndef DGEMM_MR
#define DGEMM_MR    4
#endif

int
main()
{
    size_t      m       = 6;
    size_t      k       = 7;
    double      *A      = malloc(m*k*sizeof(*A));
    ptrdiff_t   incRowA = 1;
    ptrdiff_t   incColA = m;

    double      *p      = malloc(DGEMM_MC*DGEMM_KC*sizeof(*p));

    initGeMatrix(m, k, A, incRowA, incColA);
    printGeMatrix(m, k, A, incRowA, incColA);
    printGeMatrix(1, DGEMM_MC*DGEMM_KC, p, 1, 1);

    free(p);
    free(A);
}





































































































Home