mpi/kokkos/mpiaijkok.kokkos.cxx

451       PetscInt  alen = Ai[i + 1] - Ai[i], blen = Bi[i + 1] - Bi[i];
452       for (PetscInt j = 0; j < alen + blen; j++) {
455         } else if (j < nzLeft + alen) {
458           buf[j] = garray1[Bj[Bi[i] + j - alen]]; // right B, in global
652           PetscInt alen   = Ai(i + 1) - Ai(i);
656           Kokkos::parallel_for(Kokkos::ThreadVectorRange(t, alen + blen), [&](PetscInt j) {
660             } else if (j < nzleft + alen) { // diag A
663               val = Ba(Bi(i) + j - alen);
846       PetscInt  alen = Ai[i + 1] - Ai[i], blen = Bi[i + 1] - Bi[i];
847       for (PetscInt j = 0; j < alen + blen; j++) {
850         } else if (j < nzLeft + alen) {
853           buf[j] = empi->garray[Bj[Bi[i] + j - alen]]; // right B, in global
980           PetscInt alen   = Ai(i + 1) - Ai(i);
984           Kokkos::parallel_for(Kokkos::ThreadVectorRange(t, alen + blen), [&](PetscInt j) {
987             } else if (j < nzleft + alen) { // diag A
990               rootBuf(disp + j) = Ba(Bi(i) + j - alen);
1030           PetscInt alen   = Fdi(i + 1) - Fdi(i);
1034           Kokkos::parallel_for(Kokkos::ThreadVectorRange(t, alen + blen), [&](PetscInt j) {
1038             } else if (j < nzLeft + alen) { // diag
1041               Foa(Foi(i) + j - alen) = val;