diff options
| -rw-r--r-- | src/routines/levelx/xgemmstridedbatched.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/routines/levelx/xgemmstridedbatched.cpp b/src/routines/levelx/xgemmstridedbatched.cpp index d9e3ebba..5d8945ce 100644 --- a/src/routines/levelx/xgemmstridedbatched.cpp +++ b/src/routines/levelx/xgemmstridedbatched.cpp @@ -62,6 +62,9 @@ void XgemmStridedBatched<T>::DoGemmStridedBatched(const Layout layout, const Tra throw BLASError(StatusCode::kInvalidBatchCount); } + // Makes sure the strides are valid + if (c_stride == 0) { throw BLASError(StatusCode::kInvalidDimension); } + // Two methods to choose from, select which one to run const auto do_gemm_direct = Xgemm<T>::UseDirectKernel(m, n, k, db_["XGEMM_MIN_INDIRECT_SIZE"]); const auto gemm_kernel_id = (do_gemm_direct) ? 0 : db_["GEMMK"]; |