Fixed a bug in cuda::gemm()

93410af3 · Davis King · dae8929a · 93410af3
Commit 93410af3 authored Nov 04, 2015 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

cublas_dlibapi.cpp dlib/dnn/cublas_dlibapi.cpp +2 -2

No files found.
--- a/dlib/dnn/cublas_dlibapi.cpp
+++ b/dlib/dnn/cublas_dlibapi.cpp
@@ -110,8 +110,8 @@ namespace dlib
                              mat(lhs).nc() == mat(rhs).nr(),"")
            }
-            const int m = mat(dest).nr();
+            const int m = mat(dest).nc();
-            const int n = mat(dest).nc();
+            const int n = mat(dest).nr();
            const int k = trans_rhs ? mat(rhs).nc() : mat(rhs).nr();
            check(cublasSgemm(context(),
                              transb,