OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Fixed a bug in cuda::gemm()

This commit is contained in:
Davis King 2015-11-03 21:37:18 -05:00
parent dae8929a64
commit 93410af39e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dlib/dnn/cublas_dlibapi.cpp
View File

@ -110,8 +110,8 @@ namespace dlib
mat(lhs).nc() == mat(rhs).nr(),"")
}
const int m = mat(dest).nr();
const int n = mat(dest).nc();
const int m = mat(dest).nc();
const int n = mat(dest).nr();
const int k = trans_rhs ? mat(rhs).nc() : mat(rhs).nr();
check(cublasSgemm(context(),
transb,