| /petsc/src/ksp/pc/tutorials/output/ |
| H A D | ex3_1.out | 1 [0] A Istart,Iend: 0 100; nloc 100
2 [1] A Istart,Iend: 100 200; nloc 100
3 [2] A Istart,Iend: 200 1200; nloc 1000
4 [3] A Istart,Iend: 1200 2200; nloc 1000
5 [4] A Istart,Iend: 2200 3200; nloc 1000
6 [5] A Istart,Iend: 3200 4200; nloc 1000
7 [6] A Istart,Iend: 4200 5200; nloc 1000
8 [7] A Istart,Iend: 5200 6200; nloc 1000
|
| H A D | ex3_2.out | 1 [0] A Istart,Iend: 0 100; nloc 100
2 [1] A Istart,Iend: 100 200; nloc 100
3 [2] A Istart,Iend: 200 1200; nloc 1000
4 [3] A Istart,Iend: 1200 2200; nloc 1000
5 [4] A Istart,Iend: 2200 3200; nloc 1000
6 [5] A Istart,Iend: 3200 4200; nloc 1000
7 [6] A Istart,Iend: 4200 5200; nloc 1000
8 [7] A Istart,Iend: 5200 6200; nloc 1000
|
| /petsc/src/ksp/ksp/impls/gmres/agmres/ |
| H A D | agmresorthog.c | 124 PetscInt nloc, d, i, j, k; in KSPAGMRESRoddec() local
138 PetscCall(VecGetLocalSize(VEC_V(0), &nloc)); in KSPAGMRESRoddec()
139 PetscCall(PetscBLASIntCast(nloc, &bnloc)); in KSPAGMRESRoddec()
140 …PetscCheck(nvec <= nloc, PetscObjectComm((PetscObject)ksp), PETSC_ERR_ARG_WRONG, "In QR factorizat… in KSPAGMRESRoddec()
145 PetscCallBLAS("BLAScopy", BLAScopy_(&bnloc, col, &pas, &Qloc[j * nloc], &pas)); in KSPAGMRESRoddec()
150 PetscCall(PetscBLASIntCast(nloc - j, &len)); in KSPAGMRESRoddec()
151 Ajj = Qloc[j * nloc + j]; in KSPAGMRESRoddec()
152 PetscCallBLAS("BLASnrm2", rho = -PetscSign(Ajj) * BLASnrm2_(&len, &Qloc[j * nloc + j], &pas)); in KSPAGMRESRoddec()
158 PetscCallBLAS("BLASscal", BLASscal_(&len, &val, &Qloc[j * nloc + j + 1], &pas)); in KSPAGMRESRoddec()
159 Qloc[j * nloc + j] = 1.0; in KSPAGMRESRoddec()
[all …]
|
| /petsc/src/ksp/pc/impls/asm/ftn-custom/ |
| H A D | zasmf.c | 34 PetscInt nloc; in pcasmgetlocalsubmatrices_() local
39 *ierr = PCASMGetLocalSubmatrices(*pc, &nloc, &tmat); in pcasmgetlocalsubmatrices_()
40 if (n) *n = nloc; in pcasmgetlocalsubmatrices_()
41 …if (mat) *ierr = F90Array1dCreate(tmat, MPIU_FORTRANADDR, 1, nloc, mat PETSC_F90_2PTR_PARAM(ptrd)); in pcasmgetlocalsubmatrices_()
46 PetscInt nloc; in pcasmgetlocalsubdomains_() local
52 *ierr = PCASMGetLocalSubdomains(*pc, &nloc, &tis, &tis_local); in pcasmgetlocalsubdomains_()
54 if (n) *n = nloc; in pcasmgetlocalsubdomains_()
55 if (is) *ierr = F90Array1dCreate(tis, MPIU_FORTRANADDR, 1, nloc, is PETSC_F90_2PTR_PARAM(ptrd1)); in pcasmgetlocalsubdomains_()
57 …if (is_local) *ierr = F90Array1dCreate(tis_local, MPIU_FORTRANADDR, 1, nloc, is_local PETSC_F90_2P… in pcasmgetlocalsubdomains_()
92 PetscInt nloc, flocal; in pcasmgetsubksp_() local
[all …]
|
| /petsc/src/ksp/pc/impls/fieldsplit/ftn-custom/ |
| H A D | zfieldsplitf.c | 19 PetscInt nloc; in pcfieldsplitschurgetsubksp_() local
21 *ierr = PCFieldSplitSchurGetSubKSP(*pc, &nloc, &tksp); in pcfieldsplitschurgetsubksp_()
23 if (n_local) *n_local = nloc; in pcfieldsplitschurgetsubksp_()
24 *ierr = F90Array1dCreate(tksp, MPIU_FORTRANADDR, 1, nloc, ksp PETSC_F90_2PTR_PARAM(ptrd)); in pcfieldsplitschurgetsubksp_()
30 PetscInt nloc; in pcfieldsplitgetsubksp_() local
32 *ierr = PCFieldSplitGetSubKSP(*pc, &nloc, &tksp); in pcfieldsplitgetsubksp_()
34 if (n_local) *n_local = nloc; in pcfieldsplitgetsubksp_()
35 *ierr = F90Array1dCreate(tksp, MPIU_FORTRANADDR, 1, nloc, ksp PETSC_F90_2PTR_PARAM(ptrd)); in pcfieldsplitgetsubksp_()
|
| /petsc/src/dm/impls/moab/ |
| H A D | dmmbmat.cxx | 26 nlsiz = (tmp ? dmmoab->nloc : dmmoab->nloc * dmmoab->numFields); in DMCreateMatrix_Moab()
36 …PetscCall(MatSetSizes(A, dmmoab->nloc * dmmoab->numFields, dmmoab->nloc * dmmoab->numFields, PETSC… in DMCreateMatrix_Moab()
65 PetscInt i, f, nloc, vpere, bs, n_nnz, n_onz, ivtx = 0; in DMMoab_Compute_NNZ_From_Connectivity() local
76 nloc = dmmoab->nloc; in DMMoab_Compute_NNZ_From_Connectivity()
79 nlsiz = (isinterlaced ? nloc : nloc * nfields); in DMMoab_Compute_NNZ_From_Connectivity()
125 …nnz[f * nloc + ivtx] = n_nnz; /* leave out self to avoid repeats -> node shared by multip… in DMMoab_Compute_NNZ_From_Connectivity()
126 if (onz) onz[f * nloc + ivtx] = n_onz; /* add ghost non-owned nodes */ in DMMoab_Compute_NNZ_From_Connectivity()
139 for (ivtx = 0; ivtx < nloc; ivtx++) { in DMMoab_Compute_NNZ_From_Connectivity()
147 else nnz[ibs * nloc + ivtx] *= inbsize; in DMMoab_Compute_NNZ_From_Connectivity()
155 else onz[ibs * nloc + ivtx] *= iobsize; in DMMoab_Compute_NNZ_From_Connectivity()
[all …]
|
| H A D | dmmbvec.cxx | 170 PetscCall(PetscMalloc1((dmmoab->nloc + dmmoab->nghost) * dmmoab->numFields, varray)); in DMMoabVecGetArray()
175 …nloc + dmmoab->nghost, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and … in DMMoabVecGetArray()
246 …nloc + dmmoab->nghost, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and … in DMMoabVecRestoreArray()
329 PetscCall(PetscMalloc1((dmmoab->nloc + dmmoab->nghost) * dmmoab->numFields, varray)); in DMMoabVecGetArrayRead()
334 …nloc + dmmoab->nghost, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mismatch between local vertices and … in DMMoabVecGetArrayRead()
479 …->bs, dmmoab->numFields * dmmoab->nloc, dmmoab->numFields * dmmoab->n, dmmoab->nghost, &dmmoab->gs… in DMCreateVector_Moab_Private()
483 PetscCall(VecSetSizes(*vec, dmmoab->numFields * dmmoab->nloc, PETSC_DECIDE)); in DMCreateVector_Moab_Private()
504 …->bs, dmmoab->numFields * dmmoab->nloc, dmmoab->numFields * dmmoab->n, dmmoab->nghost, &dmmoab->gs… in DMCreateVector_Moab_Private()
|
| H A D | dmmbfield.cxx | 49 PetscCall(PetscMalloc1(dmmoab->nloc, &farray)); in DMMoabSetFieldVector()
99 PetscCall(PetscMalloc1(dmmoab->nloc, &farray)); in DMMoabSetGlobalFieldVector()
108 …for (i = 0; i < dmmoab->nloc; i++) farray[i] = (dmmoab->bs == 1 ? rarray[ifield * dmmoab->nloc + i… in DMMoabSetGlobalFieldVector()
116 PetscCall(PetscMalloc1(dmmoab->nloc * dmmoab->numFields, &varray)); in DMMoabSetGlobalFieldVector()
127 …for (i = 0; i < dmmoab->nloc; i++) farray[i] = (dmmoab->bs == 1 ? varray[ifield * dmmoab->nloc + i… in DMMoabSetGlobalFieldVector()
|
| /petsc/src/ksp/pc/tutorials/ |
| H A D | ex3.c | 22 PetscInt i, j, Ii, J, Istart, Iend, m, n = 7, its, nloc, matdistribute = 0; in main() local
43 nloc = (rank + 1) * n; in main()
45 nloc = m * n; in main()
46 for (i = 0; i < size - 1; i++) nloc -= (i + 1) * n; in main()
51 nloc = n; in main()
53 nloc = 10 * n; /* 10x larger load */ in main()
59 PetscCall(MatSetSizes(A, nloc, nloc, PETSC_DECIDE, PETSC_DECIDE)); in main()
66 nloc = Iend - Istart; in main()
67 …t,Iend: %" PetscInt_FMT " %" PetscInt_FMT "; nloc %" PetscInt_FMT "\n", rank, Istart, Iend, nloc)); in main()
104 PetscCall(VecSetSizes(u, nloc, PETSC_DECIDE)); in main()
|
| /petsc/src/mat/tests/ |
| H A D | ex69.c | 45 PetscInt k = 6, l = 0, i, Istart, Iend, nloc, bs, test = 1; in main() local
85 PetscCall(VecGetLocalSize(t, &nloc)); in main()
89 PetscCall(VecSetSizes(v, k * nloc, k * n)); in main()
99 …PetscCall(MatCreateDenseCUDA(PetscObjectComm((PetscObject)v), nloc, PETSC_DECIDE, n, k - l, vv, &B… in main()
101 …PetscCall(MatDenseCUDAPlaceArray(B, vv + l * nloc)); … in main()
105 …PetscCall(MatCreateDenseCUDA(PetscObjectComm((PetscObject)v), nloc, PETSC_DECIDE, n, k - l, NULL, … in main()
106 PetscCall(MatDenseCUDAPlaceArray(B, vv + l * nloc)); /* set the actual pointer */ in main()
110 …ll(MatCreateDenseCUDA(PetscObjectComm((PetscObject)v), nloc, PETSC_DECIDE, n, k - l, vv + l * nloc… in main()
120 PetscCall(MatCreateShell(PetscObjectComm((PetscObject)v), nloc, nloc, n, n, A, &S)); in main()
129 …PetscCall(MatCreateDenseCUDA(PetscObjectComm((PetscObject)v), nloc, PETSC_DECIDE, n, k - l, NULL, … in main()
[all …]
|
| H A D | ex171f.F90 | 12 PetscInt nloc, on
19 nloc = 12
21 …PetscCallA(VecCreateFromOptions(PETSC_COMM_WORLD, PETSC_NULL_CHARACTER, on, nloc, PETSC_DETERMINE,…
|
| H A D | ex242.c | 9 PetscInt i, j, M = 5, N, mb = 2, nb, nrows, ncols, mloc, nloc; in main() local
33 nloc = PETSC_DECIDE; in main()
34 PetscCall(PetscSplitOwnershipEqual(PETSC_COMM_WORLD, &nloc, &N)); in main()
35 PetscCall(MatSetSizes(C, mloc, nloc, M, N)); in main()
101 PetscCall(MatSetSizes(B, mloc, nloc, PETSC_DECIDE, PETSC_DECIDE)); in main()
194 nloc = PETSC_DECIDE; in main()
195 PetscCall(PetscSplitOwnershipEqual(subcomm, &nloc, &N)); in main()
196 PetscCall(MatSetSizes(Asub, mloc, nloc, M, N)); in main()
|
| H A D | ex243.c | 8 PetscInt i, j, M = 10, N = 5, nloc, mloc, nrows, ncols; in main() local
27 nloc = PETSC_DECIDE; in main()
28 PetscCall(PetscSplitOwnershipEqual(PETSC_COMM_WORLD, &nloc, &N)); in main()
29 PetscCall(MatSetSizes(A, mloc, nloc, M, N)); in main()
|
| /petsc/src/ksp/pc/impls/gamg/ |
| H A D | util.c | 27 PetscInt nnodes, num_ghosts, dir, kk, jj, my0, Iend, nloc; in PCGAMGGetDataWithGhosts() local
37 nloc = Iend - my0; in PCGAMGGetDataWithGhosts()
39 nnodes = num_ghosts + nloc; in PCGAMGGetDataWithGhosts()
46 for (kk = 0; kk < nloc; kk++) { in PCGAMGGetDataWithGhosts()
48 PetscScalar crd = data_in[dir * nloc + kk]; /* col oriented */ in PCGAMGGetDataWithGhosts()
59 …for (kk = nloc, jj = 0; jj < num_ghosts; kk++, jj++) datas[dir * nnodes + kk] = PetscRealPart(data… in PCGAMGGetDataWithGhosts()
|
| H A D | agg.c | 357 PetscInt arrsz, kk, ii, jj, nloc, ndatarows, ndf; in PCSetCoordinates_AGG() local
363 nloc = a_nloc; in PCSetCoordinates_AGG()
375 arrsz = nloc * pc_gamg->data_cell_rows * pc_gamg->data_cell_cols; in PCSetCoordinates_AGG()
382 for (kk = 0; kk < nloc; kk++) { in PCSetCoordinates_AGG()
383 const PetscInt M = nloc * pc_gamg->data_cell_rows; /* stride into data */ in PCSetCoordinates_AGG()
513 …PetscInt Istart, my0, Iend, nloc, clid, flid = 0, aggID, kk, jj, ii, mm, nSelected, minsz, ng… in formProl0() local
522 nloc = (Iend - Istart) / bs; in formProl0()
526 nghosts = data_stride / bs - nloc; in formProl0()
530 …for (kk = 0; kk < nghosts; kk++) PetscCall(PetscHMapISet(fgid_flid, flid_fgid[nloc + kk], nloc + k… in formProl0()
533 for (nSelected = mm = 0; mm < nloc; mm++) { in formProl0()
[all …]
|
| H A D | geo.c | 38 PetscInt arrsz, bs, my0, kk, ii, nloc, Iend, aloc; in PCSetCoordinates_GEO() local
46 nloc = (Iend - my0) / bs; in PCSetCoordinates_GEO()
48 …nloc == a_nloc || aloc == a_nloc, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of local blocks %"… in PCSetCoordinates_GEO()
51 …PetscCheck(coords || (nloc <= 0), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need coordinates for pc_g… in PCSetCoordinates_GEO()
54 arrsz = nloc * pc_gamg->data_cell_rows * pc_gamg->data_cell_cols; in PCSetCoordinates_GEO()
64 if (nloc == a_nloc) { in PCSetCoordinates_GEO()
65 for (kk = 0; kk < nloc; kk++) { in PCSetCoordinates_GEO()
66 for (ii = 0; ii < ndm; ii++) pc_gamg->data[ii * nloc + kk] = coords[kk * ndm + ii]; in PCSetCoordinates_GEO()
69 for (kk = 0; kk < nloc; kk++) { in PCSetCoordinates_GEO()
70 for (ii = 0; ii < ndm; ii++) pc_gamg->data[ii * nloc + kk] = coords[bs * kk * ndm + ii]; in PCSetCoordinates_GEO()
[all …]
|
| /petsc/src/mat/graphops/coarsen/impls/mis/ |
| H A D | mis.c | 33 const PetscInt nloc = Gmat->rmap->n; in MatCoarsenApply_MIS_private() local
42 PetscCall(PetscInfo(info_is, "mis: nloc = %d\n", (int)nloc)); in MatCoarsenApply_MIS_private()
57 PetscCall(PetscMalloc4(nloc, &lid_gid, nloc, &lid_cprowID, nloc, &lid_removed, nloc, &lid_state)); in MatCoarsenApply_MIS_private()
58 if (strict_aggs) PetscCall(PetscMalloc1(nloc, &lid_parent_gid)); in MatCoarsenApply_MIS_private()
60 for (kk = 0, gid = my0; kk < nloc; kk++, gid++) lid_gid[kk] = gid; in MatCoarsenApply_MIS_private()
72 PetscCall(PetscCDCreate(strict_aggs ? nloc : num_fine_ghosts + nloc, &agg_lists)); in MatCoarsenApply_MIS_private()
76 for (kk = 0; kk < nloc; kk++) { in MatCoarsenApply_MIS_private()
93 while (nDone < nloc || PETSC_TRUE) { /* asynchronous not implemented */ in MatCoarsenApply_MIS_private()
95 for (kk = 0; kk < nloc; kk++) { in MatCoarsenApply_MIS_private()
165 if (statej == MIS_NOT_DONE) PetscCall(PetscCDAppendID(agg_lists, lid, nloc + cpid)); in MatCoarsenApply_MIS_private()
[all …]
|
| /petsc/src/ksp/pc/impls/bjacobi/ftn-custom/ |
| H A D | zbjacobif.c | 15 PetscInt nloc, flocal; in pcbjacobigetsubksp_() local
19 *ierr = PCBJacobiGetSubKSP(*pc, &nloc, &flocal, &tksp); in pcbjacobigetsubksp_()
20 if (n_local) *n_local = nloc; in pcbjacobigetsubksp_()
22 *ierr = F90Array1dCreate(tksp, MPIU_FORTRANADDR, 1, nloc, ksp PETSC_F90_2PTR_PARAM(ptrd)); in pcbjacobigetsubksp_()
|
| /petsc/src/ksp/pc/impls/gasm/ftn-custom/ |
| H A D | zgasmf.c | 45 PetscInt nloc, flocal; in pcgasmgetsubksp_() local
49 *ierr = PCGASMGetSubKSP(*pc, &nloc, &flocal, &tksp); in pcgasmgetsubksp_()
50 if (n_local) *n_local = nloc; in pcgasmgetsubksp_()
52 *ierr = F90Array1dCreate(tksp, MPIU_FORTRANADDR, 1, nloc, ksp PETSC_F90_2PTR_PARAM(ptrd)); in pcgasmgetsubksp_()
|
| /petsc/src/dm/impls/plex/tutorials/ |
| H A D | ex20.c | 130 PetscInt nloc; in main() local
132 PetscCall(VecGetLocalSize(Vecxyz[0], &nloc)); in main()
133 PetscCall(PetscMalloc1(nloc, &isObs)); in main()
140 for (PetscInt v = 0; v < nloc; v++) { in main()
174 for (PetscInt v = 0; v < nloc; v++) { in main()
191 PetscInt nloc; in main() local
194 PetscCall(VecGetLocalSize(Vecxyz[0], &nloc)); in main()
197 for (PetscInt v = 0; v < nloc; v++) { in main()
|
| /petsc/src/ksp/pc/impls/ml/ |
| H A D | ml.c | 65 PetscInt nloc; member
407 PetscInt arrsz, oldarrsz, bs, my0, kk, ii, nloc, Iend, aloc; in PCSetCoordinates_ML() local
417 nloc = (Iend - my0) / bs; in PCSetCoordinates_ML()
419 …nloc == a_nloc) || (aloc == a_nloc), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of local blocks… in PCSetCoordinates_ML()
421 oldarrsz = pc_ml->dim * pc_ml->nloc; in PCSetCoordinates_ML()
423 pc_ml->nloc = nloc; in PCSetCoordinates_ML()
424 arrsz = ndm * nloc; in PCSetCoordinates_ML()
433 if (nloc == a_nloc) { in PCSetCoordinates_ML()
434 for (kk = 0; kk < nloc; kk++) { in PCSetCoordinates_ML()
435 for (ii = 0; ii < ndm; ii++) pc_ml->coords[ii * nloc + kk] = coords[kk * ndm + ii]; in PCSetCoordinates_ML()
[all …]
|
| /petsc/src/mat/graphops/coarsen/impls/hem/ |
| H A D | hem.c | 422 …const PetscInt nloc = a_Gmat->rmap->n, request_size = PetscCeilInt((int)sizeof(MPI_Request), (i… in MatCoarsenApply_HEM_private() local
441 PetscCall(PetscMalloc3(nloc, &lid_matched, nloc, &lid_cprowID, nloc, &lid_max_pe)); in MatCoarsenApply_HEM_private()
442 PetscCall(PetscCDCreate(nloc, &agg_llists)); in MatCoarsenApply_HEM_private()
443 PetscCall(PetscCDSetChunkSize(agg_llists, nloc + 1)); in MatCoarsenApply_HEM_private()
446 for (PetscInt kk = 0; kk < nloc; kk++) PetscCall(PetscCDAppendID(agg_llists, kk, Istart + kk)); in MatCoarsenApply_HEM_private()
466 for (PetscInt kk = 0; kk < nloc; kk++) lid_cprowID[kk] = -1; in MatCoarsenApply_HEM_private()
484 for (PetscInt kk = 0; kk < nloc; kk++) { in MatCoarsenApply_HEM_private()
501 for (PetscInt kk = 0, gid = Istart; kk < nloc; kk++, gid++) { in MatCoarsenApply_HEM_private()
517 …for (PetscInt kk = 0, gid = Istart; kk < nloc; kk++, gid++) PetscCall(VecSetValues(vec, 1, &gid, &… in MatCoarsenApply_HEM_private()
526 for (PetscInt kk = 0, gid = Istart; kk < nloc; kk++, gid++) { in MatCoarsenApply_HEM_private()
[all …]
|
| /petsc/src/dm/tests/ |
| H A D | ex4.c | 10 PetscInt s = 2, w = 2, n = PETSC_DECIDE, nloc, l, i, j, kk; in main() local
124 PetscCall(ISLocalToGlobalMappingGetSize(ltogm, &nloc)); in main()
128 PetscCall(PetscMalloc1(nloc, &iglobal)); in main()
142 PetscCall(AOPetscToApplication(ao, nloc, iglobal)); in main()
145 PetscCall(AOApplicationToPetsc(ao, nloc, iglobal)); in main()
|
| H A D | ex6.c | 10 …PetscInt M = 3, N = 5, P = 3, s = 1, w = 2, nloc, l, i, j, k, kk, m = PETSC_DECIDE, n = PET… in main() local
143 PetscCall(ISLocalToGlobalMappingGetSize(ltogm, &nloc)); in main()
149 PetscCall(PetscMalloc1(nloc, &iglobal)); in main()
165 PetscCall(AOPetscToApplication(ao, nloc, iglobal)); in main()
168 PetscCall(AOApplicationToPetsc(ao, nloc, iglobal)); in main()
|
| /petsc/src/mat/graphops/coarsen/impls/misk/ |
| H A D | misk.c | 301 const PetscInt nloc = Gmat->rmap->n; in MatCoarsenApply_MISK_private() local
305 PetscCall(PetscCDCreate(nloc, &agg_lists)); in MatCoarsenApply_MISK_private()
322 for (PetscInt lid = 0; lid < nloc; lid++) { in MatCoarsenApply_MISK_private()
327 if (max_osz > MM - nloc) max_osz = MM - nloc; in MatCoarsenApply_MISK_private()
330 …PetscCall(MatCreateAIJ(comm, nloc, nloc, PETSC_DETERMINE, PETSC_DETERMINE, 0, NULL, max_osz, NULL,… in MatCoarsenApply_MISK_private()
331 for (PetscInt lid = 0, gidi = Istart; lid < nloc; lid++, gidi++) { in MatCoarsenApply_MISK_private()
340 …if (gidj < Istart || gidj >= Istart + nloc) PetscCall(MatSetValues(mat, 1, &gidi, 1, &gidj, &one, … in MatCoarsenApply_MISK_private()
|