Searched refs:vector_length (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 61 …t VectorViewType &B, const int N_team, const int team_size, const int vector_length, const int N_i… in Functor_TestBatchedTeamVectorGMRES()
62 … _X(X), _B(B), _N_team(N_team), _team_size(team_size), _vector_length(vector_length), _N_iteration… in Functor_TestBatchedTeamVectorGMRES()
67 …t VectorViewType &B, const int N_team, const int team_size, const int vector_length, const int N_i… in Functor_TestBatchedTeamVectorGMRES()
68 … _X(X), _B(B), _N_team(N_team), _team_size(team_size), _vector_length(vector_length), _N_iteration… in Functor_TestBatchedTeamVectorGMRES()
249 const int vector_length = -1; in PCApply_BJKOKKOSKERNELS() local
255 …inv_diag, rowOffsets, colIndices, batch_x, batch_b, Nsolves_team, -1, vector_length, n_iterations,… in PCApply_BJKOKKOSKERNELS()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 184 …etscInt nnz, PetscInt rows_per_thread, PetscInt &team_size, PetscInt &vector_length, PetscInt &row… in MatMergeGetLaunchParameters() argument
200 if (vector_length < 1) { in MatMergeGetLaunchParameters()
201 vector_length = 1; in MatMergeGetLaunchParameters()
202 while (vector_length < max_vector_length && vector_length * 6 < nnz_per_row) vector_length *= 2; in MatMergeGetLaunchParameters()
217 team_size = 256 / vector_length; in MatMergeGetLaunchParameters()
|