Searched refs:recvRowCnt (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 397 PetscInt sendRowCnt = roffset[nranks], recvRowCnt = ioffset[niranks]; in MatMPIAIJKokkosReduceBegin() local
400 …PetscCall(PetscMalloc5(sendRowCnt, &sendRowLen, recvRowCnt + 1, &recvRowLen, nranks, &sdisp, niran… in MatMPIAIJKokkosReduceBegin()
416 for (PetscInt i = 0; i < recvRowCnt; i++) recvRowLen[i + 1] += recvRowLen[i]; in MatMPIAIJKokkosReduceBegin()
468 …PetscCall(PetscMalloc4(recvRowCnt, &recvRowPerm, nroots, &recvColSorted, nroots, &recvNzPerm, nroo… in MatMPIAIJKokkosReduceBegin()
471 …for (PetscInt i = 0; i < recvRowCnt; i++) recvRowPerm[i] = i; // put up a permutation… in MatMPIAIJKokkosReduceBegin()
472 …PetscCall(PetscSortIntWithPermutation(recvRowCnt, irootloc, recvRowPerm)); // irootloc[] (owned by… in MatMPIAIJKokkosReduceBegin()
485 while (iter < recvRowCnt) { // iter over received rows in MatMPIAIJKokkosReduceBegin()
489 …while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows+… in MatMPIAIJKokkosReduceBegin()
554 while (iter < recvRowCnt) { in MatMPIAIJKokkosReduceBegin()
559 …while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows+… in MatMPIAIJKokkosReduceBegin()
|