Lines Matching refs:ioffset
384 const PetscInt *ioffset, *irootloc, *roffset, *rmine;
392 PetscCall(PetscSFGetLeafRanks(ownerSF, &niranks, &iranks, &ioffset, &irootloc)); // get leaf ranks connecting to roots on this process (I'll recv from them)
397 PetscInt sendRowCnt = roffset[nranks], recvRowCnt = ioffset[niranks];
405 for (PetscInt i = 0; i < niranks; i++) PetscCallMPI(MPIU_Irecv(&recvRowLen[ioffset[i]], ioffset[i + 1] - ioffset[i], MPIU_INT, iranks[i], tag, comm, &reqs[nranks + i]));
413 for (PetscInt j = ioffset[i]; j < ioffset[i + 1]; j++) rdisp[i + 1] += recvRowLen[j];
793 const PetscInt *ioffset, *irootloc, *roffset;
799 PetscCall(PetscSFGetLeafRanks(ownerSF, &niranks, &iranks, &ioffset, &irootloc)); // get leaf ranks referencing roots on this process
806 for (PetscInt j = ioffset[i]; j < ioffset[i + 1]; j++) {
834 PetscIntKokkosViewHost rowoffset_h(NoInit("rowoffset_h"), ioffset[niranks] + 1);
837 for (PetscInt i = 0; i < ioffset[niranks]; i++) rowoffset[i + 1] = rowoffset[i] + E_RowLen[irootloc[i]];
842 for (PetscInt k = 0; k < ioffset[niranks]; k++) {
910 PetscIntKokkosViewHost tmp(const_cast<PetscInt *>(irootloc), ioffset[niranks]); // irootloc[] is owned by ownerSF. We create a copy for safety
911 PetscIntKokkosViewHost irootloc_h(NoInit("irootloc_h"), ioffset[niranks]);