bjacobi/bjkokkos/bjkokkos.kokkos.cxx

48   Kokkos::parallel_for(Kokkos::TeamThreadRange(team, start, end), [=](const int rowb) {
49     int                rowa = ic[rowb];
55     Kokkos::single(Kokkos::PerThread(team), [=]() { y_loc[rowb - start] = sum; });
66   Kokkos::parallel_for(Kokkos::TeamThreadRange(team, start, end), [=](const int rowb) {
67     int                rowa = ic[rowb];
71     const PetscScalar  xx   = x_loc[rowb - start]; // rowb = ic[rowa] = ic[r[rowb]]
158   parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](int rowb) {
159     int rowa         = ic[rowb];
160     R[rowb - start]  = glb_b[rowa];
161     XX[rowb - start] = 0;
314   parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](int rowb) {
315     int rowa    = ic[rowb];
316     glb_x[rowa] = XX[rowb - start];
383   parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](int rowb) {
384     int rowa         = ic[rowb];
385     Rl[rowb - start] = Rr[rowb - start] = glb_b[rowa];
386     XX[rowb - start]                    = 0;
510   parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](int rowb) {
511     int rowa    = ic[rowb];
512     glb_x[rowa] = XX[rowb - start];
983           Kokkos::parallel_for(Kokkos::TeamThreadRange(team, d_bid_eqOffset[blkID], d_bid_eqOffset[blkID + 1]), [=](const int rowb) {
984             const PetscInt     rowa = ic[rowb], ai = d_ai[rowa], *aj = d_aj + ai; // grab original data
992                 if (colb == rowb) {
993                   d_idiag[rowb] = 1. / aa[j];
999             if (found != 1) Kokkos::single(Kokkos::PerThread(team), [=]() { printf("ERRORrow %d) found = %d\n", rowb, found); });