summaryrefslogtreecommitdiff
path: root/src/routines
diff options
context:
space:
mode:
authorCedric Nugteren <web@cedricnugteren.nl>2017-08-21 20:06:29 +0200
committerGitHub <noreply@github.com>2017-08-21 20:06:29 +0200
commite5eb6b1d3a66358093cb40f0fad51ecdc4654771 (patch)
treee5eb03736240ec07534319fdd15661e1093f04ac /src/routines
parentd67fd6604b4a6584c4f9e856057fcc8076ce377d (diff)
parentdfd332524ab0e66a04d803bb075787e35cd2db1a (diff)
Merge pull request #173 from mcian/PSO_params
Add PSO parameters support and search strategy selection from command…
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,