From 93410af39e68487efbee0d0be019bdb4707a7d58 Mon Sep 17 00:00:00 2001 From: Davis King Date: Tue, 3 Nov 2015 21:37:18 -0500 Subject: [PATCH] Fixed a bug in cuda::gemm() --- dlib/dnn/cublas_dlibapi.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dlib/dnn/cublas_dlibapi.cpp b/dlib/dnn/cublas_dlibapi.cpp index 7ae40f1ca..cfde29057 100644 --- 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 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,