Searched refs:_N_team (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 53 const int _N_team, _team_size, _vector_length; member
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()
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()
75 const int first_matrix = static_cast<int>(member.league_rank()) * _N_team; in operator ()()
77 …atic_cast<int>(member.league_rank() + 1) * _N_team < N ? static_cast<int>(member.league_rank() + 1… in operator ()()
111 …Kokkos::TeamPolicy<DeviceType> auto_policy(ceil(1. * _D.extent(0) / _N_team), Kokkos::AUTO(), Kokk… in run()
112 …Kokkos::TeamPolicy<DeviceType> tuned_policy(ceil(1. * _D.extent(0) / _N_team), _team_size, _vector… in run()
133 size_t bytes_1D = ViewType2D::shmem_size(_N_team, 1); in run()
136 size_t bytes_2D_1 = ViewType2D::shmem_size(_N_team, _X.extent(1)); in run()
137 size_t bytes_2D_2 = ViewType2D::shmem_size(_N_team, maximum_iteration + 1); in run()
|