Lines Matching refs:exec
143 auto exec = PetscGetKokkosExecutionSpace(); in MatSeqAIJKokkosSyncHost() local
150 PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, exec)); in MatSeqAIJKokkosSyncHost()
1265 auto exec = PetscGetKokkosExecutionSpace(); in MatAXPY_SeqAIJKokkos() local
1294 KokkosBlas::axpy(exec, alpha, Xa, Ya); in MatAXPY_SeqAIJKokkos()
1301 Kokkos::TeamPolicy<>(exec, Y->rmap->n, 1), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatAXPY_SeqAIJKokkos()
1557 auto exec = PetscGetKokkosExecutionSpace(); in MatSetSeqAIJKokkosWithCSRMatrix() local
1571 PetscCall(KokkosDualViewSyncHost(akok->i_dual, exec)); /* We always need sync'ed i, j on host */ in MatSetSeqAIJKokkosWithCSRMatrix()
1572 PetscCall(KokkosDualViewSyncHost(akok->j_dual, exec)); in MatSetSeqAIJKokkosWithCSRMatrix()
1782 auto exec = PetscGetKokkosExecutionSpace(); in MatSolve_SeqAIJKokkos_Cholesky() local
1805 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAMBDA(const PetscInt … in MatSolve_SeqAIJKokkos_Cholesky()
1808 …PetscCallCXX(sptrsv_solve(exec, &factors->khUt, factors->iUt_d, factors->jUt_d, factors->aUt_d, B,… in MatSolve_SeqAIJKokkos_Cholesky()
1812 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAMBDA(const PetscInt … in MatSolve_SeqAIJKokkos_Cholesky()
1820 …PetscCallCXX(sptrsv_solve(exec, &factors->khU, factors->iU_d, factors->jU_d, factors->aU_d, Y, X)); in MatSolve_SeqAIJKokkos_Cholesky()
1823 …if (!identity) PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAMBDA(… in MatSolve_SeqAIJKokkos_Cholesky()
1836 auto exec = PetscGetKokkosExecutionSpace(); in MatSolve_SeqAIJKokkos_LU() local
1859 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAMBDA(const PetscInt … in MatSolve_SeqAIJKokkos_LU()
1862 …PetscCallCXX(sptrsv_solve(exec, &factors->khL, factors->iL_d, factors->jL_d, factors->aL_d, B, Y)); in MatSolve_SeqAIJKokkos_LU()
1870 …PetscCallCXX(sptrsv_solve(exec, &factors->khU, factors->iU_d, factors->jU_d, factors->aU_d, Y, X)); in MatSolve_SeqAIJKokkos_LU()
1873 …if (!col_identity) PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAM… in MatSolve_SeqAIJKokkos_LU()
1887 auto exec = PetscGetKokkosExecutionSpace(); in MatSolveTranspose_SeqAIJKokkos_LU() local
1910 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAMBDA(const PetscInt … in MatSolveTranspose_SeqAIJKokkos_LU()
1913 …PetscCallCXX(sptrsv_solve(exec, &factors->khUt, factors->iUt_d, factors->jUt_d, factors->aUt_d, B,… in MatSolveTranspose_SeqAIJKokkos_LU()
1921 …PetscCallCXX(sptrsv_solve(exec, &factors->khLt, factors->iLt_d, factors->jLt_d, factors->aLt_d, Y,… in MatSolveTranspose_SeqAIJKokkos_LU()
1924 …if (!row_identity) PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAM… in MatSolveTranspose_SeqAIJKokkos_LU()