diff options
| author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-18 19:41:02 +0100 |
|---|---|---|
| committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-18 19:41:02 +0100 |
| commit | bc54411d193b700ef5e0e1ad08c6311597bd433d (patch) | |
| tree | dd85d23aa34db10083a5a9acd40ad4b8a0a655b0 /src/routines/levelx/xgemmbatched.cpp | |
| parent | 0e5eaa6eb9d3049da4d04ebd22d413883c036142 (diff) | |
Made the batched routines also chose direct/indirect kernel like the main GEMM routine
Diffstat (limited to 'src/routines/levelx/xgemmbatched.cpp')
| -rw-r--r-- | src/routines/levelx/xgemmbatched.cpp | 2 |
1 files changed, 1 insertions, 1 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, |