Home
last modified time | relevance | path

Searched refs:m_team_size (Results 1 – 25 of 36) sorted by relevance

12

/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/HIP/
H A DKokkos_HIP_Parallel_Team.hpp75 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
89 m_team_size = p.m_team_size; in TeamPolicyInternal()
237 m_team_size(-1), in TeamPolicyInternal()
430 int m_team_size; member in Kokkos::Impl::ParallelFor
525 m_team_size = in ParallelFor()
526 m_team_size >= 0 in ParallelFor()
546 m_team_size <= 0 in ParallelFor()
647 int m_team_size; member in Kokkos::Impl::ParallelReduce
895 m_team_size =
896 m_team_size >= 0
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/HIP/
H A DKokkos_HIP_Parallel_Team.hpp75 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
89 m_team_size = p.m_team_size; in TeamPolicyInternal()
237 m_team_size(-1), in TeamPolicyInternal()
430 int m_team_size; member in Kokkos::Impl::ParallelFor
525 m_team_size = in ParallelFor()
526 m_team_size >= 0 in ParallelFor()
546 m_team_size <= 0 in ParallelFor()
647 int m_team_size; member in Kokkos::Impl::ParallelReduce
895 m_team_size =
896 m_team_size >= 0
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/HIP/
H A DKokkos_HIP_Parallel_Team.hpp75 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
89 m_team_size = p.m_team_size; in TeamPolicyInternal()
237 m_team_size(-1), in TeamPolicyInternal()
430 int m_team_size; member in Kokkos::Impl::ParallelFor
525 m_team_size = in ParallelFor()
526 m_team_size >= 0 in ParallelFor()
546 m_team_size <= 0 in ParallelFor()
647 int m_team_size; member in Kokkos::Impl::ParallelReduce
895 m_team_size =
896 m_team_size >= 0
[all …]
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/SYCL/
H A DKokkos_SYCL_Parallel_Team.hpp70 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
84 m_team_size = p.m_team_size; in TeamPolicyInternal()
186 m_team_size(-1), in TeamPolicyInternal()
199 m_team_size(team_size_request), in TeamPolicyInternal()
371 int m_team_size; member in Kokkos::Impl::ParallelFor
442 if (m_team_size < 0) m_team_size = 32; in ParallelFor()
528 int m_team_size; member in Kokkos::Impl::ParallelReduce
556 const size_t wgroup_size = m_team_size; in sycl_direct_launch()
763 if (m_team_size < 0) m_team_size = 32; in initialize()
766 if ((m_team_size & m_team_size - 1) || m_team_size < 2) { in initialize()
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/SYCL/
H A DKokkos_SYCL_Parallel_Team.hpp70 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
84 m_team_size = p.m_team_size; in TeamPolicyInternal()
186 m_team_size(-1), in TeamPolicyInternal()
199 m_team_size(team_size_request), in TeamPolicyInternal()
371 int m_team_size; member in Kokkos::Impl::ParallelFor
442 if (m_team_size < 0) m_team_size = 32; in ParallelFor()
528 int m_team_size; member in Kokkos::Impl::ParallelReduce
556 const size_t wgroup_size = m_team_size; in sycl_direct_launch()
763 if (m_team_size < 0) m_team_size = 32; in initialize()
766 if ((m_team_size & m_team_size - 1) || m_team_size < 2) { in initialize()
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/SYCL/
H A DKokkos_SYCL_Parallel_Team.hpp70 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
84 m_team_size = p.m_team_size; in TeamPolicyInternal()
186 m_team_size(-1), in TeamPolicyInternal()
199 m_team_size(team_size_request), in TeamPolicyInternal()
371 int m_team_size; member in Kokkos::Impl::ParallelFor
442 if (m_team_size < 0) m_team_size = 32; in ParallelFor()
528 int m_team_size; member in Kokkos::Impl::ParallelReduce
556 const size_t wgroup_size = m_team_size; in sycl_direct_launch()
763 if (m_team_size < 0) m_team_size = 32; in initialize()
766 if ((m_team_size & m_team_size - 1) || m_team_size < 2) { in initialize()
[all …]
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/impl/
H A DKokkos_HostThreadTeam.cpp90 mem->m_team_size = 1; in organize_pool()
116 m_team_size = 1; in disband_pool()
130 m_team_rank == 0 && m_team_size == 1 && m_team_alloc == 1 && in organize_team()
161 m_team_size = team_size; in organize_team()
196 m_team_size = 1; in disband_team()
213 if (1 == m_team_size || team_rendezvous()) { in get_work_stealing()
274 m_steal_rank + m_team_alloc + m_team_size <= m_pool_size in get_work_stealing()
288 if (1 < m_team_size) { in get_work_stealing()
294 } else if (1 < m_team_size) { in get_work_stealing()
H A DKokkos_HostThreadTeam.hpp113 int m_team_size; member in Kokkos::Impl::HostThreadTeamData
132 HostBarrier::wait(ptr, m_team_size, m_team_rendezvous_step); in team_rendezvous()
144 HostBarrier::wait(ptr, m_team_size, m_team_rendezvous_step); in team_rendezvous()
190 m_team_size(1), in HostThreadTeamData()
457 int team_size() const noexcept { return m_data.m_team_size; } in team_size()
507 if (1 < m_data.m_team_size) { in team_broadcast()
554 if (1 < m_data.m_team_size) { in team_broadcast()
593 if (1 < m_data.m_team_size) { in team_reduce()
612 for (int i = 1; i < m_data.m_team_size; ++i) { in team_reduce()
716 for (int i = 1; i < m_data.m_team_size; ++i) { in team_scan()
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/impl/
H A DKokkos_HostThreadTeam.cpp90 mem->m_team_size = 1; in organize_pool()
116 m_team_size = 1; in disband_pool()
130 m_team_rank == 0 && m_team_size == 1 && m_team_alloc == 1 && in organize_team()
161 m_team_size = team_size; in organize_team()
196 m_team_size = 1; in disband_team()
213 if (1 == m_team_size || team_rendezvous()) { in get_work_stealing()
274 m_steal_rank + m_team_alloc + m_team_size <= m_pool_size in get_work_stealing()
288 if (1 < m_team_size) { in get_work_stealing()
294 } else if (1 < m_team_size) { in get_work_stealing()
H A DKokkos_HostThreadTeam.hpp113 int m_team_size; member in Kokkos::Impl::HostThreadTeamData
132 HostBarrier::wait(ptr, m_team_size, m_team_rendezvous_step); in team_rendezvous()
144 HostBarrier::wait(ptr, m_team_size, m_team_rendezvous_step); in team_rendezvous()
190 m_team_size(1), in HostThreadTeamData()
457 int team_size() const noexcept { return m_data.m_team_size; } in team_size()
507 if (1 < m_data.m_team_size) { in team_broadcast()
554 if (1 < m_data.m_team_size) { in team_broadcast()
593 if (1 < m_data.m_team_size) { in team_reduce()
612 for (int i = 1; i < m_data.m_team_size; ++i) { in team_reduce()
716 for (int i = 1; i < m_data.m_team_size; ++i) { in team_scan()
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/impl/
H A DKokkos_HostThreadTeam.cpp90 mem->m_team_size = 1; in organize_pool()
116 m_team_size = 1; in disband_pool()
130 m_team_rank == 0 && m_team_size == 1 && m_team_alloc == 1 && in organize_team()
161 m_team_size = team_size; in organize_team()
196 m_team_size = 1; in disband_team()
213 if (1 == m_team_size || team_rendezvous()) { in get_work_stealing()
274 m_steal_rank + m_team_alloc + m_team_size <= m_pool_size in get_work_stealing()
288 if (1 < m_team_size) { in get_work_stealing()
294 } else if (1 < m_team_size) { in get_work_stealing()
H A DKokkos_HostThreadTeam.hpp113 int m_team_size; member in Kokkos::Impl::HostThreadTeamData
132 HostBarrier::wait(ptr, m_team_size, m_team_rendezvous_step); in team_rendezvous()
144 HostBarrier::wait(ptr, m_team_size, m_team_rendezvous_step); in team_rendezvous()
190 m_team_size(1), in HostThreadTeamData()
457 int team_size() const noexcept { return m_data.m_team_size; } in team_size()
507 if (1 < m_data.m_team_size) { in team_broadcast()
554 if (1 < m_data.m_team_size) { in team_broadcast()
593 if (1 < m_data.m_team_size) { in team_reduce()
612 for (int i = 1; i < m_data.m_team_size; ++i) { in team_reduce()
716 for (int i = 1; i < m_data.m_team_size; ++i) { in team_scan()
[all …]
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/Threads/
H A DKokkos_ThreadsTeam.hpp86 int m_team_size; member in Kokkos::Impl::ThreadsExecTeamMember
358 for (int i = m_team_size; i--;) {
365 for (int i = m_team_size; i--;) {
403 m_team_size(team.team_size()), in ThreadsExecTeamMember()
439 if ((team.team_alloc() > m_team_size) in ThreadsExecTeamMember()
453 m_team_size = team.team_size(); in ThreadsExecTeamMember()
485 m_team_size(1), in ThreadsExecTeamMember()
570 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
598 (m_team_size + team_grain - 1 <= 0) in init()
631 m_team_size = p.m_team_size; in TeamPolicyInternal()
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/Threads/
H A DKokkos_ThreadsTeam.hpp86 int m_team_size; member in Kokkos::Impl::ThreadsExecTeamMember
358 for (int i = m_team_size; i--;) {
365 for (int i = m_team_size; i--;) {
403 m_team_size(team.team_size()), in ThreadsExecTeamMember()
439 if ((team.team_alloc() > m_team_size) in ThreadsExecTeamMember()
453 m_team_size = team.team_size(); in ThreadsExecTeamMember()
485 m_team_size(1), in ThreadsExecTeamMember()
570 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
598 (m_team_size + team_grain - 1 <= 0) in init()
631 m_team_size = p.m_team_size; in TeamPolicyInternal()
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/Threads/
H A DKokkos_ThreadsTeam.hpp86 int m_team_size; member in Kokkos::Impl::ThreadsExecTeamMember
358 for (int i = m_team_size; i--;) {
365 for (int i = m_team_size; i--;) {
403 m_team_size(team.team_size()), in ThreadsExecTeamMember()
439 if ((team.team_alloc() > m_team_size) in ThreadsExecTeamMember()
453 m_team_size = team.team_size(); in ThreadsExecTeamMember()
485 m_team_size(1), in ThreadsExecTeamMember()
570 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
598 (m_team_size + team_grain - 1 <= 0) in init()
631 m_team_size = p.m_team_size; in TeamPolicyInternal()
[all …]
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/OpenMP/
H A DKokkos_OpenMP_Team.hpp77 m_team_size = p.m_team_size; in TeamPolicyInternal()
138 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
161 m_team_size = team_size_request < team_max ? team_size_request : team_max; in init()
165 team_grain * ((m_team_size + team_grain - 1) / team_grain); in init()
178 inline int team_size() const { return m_team_size; } in team_size()
183 m_team_size = new_team_size; in impl_set_team_size()
187 if (team_size_ < 0) team_size_ = m_team_size; in scratch_size()
/dports/devel/kokkos/kokkos-3.4.01/core/src/OpenMP/
H A DKokkos_OpenMP_Team.hpp77 m_team_size = p.m_team_size; in TeamPolicyInternal()
138 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
161 m_team_size = team_size_request < team_max ? team_size_request : team_max; in init()
165 team_grain * ((m_team_size + team_grain - 1) / team_grain); in init()
178 inline int team_size() const { return m_team_size; } in team_size()
183 m_team_size = new_team_size; in impl_set_team_size()
187 if (team_size_ < 0) team_size_ = m_team_size; in scratch_size()
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/OpenMP/
H A DKokkos_OpenMP_Team.hpp77 m_team_size = p.m_team_size; in TeamPolicyInternal()
138 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
161 m_team_size = team_size_request < team_max ? team_size_request : team_max; in init()
165 team_grain * ((m_team_size + team_grain - 1) / team_grain); in init()
178 inline int team_size() const { return m_team_size; } in team_size()
183 m_team_size = new_team_size; in impl_set_team_size()
187 if (team_size_ < 0) team_size_ = m_team_size; in scratch_size()
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/Cuda/
H A DKokkos_Cuda_Parallel.hpp99 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
114 m_team_size = p.m_team_size; in TeamPolicyInternal()
272 m_team_size(-1), in TeamPolicyInternal()
684 int m_team_size; member in Kokkos::Impl::ParallelFor
781 m_team_size = in ParallelFor()
782 m_team_size >= 0 in ParallelFor()
783 ? m_team_size in ParallelFor()
801 m_team_size <= 0 in ParallelFor()
1584 int m_team_size; member in Kokkos::Impl::ParallelReduce
1877 m_team_size =
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/Cuda/
H A DKokkos_Cuda_Parallel.hpp99 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
114 m_team_size = p.m_team_size; in TeamPolicyInternal()
272 m_team_size(-1), in TeamPolicyInternal()
684 int m_team_size; member in Kokkos::Impl::ParallelFor
781 m_team_size = in ParallelFor()
782 m_team_size >= 0 in ParallelFor()
783 ? m_team_size in ParallelFor()
801 m_team_size <= 0 in ParallelFor()
1584 int m_team_size; member in Kokkos::Impl::ParallelReduce
1877 m_team_size =
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/Cuda/
H A DKokkos_Cuda_Parallel.hpp99 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
114 m_team_size = p.m_team_size; in TeamPolicyInternal()
272 m_team_size(-1), in TeamPolicyInternal()
684 int m_team_size; member in Kokkos::Impl::ParallelFor
781 m_team_size = in ParallelFor()
782 m_team_size >= 0 in ParallelFor()
783 ? m_team_size in ParallelFor()
801 m_team_size <= 0 in ParallelFor()
1584 int m_team_size; member in Kokkos::Impl::ParallelReduce
1877 m_team_size =
[all …]
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/OpenMPTarget/
H A DKokkos_OpenMPTarget_Task.cpp69 m_team_size(1) {} in TaskExec()
80 m_team_size(arg_team_size) { in TaskExec()
90 for (int i = 0; i < m_team_size; ++i) { in TaskExec()
H A DKokkos_OpenMPTarget_Exec.hpp554 int m_team_size; member in Kokkos::Impl::OpenMPTargetExecTeamMember
614 KOKKOS_INLINE_FUNCTION int team_size() const { return m_team_size; } in team_size()
660 for (int i = m_team_rank; i < n_values; i += m_team_size) { in team_reduce()
666 for (int k = 0; k < m_team_size; k += n_values) { in team_reduce()
766 m_team_size(team_size),
850 int m_team_size; member in Kokkos::Impl::TeamPolicyInternal
870 m_team_size = team_size_request; in init()
882 inline void impl_set_team_size(const size_t size) { m_team_size = size; } in impl_set_team_size()
890 inline int team_size() const { return m_team_size; } in team_size()
893 if (team_size_ < 0) team_size_ = m_team_size; in scratch_size()
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/OpenMPTarget/
H A DKokkos_OpenMPTarget_Task.cpp69 m_team_size(1) {} in TaskExec()
80 m_team_size(arg_team_size) { in TaskExec()
90 for (int i = 0; i < m_team_size; ++i) { in TaskExec()
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/OpenMPTarget/
H A DKokkos_OpenMPTarget_Task.cpp69 m_team_size(1) {} in TaskExec()
80 m_team_size(arg_team_size) { in TaskExec()
90 for (int i = 0; i < m_team_size; ++i) { in TaskExec()

12