Home
last modified time | relevance | path

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

/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx61 …VectorViewType &X, const VectorViewType &B, const int N_team, const int team_size, const int vecto… in Functor_TestBatchedTeamVectorGMRES()
62 …_D(D), _r(r), _c(c), _X(X), _B(B), _N_team(N_team), _team_size(team_size), _vector_length(vector_l… in Functor_TestBatchedTeamVectorGMRES()
67 …VectorViewType &X, const VectorViewType &B, const int N_team, const int team_size, const int vecto… in Functor_TestBatchedTeamVectorGMRES()
68 …_D(D), _diag(diag), _r(r), _c(c), _X(X), _B(B), _N_team(N_team), _team_size(team_size), _vector_le… in Functor_TestBatchedTeamVectorGMRES()
155 …ai, const PetscInt *glb_Aaj, const PetscScalar *glb_Aaa, const PetscInt team_size, MatInfo info, c… in PCApply_BJKOKKOSKERNELS() argument
196 …"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()
H A Dbjkokkos.kokkos.cxx538 …okkos::DefaultExecutionSpace().concurrency(), openmp = !!(conc < 1000), team_size = (openmp == 0 &… in PCApply_BJKOKKOS() local
593 …ly_BJKOKKOSKERNELS(pc, glb_bdata, glb_xdata, glb_Aai, glb_Aaj, glb_Aaa, team_size, info, batch_sz,… in PCApply_BJKOKKOS()
628 …jac->n, scr_bytes_team_shared, global_buff_words, rtol, (int)nBlk, (int)team_size, PCBJKOKKOS_VEC_… in PCApply_BJKOKKOS()
632 …"Solve", Kokkos::TeamPolicy<Kokkos::LaunchBounds<256, 4>>(nBlk, team_size, PCBJKOKKOS_VEC_SIZE).se… in PCApply_BJKOKKOS()
974 …okkos::DefaultExecutionSpace().concurrency(), openmp = !!(conc < 1000), team_size = (openmp == 0 &… in PCSetUp_BJKOKKOS() local
981 …"Diag", Kokkos::TeamPolicy<>(jac->nBlocks, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const te… in PCSetUp_BJKOKKOS()
/petsc/src/ts/utils/dmplexlandau/kokkos/
H A Dlandau.kokkos.cxx394 …okkos::DefaultExecutionSpace().concurrency(), openmp = !!(conc < 1000), team_size = (openmp == 0) … in LandauKokkosJacobian() local
512 … scr_bytes_fdf, KOKKOS_SHARED_LEVEL, (int)(num_cells_batch * batch_sz), team_size, vector_size, nf… in LandauKokkosJacobian()
514 …"f, df", Kokkos::TeamPolicy<>(num_cells_batch * batch_sz, team_size, vector_size).set_scratch_size… in LandauKokkosJacobian()
727 …: "global", maximum_shared_mem_size, (int)(num_cells_batch * batch_sz), team_size, vector_size, nf… in LandauKokkosJacobian()
728 …mPolicy<Kokkos::LaunchBounds<256, lbound2>>(num_cells_batch * batch_sz, team_size, vector_size).se… in LandauKokkosJacobian()
736 …m size=%d vector size=%d #face=%d Nb=%" PetscInt_FMT ", %s assembly\n", team_size, vector_size, nf… in LandauKokkosJacobian()
738 …"Mass", Kokkos::TeamPolicy<>(num_cells_batch * batch_sz, team_size, vector_size), KOKKOS_LAMBDA(co… in LandauKokkosJacobian()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx184 …ers(PetscInt numRows, PetscInt nnz, PetscInt rows_per_thread, PetscInt &team_size, PetscInt &vecto… in MatMergeGetLaunchParameters() argument
215 if (team_size < 1) { in MatMergeGetLaunchParameters()
217 team_size = 256 / vector_length; in MatMergeGetLaunchParameters()
219 team_size = 1; in MatMergeGetLaunchParameters()
223 rows_per_team = rows_per_thread * team_size; in MatMergeGetLaunchParameters()