Home
last modified time | relevance | path

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

/petsc/include/petsc/private/
H A Dpcbjkokkosimpl.h67 typedef Kokkos::TeamPolicy<>::member_type team_member; typedef
/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx196 …kos::TeamPolicy<>(Nsolves, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const team_member team) { in PCApply_BJKOKKOSKERNELS()
216 …x sum", Kokkos::TeamPolicy<>(Nsolves / Nsolves_team, 1, 1), KOKKOS_LAMBDA(const team_member team) { in PCApply_BJKOKKOSKERNELS()
224 …>(Nsolves /* /batch_sz */, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const team_member team) { in PCApply_BJKOKKOSKERNELS()
259 …Policy<>(Nsolves /* /batch_sz */, -1, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const team_member team) { in PCApply_BJKOKKOSKERNELS()
H A Dbjkokkos.kokkos.cxx46 KOKKOS_INLINE_FUNCTION PetscErrorCode MatMult(const team_member team, const PetscInt *glb_Aai, cons… in MatMult()
62 KOKKOS_INLINE_FUNCTION PetscErrorCode MatMultTranspose(const team_member team, const PetscInt *glb_… in MatMultTranspose()
88 static KOKKOS_INLINE_FUNCTION PetscErrorCode BJSolve_TFQMR(const team_member team, const PetscInt *… in BJSolve_TFQMR()
330 static KOKKOS_INLINE_FUNCTION PetscErrorCode BJSolve_BICG(const team_member team, const PetscInt *g… in BJSolve_BICG()
632 …KKOS_SHARED_LEVEL, Kokkos::PerTeam(scr_bytes_team_shared)), KOKKOS_LAMBDA(const team_member team) { in PCApply_BJKOKKOS()
981 …TeamPolicy<>(jac->nBlocks, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const team_member team) { in PCSetUp_BJKOKKOS()
/petsc/src/ts/utils/dmplexlandau/kokkos/
H A Dlandau.kokkos.cxx37 typedef Kokkos::TeamPolicy<>::member_type team_member; typedef
514 …_size(KOKKOS_SHARED_LEVEL, Kokkos::PerTeam(scr_bytes_fdf)), KOKKOS_LAMBDA(const team_member team) { in LandauKokkosJacobian()
585 auto jac_lambda = KOKKOS_LAMBDA(const team_member team) in LandauKokkosJacobian()
738 …licy<>(num_cells_batch * batch_sz, team_size, vector_size), KOKKOS_LAMBDA(const team_member team) { in LandauKokkosJacobian()