diff options
Diffstat (limited to 'src/routines/levelx/xgemmstridedbatched.cpp')
-rw-r--r-- | src/routines/levelx/xgemmstridedbatched.cpp | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/routines/levelx/xgemmstridedbatched.cpp b/src/routines/levelx/xgemmstridedbatched.cpp index 5d8945ce..8fb83141 100644 --- a/src/routines/levelx/xgemmstridedbatched.cpp +++ b/src/routines/levelx/xgemmstridedbatched.cpp @@ -78,11 +78,9 @@ void XgemmStridedBatched<T>::DoGemmStridedBatched(const Layout layout, const Tra gemm_kernel_id); // Tests the matrices for validity - for (auto batch = size_t{0}; batch < batch_count; ++batch) { - TestMatrixA(a_one, a_two, a_buffer, a_offset + a_stride * batch, a_ld); - TestMatrixB(b_one, b_two, b_buffer, b_offset + b_stride * batch, b_ld); - TestMatrixC(c_one, c_two, c_buffer, c_offset + c_stride * batch, c_ld); - } + TestStridedBatchedMatrixA(a_one, a_two, a_buffer, a_offset, a_stride, batch_count, a_ld); + TestStridedBatchedMatrixB(b_one, b_two, b_buffer, b_offset, b_stride, batch_count, b_ld); + TestStridedBatchedMatrixC(c_one, c_two, c_buffer, c_offset, c_stride, batch_count, c_ld); // Selects which version of the batched GEMM to run if (do_gemm_direct) { // single generic kernel |