Home
last modified time | relevance | path

Searched refs:PCBJKOKKOS_VEC_SIZE (Results 1 – 3 of 3) sorted by relevance

/petsc/include/petsc/private/
H A Dpcbjkokkosimpl.h19 #define PCBJKOKKOS_VEC_SIZE 16 macro
/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx196 …"rowOffsets+map", Kokkos::TeamPolicy<>(Nsolves, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(con… in PCApply_BJKOKKOSKERNELS()
224 …"copy matrix", Kokkos::TeamPolicy<>(Nsolves /* /batch_sz */, team_size, PCBJKOKKOS_VEC_SIZE), KOKK… in PCApply_BJKOKKOSKERNELS()
259 …"map", Kokkos::TeamPolicy<>(Nsolves /* /batch_sz */, -1, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const… in PCApply_BJKOKKOSKERNELS()
H A Dbjkokkos.kokkos.cxx538 …ncurrency(), openmp = !!(conc < 1000), team_size = (openmp == 0 && PCBJKOKKOS_VEC_SIZE != 1) ? PCB… in PCApply_BJKOKKOS()
628 …s_team_shared, global_buff_words, rtol, (int)nBlk, (int)team_size, PCBJKOKKOS_VEC_SIZE, nShareVec,… in PCApply_BJKOKKOS()
632 … Kokkos::TeamPolicy<Kokkos::LaunchBounds<256, 4>>(nBlk, team_size, PCBJKOKKOS_VEC_SIZE).set_scratc… in PCApply_BJKOKKOS()
974 …ncurrency(), openmp = !!(conc < 1000), team_size = (openmp == 0 && PCBJKOKKOS_VEC_SIZE != 1) ? PCB… in PCSetUp_BJKOKKOS()
981 …"Diag", Kokkos::TeamPolicy<>(jac->nBlocks, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const te… in PCSetUp_BJKOKKOS()