diff --git a/cgo/lapack.go b/cgo/lapack.go index 024173e..a876aec 100644 --- a/cgo/lapack.go +++ b/cgo/lapack.go @@ -234,7 +234,7 @@ func (impl Implementation) Dgels(trans blas.Transpose, m, n, nrhs int, a []float return true } checkMatrix(m, n, a, lda) - checkMatrix(mn, nrhs, b, ldb) + checkMatrix(max(m, n), nrhs, b, ldb) if len(work) < lwork { panic(shortWork) } diff --git a/native/dgels.go b/native/dgels.go index 47bfa68..a06c7ed 100644 --- a/native/dgels.go +++ b/native/dgels.go @@ -42,7 +42,7 @@ func (impl Implementation) Dgels(trans blas.Transpose, m, n, nrhs int, a []float notran := trans == blas.NoTrans checkMatrix(m, n, a, lda) mn := min(m, n) - checkMatrix(mn, nrhs, b, ldb) + checkMatrix(max(m, n), nrhs, b, ldb) // Find optimal block size. tpsd := true