Home
last modified time | relevance | path

Searched refs:KokkosTeamMemberType (Results 1 – 4 of 4) sorted by relevance

/petsc/src/ksp/pc/impls/vpbjacobi/kokkos/
H A Dvpbjacobi_kok.kokkos.cxx91 …sExecutionSpace(), jac->nblocks, Kokkos::AUTO()), KOKKOS_LAMBDA(const KokkosTeamMemberType &team) { in PCApplyOrTranspose_VPBJacobi_Kokkos()
101 …KokkosBlas::TeamGemv<KokkosTeamMemberType, KokkosBlas::Trans::Transpose>::invoke(team, 1., B, x1, … in PCApplyOrTranspose_VPBJacobi_Kokkos()
103 …KokkosBlas::TeamGemv<KokkosTeamMemberType, KokkosBlas::Trans::NoTranspose>::invoke(team, 1., B, x1… in PCApplyOrTranspose_VPBJacobi_Kokkos()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp55 using KokkosTeamMemberType = Kokkos::TeamPolicy<DefaultExecutionSpace>::member_type; typedef
H A Daijkok.kokkos.cxx752 …(PetscGetKokkosExecutionSpace(), m, Kokkos::AUTO()), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatSeqAIJKokkosMergeMats()
781 …(PetscGetKokkosExecutionSpace(), m, Kokkos::AUTO()), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatSeqAIJKokkosMergeMats()
1110 …(PetscGetKokkosExecutionSpace(), m, Kokkos::AUTO()), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatDiagonalScale_SeqAIJKokkos()
1301 Kokkos::TeamPolicy<>(exec, Y->rmap->n, 1), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatAXPY_SeqAIJKokkos()
1512 …PetscGetKokkosExecutionSpace(), nblocks, ts), KOKKOS_LAMBDA(const KokkosTeamMemberType &teamMember… in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1546 …KokkosBatched::TeamLU<KokkosTeamMemberType, KokkosBatched::Algo::LU::Unblocked>::invoke(teamMember… in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1547 …KokkosBatched::TeamInverseLU<KokkosTeamMemberType, KokkosBatched::Algo::InverseLU::Unblocked>::inv… in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx647 …ExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatMPIAIJKokkosReduceBegin()
974 …ExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatMPIAIJKokkosBcastBegin()
1025 …ExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatMPIAIJKokkosBcastEnd()