stepan
/
dynare
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Amending assertions in BlasBindings.hh

time-shift
George Perendia 2010-03-15 12:12:43 +00:00
parent 195fdece71
commit 882ea0ed75
1 changed files with 34 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
mex/sources/estimation/libmat/BlasBindings.hh
View File

@ -50,10 +50,41 @@ namespace blas
double alpha, const Mat1 &A, const Mat2 &B,
double beta, Mat3 &C)
{
assert(A.getRows() == C.getRows());
assert(A.getCols() == B.getRows());
assert(B.getCols() == C.getCols());
blas_int m = A.getRows(), n = B.getCols(), k = A.getCols();
if (*transa == 'N')
{
if (*transb == 'N')
{
assert(A.getRows() == C.getRows());
assert(A.getCols() == B.getRows());
assert(B.getCols() == C.getCols());
}
else if (*transb == 'T')
{
assert(A.getRows() == C.getRows());
assert(A.getCols() == B.getCols());
assert(B.getRows() == C.getCols());
n = B.getRows();
}
}
else if (*transa == 'T')
{
m = A.getCols();
k = A.getRows();
if (*transb == 'N')
{
assert(A.getCols() == C.getRows());
assert(A.getRows() == B.getRows());
assert(B.getCols() == C.getCols());
}
else if (*transb == 'T')
{
assert(A.getCols() == C.getRows());
assert(A.getRows() == B.getCols());
assert(B.getRows() == C.getCols());
n = B.getRows();
}
}
blas_int lda = A.getLd(), ldb = B.getLd(), ldc = C.getLd();
dgemm(transa, transb, &m, &n, &k, &alpha, A.getData(), &lda,
B.getData(), &ldb, &beta, C.getData(), &ldc);