summaryrefslogtreecommitdiff
path: root/src/routines
diff options
context:
space:
mode:
authorCedric Nugteren <web@cedricnugteren.nl>2017-08-24 21:15:14 +0200
committerCedric Nugteren <web@cedricnugteren.nl>2017-08-24 21:15:14 +0200
commit161fd8514d75d61045e1683e091f1851656c28da (patch)
treed0027323a9f45ea4fd2a82bb029f5ba9d58adf44 /src/routines
parent4d9d03ba512fb404537180a42cf17fb3e10a8033 (diff)
parentda28cc5e9315729bc7f9078f1ae43ac68740ec87 (diff)
Merge branch 'master' into im_to_col
Diffstat (limited to 'src/routines')
-rw-r--r--src/routines/level3/xgemm.cpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/src/routines/level3/xgemm.cpp b/src/routines/level3/xgemm.cpp
index 136eec43..3909c308 100644
--- a/src/routines/level3/xgemm.cpp
+++ b/src/routines/level3/xgemm.cpp
@@ -102,9 +102,9 @@ void Xgemm<T>::DoGemm(const Layout layout,
TestMatrixC(c_one, c_two, c_buffer, c_offset, c_ld);
// Selects which version of GEMM to run
- const auto m_n_k = static_cast<unsigned long>(m) * static_cast<unsigned long>(n) *
- static_cast<unsigned long>(k);
- const auto do_gemm_direct = (m_n_k < static_cast<unsigned long>(db_["XGEMM_MIN_INDIRECT_SIZE"]));
+ const auto m_n_k = static_cast<unsigned long long>(m) * static_cast<unsigned long long>(n) *
+ static_cast<unsigned long long>(k);
+ const auto do_gemm_direct = (m_n_k < static_cast<unsigned long long>(db_["XGEMM_MIN_INDIRECT_SIZE"]));
if (do_gemm_direct) { // for small sizes (single kernel)
GemmDirect(m, n, k, alpha,
a_buffer, a_offset, a_ld, b_buffer, b_offset, b_ld, beta,