Lines Matching refs:member

73   KOKKOS_INLINE_FUNCTION void operator()(const MemberType &member) const
75     const int first_matrix = static_cast<int>(member.league_rank()) * _N_team;
77     const int last_matrix  = (static_cast<int>(member.league_rank() + 1) * _N_team < N ? static_cast<int>(member.league_rank() + 1) * _N_team : N);
78     const int graphID      = static_cast<int>(member.league_rank());
88     ScratchPadIntViewType r(member.team_scratch(1), _r.extent(1));
89     ScratchPadIntViewType c(member.team_scratch(1), _c.extent(1));
91     TeamVectorCopy1D::invoke(member, Kokkos::subview(_r, graphID, Kokkos::ALL), r);
92     TeamVectorCopy1D::invoke(member, Kokkos::subview(_c, graphID, Kokkos::ALL), c);
95     ScratchPadValuesViewType diag(member.team_scratch(1), last_matrix - first_matrix, _diag.extent(1));
98     KokkosBatched::TeamVectorCopy<MemberType>::invoke(member, Kokkos::subview(_diag, Kokkos::make_pair(first_matrix, last_matrix), Kokkos::ALL), diag);
102     KokkosBatched::TeamVectorGMRES<MemberType>::template invoke<Operator, VectorViewType, PrecOperator, KrylovHandleType>(member, A, b, x, P, _handle);
199         if (blkID % Nsolves_team == 0) {                                                        // first matrix on this member