Home
last modified time | relevance | path

Searched refs:recvcounts (Results 1 – 13 of 13) sorted by relevance

/petsc/src/mat/impls/dense/mpi/
H A Dmpidense.h15 PetscMPIInt *recvcounts; member
22 PetscMPIInt *recvcounts; member
H A Dmpidense.c2050 PetscCall(PetscFree2(atb->sendbuf, atb->recvcounts)); in MatProductCtxDestroy_MatTransMatMult_MPIDense_MPIDense()
2062 PetscCall(PetscFree2(abt->recvcounts, abt->recvdispls)); in MatProductCtxDestroy_MatMatTransMult_MPIDense_MPIDense()
2072 PetscMPIInt size, *recvcounts; in MatTransposeMatMultNumeric_MPIDense_MPIDense() local
2082 recvcounts = atb->recvcounts; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2164 PetscCallMPI(MPI_Reduce_scatter(sendbuf, carray, recvcounts, MPIU_SCALAR, MPIU_SUM, comm)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2203 PetscCall(PetscMalloc2(cM * cN, &atb->sendbuf, size, &atb->recvcounts)); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2205 …; i < size; i++) PetscCall(PetscMPIIntCast((ranges[i + 1] - ranges[i]) * cN, &atb->recvcounts[i])); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2251 PetscCall(PetscMalloc2(size, &abt->recvcounts, size + 1, &abt->recvdispls)); in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2253 … i++) PetscCall(PetscMPIIntCast(abt->recvdispls[i + 1] - abt->recvdispls[i], &abt->recvcounts[i])); in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2382 …PetscCallMPI(MPI_Allgatherv(sendbuf, ibn, MPIU_SCALAR, recvbuf, abt->recvcounts, abt->recvdispls, … in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
/petsc/src/vec/is/sf/impls/basic/gatherv/
H A Dsfgatherv.c29 …PetscCallMPI(MPIU_Igatherv(rootbuf, count, unit, leafbuf, dat->recvcounts, dat->displs, unit, 0 /*… in PetscSFLinkStartCommunication_Gatherv()
31 …PetscCallMPI(MPIU_Iscatterv(leafbuf, dat->recvcounts, dat->displs, unit, rootbuf, count, unit, 0, … in PetscSFLinkStartCommunication_Gatherv()
/petsc/include/petsc/mpiuni/
H A Dmpi.h494 …#define MPI_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, com… argument
495 …(MPIUNI_ARG(recvcounts), MPIUNI_ARG(displs), MPIUNI_ARG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(co…
496 …#define MPI_Gatherv_c(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, c… argument
497 …(MPIUNI_ARG(recvcounts), MPIUNI_ARG(displs), MPIUNI_ARG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(co…
505 …#define MPI_Allgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, comm) \ argument
506 …(MPIUNI_ARG(recvcounts), MPIUNI_ARG(displs), MPIUNI_ARG(recvtype), MPIUNI_ARG(comm), MPIUNI_Memcpy…
508 …#define MPI_Alltoallv(sendbuf, sendcounts, sdispls, sendtype, recvbuf, recvcounts, rdispls, recvty… argument
509 …NI_ARG(sdispls), MPIUNI_ARG(sendtype), MPIUNI_ARG(recvbuf), MPIUNI_ARG(recvcounts), MPIUNI_ARG(rdi…
510 …#define MPI_Alltoallw(sendbuf, sendcounts, sdispls, sendtypes, recvbuf, recvcounts, rdispls, recvt… argument
521 …er(sendbuf, recvbuf, recvcounts, datatype, op, comm) (MPIUNI_ARG(op), MPIUNI_ARG(comm), MPIUNI_Mem… argument
/petsc/src/dm/impls/plex/
H A Dplexorient.c554 PetscMPIInt *recvcounts = NULL, *displs = NULL, *Nc, p, o, itotNeighbors; in DMPlexOrient() local
559 PetscCall(PetscCalloc4(size, &recvcounts, size + 1, &displs, size, &Nc, size + 1, &Noff)); in DMPlexOrient()
565 recvcounts[p] = 0; in DMPlexOrient()
566 for (c = 0; c < Nc[p]; ++c, ++o) recvcounts[p] += N[o]; in DMPlexOrient()
567 displs[p + 1] = displs[p] + recvcounts[p]; in DMPlexOrient()
571 …PetscCallMPI(MPI_Gatherv(nrankComp, itotNeighbors, MPIU_SF_NODE, adj, recvcounts, displs, MPIU_SF_… in DMPlexOrient()
572 …PetscCallMPI(MPI_Gatherv(match, itotNeighbors, MPI_C_BOOL, val, recvcounts, displs, MPI_C_BOOL, 0,… in DMPlexOrient()
668 PetscCall(PetscFree4(recvcounts, displs, Nc, Noff)); in DMPlexOrient()
933 PetscMPIInt *recvcounts = NULL, *displs = NULL, *Nc; in DMPlexOrientCells_Internal() local
938 PetscCall(PetscCalloc4(size, &recvcounts, size + 1, &displs, size, &Nc, size + 1, &Noff)); in DMPlexOrientCells_Internal()
[all …]
/petsc/src/vec/is/sf/impls/basic/allgatherv/
H A Dsfallgatherv.h9 PetscMPIInt *displs, *recvcounts; member
H A Dsfallgatherv.c47 PetscCall(PetscMalloc1(size, &dat->recvcounts)); in PetscSFSetUp_Allgatherv()
53 PetscCall(PetscMPIIntCast(range[i + 1] - range[i], &dat->recvcounts[i])); in PetscSFSetUp_Allgatherv()
83 PetscCall(PetscFree(dat->recvcounts)); in PetscSFReset_Allgatherv()
125 …else PetscCallMPI(MPIU_Iallgatherv(rootbuf, sendcount, unit, leafbuf, dat->recvcounts, dat->displs… in PetscSFBcastBegin_Allgatherv()
189 …catterv(link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], dat->recvcounts, dat->displs, un… in PetscSFReduceBegin_Allgatherv()
233 …PetscCallMPI(MPIU_Igatherv(rootbuf, sendcount, unit, leafbuf, dat->recvcounts, dat->displs, unit, … in PetscSFBcastToZero_Allgatherv()
/petsc/src/sys/mpiuni/
H A Dmpi.c689 PETSC_EXTERN void petsc_mpi_reduce_scatter_(void *sendbuf, void *recvbuf, int *recvcounts, int *dat… in petsc_mpi_reduce_scatter_() argument
724 …erv_(void *sendbuf, int *sendcount, int *sendtype, void *recvbuf, int *recvcounts, int *displs, in… in petsc_mpi_allgatherv_() argument
726 …*ierr = MPI_Allgatherv(sendbuf, *sendcount, *sendtype, recvbuf, recvcounts, displs, *recvtype, *co… in petsc_mpi_allgatherv_()
729 …buf, int *sendcounts, int *sdispls, int *sendtype, void *recvbuf, int *recvcounts, int *rdispls, i… in petsc_mpi_alltoallv_() argument
731 …*ierr = MPI_Alltoallv(sendbuf, sendcounts, sdispls, *sendtype, recvbuf, recvcounts, rdispls, *recv… in petsc_mpi_alltoallv_()
/petsc/src/sys/objects/
H A Dpinit.c1883 PetscMPIInt size, rank, ilen, *recvcounts = NULL, *displs = NULL; in PetscCheckAllreduceSameLineAndCount_Private() local
1894 if (rank == 0) PetscCall(PetscMalloc2(size, &recvcounts, size + 1, &displs)); in PetscCheckAllreduceSameLineAndCount_Private()
1895 PetscCallMPI(MPI_Gather(&ilen, 1, MPI_INT, recvcounts, 1, MPI_INT, 0, comm)); in PetscCheckAllreduceSameLineAndCount_Private()
1898 for (PetscMPIInt i = 0; i < size; i++) displs[i + 1] = displs[i] + recvcounts[i]; in PetscCheckAllreduceSameLineAndCount_Private()
1901 PetscCallMPI(MPI_Gatherv(str, ilen, MPI_CHAR, str0, recvcounts, displs, MPI_CHAR, 0, comm)); in PetscCheckAllreduceSameLineAndCount_Private()
1904 if (rank == 0) PetscCall(PetscFree2(recvcounts, displs)); in PetscCheckAllreduceSameLineAndCount_Private()
/petsc/include/
H A Dpetscsys.h2811 …unt, sendtype, recvbuf, recvcounts, displs, recvtype, root, comm) MPI_Gatherv(sendbuf, sendcount,… argument
2815 …#define MPIU_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, co… argument
2886 …unt, sendtype, recvbuf, recvcounts, displs, recvtype, root, comm) MPI_Gatherv(sendbuf, sendcount,… argument
2889 …pe, recvbuf, recvcounts, displs, recvtype, root, comm) MPI_Gatherv_c(sendbuf, sendcount, sendtype… argument
/petsc/src/mat/impls/baij/mpi/
H A Dmpibaij.c2093 PetscMPIInt size, rank, *recvcounts = NULL, *displs = NULL; in MatGetSeqNonzeroStructure_MPIBAIJ() local
2104 PetscCall(PetscMalloc1(2 * size, &recvcounts)); in MatGetSeqNonzeroStructure_MPIBAIJ()
2105 displs = recvcounts + size; in MatGetSeqNonzeroStructure_MPIBAIJ()
2107 PetscCall(PetscMPIIntCast(A->rmap->range[i + 1] / bs - A->rmap->range[i] / bs, &recvcounts[i])); in MatGetSeqNonzeroStructure_MPIBAIJ()
2110 …PetscCallMPI(MPI_Allgatherv(MPI_IN_PLACE, 0, MPI_DATATYPE_NULL, lens, recvcounts, displs, MPIU_INT… in MatGetSeqNonzeroStructure_MPIBAIJ()
2145 recvcounts[i] = 0; in MatGetSeqNonzeroStructure_MPIBAIJ()
2146 for (j = A->rmap->range[i] / bs; j < A->rmap->range[i + 1] / bs; j++) recvcounts[i] += lens[j]; in MatGetSeqNonzeroStructure_MPIBAIJ()
2149 for (i = 1; i < size; i++) displs[i] = displs[i - 1] + recvcounts[i - 1]; in MatGetSeqNonzeroStructure_MPIBAIJ()
2150 …PetscCallMPI(MPI_Allgatherv(MPI_IN_PLACE, 0, MPI_DATATYPE_NULL, b->j, recvcounts, displs, MPIU_INT… in MatGetSeqNonzeroStructure_MPIBAIJ()
2160 PetscCall(PetscFree(recvcounts)); in MatGetSeqNonzeroStructure_MPIBAIJ()
/petsc/src/dm/impls/network/
H A Dnetwork.c2881 PetscMPIInt rank, size, *displs = NULL, *recvcounts = NULL, remoterank; in DMNetworkSetVertexLocalToGlobalOrdering() local
2911 PetscCall(PetscMalloc3(size + 1, &vrange, size, &displs, size, &recvcounts)); in DMNetworkSetVertexLocalToGlobalOrdering()
2916 recvcounts[i] = 1; in DMNetworkSetVertexLocalToGlobalOrdering()
2921 PetscCallMPI(MPI_Allgatherv(&ii, 1, MPIU_INT, vrange + 1, recvcounts, displs, MPIU_INT, comm)); in DMNetworkSetVertexLocalToGlobalOrdering()
2934 PetscCall(PetscFree3(vrange, displs, recvcounts)); in DMNetworkSetVertexLocalToGlobalOrdering()
/petsc/src/mat/impls/aij/mpi/mumps/
H A Dmumps.c2768 PetscMPIInt *recvcounts, *displs, n; in MatSetFromOptions_MUMPS() local
2781 PetscCall(PetscCalloc2(size, &recvcounts, size + 1, &displs)); in MatSetFromOptions_MUMPS()
2783 …PetscCallMPI(MPI_Gather(&n, 1, MPI_INT, recvcounts, 1, MPI_INT, 0, PetscObjectComm((PetscObject)A)… in MatSetFromOptions_MUMPS()
2784 for (PetscInt p = 0; p < size; ++p) displs[p + 1] = displs[p] + recvcounts[p]; in MatSetFromOptions_MUMPS()
2787 …PetscCallMPI(MPI_Gatherv(bsizes, n, MPIU_INT, blkptr + 1, recvcounts, displs, MPIU_INT, 0, PetscOb… in MatSetFromOptions_MUMPS()
2793 PetscCall(PetscFree2(recvcounts, displs)); in MatSetFromOptions_MUMPS()