| /petsc/src/vec/is/utils/ |
| H A D | psort.c | 239 PetscMPIInt *keys_per_snd, *keys_per_rcv, *offsets_snd, *offsets_rcv, nrecv; in PetscParallelSortInt_Samplesort() local 265 nrecv = offsets_rcv[size]; in PetscParallelSortInt_Samplesort() 267 PetscCall(PetscMalloc1(nrecv, &buffer)); in PetscParallelSortInt_Samplesort() 273 PetscCall(PetscSortInt(nrecv, buffer)); in PetscParallelSortInt_Samplesort() 278 PetscCall(PetscParallelSortedInt(mapin->comm, nrecv, buffer, &sorted)); in PetscParallelSortInt_Samplesort() 284 PetscCall(PetscParallelRedistribute(mapout, nrecv, buffer, keysout)); in PetscParallelSortInt_Samplesort()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimatmatmult.c | 1274 PetscMPIInt tagi, tagj, *len_si, *len_s, *len_ri, nrecv, proc, nsend; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() local 1355 PetscCall(PetscGatherNumberOfMessages(comm, NULL, len_s, &nrecv)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1356 PetscCall(PetscGatherMessageLengths2(comm, nsend, nrecv, len_s, len_si, &id_r, &len_r, &len_ri)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1360 PetscCall(PetscPostIrecvInt(comm, tagj, nrecv, id_r, len_r, &buf_rj, &rwaits)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1377 for (i = 0; i < nrecv; i++) PetscCallMPI(MPI_Waitany(nrecv, rwaits, &icompleted, &rstatus)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1388 for (k = 0; k < nrecv; k++) { /* k-th received message */ in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1397 PetscCall(PetscPostIrecvInt(comm, tagi, nrecv, id_r, len_ri, &buf_ri, &rwaits)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1422 for (i = 0; i < nrecv; i++) PetscCallMPI(MPI_Waitany(nrecv, rwaits, &icompleted, &rstatus)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1436 PetscCall(PetscMalloc3(nrecv, &buf_ri_k, nrecv, &nextrow, nrecv, &nextci)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() 1437 for (k = 0; k < nrecv; k++) { in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() [all …]
|
| H A D | mpiptap.c | 225 PetscMPIInt tagi, tagj, *len_si, *len_s, *len_ri, nrecv, nsend, proc; in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() local 419 PetscCall(PetscGatherNumberOfMessages(comm, NULL, len_s, &nrecv)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 420 PetscCall(PetscGatherMessageLengths2(comm, nsend, nrecv, len_s, len_si, &id_r, &len_r, &len_ri)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 424 PetscCall(PetscPostIrecvInt(comm, tagj, nrecv, id_r, len_r, &buf_rj, &rwaits)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 449 for (i = 0; i < nrecv; i++) PetscCallMPI(MPI_Waitany(nrecv, rwaits, &icompleted, &rstatus)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 454 for (k = 0; k < nrecv; k++) { /* k-th received message */ in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 463 PetscCall(PetscPostIrecvInt(comm, tagi, nrecv, id_r, len_ri, &buf_ri, &rwaits)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 488 for (i = 0; i < nrecv; i++) PetscCallMPI(MPI_Waitany(nrecv, rwaits, &icompleted, &rstatus)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 502 PetscCall(PetscMalloc3(nrecv, &buf_ri_k, nrecv, &nextrow, nrecv, &nextci)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() 503 for (k = 0; k < nrecv; k++) { in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() [all …]
|
| H A D | mpiaij.c | 4689 …PetscCall(PetscPostIrecvScalar(comm, taga, merge->nrecv, merge->id_r, merge->len_r, &abuf_r, &r_wa… in MatCreateMPIAIJSumSeqAIJNumeric() 4699 if (merge->nrecv) PetscCallMPI(MPI_Waitall(merge->nrecv, r_waits, status)); in MatCreateMPIAIJSumSeqAIJNumeric() 4708 PetscCall(PetscMalloc3(merge->nrecv, &buf_ri_k, merge->nrecv, &nextrow, merge->nrecv, &nextai)); in MatCreateMPIAIJSumSeqAIJNumeric() 4710 for (k = 0; k < merge->nrecv; k++) { in MatCreateMPIAIJSumSeqAIJNumeric() 4737 for (k = 0; k < merge->nrecv; k++) { /* k-th received message */ in MatCreateMPIAIJSumSeqAIJNumeric() 4832 PetscCall(PetscGatherNumberOfMessages(comm, NULL, len_s, &merge->nrecv)); in MatCreateMPIAIJSumSeqAIJSymbolic() 4833 …PetscCall(PetscGatherMessageLengths2(comm, merge->nsend, merge->nrecv, len_s, len_si, &merge->id_r… in MatCreateMPIAIJSumSeqAIJSymbolic() 4837 …PetscCall(PetscPostIrecvInt(comm, tagj, merge->nrecv, merge->id_r, merge->len_r, &buf_rj, &rj_wait… in MatCreateMPIAIJSumSeqAIJSymbolic() 4850 if (merge->nrecv) PetscCallMPI(MPI_Waitall(merge->nrecv, rj_waits, status)); in MatCreateMPIAIJSumSeqAIJSymbolic() 4855 PetscCall(PetscPostIrecvInt(comm, tagi, merge->nrecv, merge->id_r, len_ri, &buf_ri, &ri_waits)); in MatCreateMPIAIJSumSeqAIJSymbolic() [all …]
|
| H A D | mpiaij.h | 9 PetscMPIInt nsend, nrecv; member
|
| H A D | mpiov.c | 1089 PetscInt sendcount, *rstarts = A->rmap->range, n, cnt, j, nrecv = 0; in MatCreateSubMatrix_MPIAIJ_All() local 1114 nrecv = 0; in MatCreateSubMatrix_MPIAIJ_All() 1116 for (j = A->rmap->range[i]; j < A->rmap->range[i + 1]; j++) nrecv += lens[j]; in MatCreateSubMatrix_MPIAIJ_All() 1118 PetscCall(PetscArrayzero(b->j, nrecv)); in MatCreateSubMatrix_MPIAIJ_All() 1146 …PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, b->j, nrecv, MPIU_INT, MPI_SUM, PetscObjectComm((PetscOb… in MatCreateSubMatrix_MPIAIJ_All()
|
| /petsc/src/vec/vec/utils/ |
| H A D | vscat.c | 933 PetscMPIInt nsend, nrecv, nreq, yrank, *sendto, *recvfrom, tag1, tag2; in VecScatterCreate() local 995 PetscCall(PetscGatherNumberOfMessages_Private(ycomm, NULL, slens, &nrecv)); in VecScatterCreate() 996 PetscCall(PetscGatherMessageLengths_Private(ycomm, nsend, nrecv, slens, &recvfrom, &rlens)); in VecScatterCreate() 999 for (i = 0; i < nrecv; i++) rlentotal += rlens[i]; in VecScatterCreate() 1007 PetscCall(PetscMPIIntCast((nsend + nrecv) * 2, &nreq)); in VecScatterCreate() 1012 for (i = disp = 0; i < nrecv; i++) { in VecScatterCreate() 1015 …allMPI(MPIU_Irecv(ryindices + disp, count, MPIU_INT, recvfrom[i], tag2, ycomm, rreqs + nrecv + i)); in VecScatterCreate() 1031 for (i = disp = 0; i < nrecv; i++) { in VecScatterCreate()
|
| /petsc/src/vec/is/sf/impls/basic/neighbor/ |
| H A D | sfneighbor.c | 34 …s(PetscInt nsend, PetscSFCount *sendcnts, MPI_Datatype sendtype, PetscInt nrecv, PetscSFCount *rec… in PetscLogMPIMessages() argument 39 petsc_irecv_ct += (PetscLogDouble)nrecv; in PetscLogMPIMessages() 50 for (i = 0; i < nrecv; i++) petsc_irecv_len += (PetscLogDouble)(recvcnts[i] * typesize); in PetscLogMPIMessages()
|
| /petsc/src/dm/impls/plex/ |
| H A D | plexvtu.c | 40 PetscMPIInt nrecv; in TransferWrite() local 42 PetscCallMPI(MPI_Get_count(&status, mpidatatype, &nrecv)); in TransferWrite() 43 PetscCheck(count == nrecv, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Array size mismatch"); in TransferWrite()
|