diff options
Diffstat (limited to 'src/routines')
| -rw-r--r-- | src/routines/levelx/xgemmbatched.cpp | 2 | ||||
| -rw-r--r-- | src/routines/levelx/xgemmstridedbatched.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/src/routines/levelx/xgemmbatched.cpp b/src/routines/levelx/xgemmbatched.cpp index 1c0953e8..6a089b8a 100644 --- a/src/routines/levelx/xgemmbatched.cpp +++ b/src/routines/levelx/xgemmbatched.cpp @@ -96,7 +96,7 @@ void XgemmBatched<T>::DoGemmBatched(const Layout layout, const Transpose a_trans } // Selects which version of the batched GEMM to run - const auto do_gemm_direct = true; + const auto do_gemm_direct = Xgemm<T>::UseDirectKernel(m, n, k, db_["XGEMM_MIN_INDIRECT_SIZE"]); if (do_gemm_direct) { // single generic kernel BatchedGemmDirect(m, n, k, alphas_device, a_buffer, a_offsets_int, a_ld, b_buffer, b_offsets_int, b_ld, diff --git a/src/routines/levelx/xgemmstridedbatched.cpp b/src/routines/levelx/xgemmstridedbatched.cpp index affbceee..6165a396 100644 --- a/src/routines/levelx/xgemmstridedbatched.cpp +++ b/src/routines/levelx/xgemmstridedbatched.cpp @@ -76,7 +76,7 @@ void XgemmStridedBatched<T>::DoGemmStridedBatched(const Layout layout, const Tra } // Selects which version of the batched GEMM to run - const auto do_gemm_direct = true; + const auto do_gemm_direct = Xgemm<T>::UseDirectKernel(m, n, k, db_["XGEMM_MIN_INDIRECT_SIZE"]);; if (do_gemm_direct) { // single generic kernel BatchedGemmDirect(m, n, k, alpha, a_buffer, a_offset, a_ld, a_stride, |