Searched refs:rowb (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 200 … Kokkos::parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](const int rowb) { // Nloc in PCApply_BJKOKKOSKERNELS() argument
201 int rowa = d_isicol[rowb]; in PCApply_BJKOKKOSKERNELS()
203 rowOffsets(blkID / Nsolves_team, rowb + 1 - start) = n; // save sizes in PCApply_BJKOKKOSKERNELS()
208 Kokkos::parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](int rowb) { in PCApply_BJKOKKOSKERNELS() argument
209 int rowa = d_isicol[rowb]; in PCApply_BJKOKKOSKERNELS()
210 batch_b(blkID, rowb - start) = glb_bdata[rowa]; in PCApply_BJKOKKOSKERNELS()
226 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, start, end), [=](const int rowb) { in PCApply_BJKOKKOSKERNELS() argument
227 int rowa = d_isicol[rowb]; in PCApply_BJKOKKOSKERNELS()
233 …if (fill_idx && blkID % Nsolves_team == 0) colIndices(graphID, rowOffsets(graphID, rowb - start) +… in PCApply_BJKOKKOSKERNELS()
234 batch_values(blkID, rowOffsets(graphID, rowb - start) + i) = val; in PCApply_BJKOKKOSKERNELS()
[all …]
|
| H A D | bjkokkos.kokkos.cxx | 48 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, start, end), [=](const int rowb) { in MatMult() argument
49 int rowa = ic[rowb]; in MatMult()
55 Kokkos::single(Kokkos::PerThread(team), [=]() { y_loc[rowb - start] = sum; }); in MatMult()
66 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, start, end), [=](const int rowb) { in MatMultTranspose() argument
67 int rowa = ic[rowb]; in MatMultTranspose()
71 const PetscScalar xx = x_loc[rowb - start]; // rowb = ic[rowa] = ic[r[rowb]] in MatMultTranspose()
158 parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](int rowb) { in BJSolve_TFQMR() argument
159 int rowa = ic[rowb]; in BJSolve_TFQMR()
160 R[rowb - start] = glb_b[rowa]; in BJSolve_TFQMR()
161 XX[rowb - start] = 0; in BJSolve_TFQMR()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 5685 PetscErrorCode MatGetBrowsOfAcols(Mat A, Mat B, MatReuse scall, IS *rowb, IS *colb, Mat *B_seq) in MatGetBrowsOfAcols() argument
5714 …PetscCheck(rowb && colb, PETSC_COMM_SELF, PETSC_ERR_SUP, "IS rowb and colb must be provided for MA… in MatGetBrowsOfAcols()
5715 isrowb = *rowb; in MatGetBrowsOfAcols()
5723 if (!rowb) { in MatGetBrowsOfAcols()
5726 *rowb = isrowb; in MatGetBrowsOfAcols()
|