| /petsc/src/mat/utils/ |
| H A D | zerorows.c | 7 PetscInt *owners = A->rmap->range; in MatZeroRowsMapLocal_Private() local
24 …if (idx < owners[p] || owners[p + 1] <= idx) { /* short-circuit the search if the last p owns this… in MatZeroRowsMapLocal_Private()
29 lrows[len++] = idx - owners[p]; in MatZeroRowsMapLocal_Private()
32 rrows[r].index = rows[r] - owners[p]; in MatZeroRowsMapLocal_Private()
|
| H A D | matstash.c | 437 PetscErrorCode MatStashScatterBegin_Private(Mat mat, MatStash *stash, PetscInt *owners) in MatStashScatterBegin_Private() argument
440 PetscCall((*stash->ScatterBegin)(mat, stash, owners)); in MatStashScatterBegin_Private()
444 static PetscErrorCode MatStashScatterBegin_Ref(Mat mat, MatStash *stash, PetscInt *owners) in MatStashScatterBegin_Ref() argument
482 if (idx >= owners[j] && idx < owners[j + 1]) { in MatStashScatterBegin_Ref()
821 static PetscErrorCode MatStashScatterBegin_BTS(Mat mat, MatStash *stash, PetscInt owners[]) in MatStashScatterBegin_BTS() argument
846 …PetscCheck(sendblock_b->row >= owners[stash->sendranks[i]], stash->comm, PETSC_ERR_ARG_WRONG, "MAT… in MatStashScatterBegin_BTS()
847 if (sendblock_b->row >= owners[stash->sendranks[i] + 1]) break; in MatStashScatterBegin_BTS()
861 PetscCall(PetscFindInt(sendblock_rowstart->row, stash->size + 1, owners, &owner)); in MatStashScatterBegin_BTS()
866 if (sendblock_i->row >= owners[owner + 1]) break; in MatStashScatterBegin_BTS()
880 PetscCall(PetscFindInt(sendblock_rowstart->row, stash->size + 1, owners, &iowner)); in MatStashScatterBegin_BTS()
[all …]
|
| /petsc/src/vec/is/sf/utils/ |
| H A D | sfutils.c | 566 PetscInt *owners = map->range; in PetscLayoutMapLocal() local
587 …if (idx < owners[p] || owners[p + 1] <= idx) { /* short-circuit the search if the last p owns this… in PetscLayoutMapLocal()
591 ridxs[nleaves].index = idxs[r] - owners[p]; in PetscLayoutMapLocal()
733 PetscSFNode *owners, *buffer, *iremote; in PetscSFCreateByMatchingIndices() local
774 PetscCall(PetscMalloc1(numRootIndices, &owners)); in PetscSFCreateByMatchingIndices()
776 owners[i].rank = rank; in PetscSFCreateByMatchingIndices()
777 owners[i].index = rootLocalOffset + (rootLocalIndices ? rootLocalIndices[i] : i); in PetscSFCreateByMatchingIndices()
783 PetscCall(PetscSFReduceBegin(sf1, MPIU_SF_NODE, owners, buffer, MPI_MAXLOC)); in PetscSFCreateByMatchingIndices()
784 PetscCall(PetscSFReduceEnd(sf1, MPIU_SF_NODE, owners, buffer, MPI_MAXLOC)); in PetscSFCreateByMatchingIndices()
787 PetscCall(PetscFree(owners)); in PetscSFCreateByMatchingIndices()
[all …]
|
| /petsc/src/vec/is/ao/impls/memscalable/ |
| H A D | aomemscalable.c | 98 const PetscInt *owners = aomems->map->range; in AOMap_MemoryScalable_private() local
125 if (idx >= owners[j] && idx < owners[j + 1]) { in AOMap_MemoryScalable_private()
173 ia[i] = maploc[ia[i] - owners[rank]]; in AOMap_MemoryScalable_private()
203 for (PetscCount i = 0; i < nindices; i++) sbuf[i] = maploc[rbuf[i] - owners[rank]]; in AOMap_MemoryScalable_private()
274 PetscInt *owners = aomems->map->range; in AOCreateMemoryScalable_private() local
301 if (idx >= owners[j] && idx < owners[j + 1]) { in AOCreateMemoryScalable_private()
339 ip = from_array[i] - owners[rank]; in AOCreateMemoryScalable_private()
370 ip = rbuf[i] - owners[rank]; /* local index */ in AOCreateMemoryScalable_private()
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | mmsbaij.c | 15 PetscInt *owners = sbaij->rangebs, *ec_owner, k; in MatSetUpMultiply_MPISBAIJ() local
63 while (garray[i] >= owners[j + 1]) j++; in MatSetUpMultiply_MPISBAIJ()
83 for (i = owners[j]; i < owners[j + 1]; i++) { in MatSetUpMultiply_MPISBAIJ()
136 sgarray[i] = garray[i] + (sowners[j] / bs - owners[j]); in MatSetUpMultiply_MPISBAIJ()
|
| /petsc/src/dm/impls/da/ |
| H A D | dasub.c | 203 PetscInt flag, *owners, xs, xm, ys, ym, zs, zm; in DMDAGetProcessorSubset() local
225 PetscCall(PetscMalloc2(size, &owners, size, &ranks)); in DMDAGetProcessorSubset()
226 …PetscCallMPI(MPI_Allgather(&flag, 1, MPIU_INT, owners, 1, MPIU_INT, PetscObjectComm((PetscObject)d… in DMDAGetProcessorSubset()
230 if (owners[i]) { in DMDAGetProcessorSubset()
242 PetscCall(PetscFree2(owners, ranks)); in DMDAGetProcessorSubset()
|
| /petsc/src/vec/vec/utils/ |
| H A D | vecstash.c | 220 PetscErrorCode VecStashScatterBegin_Private(VecStash *stash, const PetscInt *owners) in VecStashScatterBegin_Private() argument
241 if (idx >= owners[j] && idx < owners[j + 1]) { in VecStashScatterBegin_Private()
438 …hGetOwnerList_Private(VecStash *stash, PetscLayout map, PetscMPIInt *nowners, PetscMPIInt **owners) in VecStashGetOwnerList_Private() argument
457 PetscCall(PetscSegBufferExtractAlloc(seg, owners)); in VecStashGetOwnerList_Private()
|
| /petsc/src/mat/graphops/partition/impls/hierarchical/ |
| H A D | hierarchical.c | 246 PetscMPIInt rank, *owners; in MatPartitioningHierarchical_ReassembleFineparts() local
262 PetscCall(PetscCalloc1(localsize, &owners)); in MatPartitioningHierarchical_ReassembleFineparts()
264 …for (i = 0; i < localsize; i++) PetscCall(PetscLayoutFindOwner(rmap, global_indices[i], &owners[i]… in MatPartitioningHierarchical_ReassembleFineparts()
274 remote[i].rank = owners[i]; in MatPartitioningHierarchical_ReassembleFineparts()
275 remote[i].index = global_indices[i] - ranges[owners[i]]; in MatPartitioningHierarchical_ReassembleFineparts()
287 PetscCall(PetscFree(owners)); in MatPartitioningHierarchical_ReassembleFineparts()
|
| /petsc/src/dm/impls/plex/adaptors/parmmg/ |
| H A D | parmmgadapt.c | 39 PetscInt *gv_new, *owners, *verticesNewSorted, pStart, pEnd; in DMAdaptMetric_ParMmg_Plex() local
388 PetscCall(PetscMalloc2(numVerticesNew, &owners, numVerticesNew, &gv_new)); in DMAdaptMetric_ParMmg_Plex()
390 PetscCallMMG_NONSTANDARD(PMMG_Get_verticesGloNum, parmesh, gv_new, owners); in DMAdaptMetric_ParMmg_Plex()
394 if (owners[i] == rank) numVerticesNewLoc++; in DMAdaptMetric_ParMmg_Plex()
398 if (owners[i] == rank) { in DMAdaptMetric_ParMmg_Plex()
444 PetscCall(PetscFree2(owners, gv_new)); in DMAdaptMetric_ParMmg_Plex()
|
| /petsc/src/vec/vec/impls/mpi/ |
| H A D | pdvec.c | 733 const PetscInt *owners = xin->map->range; in VecSetValues_MPI() local
734 const PetscInt start = owners[rank], end = owners[rank + 1]; in VecSetValues_MPI()
768 PetscInt *owners = xin->map->range, start = owners[rank]; in VecSetValuesBlocked_MPI() local
769 PetscInt end = owners[rank + 1], i, row, bs = xin->map->bs, j; in VecSetValuesBlocked_MPI()
819 PetscInt *owners = xin->map->range, *bowners, i, bs, nstash, reallocs; in VecAssemblyBegin_MPI() local
837 for (i = 0; i < size + 1; i++) bowners[i] = owners[i] / bs; in VecAssemblyBegin_MPI()
841 PetscCall(VecStashScatterBegin_Private(&xin->stash, owners)); in VecAssemblyBegin_MPI()
|
| H A D | pbvec.c | 226 PetscMPIInt nowners, bnowners, *owners, *bowners; in VecAssemblyBegin_MPI_BTS() local
229 PetscCall(VecStashGetOwnerList_Private(&X->stash, X->map, &nowners, &owners)); in VecAssemblyBegin_MPI_BTS()
231 PetscCall(PetscMergeMPIIntArray(nowners, owners, bnowners, bowners, &ntmp, &x->sendranks)); in VecAssemblyBegin_MPI_BTS()
233 PetscCall(PetscFree(owners)); in VecAssemblyBegin_MPI_BTS()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimatmatmult.c | 1277 PetscInt len, *dnz, *onz, *owners, nzi; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() local
1324 owners = rowmap->range; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1337 while (prmap[i] >= owners[proc + 1]) proc++; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1415 buf_si[nrows + 1] = prmap[i] - owners[proc]; /* local row index */ in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1465 k = i + owners[rank]; /* column index */ in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1473 PetscCall(MatPreallocateSet(i + owners[rank], nzi, current_space->array, dnz, onz)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1601 PetscInt *owners, nrows, **buf_ri_k, **nextrow, **nextci; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ() local
1630 owners = merge->rowmap->range; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1723 row = owners[rank] + i; /* global row index of C_seq */ in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1770 PetscInt len, *dnz, *onz, *owners; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ() local
[all …]
|
| H A D | mpiptap.c | 228 const PetscInt *owners; in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() local
387 PetscCall(PetscLayoutGetRanges(rowmap, &owners)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
401 while (prmap[i] >= owners[proc + 1]) proc++; in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
481 buf_si[nrows + 1] = prmap[i] - owners[proc]; /* local row index */ in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
532 PetscCall(MatPreallocateSet(i + owners[rank], nzi, current_space->array, dnz, onz)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
1516 PetscInt len, *dnz, *onz, *owners, nzi, nspacedouble; in MatPtAPSymbolic_MPIAIJ_MPIAIJ() local
1677 owners = rowmap->range; in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1690 while (prmap[i] >= owners[proc + 1]) proc++; in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1770 buf_si[nrows + 1] = prmap[i] - owners[proc]; /* local row index */ in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1821 PetscCall(MatPreallocateSet(i + owners[rank], nzi, current_space->array, dnz, onz)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
|
| H A D | mpiaij.c | 938 PetscInt *lrows, *owners = A->rmap->range; in MatZeroRowsColumns_MPIAIJ() local
957 …if (idx < owners[p] || owners[p + 1] <= idx) { /* short-circuit the search if the last p owns this… in MatZeroRowsColumns_MPIAIJ()
961 rrows[r].index = rows[r] - owners[p]; in MatZeroRowsColumns_MPIAIJ()
4653 PetscInt N = mpimat->cmap->N, i, j, *owners, *ai = a->i, *aj, m; in MatCreateMPIAIJSumSeqAIJNumeric() local
4684 owners = merge->rowmap->range; in MatCreateMPIAIJSumSeqAIJNumeric()
4694 i = owners[proc]; in MatCreateMPIAIJSumSeqAIJNumeric()
4720 arow = owners[rank] + i; in MatCreateMPIAIJSumSeqAIJNumeric()
4773 PetscInt M = seqmat->rmap->n, N = seqmat->cmap->n, i, *owners, *ai = a->i, *aj = a->j; in MatCreateMPIAIJSumSeqAIJSymbolic() local
4805 owners = merge->rowmap->range; in MatCreateMPIAIJSumSeqAIJSymbolic()
4817 PetscCall(PetscMPIIntCast(owners[proc + 1] - owners[proc] + 1, &len_si[proc])); in MatCreateMPIAIJSumSeqAIJSymbolic()
[all …]
|
| /petsc/src/dm/impls/plex/hdf5/ |
| H A D | plexhdf5.c | 782 PetscInt *chartSize, *owners, *gpoints; in DMPlexDistributionView_HDF5_Private() local
800 PetscCall(PetscMalloc1(*chartSize, &owners)); in DMPlexDistributionView_HDF5_Private()
807 owners[p - pStart] = rank; in DMPlexDistributionView_HDF5_Private()
813 owners[ilocalp] = iremote[p].rank; in DMPlexDistributionView_HDF5_Private()
816 PetscCall(ISCreateGeneral(comm, *chartSize, owners, PETSC_OWN_POINTER, &ownersIS)); in DMPlexDistributionView_HDF5_Private()
1958 const PetscInt *chartSize, *owners, *gpoints; in DMPlexDistributionLoad_HDF5_Private() local
1993 PetscCall(ISGetIndices(ownersIS, &owners)); in DMPlexDistributionLoad_HDF5_Private()
2063 if (owners[p] == rank) { in DMPlexDistributionLoad_HDF5_Private()
2105 PetscCall(ISRestoreIndices(ownersIS, &owners)); in DMPlexDistributionLoad_HDF5_Private()
2975 PetscSFNode *owners, *buffer; in DMPlexSectionLoad_HDF5_Internal() local
[all …]
|
| /petsc/src/dm/impls/plex/ |
| H A D | plexvtu.c | 525 PetscVTKInt *owners; in DMPlexVTKWriteAll_VTU() local
529 PetscCall(PetscMalloc1(piece.ncells, &owners)); in DMPlexVTKWriteAll_VTU()
530 for (i = 0; i < piece.ncells; i++) owners[i] = orank; in DMPlexVTKWriteAll_VTU()
531 … PetscCall(TransferWrite(comm, viewer, fp, r, 0, owners, buffer, piece.ncells, MPI_INT, tag)); in DMPlexVTKWriteAll_VTU()
532 PetscCall(PetscFree(owners)); in DMPlexVTKWriteAll_VTU()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 1659 PetscInt *lrows, *owners = A->rmap->range; in MatZeroRowsColumns_MPIBAIJ() local
1678 …if (idx < owners[p] || owners[p + 1] <= idx) { /* short-circuit the search if the last p owns this… in MatZeroRowsColumns_MPIBAIJ()
1682 rrows[r].index = rows[r] - owners[p]; in MatZeroRowsColumns_MPIBAIJ()
|
| /petsc/src/mat/impls/scalapack/ |
| H A D | matscalapack.c | 1514 static PetscErrorCode MatStashScatterBegin_ScaLAPACK(Mat mat, MatStash *stash, PetscInt *owners) in MatStashScatterBegin_ScaLAPACK() argument
|