Skip to content

Commit db5f26f

Browse files
committed
fix following parameter specification
1 parent 03552e2 commit db5f26f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/linalg/impl_linalg.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -588,8 +588,8 @@ pub fn general_mat_vec_mul<A, S1, S2, S3>(alpha: A,
588588
{
589589
let a_trans = CblasNoTrans;
590590
let a_stride = match layout {
591-
CBLAS_LAYOUT::CblasRowMajor => a.strides()[0] as blas_index,
592-
CBLAS_LAYOUT::CblasColMajor => a.strides()[1] as blas_index,
591+
CBLAS_LAYOUT::CblasRowMajor => a.strides()[0].max(k as isize) as blas_index,
592+
CBLAS_LAYOUT::CblasColMajor => a.strides()[1].max(m as isize) as blas_index,
593593
};
594594

595595
let x_stride = x.strides()[0] as blas_index;

0 commit comments

Comments
 (0)