From afcf5dc6ebc287b392edcb6bd3ac48966ba98e3c Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Sat, 5 Jan 2019 10:56:35 +0100 Subject: Added a check to prevent the stride of matrix C being set to 0 for the strided-batched-GEMM routine --- src/routines/levelx/xgemmstridedbatched.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/routines/levelx') 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::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::UseDirectKernel(m, n, k, db_["XGEMM_MIN_INDIRECT_SIZE"]); const auto gemm_kernel_id = (do_gemm_direct) ? 0 : db_["GEMMK"]; -- cgit v1.2.3