Searched refs:y_loc (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkos.kokkos.cxx | 46 …tscInt *ic, const PetscInt start, const PetscInt end, const PetscScalar *x_loc, PetscScalar *y_loc) in MatMult() argument
55 Kokkos::single(Kokkos::PerThread(team), [=]() { y_loc[rowb - start] = sum; }); in MatMult()
62 …tscInt *ic, const PetscInt start, const PetscInt end, const PetscScalar *x_loc, PetscScalar *y_loc) in MatMultTranspose() argument
64 Kokkos::parallel_for(Kokkos::TeamVectorRange(team, end - start), [=](int i) { y_loc[i] = 0; }); in MatMultTranspose()
74 Kokkos::atomic_fetch_add(&y_loc[r[aj[i]] - start], val); in MatMultTranspose()
|