diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/routines/level3/xgemm.cpp | 6 | ||||
-rw-r--r-- | src/tuning/kernels/xgemm.cpp | 6 |
2 files changed, 6 insertions, 6 deletions
diff --git a/src/routines/level3/xgemm.cpp b/src/routines/level3/xgemm.cpp index 30e5999c..4c8e0f79 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, diff --git a/src/tuning/kernels/xgemm.cpp b/src/tuning/kernels/xgemm.cpp index e2032256..a6a5e2e7 100644 --- a/src/tuning/kernels/xgemm.cpp +++ b/src/tuning/kernels/xgemm.cpp @@ -63,7 +63,7 @@ class TuneXgemm { static double DefaultInfluenceGlobalPSO(){ return 0.1; } static double DefaultInfluenceLocalPSO(){ return 0.3; } static double DefaultInfluenceRandomPSO(){ return 0.6; } - static size_t DefaultHeuristic(){ return size_t{3};} // PSO + static size_t DefaultHeuristic(){ return (size_t)cltune::SearchMethod::PSO;} // PSO static double DefaultMaxTempAnn(){ return 1.0;} // Describes how to obtain the sizes of the buffers @@ -187,7 +187,7 @@ class TuneXgemm { // Multi Search Strategy is enable if( args.multi_search_strategy){ if( V == 1){ - return size_t{0}; + return (size_t) cltune::SearchMethod::FullSearch; } else{ return args.heuristic_selection; @@ -197,7 +197,7 @@ class TuneXgemm { // Use full-search to explore all parameter combinations or random-search to search only a part of // the parameter values. The fraction is set as a command-line argument. if (args.fraction == 1.0 || args.fraction == 0.0) { - return size_t{0}; // Full search + return (size_t) cltune::SearchMethod::FullSearch; } else { return args.heuristic_selection; |