func checkGbtrs(ind *linalg.IndexOpts, A, B matrix.Matrix, ipiv []int32) error { if ind.Kl < 0 { return errors.New("Gbtrs: invalid kl") } if ind.N < 0 { ind.N = A.Rows() } if ind.Nrhs < 0 { ind.Nrhs = A.Cols() } if ind.N == 0 || ind.Nrhs == 0 { return nil } if ind.Ku < 0 { ind.Ku = A.Rows() - 2*ind.Kl - 1 } if ind.Ku < 0 { return errors.New("Gbtrs: invalid ku") } if ind.LDa == 0 { ind.LDa = max(1, A.Rows()) } if ind.LDa < 2*ind.Kl+ind.Ku+1 { return errors.New("Gbtrs: lda") } if ind.OffsetA < 0 { return errors.New("Gbtrs: offsetA") } sizeA := A.NumElements() if sizeA < ind.OffsetA+(ind.N-1)*ind.LDa+2*ind.Kl+ind.Ku+1 { return errors.New("Gbtrs: sizeA") } if ind.LDb == 0 { ind.LDb = max(1, B.Rows()) } if ind.OffsetB < 0 { return errors.New("Gbtrs: offsetB") } sizeB := B.NumElements() if sizeB < ind.OffsetB+(ind.Nrhs-1)*ind.LDb+ind.N { return errors.New("Gbtrs: sizeB") } if ipiv != nil && len(ipiv) < ind.N { return errors.New("Gbtrs: size ipiv") } return nil }