| /petsc/src/mat/impls/baij/seq/ |
| H A D | baijfact3.c | 81 PetscErrorCode MatSeqBAIJSetNumericFactorization_inplace(Mat inA, PetscBool natural) in MatSeqBAIJSetNumericFactorization_inplace() argument
85 switch (inA->rmap->bs) { in MatSeqBAIJSetNumericFactorization_inplace()
87 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_1_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
90 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_2_NaturalOrdering_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
93 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_3_NaturalOrdering_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
96 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_4_NaturalOrdering_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
99 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_5_NaturalOrdering_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
102 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_6_NaturalOrdering_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
105 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_7_NaturalOrdering_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
108 inA->ops->lufactornumeric = MatILUFactorNumeric_SeqBAIJ_N_inplace; in MatSeqBAIJSetNumericFactorization_inplace()
[all …]
|
| H A D | baij.c | 2524 static PetscErrorCode MatILUFactor_SeqBAIJ(Mat inA, IS row, IS col, const MatFactorInfo *info) in MatILUFactor_SeqBAIJ() argument
2526 Mat_SeqBAIJ *a = (Mat_SeqBAIJ *)inA->data; in MatILUFactor_SeqBAIJ()
2536 outA = inA; in MatILUFactor_SeqBAIJ()
2537 inA->factortype = MAT_FACTOR_LU; in MatILUFactor_SeqBAIJ()
2538 PetscCall(PetscFree(inA->solvertype)); in MatILUFactor_SeqBAIJ()
2539 PetscCall(PetscStrallocpy(MATSOLVERPETSC, &inA->solvertype)); in MatILUFactor_SeqBAIJ()
2552 …PetscCall(MatSeqBAIJSetNumericFactorization_inplace(inA, (PetscBool)(row_identity && col_identity)… in MatILUFactor_SeqBAIJ()
2553 if (!a->solve_work) PetscCall(PetscMalloc1(inA->rmap->N + inA->rmap->bs, &a->solve_work)); in MatILUFactor_SeqBAIJ()
2554 PetscCall(MatLUFactorNumeric(outA, inA, info)); in MatILUFactor_SeqBAIJ()
|
| H A D | baij2.c | 3144 PetscErrorCode MatScale_SeqBAIJ(Mat inA, PetscScalar alpha) in MatScale_SeqBAIJ() argument
3146 Mat_SeqBAIJ *a = (Mat_SeqBAIJ *)inA->data; in MatScale_SeqBAIJ()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mmaij.c | 206 static PetscErrorCode MatMPIAIJDiagonalScaleLocalSetUp(Mat inA, Vec scale) in MatMPIAIJDiagonalScaleLocalSetUp() argument
208 Mat_MPIAIJ *ina = (Mat_MPIAIJ *)inA->data; /*access private part of matrix */ in MatMPIAIJDiagonalScaleLocalSetUp()
209 PetscInt i, j, n, nt, cstart, cend, no, *garray = ina->garray, *lindices, bs = inA->rmap->bs; in MatMPIAIJDiagonalScaleLocalSetUp()
213 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPIAIJDiagonalScaleLocalSetUp()
215 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPIAIJDiagonalScaleLocalSetUp()
217 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIAIJDiagonalScaleLocalSetUp()
218 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) { in MatMPIAIJDiagonalScaleLocalSetUp()
220 r_rmapd[i] = inA->rmap->mapping->indices[i] + 1; in MatMPIAIJDiagonalScaleLocalSetUp()
225 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIAIJDiagonalScaleLocalSetUp()
237 PetscCall(PetscCalloc1(inA->cmap->N / bs, &lindices)); in MatMPIAIJDiagonalScaleLocalSetUp()
[all …]
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mmsell.c | 200 static PetscErrorCode MatMPISELLDiagonalScaleLocalSetUp(Mat inA, Vec scale) in MatMPISELLDiagonalScaleLocalSetUp() argument
202 Mat_MPISELL *ina = (Mat_MPISELL *)inA->data; /*access private part of matrix */ in MatMPISELLDiagonalScaleLocalSetUp()
207 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPISELLDiagonalScaleLocalSetUp()
209 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPISELLDiagonalScaleLocalSetUp()
211 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()
212 if (inA->rmap->mapping->indices[i] >= cstart && inA->rmap->mapping->indices[i] < cend) { in MatMPISELLDiagonalScaleLocalSetUp()
214 r_rmapd[i] = inA->rmap->mapping->indices[i] + 1; in MatMPISELLDiagonalScaleLocalSetUp()
219 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPISELLDiagonalScaleLocalSetUp()
224 PetscCall(PetscCalloc1(inA->cmap->N, &lindices)); in MatMPISELLDiagonalScaleLocalSetUp()
226 no = inA->rmap->mapping->n - nt; in MatMPISELLDiagonalScaleLocalSetUp()
[all …]
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mmbaij.c | 199 static PetscErrorCode MatMPIBAIJDiagonalScaleLocalSetUp(Mat inA, Vec scale) in MatMPIBAIJDiagonalScaleLocalSetUp() argument
201 Mat_MPIBAIJ *ina = (Mat_MPIBAIJ *)inA->data; /*access private part of matrix */ in MatMPIBAIJDiagonalScaleLocalSetUp()
203 PetscInt bs = inA->rmap->bs, i, n, nt, j, cstart, cend, no, *garray = ina->garray, *lindices; in MatMPIBAIJDiagonalScaleLocalSetUp()
207 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPIBAIJDiagonalScaleLocalSetUp()
209 PetscCall(PetscCalloc1(inA->rmap->mapping->n, &r_rmapd)); in MatMPIBAIJDiagonalScaleLocalSetUp()
211 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIBAIJDiagonalScaleLocalSetUp()
212 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) { in MatMPIBAIJDiagonalScaleLocalSetUp()
214 r_rmapd[i] = inA->rmap->mapping->indices[i] + 1; in MatMPIBAIJDiagonalScaleLocalSetUp()
219 for (i = 0; i < inA->rmap->mapping->n; i++) { in MatMPIBAIJDiagonalScaleLocalSetUp()
229 no = inA->rmap->mapping->n - nt; in MatMPIBAIJDiagonalScaleLocalSetUp()
[all …]
|
| /petsc/src/mat/tutorials/ |
| H A D | ex8.c | 6 static PetscErrorCode MatScaleUserImpl_SeqAIJ(Mat inA, PetscScalar alpha) in MatScaleUserImpl_SeqAIJ() argument
9 PetscCall(MatScale(inA, alpha)); in MatScaleUserImpl_SeqAIJ()
|
| /petsc/src/ksp/ksp/tests/ |
| H A D | ex21.c | 19 PetscErrorCode Create(MPI_Comm comm, Mat *inA, IS *is0, IS *is1) in Create() argument
26 *inA = 0; in Create()
92 *inA = A; in Create()
|
| /petsc/src/mat/impls/baij/seq/baijmkl/ |
| H A D | baijmkl.c | 356 static PetscErrorCode MatScale_SeqBAIJMKL(Mat inA, PetscScalar alpha) in MatScale_SeqBAIJMKL() argument
359 PetscCall(MatScale_SeqBAIJ(inA, alpha)); in MatScale_SeqBAIJMKL()
360 PetscCall(MatSeqBAIJMKL_create_mkl_handle(inA)); in MatScale_SeqBAIJMKL()
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaij.c | 825 static PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA, IS row, const MatFactorInfo *info) in MatICCFactor_SeqSBAIJ() argument
827 Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ *)inA->data; in MatICCFactor_SeqSBAIJ()
835 …PetscCheck(inA->rmap->bs == 1, PETSC_COMM_SELF, PETSC_ERR_SUP, "Matrix block size %" PetscInt_FMT … in MatICCFactor_SeqSBAIJ()
837 outA = inA; in MatICCFactor_SeqSBAIJ()
838 PetscCall(PetscFree(inA->solvertype)); in MatICCFactor_SeqSBAIJ()
839 PetscCall(PetscStrallocpy(MATSOLVERPETSC, &inA->solvertype)); in MatICCFactor_SeqSBAIJ()
841 inA->factortype = MAT_FACTOR_ICC; in MatICCFactor_SeqSBAIJ()
842 PetscCall(MatSeqSBAIJSetNumericFactorization_inplace(inA, row_identity)); in MatICCFactor_SeqSBAIJ()
854 if (!a->solve_work) PetscCall(PetscMalloc1(inA->rmap->N + inA->rmap->bs, &a->solve_work)); in MatICCFactor_SeqSBAIJ()
856 PetscCall(MatCholeskyFactorNumeric(outA, inA, info)); in MatICCFactor_SeqSBAIJ()
|
| H A D | sbaij2.c | 1261 PetscErrorCode MatScale_SeqSBAIJ(Mat inA, PetscScalar alpha) in MatScale_SeqSBAIJ() argument
1263 Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ *)inA->data; in MatScale_SeqSBAIJ()
|
| /petsc/src/mat/impls/dense/mpi/ |
| H A D | mmdense.c | 400 PETSC_INTERN PetscErrorCode MatScale_MPIDense(Mat inA, PetscScalar alpha) in MatScale_MPIDense() argument
402 Mat_MPIDense *A = (Mat_MPIDense *)inA->data; in MatScale_MPIDense()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 2576 static PetscErrorCode MatILUFactor_SeqAIJ(Mat inA, IS row, IS col, const MatFactorInfo *info) in MatILUFactor_SeqAIJ() argument
2578 Mat_SeqAIJ *a = (Mat_SeqAIJ *)inA->data; in MatILUFactor_SeqAIJ()
2588 outA = inA; in MatILUFactor_SeqAIJ()
2589 PetscCall(PetscFree(inA->solvertype)); in MatILUFactor_SeqAIJ()
2590 PetscCall(PetscStrallocpy(MATSOLVERPETSC, &inA->solvertype)); in MatILUFactor_SeqAIJ()
2607 PetscCall(PetscMalloc1(inA->rmap->n, &a->solve_work)); in MatILUFactor_SeqAIJ()
2611 PetscCall(MatLUFactorNumeric_SeqAIJ_inplace(outA, inA, info)); in MatILUFactor_SeqAIJ()
2613 PetscCall(MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(outA, inA, info)); in MatILUFactor_SeqAIJ()
2619 PetscErrorCode MatScale_SeqAIJ(Mat inA, PetscScalar alpha) in MatScale_SeqAIJ() argument
2621 Mat_SeqAIJ *a = (Mat_SeqAIJ *)inA->data; in MatScale_SeqAIJ()
[all …]
|
| /petsc/src/mat/impls/sell/seq/ |
| H A D | sell.c | 1660 PetscErrorCode MatScale_SeqSELL(Mat inA, PetscScalar alpha) in MatScale_SeqSELL() argument
1662 Mat_SeqSELL *a = (Mat_SeqSELL *)inA->data; in MatScale_SeqSELL()
1672 if (inA->offloadmask != PETSC_OFFLOAD_UNALLOCATED) inA->offloadmask = PETSC_OFFLOAD_CPU; in MatScale_SeqSELL()
|