| /petsc/src/ksp/ksp/tests/ ! |
| H A D | ex57.c | 11 Mat subA[9]; in main() local
24 …MatCreateFromOptions(PETSC_COMM_WORLD, NULL, 1, 5, 5, PETSC_DETERMINE, PETSC_DETERMINE, &subA[0])); in main()
25 PetscCall(MatGetOwnershipRange(subA[0], &mstart, &mend)); in main()
29 PetscCall(MatSetValues(subA[0], 1, &row, 1, &col, &val, INSERT_VALUES)); in main()
31 PetscCall(MatAssemblyBegin(subA[0], MAT_FINAL_ASSEMBLY)); in main()
32 PetscCall(MatAssemblyEnd(subA[0], MAT_FINAL_ASSEMBLY)); in main()
34 …MatCreateFromOptions(PETSC_COMM_WORLD, NULL, 1, 5, 3, PETSC_DETERMINE, PETSC_DETERMINE, &subA[1])); in main()
35 PetscCall(MatGetOwnershipRange(subA[1], &mstart, &mend)); in main()
39 PetscCall(MatSetValues(subA[1], 1, &row, 1, &col, &val, INSERT_VALUES)); in main()
41 PetscCall(MatAssemblyBegin(subA[1], MAT_FINAL_ASSEMBLY)); in main()
[all …]
|
| H A D | ex37.c | 14 Mat A, subA; in main() local
96 PetscCall(MatCreate(subcomm, &subA)); in main()
97 PetscCall(MatSetSizes(subA, PETSC_DECIDE, PETSC_DECIDE, 10, 10)); in main()
98 PetscCall(MatSetFromOptions(subA)); in main()
99 PetscCall(MatSetUp(subA)); in main()
100 PetscCall(MatAssemblyBegin(subA, MAT_FINAL_ASSEMBLY)); in main()
101 PetscCall(MatAssemblyEnd(subA, MAT_FINAL_ASSEMBLY)); in main()
102 PetscCall(MatZeroEntries(subA)); in main()
105 PetscCall(MatCreateVecs(subA, &subx, &subb)); in main()
107 PetscCall(MatMult(subA, subx, subb)); in main()
[all …]
|
| /petsc/src/snes/tutorials/ ! |
| H A D | ex70.c | 58 Mat subA[4]; /* the four blocks */ member
260 PetscCall(MatCreate(PETSC_COMM_WORLD, &s->subA[0])); in StokesSetupMatBlock00()
261 PetscCall(MatSetOptionsPrefix(s->subA[0], "a00_")); in StokesSetupMatBlock00()
262 …PetscCall(MatSetSizes(s->subA[0], PETSC_DECIDE, PETSC_DECIDE, 2 * s->nx * s->ny, 2 * s->nx * s->ny… in StokesSetupMatBlock00()
263 PetscCall(MatSetType(s->subA[0], MATMPIAIJ)); in StokesSetupMatBlock00()
264 PetscCall(MatMPIAIJSetPreallocation(s->subA[0], 5, NULL, 5, NULL)); in StokesSetupMatBlock00()
265 PetscCall(MatGetOwnershipRange(s->subA[0], &start, &end)); in StokesSetupMatBlock00()
276 PetscCall(MatSetValues(s->subA[0], 1, &row, sz, cols, vals, INSERT_VALUES)); in StokesSetupMatBlock00()
278 PetscCall(MatAssemblyBegin(s->subA[0], MAT_FINAL_ASSEMBLY)); in StokesSetupMatBlock00()
279 PetscCall(MatAssemblyEnd(s->subA[0], MAT_FINAL_ASSEMBLY)); in StokesSetupMatBlock00()
[all …]
|
| /petsc/src/ksp/pc/impls/hpddm/ ! |
| H A D | pchpddm.cxx | 1770 … *sub, A, P, N, C = nullptr, uaux = nullptr, weighted, subA[2], S; in PCSetUp_HPDDM() local
2522 *subA = A0; in PCSetUp_HPDDM()
2523 sub = subA; in PCSetUp_HPDDM()
2610 PetscCall(KSPGetOperators(ksp[0], subA, subA + 1)); in PCSetUp_HPDDM()
2611 PetscCall(PetscObjectReference((PetscObject)subA[0])); in PCSetUp_HPDDM()
2612 PetscCall(MatDuplicate(subA[1], MAT_SHARE_NONZERO_PATTERN, &D)); in PCSetUp_HPDDM()
2613 PetscCall(MatGetOptionsPrefix(subA[1], &matpre)); in PCSetUp_HPDDM()
2664 PetscCall(KSPGetOperators(ksp[0], subA, subA + 1)); in PCSetUp_HPDDM()
2666 if (n != subA[0]->rmap->n || n != subA[0]->cmap->n) { in PCSetUp_HPDDM()
2669 …PetscCheck(n == subA[0]->rmap->n && n == subA[0]->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, … in PCSetUp_HPDDM()
[all …]
|
| /petsc/src/mat/interface/ ! |
| H A D | matrix.c | 7976 PetscErrorCode MatSelectVariableBlockSizes(Mat subA, Mat A, IS isrow) in MatSelectVariableBlockSizes() argument
7996 …I(MPIU_Allreduce(MPI_IN_PLACE, &flg, 1, MPI_C_BOOL, MPI_LAND, PetscObjectComm((PetscObject)subA))); in MatSelectVariableBlockSizes()
8017 subA->nblocks = Nb; in MatSelectVariableBlockSizes()
8018 PetscCall(PetscFree(subA->bsizes)); in MatSelectVariableBlockSizes()
8019 PetscCall(PetscMalloc1(subA->nblocks, &subA->bsizes)); in MatSelectVariableBlockSizes()
8022 if (sb < subA->nblocks) subA->bsizes[sb] = 0; in MatSelectVariableBlockSizes()
8028 ++subA->bsizes[sb]; in MatSelectVariableBlockSizes()
8034 if (sb < subA->nblocks && subA->bsizes[sb]) ++sb; in MatSelectVariableBlockSizes()
8036 …etscCheck(sb == subA->nblocks, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of blocks %" Petsc… in MatSelectVariableBlockSizes()
8038 PetscCall(MatGetLocalSize(subA, &nlocal, NULL)); in MatSelectVariableBlockSizes()
[all …]
|