| /petsc/src/sys/utils/ |
| H A D | mpimesg.c | 33 …sages(MPI_Comm comm, const PetscMPIInt iflags[], const PetscMPIInt ilengths[], PetscMPIInt *nrecvs) in PetscGatherNumberOfMessages() argument
55 *nrecvs = recv_buf[rank]; in PetscGatherNumberOfMessages()
90 PetscErrorCode PetscGatherMessageLengths(MPI_Comm comm, PetscMPIInt nsends, PetscMPIInt nrecvs, con… in PetscGatherMessageLengths() argument
102 PetscCall(PetscMalloc2(nrecvs + nsends, &r_waits, nrecvs + nsends, &w_status)); in PetscGatherMessageLengths()
103 s_waits = PetscSafePointerPlusOffset(r_waits, nrecvs); in PetscGatherMessageLengths()
106 PetscCall(PetscMalloc1(nrecvs, olengths)); in PetscGatherMessageLengths()
107 …for (i = 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((*olengths) + i, 1, MPI_INT, MPI_ANY_SOURCE, … in PetscGatherMessageLengths()
118 if (nrecvs + nsends) PetscCallMPI(MPI_Waitall(nrecvs + nsends, r_waits, w_status)); in PetscGatherMessageLengths()
121 PetscCall(PetscMalloc1(nrecvs, onodes)); in PetscGatherMessageLengths()
122 for (i = 0; i < nrecvs; ++i) { in PetscGatherMessageLengths()
[all …]
|
| H A D | mpits.c | 75 PetscMPIInt nrecvs, tag, done, i; in PetscCommBuildTwoSided_Ibarrier() local
92 nrecvs = 0; in PetscCommBuildTwoSided_Ibarrier()
108 nrecvs++; in PetscCommBuildTwoSided_Ibarrier()
123 *nfrom = nrecvs; in PetscCommBuildTwoSided_Ibarrier()
135 PetscMPIInt size, rank, *iflags, nrecvs, tag, *franks, i, flg; in PetscCommBuildTwoSided_Allreduce() local
157 nrecvs = iflags[rank]; in PetscCommBuildTwoSided_Allreduce()
160 PetscCall(PetscMalloc(nrecvs * count * unitbytes, &fdata)); in PetscCommBuildTwoSided_Allreduce()
162 PetscCall(PetscMalloc2(nto + nrecvs, &reqs, nto + nrecvs, &statuses)); in PetscCommBuildTwoSided_Allreduce()
163 sendreqs = PetscSafePointerPlusOffset(reqs, nrecvs); in PetscCommBuildTwoSided_Allreduce()
164 …for (i = 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((void *)(fdata + count * unitbytes * i), coun… in PetscCommBuildTwoSided_Allreduce()
[all …]
|
| /petsc/src/vec/is/is/utils/ |
| H A D | iscoloring.c | 409 PetscInt *send_indices, rstart, *recv_indices, nrecvs, nsends; in ISBuildTwoSided() local
462 nrecvs = 0; in ISBuildTwoSided()
463 for (PetscMPIInt i = 0; i < nfrom; i++) nrecvs += fromsizes[i * 2]; in ISBuildTwoSided()
464 PetscCall(PetscCalloc1(nrecvs, &recv_indices)); in ISBuildTwoSided()
465 PetscCall(PetscMalloc1(nrecvs, &iremote)); in ISBuildTwoSided()
466 nrecvs = 0; in ISBuildTwoSided()
469 iremote[nrecvs].rank = fromranks[i]; in ISBuildTwoSided()
470 iremote[nrecvs++].index = fromsizes[2 * fromperm_newtoold[i] + 1] + j; in ISBuildTwoSided()
474 …PetscCall(PetscSFSetGraph(sf, nsends, nrecvs, NULL, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER)… in ISBuildTwoSided()
486 PetscCall(PetscSortInt(nrecvs, recv_indices)); in ISBuildTwoSided()
[all …]
|
| /petsc/src/ksp/pc/impls/redistribute/ |
| H A D | redistribute.c | 79 PetscMPIInt *sizes = NULL, nrecvs, nsends; in PCSetUp_Redistribute() local
183 PetscCall(PetscGatherNumberOfMessages(comm, NULL, sizes, &nrecvs)); in PCSetUp_Redistribute()
184 PetscCall(PetscGatherMessageLengths(comm, nsends, nrecvs, sizes, &onodes1, &olengths1)); in PCSetUp_Redistribute()
185 PetscCall(PetscSortMPIIntWithArray(nrecvs, onodes1, olengths1)); in PCSetUp_Redistribute()
187 for (PetscMPIInt i = 0; i < nrecvs; i++) recvtotal += olengths1[i]; in PCSetUp_Redistribute()
190 PetscCall(PetscMalloc3(recvtotal, &rvalues, nrecvs, &source, nrecvs, &recv_waits)); in PCSetUp_Redistribute()
192 for (PetscMPIInt i = 0; i < nrecvs; i++) { in PCSetUp_Redistribute()
218 count = nrecvs; in PCSetUp_Redistribute()
221 PetscCallMPI(MPI_Waitany(nrecvs, recv_waits, &imdex, &recv_status)); in PCSetUp_Redistribute()
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mpisell.h | 15 PetscInt nsends, nrecvs; /* numbers of sends and receives */ member
|
| /petsc/src/vec/vec/utils/ |
| H A D | vecstash.c | 60 stash->nrecvs = 0; in VecStashCreate_Private()
306 stash->nrecvs = nreceives; in VecStashScatterBegin_Private()
342 if (stash->nprocessed == stash->nrecvs) PetscFunctionReturn(PETSC_SUCCESS); in VecStashScatterGetMesg_Private()
348 PetscCallMPI(MPI_Waitany(2 * stash->nrecvs, stash->recv_waits, &i, &recv_status)); in VecStashScatterGetMesg_Private()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimatmatmult.c | 393 PetscInt nsends, nrecvs; member
407 for (i = 0; i < contents->nrecvs; i++) PetscCallMPI(MPI_Type_free(&contents->rtype[i])); in MatMPIAIJ_MPIDenseDestroy()
423 PetscMPIInt *disp, nsends, nrecvs, nrows_to, nrows_from; in MatMatMultSymbolic_MPIAIJ_MPIDense() local
441 …PetscCall(VecScatterGetRemoteOrdered_Private(ctx, PETSC_FALSE /*recv*/, &nrecvs, &rstarts, NULL, N… in MatMatMultSymbolic_MPIAIJ_MPIDense()
480 …PetscCall(PetscMalloc4(nsends, &stype, nrecvs, &rtype, nrecvs, &contents->rwaits, nsends, &content… in MatMatMultSymbolic_MPIAIJ_MPIDense()
485 contents->nrecvs = nrecvs; in MatMatMultSymbolic_MPIAIJ_MPIDense()
498 for (PetscMPIInt i = 0; i < nrecvs; i++) { in MatMatMultSymbolic_MPIAIJ_MPIDense()
510 …PetscCall(VecScatterRestoreRemoteOrdered_Private(ctx, PETSC_FALSE /*recv*/, &nrecvs, &rstarts, NUL… in MatMatMultSymbolic_MPIAIJ_MPIDense()
544 PetscMPIInt nsends, nrecvs; in MatMPIDenseScatter() local
560 …PetscCall(VecScatterGetRemoteOrdered_Private(ctx, PETSC_FALSE /*recv*/, &nrecvs, &rstarts, NULL, &… in MatMPIDenseScatter()
[all …]
|
| H A D | mpiaij.h | 49 PetscInt nsends, nrecvs; /* numbers of sends and receives */ \
|
| H A D | mpiov.c | 232 …verlap_MPIAIJ_Receive_Scalable(Mat mat, PetscInt nidx, IS is[], PetscInt nrecvs, PetscInt *recvdat… in MatIncreaseOverlap_MPIAIJ_Receive_Scalable() argument
247 PetscCall(PetscMalloc2((max_lsize + nrecvs) * nidx, &indices_temp, nidx, &iscomms)); in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
251 …PetscCall(PetscArraycpy(PetscSafePointerPlusOffset(indices_temp, i * (max_lsize + nrecvs)), indice… in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
256 for (i = 0; i < nrecvs;) { in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
259 indices_i = indices_temp + (max_lsize + nrecvs) * is_id; in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
269 indices_i = PetscSafePointerPlusOffset(indices_temp, (max_lsize + nrecvs) * i); in MatIncreaseOverlap_MPIAIJ_Receive_Scalable()
|
| H A D | mpiaij.c | 5767 PetscMPIInt nrecvs, nsends; in MatGetBrowsOfAoCols_MPIAIJ() local
5796 …PetscCall(VecScatterGetRemoteOrdered_Private(ctx, PETSC_FALSE /*recv*/, &nrecvs, &rstarts, NULL /*… in MatGetBrowsOfAoCols_MPIAIJ()
5797 PetscCall(PetscMPIIntCast(nsends + nrecvs, &nreqs)); in MatGetBrowsOfAoCols_MPIAIJ()
5800 swaits = PetscSafePointerPlusOffset(reqs, nrecvs); in MatGetBrowsOfAoCols_MPIAIJ()
5806 …if (nrecvs) PetscCall(PetscMalloc1(rbs * (rstarts[nrecvs] - rstarts[0]), &rvalues)); /* rstarts ca… in MatGetBrowsOfAoCols_MPIAIJ()
5807 for (i = 0; i < nrecvs; i++) { in MatGetBrowsOfAoCols_MPIAIJ()
5814 PetscCall(PetscMalloc2(nsends + 1, &sstartsj, nrecvs + 1, &rstartsj)); in MatGetBrowsOfAoCols_MPIAIJ()
5856 for (i = 0; i < nrecvs; i++) { in MatGetBrowsOfAoCols_MPIAIJ()
5874 for (i = 0; i < nrecvs; i++) { in MatGetBrowsOfAoCols_MPIAIJ()
5906 for (i = 0; i < nrecvs; i++) { in MatGetBrowsOfAoCols_MPIAIJ()
[all …]
|
| /petsc/src/mat/utils/ |
| H A D | matstash.c | 71 stash->nrecvs = 0; in MatStashCreate_Private()
597 stash->nrecvs = nreceives; in MatStashScatterBegin_Ref()
637 if (stash->nprocessed == stash->nrecvs) PetscFunctionReturn(PETSC_SUCCESS); in MatStashScatterGetMesg_Ref()
647 PetscCallMPI(MPI_Waitany(2 * stash->nrecvs, stash->recv_waits, &i, &recv_status)); in MatStashScatterGetMesg_Ref()
|
| /petsc/include/petsc/private/ |
| H A D | matimpl.h | 356 PetscMPIInt nsends, nrecvs; /* numbers of sends and receives */ member
427 PetscInt nzlocal, nsends, nrecvs; member
740 PetscInt nsends, nrecvs; /* numbers of sends and receives */ member
|
| H A D | vecimpl.h | 142 PetscMPIInt nsends, nrecvs; /* numbers of sends and receives */ member
|
| /petsc/src/mat/impls/scalapack/ |
| H A D | matscalapack.c | 1665 stash->nrecvs = nreceives; in MatStashScatterBegin_ScaLAPACK()
|
| /petsc/src/mat/interface/ |
| H A D | matrix.c | 1416 for (PetscInt i = 0; i < redund->nrecvs; i++) { in MatDestroy_Redundant()
|