Searched refs:recvRowPerm (Results 1 – 1 of 1) sorted by relevance
466 PetscInt *recvRowPerm, *recvColSorted; in MatMPIAIJKokkosReduceBegin() local468 …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()486 PetscInt curRowIdx = irootloc[recvRowPerm[iter]]; in MatMPIAIJKokkosReduceBegin()489 …while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows+… in MatMPIAIJKokkosReduceBegin()498 PetscInt i = recvRowPerm[iter + d]; in MatMPIAIJKokkosReduceBegin()555 PetscInt curRowIdx = irootloc[recvRowPerm[iter]]; // current row idx in MatMPIAIJKokkosReduceBegin()559 …while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows+… in MatMPIAIJKokkosReduceBegin()561 PetscInt i = recvRowPerm[iter + d]; in MatMPIAIJKokkosReduceBegin()[all …]