/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/HIP/ |
H A D | Kokkos_HIP_Parallel_Team.hpp | 75 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 D | Kokkos_HIP_Parallel_Team.hpp | 75 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 D | Kokkos_HIP_Parallel_Team.hpp | 75 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 D | Kokkos_SYCL_Parallel_Team.hpp | 70 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 D | Kokkos_SYCL_Parallel_Team.hpp | 70 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 D | Kokkos_SYCL_Parallel_Team.hpp | 70 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 D | Kokkos_HostThreadTeam.cpp | 90 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 D | Kokkos_HostThreadTeam.hpp | 113 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 D | Kokkos_HostThreadTeam.cpp | 90 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 D | Kokkos_HostThreadTeam.hpp | 113 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 D | Kokkos_HostThreadTeam.cpp | 90 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 D | Kokkos_HostThreadTeam.hpp | 113 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 D | Kokkos_ThreadsTeam.hpp | 86 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 D | Kokkos_ThreadsTeam.hpp | 86 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 D | Kokkos_ThreadsTeam.hpp | 86 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 D | Kokkos_OpenMP_Team.hpp | 77 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 D | Kokkos_OpenMP_Team.hpp | 77 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 D | Kokkos_OpenMP_Team.hpp | 77 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 D | Kokkos_Cuda_Parallel.hpp | 99 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 D | Kokkos_Cuda_Parallel.hpp | 99 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 D | Kokkos_Cuda_Parallel.hpp | 99 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 D | Kokkos_OpenMPTarget_Task.cpp | 69 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 D | Kokkos_OpenMPTarget_Exec.hpp | 554 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 D | Kokkos_OpenMPTarget_Task.cpp | 69 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 D | Kokkos_OpenMPTarget_Task.cpp | 69 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()
|