mpi/kokkos/mpiaijkok.kokkos.cxx

149   PetscIntKokkosView irootloc; // plans to put E (i.e., Bd, Bo) into rootBuf
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)
472     PetscCall(PetscSortIntWithPermutation(recvRowCnt, irootloc, recvRowPerm)); // irootloc[] (owned by ownerSF) won't be changed
486       PetscInt curRowIdx = irootloc[recvRowPerm[iter]];
489       while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows++;
555       PetscInt curRowIdx = irootloc[recvRowPerm[iter]]; // current row idx
559       while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows++;
793     const PetscInt    *ioffset, *irootloc, *roffset;
799     PetscCall(PetscSFGetLeafRanks(ownerSF, &niranks, &iranks, &ioffset, &irootloc)); // get leaf ranks referencing roots on this process
807         PetscInt r = irootloc[j]; // row to be sent
833     // Build a plan (rowoffset, irootloc, E_NzLeft) to copy rows in E to rootdata of bcastSF in parallel
835     PetscInt              *rowoffset = rowoffset_h.data(); // for each entry (row) indicated in irootloc[], we calculate its destinate offset in copying
837     for (PetscInt i = 0; i < ioffset[niranks]; i++) rowoffset[i + 1] = rowoffset[i] + E_RowLen[irootloc[i]];
843       PetscInt  i      = irootloc[k]; // row to be copied
910     PetscIntKokkosViewHost tmp(const_cast<PetscInt *>(irootloc), ioffset[niranks]); // irootloc[] is owned by ownerSF. We create a copy for safety
916     mm->irootloc  = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), irootloc_h);
937     PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(mm->irootloc.extent(0), mm->rootBuf.extent(0), -1, teamSize, vectorLength, rowsPerTeam));
964   PetscIntKokkosView  &irootloc  = mm->irootloc;
970   PetscInt workSets     = (irootloc.extent(0) + rowsPerTeam - 1) / rowsPerTeam;
976         size_t r = t.league_rank() * rowsPerTeam + k; // r-th entry in irootloc[]
977         if (r < irootloc.extent(0)) {
978           PetscInt i      = irootloc(r); // row i of E