Searched refs:Nblk (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkos.kokkos.cxx | 92 int Nblk = end - start, it, m, stride = stride_shared, idx = 0; in BJSolve_TFQMR() local
164 …parallel_reduce(Kokkos::TeamVectorRange(team, Nblk), [=](const int idx, PetscScalar &lsum) { lsum … in BJSolve_TFQMR()
183 parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { RP[idx] = R[idx]; }); in BJSolve_TFQMR()
192 …parallel_reduce(Kokkos::TeamVectorRange(team, Nblk), [=](const int idx, PetscScalar &dot) { dot +=… in BJSolve_TFQMR()
194 parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { in BJSolve_TFQMR()
206 …parallel_reduce(Kokkos::TeamVectorRange(team, Nblk), [=](const int idx, PetscScalar &dot) { dot +=… in BJSolve_TFQMR()
215 parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { in BJSolve_TFQMR()
221 … parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { T[idx] = Diag[idx] * T[idx]; }); in BJSolve_TFQMR()
225 …parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { R[idx] = R[idx] - a * AUQ[idx]; }… in BJSolve_TFQMR()
227 …parallel_reduce(Kokkos::TeamVectorRange(team, Nblk), [=](const int idx, PetscScalar &lsum) { lsum … in BJSolve_TFQMR()
[all …]
|