summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/atlas/test_cblas_dgemm.c
blob: 2cb90fb8830c56a0089f3c027c4e114acc06e4f5 (plain) (blame)
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
#include <cblas.h>
#include <stdio.h>

double m[] = {
  3, 1, 3,
  1, 5, 9,
  2, 6, 5
};

double x[] = {
  -1, 3, -3
};

#ifdef __cplusplus
extern "C" {
#endif

     void dgesv_(int *n, int *nrhs,  double *a,  int  *lda,
           int *ipivot, double *b, int *ldb, int *info);

#ifdef __cplusplus
}
#endif

int main(void) {
  int i;
  // blas:
  double A[6] = {1.0, 2.0, 1.0, -3.0, 4.0, -1.0};
  double B[6] = {1.0, 2.0, 1.0, -3.0, 4.0, -1.0};
  double C[9] = {.5, .5, .5, .5, .5, .5, .5, .5, .5};
  cblas_dgemm(CblasColMajor, CblasNoTrans, CblasTrans,
              3, 3, 2, 1, A, 3, B, 3, 2, C, 3);
  for (i = 0; i < 9; i++)
    printf("%f\n", C[i]);

  // lapack:
  int ipiv[3];
  int j;
  int info;
  int n = 1;
  int nrhs = 1;
  int lda = 3;
  int ldb = 3;
  dgesv_(&n,&nrhs, &m[0], &lda, ipiv, &x[0], &ldb, &info);
  for (i=0; i<3; ++i)
    printf("%5.1f\n", x[i]);

  return 0;
}