| /petsc/src/vec/is/ao/impls/memscalable/ |
| H A D | aomemscalable.c | 97 PetscMPIInt nreceives, nsends; in AOMap_MemoryScalable_private() local
140 PetscCall(PetscMPIIntCast(inreceives, &nreceives)); in AOMap_MemoryScalable_private()
146 PetscCall(PetscMalloc2(nreceives * nmax, &rindices, nreceives, &recv_waits)); in AOMap_MemoryScalable_private()
150 PetscCall(PetscMalloc3(n, &sindices2, nreceives, &send_waits2, nreceives, &send_status2)); in AOMap_MemoryScalable_private()
157 …for (PetscMPIInt i = 0; i < nreceives; i++) PetscCallMPI(MPIU_Irecv(rindices + nmax * i, nmax, MPI… in AOMap_MemoryScalable_private()
195 for (j = 0; j < nreceives; j++) { in AOMap_MemoryScalable_private()
196 …PetscCallMPI(MPI_Waitany(nreceives, recv_waits, &widx, &recv_status)); /* idx: index of handle for… in AOMap_MemoryScalable_private()
210 if (nreceives) PetscCallMPI(MPI_Waitall(nreceives, send_waits2, send_status2)); in AOMap_MemoryScalable_private()
272 PetscInt *owner, *start, *sizes, nsends, nreceives; in AOCreateMemoryScalable_private() local
314 PetscCall(PetscMaxSum(comm, sizes, &nmax, &nreceives)); in AOCreateMemoryScalable_private()
[all …]
|
| /petsc/src/vec/vec/utils/ |
| H A D | vecstash.c | 222 …scMPIInt size = stash->size, tag1 = stash->tag1, tag2 = stash->tag2, j, *owner, nsends, nreceives; in VecStashScatterBegin_Private() local
254 PetscCall(PetscMPIIntCast(inreceives, &nreceives)); in VecStashScatterBegin_Private()
261 PetscCall(PetscMalloc2(nreceives * nmax * bs, &rvalues, nreceives * nmax, &rindices)); in VecStashScatterBegin_Private()
262 PetscCall(PetscMalloc1(2 * nreceives, &recv_waits)); in VecStashScatterBegin_Private()
263 for (PetscMPIInt i = 0, count = 0; i < nreceives; i++) { in VecStashScatterBegin_Private()
306 stash->nrecvs = nreceives; in VecStashScatterBegin_Private()
|
| /petsc/src/mat/utils/ |
| H A D | matstash.c | 452 PetscMPIInt *sizes, *nlengths, nreceives, nsends, tag1 = stash->tag1, tag2 = stash->tag2; in MatStashScatterBegin_Ref() local
506 PetscCall(PetscGatherNumberOfMessages(comm, sizes, nlengths, &nreceives)); in MatStashScatterBegin_Ref()
507 PetscCall(PetscGatherMessageLengths(comm, nsends, nreceives, nlengths, &onodes, &olengths)); in MatStashScatterBegin_Ref()
509 for (PetscMPIInt i = 0; i < nreceives; i++) olengths[i] *= 2; in MatStashScatterBegin_Ref()
510 PetscCall(PetscPostIrecvInt(comm, tag1, nreceives, onodes, olengths, &rindices, &recv_waits1)); in MatStashScatterBegin_Ref()
512 …for (PetscMPIInt i = 0; i < nreceives; i++) PetscCall(PetscMPIIntCast(olengths[i] * bs2 / 2, &olen… in MatStashScatterBegin_Ref()
513 … PetscCall(PetscPostIrecvScalar(comm, tag2, nreceives, onodes, olengths, &rvalues, &recv_waits2)); in MatStashScatterBegin_Ref()
580 PetscCall(PetscMalloc1(2 * nreceives, &recv_waits)); in MatStashScatterBegin_Ref()
582 for (PetscMPIInt i = 0; i < nreceives; i++) { in MatStashScatterBegin_Ref()
597 stash->nrecvs = nreceives; in MatStashScatterBegin_Ref()
|
| /petsc/src/mat/impls/scalapack/ |
| H A D | matscalapack.c | 1522 …PetscMPIInt tag1 = stash->tag1, tag2 = stash->tag2, *sizes, *nlengths, nreceives, insends, … in MatStashScatterBegin_ScaLAPACK() local
1573 PetscCall(PetscGatherNumberOfMessages(comm, sizes, nlengths, &nreceives)); in MatStashScatterBegin_ScaLAPACK()
1575 PetscCall(PetscGatherMessageLengths(comm, insends, nreceives, nlengths, &onodes, &olengths)); in MatStashScatterBegin_ScaLAPACK()
1577 for (PetscMPIInt i = 0; i < nreceives; i++) olengths[i] *= 2; in MatStashScatterBegin_ScaLAPACK()
1578 PetscCall(PetscPostIrecvInt(comm, tag1, nreceives, onodes, olengths, &rindices, &recv_waits1)); in MatStashScatterBegin_ScaLAPACK()
1580 for (PetscMPIInt i = 0; i < nreceives; i++) olengths[i] = (PetscMPIInt)(olengths[i] * bs2 / 2); in MatStashScatterBegin_ScaLAPACK()
1581 … PetscCall(PetscPostIrecvScalar(comm, tag2, nreceives, onodes, olengths, &rvalues, &recv_waits2)); in MatStashScatterBegin_ScaLAPACK()
1648 PetscCall(PetscMalloc1(2 * nreceives, &recv_waits)); in MatStashScatterBegin_ScaLAPACK()
1650 for (PetscMPIInt i = 0; i < nreceives; i++) { in MatStashScatterBegin_ScaLAPACK()
1665 stash->nrecvs = nreceives; in MatStashScatterBegin_ScaLAPACK()
|