mpi/kokkos/mpiaijkok.kokkos.cxx

327 .  cstart      - (global) start column of Ed
328 .  cend        - (global) end column + 1 of Ed.  In other words, E's column ownership is in range of [cstart, cend)
339   When reuse = MAT_REUSE_MATRIX, cstart, cend, garray1, ownerSF, map are not significant.
342 static PetscErrorCode MatMPIAIJKokkosReduceBegin(MPI_Comm comm, KokkosCsrMatrix A, KokkosCsrMatrix B, PetscInt cstart, PetscInt cend, const PetscInt *garray1, PetscSF ownerSF, MatReuse reuse, PetscInt *map, MatMatStruct_AtB *mm)
359     // Count how many nonzeros of each row in E are in the left of the diag block [cstart,cend)
365       // std::lower_bound(first,last,cstart), but need to use global column indices
373         if (garray1[*it] < cstart) { // map local to global
456           buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global
515         if (curColIdx >= cstart && curColIdx < cend) {
573         if (curColIdx >= cstart && curColIdx < cend) {
574           Fdj[Fdnz]        = curColIdx - cstart; // easily convert to local
617     PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d));
674 static PetscErrorCode MatMPIAIJKokkosReduceEnd(MPI_Comm comm, KokkosCsrMatrix A, KokkosCsrMatrix B, PetscInt cstart, PetscInt cend, const PetscInt *garray1, PetscSF ownerSF, MatReuse reuse, PetscInt *map, MatMatStruct_AtB *mm)
753     PetscInt        cstart, cend;
756     PetscCall(MatGetOwnershipRangeColumn(E, &cstart, &cend));
758     // Count how many nonzeros of each row in E are in the left of the diag block [cstart,cend)
764       // std::lower_bound(first,last,cstart), but need to use global column indices
772         if (empi->garray[*it] < cstart) { // map local to global
851           buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global
869       // cut the row into: Left, [cstart, cend), Right
872       lb1         = std::lower_bound(first, last, cstart);
896           Fdj[Fdi[i] + j - nzLeft] = gid - cstart;
930     PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d));
1055   PetscInt        cstart, cend;
1097   PetscCall(MatGetOwnershipRangeColumn(B, &cstart, &cend));
1098   PetscCall(MatMPIAIJKokkosReduceBegin(comm, mm->C3, mm->C4, cstart, cend, bmpi->garray, ampi->Mvctx, MAT_INITIAL_MATRIX, map_h.data(), mm));
1110   PetscCall(MatMPIAIJKokkosReduceEnd(comm, mm->C3, mm->C4, cstart, cend, bmpi->garray, ampi->Mvctx, MAT_INITIAL_MATRIX, map_h.data(), mm));