Searched refs:olengths (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/sys/utils/ |
| H A D | mpimesg.c | 90 …ds, PetscMPIInt nrecvs, const PetscMPIInt ilengths[], PetscMPIInt **onodes, PetscMPIInt **olengths) in PetscGatherMessageLengths() argument
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()
131 if (w_status[i].MPI_SOURCE == rank) (*olengths)[i] = ilengths[rank]; in PetscGatherMessageLengths()
168 …t nsends, PetscMPIInt nrecvs, const PetscInt ilengths[], PetscMPIInt **onodes, PetscInt **olengths) in PetscGatherMessageLengths_Private() argument
184 PetscCall(PetscMalloc1(nrecvs, olengths)); in PetscGatherMessageLengths_Private()
185 …for (i = 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((*olengths) + i, 1, MPIU_INT, MPI_ANY_SOURCE,… in PetscGatherMessageLengths_Private()
202 …if (w_status[i].MPI_SOURCE == rank) (*olengths)[i] = ilengths[rank]; /* See comments in PetscGathe… in PetscGatherMessageLengths_Private()
293 … tag, PetscMPIInt nrecvs, const PetscMPIInt onodes[], const PetscMPIInt olengths[], PetscInt ***rb… in PetscPostIrecvInt() argument
300 for (i = 0; i < nrecvs; i++) len += olengths[i]; /* each message length */ in PetscPostIrecvInt()
[all …]
|
| /petsc/src/mat/utils/ |
| H A D | matstash.c | 504 PetscMPIInt *onodes, *olengths; in MatStashScatterBegin_Ref() local
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 …cMPIInt i = 0; i < nreceives; i++) PetscCall(PetscMPIIntCast(olengths[i] * bs2 / 2, &olengths[i])); in MatStashScatterBegin_Ref()
513 … PetscCall(PetscPostIrecvScalar(comm, tag2, nreceives, onodes, olengths, &rvalues, &recv_waits2)); in MatStashScatterBegin_Ref()
515 PetscCall(PetscFree(olengths)); in MatStashScatterBegin_Ref()
|
| /petsc/src/mat/impls/scalapack/ |
| H A D | matscalapack.c | 1570 PetscMPIInt *onodes, *olengths; in MatStashScatterBegin_ScaLAPACK() local
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()
1583 PetscCall(PetscFree(olengths)); in MatStashScatterBegin_ScaLAPACK()
|