diff options
author | Justin Berger <j.david.berger@gmail.com> | 2018-03-17 09:32:22 -0600 |
---|---|---|
committer | Justin Berger <j.david.berger@gmail.com> | 2018-03-17 09:32:22 -0600 |
commit | 0b9e66ad2ff686a4dcf8a6838f33edb203a1bff5 (patch) | |
tree | 2ccc16e01e625901f00c59cc85535286116b1792 /redist/dclapack.h | |
parent | 7c97cfe7f63650fc79ce4fa7f081b556ce275475 (diff) | |
download | libsurvive-0b9e66ad2ff686a4dcf8a6838f33edb203a1bff5.tar.gz libsurvive-0b9e66ad2ff686a4dcf8a6838f33edb203a1bff5.tar.bz2 |
Fixed gemm
Diffstat (limited to 'redist/dclapack.h')
-rw-r--r-- | redist/dclapack.h | 33 |
1 files changed, 17 insertions, 16 deletions
diff --git a/redist/dclapack.h b/redist/dclapack.h index af8869c..af5035b 100644 --- a/redist/dclapack.h +++ b/redist/dclapack.h @@ -224,22 +224,23 @@ PRINT(Ainv,n,n); \ /* * Matrix Multiply R = alpha * A * B + beta * C * R (n by p) - * A (n by m) - * B (m by p) - * C (n by p) + * A (m by n) + * B (n by p) + * C (m by p) */ -#define GMULADD(R,A,B,C,alpha,beta,n,m,p) { \ - int i,j,k; \ - float sum; \ - for (i=0; i<n; i++) { \ - for (j=0; j<p; j++) { \ - sum = 0.0f; \ - for (k=0; k<m; k++) { \ - sum += _(A,i,k) * _(B,k,j); \ - } \ - _(R,i,j) = alpha * sum + beta * _(C,i,j); \ - } \ - } \ -} +#define GMULADD(R, A, B, C, alpha, beta, m, n, p) \ + { \ + int _i, _j, _k; \ + float sum; \ + for (_i = 0; _i < m; _i++) { \ + for (_j = 0; _j < p; _j++) { \ + sum = 0.0f; \ + for (_k = 0; _k < n; _k++) { \ + sum += _(A, _i, _k) * _(B, _k, _j); \ + } \ + _(R, _i, _j) = alpha * sum + beta * _(C, _i, _j); \ + } \ + } \ + } #endif |