Home
last modified time | relevance | path

Searched refs:lda (Results 1 – 25 of 60) sorted by relevance

123

/petsc/src/mat/tests/
H A Dex256.c9 PetscInt n = 4, lda = 5, i; in main() local
14 PetscCall(PetscOptionsGetInt(NULL, NULL, "-lda", &lda, NULL)); in main()
15 …PetscCheck(lda >= n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "lda %" PetscInt_FMT " < n %" PetscInt_… in main()
20 PetscCall(PetscCalloc2(lda * n, &a, lda * n, &b)); in main()
22 a[i + i * lda] = 1.0 + 2.0 * PETSC_i; in main()
23 if (i > 0) a[i + (i - 1) * lda] = 3.0 - 0.5 * PETSC_i; in main()
24 b[i + i * lda] = 1.0 + 2.0 * PETSC_i; in main()
25 if (i > 0) b[i + (i - 1) * lda] = 3.0 - 0.5 * PETSC_i; in main()
31 PetscCall(MatDenseSetLDA(A, lda)); in main()
37 PetscCall(MatDenseSetLDA(B, lda)); in main()
H A Dex257.c9 PetscInt n = 4, lda = 5, i, k; in main() local
15 PetscCall(PetscOptionsGetInt(NULL, NULL, "-lda", &lda, NULL)); in main()
17 …PetscCheck(lda >= n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "lda %" PetscInt_FMT " < n %" PetscInt_… in main()
20 if (cuda) PetscCall(MatCreateSeqDenseCUDA(PETSC_COMM_SELF, lda, n, NULL, &A)); in main()
23 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, lda, n, NULL, &A)); in main()
29 b[i + i * lda] = 2.0 * (i + 1); in main()
30 if (i > 0) b[i + (i - 1) * lda] = (PetscReal)(k + 1); in main()
H A Dex34.c10 PetscInt i, ia[2] = {0, 2}, ja[2] = {0, 1}, lda = 4; in main() local
24 PetscCall(PetscCalloc1(4 * lda, &data)); in main()
25 for (i = 0; i < 4; ++i) data[lda * i] = i * 1.0; in main()
30 PetscCall(MatDenseSetLDA(B, lda)); in main()
H A Dex16.c9 PetscInt M, N, rstart, rend, lda, i, j; in CheckValues() local
13 PetscCall(MatDenseGetLDA(A, &lda)); in CheckValues()
20 PetscReal w = PetscRealPart(array[ii + jj * lda]); in CheckValues()
H A Dex66.c38 …PetscInt N, n = 64, dim = 1, i, j, nrhs = 11, lda = 0, ldc = 0, ldu = 0, nlr = 7, nt, ntr… in main() local
55 PetscCall(PetscOptionsGetInt(NULL, NULL, "-lda", &lda, NULL)); in main()
98 if (lda) PetscCall(PetscMalloc1(N * (n + lda), &Adata)); in main()
100 PetscCall(MatDenseSetLDA(A, n + lda)); in main()
/petsc/src/ksp/ksp/utils/lmvm/dense/cd_cupm/
H A Dcd_cupm.cxx26 …PetscBool hermitian_transpose, PetscInt N, const PetscScalar A[], PetscInt lda, PetscScalar x[], P… in SolveInPlace() argument
38 …mitian_transpose ? CUPMBLAS_OP_C : CUPMBLAS_OP_N, CUPMBLAS_DIAG_NON_UNIT, n, A_, lda, x_, stride)); in SolveInPlace()
46 …etscInt m, PetscInt oldest, PetscInt next, const PetscScalar A[], PetscInt lda, PetscScalar x[], P… in SolveInPlaceCyclic() argument
68 …dle, CUPMBLAS_FILL_MODE_UPPER, CUPMBLAS_OP_N, CUPMBLAS_DIAG_NON_UNIT, n_new, A_, lda, x_, stride)); in SolveInPlaceCyclic()
69 …sXgemv(handle, CUPMBLAS_OP_N, n_old, n_new, &minus_one, &A_[oldest_index], lda, x_, stride, &sone,… in SolveInPlaceCyclic()
70 …R, CUPMBLAS_OP_N, CUPMBLAS_DIAG_NON_UNIT, n_old, &A_[oldest_index * (lda + 1)], lda, &x_[oldest_in… in SolveInPlaceCyclic()
72 …R, CUPMBLAS_OP_C, CUPMBLAS_DIAG_NON_UNIT, n_old, &A_[oldest_index * (lda + 1)], lda, &x_[oldest_in… in SolveInPlaceCyclic()
73 …sXgemv(handle, CUPMBLAS_OP_C, n_old, n_new, &minus_one, &A_[oldest_index], lda, &x_[oldest_index],… in SolveInPlaceCyclic()
74 …dle, CUPMBLAS_FILL_MODE_UPPER, CUPMBLAS_OP_C, CUPMBLAS_DIAG_NON_UNIT, n_new, A_, lda, x_, stride)); in SolveInPlaceCyclic()
99 …PetscBool hermitian_transpose, PetscInt n, const PetscScalar A[], PetscInt lda, PetscScalar x[], P… in MatUpperTriangularSolveInPlace_CUPM() argument
[all …]
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp177 …etscInt m, PetscInt oldest, PetscInt next, const PetscScalar A[], PetscInt lda, const PetscScalar … in trsv() argument
202 … CUPMBLAS_OP_C : CUPMBLAS_OP_N, CUPMBLAS_DIAG_NON_UNIT, n, &A_[i_oldest * (lda + 1)], lda, y_, 1)); in trsv()
204 …v(handle, CUPMBLAS_FILL_MODE_UPPER, CUPMBLAS_OP_N, CUPMBLAS_DIAG_NON_UNIT, n_new, A_, lda, y_, 1)); in trsv()
205 …mBlasXgemv(handle, CUPMBLAS_OP_N, n_old, n_new, &minus_one, &A_[i_oldest], lda, y_, 1, &sone, &y_[… in trsv()
206 …UPPER, CUPMBLAS_OP_N, CUPMBLAS_DIAG_NON_UNIT, n_old, &A_[i_oldest * (lda + 1)], lda, &y_[i_oldest]… in trsv()
208 …UPPER, CUPMBLAS_OP_C, CUPMBLAS_DIAG_NON_UNIT, n_old, &A_[i_oldest * (lda + 1)], lda, &y_[i_oldest]… in trsv()
209 …mBlasXgemv(handle, CUPMBLAS_OP_C, n_old, n_new, &minus_one, &A_[i_oldest], lda, &y_[i_oldest], 1, … in trsv()
210 …v(handle, CUPMBLAS_FILL_MODE_UPPER, CUPMBLAS_OP_C, CUPMBLAS_DIAG_NON_UNIT, n_new, A_, lda, y_, 1)); in trsv()
220 … oldest, PetscInt next, PetscScalar alpha, const PetscScalar A[], PetscInt lda, const PetscScalar … in hemv() argument
246 …asXhemv(handle, CUPMBLAS_FILL_MODE_UPPER, n, calpha, &A_[i_oldest * (lda + 1)], lda, &x_[i_oldest]… in hemv()
[all …]
H A Dblas_cyclic_cupm.cxx105 …etscInt m, PetscInt oldest, PetscInt next, const PetscScalar A[], PetscInt lda, const PetscScalar … in TRSVCyclic_CUPM_Private() argument
116 …::device::cupm::DeviceType::CUDA>::trsv(dctx, hermitian_transpose, m, oldest, next, A, lda, x, y)); in TRSVCyclic_CUPM_Private()
121 …c::device::cupm::DeviceType::HIP>::trsv(dctx, hermitian_transpose, m, oldest, next, A, lda, x, y)); in TRSVCyclic_CUPM_Private()
130 … oldest, PetscInt next, PetscScalar alpha, const PetscScalar A[], PetscInt lda, const PetscScalar … in HEMVCyclic_CUPM_Private() argument
141 …<::Petsc::device::cupm::DeviceType::CUDA>::hemv(dctx, m, oldest, next, alpha, A, lda, x, beta, y)); in HEMVCyclic_CUPM_Private()
146 …c<::Petsc::device::cupm::DeviceType::HIP>::hemv(dctx, m, oldest, next, alpha, A, lda, x, beta, y)); in HEMVCyclic_CUPM_Private()
155 … oldest, PetscInt next, PetscScalar alpha, const PetscScalar A[], PetscInt lda, const PetscScalar … in GEMVCyclic_CUPM_Private() argument
166 …m::DeviceType::CUDA>::gemv(dctx, hermitian_transpose, m, oldest, next, alpha, A, lda, x, beta, y)); in GEMVCyclic_CUPM_Private()
171 …pm::DeviceType::HIP>::gemv(dctx, hermitian_transpose, m, oldest, next, alpha, A, lda, x, beta, y)); in GEMVCyclic_CUPM_Private()
/petsc/src/mat/utils/
H A Dveccreatematdense.c25 …m, VecType vtype, PetscInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt lda, PetscScalar *data,… in MatCreateDenseFromVecType() argument
50 if (lda > 0) PetscCall(MatDenseSetLDA(*A, lda)); in MatCreateDenseFromVecType()
57 if (lda > 0) PetscCall(MatDenseSetLDA(*A, lda)); in MatCreateDenseFromVecType()
64 if (lda > 0) PetscCall(MatDenseSetLDA(*A, lda)); in MatCreateDenseFromVecType()
/petsc/src/mat/interface/ftn-custom/
H A Dzmatrixf90.c116 PetscInt m, N, lda; in matdensegetarray2d_() local
123 *ierr = MatDenseGetLDA(*mat, &lda); in matdensegetarray2d_()
125 if (m != lda) { // TODO: add F90Array2dLDACreate() in matdensegetarray2d_()
126 …ROR_INITIAL, "Array lda %" PetscInt_FMT " must match number of local rows %" PetscInt_FMT, lda, m); in matdensegetarray2d_()
143 PetscInt m, N, lda; in matdensegetarrayread2d_() local
150 *ierr = MatDenseGetLDA(*mat, &lda); in matdensegetarrayread2d_()
152 if (m != lda) { // TODO: add F90Array2dLDACreate() in matdensegetarrayread2d_()
153 …ROR_INITIAL, "Array lda %" PetscInt_FMT " must match number of local rows %" PetscInt_FMT, lda, m); in matdensegetarrayread2d_()
170 PetscInt m, N, lda; in matdensegetarraywrite2d_() local
177 *ierr = MatDenseGetLDA(*mat, &lda); in matdensegetarraywrite2d_()
[all …]
/petsc/src/tao/leastsquares/impls/pounders/
H A Dgqt.c214 PetscErrorCode gqt(PetscInt n, PetscReal *a, PetscInt lda, PetscReal *b, PetscReal delta, PetscReal… in gqt() argument
224 PetscCall(PetscBLASIntCast(lda, &blaslda)); in gqt()
225 PetscCall(PetscBLASIntCast(lda + 1, &blasldap1)); in gqt()
239 …PetscCallBLAS("BLAScopy", BLAScopy_(&iblas, &a[j + lda * (j + 1)], &blaslda, &a[j + 1 + lda * j], … in gqt()
246 PetscCallBLAS("BLASasum", wa2[j] = BLASasum_(&blasn, &a[0 + lda * j], &blas1)); in gqt()
286 …PetscCallBLAS("BLAScopy", BLAScopy_(&iblas, &a[j + 1 + j * lda], &blas1, &a[j + (j + 1) * lda], &b… in gqt()
288 for (j = 0; j < n; j++) a[j + j * lda] = wa1[j] + par; in gqt()
314 PetscCall(estsv(n, a, lda, &rznorm, z)); in gqt()
367 …llBLAS("BLAScopy", BLAScopy_(&iblas, &a[indef - 1 + 0 * lda], &blaslda, &a[0 + (indef - 1) * lda],… in gqt()
368 a[indef - 1 + (indef - 1) * lda] = wa1[indef - 1] + par; in gqt()
[all …]
/petsc/src/mat/tutorials/
H A Dex19.c13 PetscInt n = 20, lda = PETSC_DECIDE; in main() local
19 PetscCall(PetscOptionsGetInt(NULL, NULL, "-lda", &lda, NULL)); in main()
21 if (lda > 0) lda += n; in main()
29 …tCreateDenseFromVecType(PETSC_COMM_WORLD, vtype, n, n, PETSC_DECIDE, PETSC_DECIDE, lda, NULL, &A)); in main()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp207 auto &lda = MatIMPLCast(m)->lda; in SetPreallocation_() local
216 if (lda <= 0) lda = nrows; in SetPreallocation_()
225 size = lda * ncols; in SetPreallocation_()
254 const auto lda = mimpl->lda; in HostToDevice_() local
258 if (lda > nrows) { in HostToDevice_()
259 …PetscCall(PetscCUPMMemcpy2DAsync(dest, lda, src, lda, nrows, ncols, cupmMemcpyHostToDevice, stream… in HostToDevice_()
261 PetscCall(PetscCUPMMemcpyAsync(dest, src, lda * ncols, cupmMemcpyHostToDevice, stream)); in HostToDevice_()
290 const auto lda = mimpl->lda; in DeviceToHost_() local
294 if (lda > nrows) { in DeviceToHost_()
295 …PetscCall(PetscCUPMMemcpy2DAsync(dest, lda, src, lda, nrows, ncols, cupmMemcpyDeviceToHost, stream… in DeviceToHost_()
[all …]
/petsc/src/ksp/ksp/utils/lmvm/dense/
H A Dcd_utils.c88 …etscInt m, PetscInt oldest, PetscInt next, const PetscScalar A[], PetscInt lda, PetscScalar x[], P… in MatUpperTriangularSolveInPlace_Internal() argument
102 PetscCall(PetscBLASIntCast(lda, &lda_blas)); in MatUpperTriangularSolveInPlace_Internal()
107 PetscCall(MatUpperTriangularSolveInPlace_CUPM(hermitian_transpose, N, A, lda, x, 1)); in MatUpperTriangularSolveInPlace_Internal()
118 PetscCall(PetscBLASIntCast(lda, &lda_blas)); in MatUpperTriangularSolveInPlace_Internal()
122 …trsv", BLAStrsv_("U", "N", "NotUnitTriangular", &n_old, &A[oldest_index * (lda + 1)], &lda_blas, &… in MatUpperTriangularSolveInPlace_Internal()
125 …trsv", BLAStrsv_("U", "C", "NotUnitTriangular", &n_old, &A[oldest_index * (lda + 1)], &lda_blas, &… in MatUpperTriangularSolveInPlace_Internal()
133 …atUpperTriangularSolveInPlaceCyclic_CUPM(hermitian_transpose, m, oldest, next, A, lda, x, stride)); in MatUpperTriangularSolveInPlace_Internal()
148 PetscInt lda; in MatUpperTriangularSolveInPlace() local
165 PetscCall(MatDenseGetLDA(Amat, &lda)); in MatUpperTriangularSolveInPlace()
166 …egy, memtype_x, hermitian_transpose, m, oldest_update(m, num_updates), num_updates, A, lda, x, 1)); in MatUpperTriangularSolveInPlace()
/petsc/src/mat/impls/dense/seq/
H A Ddense.c24 for (j = k; j < n; j++) v[j * mat->lda + k] = v[k * mat->lda + j]; in MatSeqDenseSymmetrize_Private()
28 for (j = k; j < n; j++) v[j * mat->lda + k] = PetscConj(v[k * mat->lda + j]); in MatSeqDenseSymmetrize_Private()
50 …PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&n, mat->v, &mat->lda, mat->pivots, mat->fwork, &mat->lf… in MatSeqDenseInvertFactors_Private()
56 PetscCallBLAS("LAPACKpotri", LAPACKpotri_("L", &n, mat->v, &mat->lda, &info)); in MatSeqDenseInvertFactors_Private()
64 …PetscCallBLAS("LAPACKhetri", LAPACKhetri_("L", &n, mat->v, &mat->lda, mat->pivots, mat->fwork, &in… in MatSeqDenseInvertFactors_Private()
72 …PetscCallBLAS("LAPACKsytri", LAPACKsytri_("L", &n, mat->v, &mat->lda, mat->pivots, mat->fwork, &in… in MatSeqDenseInvertFactors_Private()
94 PetscInt m = l->lda, n = A->cmap->n, r = A->rmap->n, i, j; in MatZeroRowsColumns_SeqDense()
170 for (i = 0; i < n; i++) PetscCall(PetscArrayzero(b->v + i * b->lda, m)); in MatConvert_SeqAIJ_SeqDense()
176 b->v[*aj * b->lda + i] = *av; in MatConvert_SeqAIJ_SeqDense()
213 aa += a->lda; in MatConvert_SeqDense_SeqAIJ()
[all …]
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/
H A Dblas_cyclic.c248 …etscInt m, PetscInt oldest, PetscInt next, const PetscScalar A[], PetscInt lda, const PetscScalar … in TRSVCyclic_Private() argument
255 PetscCall(PetscBLASIntCast(lda, &blda)); in TRSVCyclic_Private()
264 …PetscCallBLAS("BLAStrsv", BLAStrsv_("U", trans, "N", &bn, &A[i_oldest * (lda + 1)], &blda, &y[i_ol… in TRSVCyclic_Private()
274 …if (bn > 0) PetscCallBLAS("BLAStrsv", BLAStrsv_("U", "N", "N", &bn, &A[i_oldest * (lda + 1)], &bld… in TRSVCyclic_Private()
276 …if (bn > 0) PetscCallBLAS("BLAStrsv", BLAStrsv_("U", "C", "N", &bn, &A[i_oldest * (lda + 1)], &bld… in TRSVCyclic_Private()
289 PetscInt m, m_local, lda; in MatSeqDenseTRSVCyclic() local
307 PetscCall(MatDenseGetLDA(A, &lda)); in MatSeqDenseTRSVCyclic()
318 …al == m) PetscCall(TRSVCyclic_CUPM_Private(hermitian_transpose, m, oldest, next, A_, lda, x_, y_)); in MatSeqDenseTRSVCyclic()
334 PetscCall(TRSVCyclic_Private(hermitian_transpose, m, oldest, next, A_, lda, x_, y_)); in MatSeqDenseTRSVCyclic()
343 … oldest, PetscInt next, PetscScalar alpha, const PetscScalar A[], PetscInt lda, const PetscScalar … in HEMVCyclic_Private() argument
[all …]
/petsc/src/ksp/ksp/tests/
H A Dex61.c14 PetscInt size = 8, lda = 10, i, j; in main() local
19 PetscCall(PetscMalloc1(lda * size, &b)); in main()
21 for (j = 0; j < size; j++) b[i + j * lda] = rand(); in main()
30 for (j = 0; j < size; j++) a[i + j * size] = b[i + j * lda]; in main()
40 PetscCall(MatDenseSetLDA(B, lda)); in main()
/petsc/src/mat/impls/aij/seq/spqr/
H A Daijspqr.c159 PetscInt lda; in MatMatSolve_SPQR() local
165 PetscCall(MatDenseGetLDA(X, &lda)); in MatMatSolve_SPQR()
166 if ((size_t)lda == Y_handle->d) { in MatMatSolve_SPQR()
167 PetscCall(PetscArraycpy(v, (PetscScalar *)Y_handle->x, lda * Y_handle->ncol)); in MatMatSolve_SPQR()
169 …for (size_t j = 0; j < Y_handle->ncol; j++) PetscCall(PetscArraycpy(&v[j * lda], &(((PetscScalar *… in MatMatSolve_SPQR()
217 PetscInt lda; in MatMatSolveTranspose_SPQR() local
223 PetscCall(MatDenseGetLDA(X, &lda)); in MatMatSolveTranspose_SPQR()
224 if ((size_t)lda == Y_handle->d) { in MatMatSolveTranspose_SPQR()
225 PetscCall(PetscArraycpy(v, (PetscScalar *)Y_handle->x, lda * Y_handle->ncol)); in MatMatSolveTranspose_SPQR()
227 …for (size_t j = 0; j < Y_handle->ncol; j++) PetscCall(PetscArraycpy(&v[j * lda], &(((PetscScalar *… in MatMatSolveTranspose_SPQR()
/petsc/src/vec/vec/impls/seq/
H A Ddvec2.c521 PetscInt64 lda = 0; in VecMultiDot_Seq_GEMV() local
541 lda = ynext - yarray; in VecMultiDot_Seq_GEMV()
542 if (lda < 0 || lda > PETSC_BLAS_INT_MAX || lda - n > 64) stop = PETSC_TRUE; in VecMultiDot_Seq_GEMV()
543 } else if (lda * (j - i) != ynext - yarray) { // not in the same stride? if so, stop here in VecMultiDot_Seq_GEMV()
554 …PetscBLASInt ione = 1, lda2 = (PetscBLASInt)lda; // the cast is safe since we've screened out thos… in VecMultiDot_Seq_GEMV()
708 PetscInt64 lda = 0; in VecMAXPY_Seq_GEMV() local
727 lda = xnext - xarray; in VecMAXPY_Seq_GEMV()
728 if (lda < 0 || lda > PETSC_BLAS_INT_MAX || lda - n > 64) stop = PETSC_TRUE; in VecMAXPY_Seq_GEMV()
729 } else if (lda * (j - i) != xnext - xarray) { // not in the same stride? if so, stop here in VecMAXPY_Seq_GEMV()
739 …PetscBLASInt incx = 1, incy = 1, lda2 = (PetscBLASInt)lda; // the cast is safe since we've screene… in VecMAXPY_Seq_GEMV()
/petsc/include/petsc/private/
H A Dcupmsolverinterface.hpp109 … cupmBlasInt_t n, cupmBlasInt_t nrhs, const cupmScalar_t *A, cupmBlasInt_t lda, cupmScalar_t *B, c… in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
111 return cupmSolverXpotrs_p(handle, uplo, n, nrhs, A, lda, B, ldb, dev_info); in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
128 …t handle, cupmBlasInt_t m, cupmBlasInt_t n, cupmScalar_t *A, cupmBlasInt_t lda, cupmScalar_t *work… in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
130 return cupmSolverXgetrf_p(handle, m, n, A, lda, work, dev_ipiv, dev_info); in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
152 …_t op, cupmBlasInt_t n, cupmBlasInt_t nrhs, cupmScalar_t *A, cupmBlasInt_t lda, cupmBlasInt_t *dev… in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
154 return cupmSolverXgetrs_p(handle, op, n, nrhs, A, lda, dev_ipiv, B, ldb, dev_info); in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
H A Dmatdensecupmimpl.h386 …MakeDiagonalIterator(T *data, PetscInt rstart, PetscInt rend, PetscInt cols, PetscInt lda) noexcept in MakeDiagonalIterator() argument
389 const std::size_t begin = rstart * lda; in MakeDiagonalIterator()
390 const std::size_t end = rend2 - rstart + rend2 * lda; in MakeDiagonalIterator()
393 return {dptr + begin, dptr + end, lda + 1}; in MakeDiagonalIterator()
412 PetscInt lda; in DiagonalUnaryTransform() local
414 PetscCall(MatDenseGetLDA(A, &lda)); in DiagonalUnaryTransform()
417 auto diagonal = detail::MakeDiagonalIterator(da.data(), rstart, rend, gcols, lda); in DiagonalUnaryTransform()
452 PetscInt lda; in GetDiagonal() local
458 PetscCall(MatDenseGetLDA(A, &lda)); in GetDiagonal()
462 auto diagonal = detail::MakeDiagonalIterator(da.data(), rstart, rend, gcols, lda); in GetDiagonal()
H A Dpetscscalapack.h58 …st char *scope, const char *top, PetscBLASInt m, PetscBLASInt n, PetscScalar *A, PetscBLASInt lda);
59 …st char *top, PetscBLASInt m, PetscBLASInt n, PetscScalar *A, PetscBLASInt lda, PetscBLASInt rsrc,…
60 …st char *top, PetscBLASInt m, PetscBLASInt n, PetscScalar *A, PetscBLASInt lda, PetscBLASInt rsrc,…
/petsc/src/ksp/ksp/utils/lmvm/
H A Dlmproducts.c171 PetscInt lda; in LMProductsUpdate_Internal() local
176 lda = 0; in LMProductsUpdate_Internal()
183 PetscCall(MatDenseGetLDA(dots->full, &lda)); in LMProductsUpdate_Internal()
220 if (dots->m_local) dot = values[(j % dots->m) * lda + (i % dots->m)]; in LMProductsUpdate_Internal()
359 PetscInt lda; in LMProductsMakeHermitian() local
361 PetscCall(MatDenseGetLDA(local, &lda)); in LMProductsMakeHermitian()
366 a[j + j * lda] = PetscRealPart(a[j + j * lda]); in LMProductsMakeHermitian()
370 a[i + j * lda] = PetscConj(a[j + i * lda]); in LMProductsMakeHermitian()
/petsc/src/mat/impls/dense/mpi/
H A Dmpidense.c59 PetscInt j, lda, rstart = A->rmap->rstart, rend = A->rmap->rend, rend2; in MatShift_MPIDense() local
64 PetscCall(MatDenseGetLDA(mat->A, &lda)); in MatShift_MPIDense()
67 for (j = rstart; j < rend2; j++) v[j - rstart + j * lda] += alpha; in MatShift_MPIDense()
186 static PetscErrorCode MatDenseGetLDA_MPIDense(Mat A, PetscInt *lda) in MatDenseGetLDA_MPIDense() argument
191 PetscCall(MatDenseGetLDA(a->A, lda)); in MatDenseGetLDA_MPIDense()
195 static PetscErrorCode MatDenseSetLDA_MPIDense(Mat A, PetscInt lda) in MatDenseSetLDA_MPIDense() argument
218 PetscCall(MatDenseSetLDA(a->A, lda)); in MatDenseSetLDA_MPIDense()
288 PetscInt lda, i, j, rstart, rend, nrows, ncols, Ncols, nlrows, nlcols; in MatCreateSubMatrix_MPIDense() local
332 PetscCall(MatDenseGetLDA(mat->A, &lda)); in MatCreateSubMatrix_MPIDense()
334 const PetscScalar *av = v + lda * icol[i]; in MatCreateSubMatrix_MPIDense()
[all …]
/petsc/src/mat/impls/h2opus/cuda/
H A Dmath2opusutils.cu20 PetscInt lda; in MatDenseGetH2OpusStridedSF() local
22 PetscCall(MatDenseGetLDA(A, &lda)); in MatDenseGetH2OpusStridedSF()
23 PetscCall(PetscSFCreateStridedSF(h2sf, A->cmap->N, lda, PETSC_DECIDE, &asf)); in MatDenseGetH2OpusStridedSF()

123