/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/Cuda/ |
H A D | Kokkos_Cuda_Parallel.hpp | 886 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1182 m_scratch_space(nullptr), in ParallelReduce() 1200 m_scratch_space(nullptr), in ParallelReduce() 1249 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1499 m_scratch_space(nullptr), in ParallelReduce() 1513 m_scratch_space(nullptr), in ParallelReduce() 1575 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1864 m_scratch_space(nullptr), in ParallelReduce() 1963 m_scratch_space(nullptr), in ParallelReduce() 2081 size_type* m_scratch_space; member in Kokkos::Impl::ParallelScan [all …]
|
H A D | Kokkos_Cuda_ReduceScan.hpp | 137 const JoinOp& join, Cuda::size_type* const m_scratch_space, in cuda_inter_block_reduction() argument 155 pointer_type global = ((pointer_type)m_scratch_space) + blockIdx.x; in cuda_inter_block_reduction() 178 pointer_type const volatile global = (pointer_type)m_scratch_space; in cuda_inter_block_reduction() 245 (void)m_scratch_space; in cuda_inter_block_reduction() 335 Cuda::size_type* const m_scratch_space, in cuda_inter_block_reduction() argument 351 pointer_type global = ((pointer_type)m_scratch_space) + blockIdx.x; in cuda_inter_block_reduction() 375 pointer_type const volatile global = (pointer_type)m_scratch_space; in cuda_inter_block_reduction() 441 (void)m_scratch_space; in cuda_inter_block_reduction()
|
/dports/devel/kokkos/kokkos-3.4.01/core/src/Cuda/ |
H A D | Kokkos_Cuda_Parallel.hpp | 886 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1182 m_scratch_space(nullptr), in ParallelReduce() 1200 m_scratch_space(nullptr), in ParallelReduce() 1249 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1499 m_scratch_space(nullptr), in ParallelReduce() 1513 m_scratch_space(nullptr), in ParallelReduce() 1575 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1864 m_scratch_space(nullptr), in ParallelReduce() 1963 m_scratch_space(nullptr), in ParallelReduce() 2081 size_type* m_scratch_space; member in Kokkos::Impl::ParallelScan [all …]
|
H A D | Kokkos_Cuda_ReduceScan.hpp | 137 const JoinOp& join, Cuda::size_type* const m_scratch_space, in cuda_inter_block_reduction() argument 155 pointer_type global = ((pointer_type)m_scratch_space) + blockIdx.x; in cuda_inter_block_reduction() 178 pointer_type const volatile global = (pointer_type)m_scratch_space; in cuda_inter_block_reduction() 245 (void)m_scratch_space; in cuda_inter_block_reduction() 335 Cuda::size_type* const m_scratch_space, in cuda_inter_block_reduction() argument 351 pointer_type global = ((pointer_type)m_scratch_space) + blockIdx.x; in cuda_inter_block_reduction() 375 pointer_type const volatile global = (pointer_type)m_scratch_space; in cuda_inter_block_reduction() 441 (void)m_scratch_space; in cuda_inter_block_reduction()
|
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/Cuda/ |
H A D | Kokkos_Cuda_Parallel.hpp | 886 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1182 m_scratch_space(nullptr), in ParallelReduce() 1200 m_scratch_space(nullptr), in ParallelReduce() 1249 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1499 m_scratch_space(nullptr), in ParallelReduce() 1513 m_scratch_space(nullptr), in ParallelReduce() 1575 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 1864 m_scratch_space(nullptr), in ParallelReduce() 1963 m_scratch_space(nullptr), in ParallelReduce() 2081 size_type* m_scratch_space; member in Kokkos::Impl::ParallelScan [all …]
|
H A D | Kokkos_Cuda_ReduceScan.hpp | 137 const JoinOp& join, Cuda::size_type* const m_scratch_space, in cuda_inter_block_reduction() argument 155 pointer_type global = ((pointer_type)m_scratch_space) + blockIdx.x; in cuda_inter_block_reduction() 178 pointer_type const volatile global = (pointer_type)m_scratch_space; in cuda_inter_block_reduction() 245 (void)m_scratch_space; in cuda_inter_block_reduction() 335 Cuda::size_type* const m_scratch_space, in cuda_inter_block_reduction() argument 351 pointer_type global = ((pointer_type)m_scratch_space) + blockIdx.x; in cuda_inter_block_reduction() 375 pointer_type const volatile global = (pointer_type)m_scratch_space; in cuda_inter_block_reduction() 441 (void)m_scratch_space; in cuda_inter_block_reduction()
|
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/HIP/ |
H A D | Kokkos_HIP_Parallel_Range.hpp | 174 size_type* m_scratch_space = nullptr; member in Kokkos::Impl::ParallelReduce 249 m_scratch_space, m_scratch_flags); in run() 259 : m_scratch_space; in run() 292 pointer_type const result = reinterpret_cast<pointer_type>(m_scratch_space); in run() 315 m_scratch_space, result, m_scratch_flags, in run() 367 m_scratch_space = in execute() 407 m_result_ptr, m_scratch_space, size); in execute() 481 size_type* m_scratch_space = nullptr; member in Kokkos::Impl::ParallelScanHIPBase 536 m_scratch_space, m_scratch_flags); in initial() 558 m_scratch_space + word_count.value * (blockIdx.x - 1); in final() [all …]
|
H A D | Kokkos_HIP_Shuffle_Reduce.hpp | 140 Kokkos::Experimental::HIP::size_type* const m_scratch_space, in hip_inter_block_shuffle_reduction() argument 159 reinterpret_cast<pointer_type>(m_scratch_space) + blockIdx.x; in hip_inter_block_shuffle_reduction() 184 reinterpret_cast<pointer_type>(m_scratch_space); in hip_inter_block_shuffle_reduction() 264 Kokkos::Experimental::HIP::size_type* const m_scratch_space, in hip_inter_block_shuffle_reduction() argument 281 reinterpret_cast<pointer_type>(m_scratch_space) + blockIdx.x; in hip_inter_block_shuffle_reduction() 307 reinterpret_cast<pointer_type>(m_scratch_space); in hip_inter_block_shuffle_reduction()
|
H A D | Kokkos_HIP_Parallel_MDRange.hpp | 243 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 282 m_scratch_space, m_scratch_flags)) { in operator ()() 290 : m_scratch_space; in operator ()() 349 m_scratch_space = in execute() 382 m_scratch_space, size); in execute() 405 m_scratch_space(nullptr), in ParallelReduce() 418 m_scratch_space(nullptr), in ParallelReduce()
|
H A D | Kokkos_HIP_Parallel_Team.hpp | 639 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 750 m_scratch_space, m_scratch_flags); in run() 760 : m_scratch_space; in run() 787 : reinterpret_cast<pointer_type>(m_scratch_space); in run() 801 m_scratch_space, result, m_scratch_flags, blockDim.y)) { in run() 827 m_scratch_space = Kokkos::Experimental::Impl::hip_internal_scratch_space( in execute() 855 m_result_ptr, m_scratch_space, size); in execute() 881 m_scratch_space(nullptr), in ParallelReduce() 983 m_scratch_space(nullptr), in ParallelReduce()
|
/dports/devel/kokkos/kokkos-3.4.01/core/src/HIP/ |
H A D | Kokkos_HIP_Parallel_Range.hpp | 174 size_type* m_scratch_space = nullptr; member in Kokkos::Impl::ParallelReduce 249 m_scratch_space, m_scratch_flags); in run() 259 : m_scratch_space; in run() 292 pointer_type const result = reinterpret_cast<pointer_type>(m_scratch_space); in run() 315 m_scratch_space, result, m_scratch_flags, in run() 367 m_scratch_space = in execute() 407 m_result_ptr, m_scratch_space, size); in execute() 481 size_type* m_scratch_space = nullptr; member in Kokkos::Impl::ParallelScanHIPBase 536 m_scratch_space, m_scratch_flags); in initial() 558 m_scratch_space + word_count.value * (blockIdx.x - 1); in final() [all …]
|
H A D | Kokkos_HIP_Shuffle_Reduce.hpp | 140 Kokkos::Experimental::HIP::size_type* const m_scratch_space, in hip_inter_block_shuffle_reduction() argument 159 reinterpret_cast<pointer_type>(m_scratch_space) + blockIdx.x; in hip_inter_block_shuffle_reduction() 184 reinterpret_cast<pointer_type>(m_scratch_space); in hip_inter_block_shuffle_reduction() 264 Kokkos::Experimental::HIP::size_type* const m_scratch_space, in hip_inter_block_shuffle_reduction() argument 281 reinterpret_cast<pointer_type>(m_scratch_space) + blockIdx.x; in hip_inter_block_shuffle_reduction() 307 reinterpret_cast<pointer_type>(m_scratch_space); in hip_inter_block_shuffle_reduction()
|
H A D | Kokkos_HIP_Parallel_MDRange.hpp | 243 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 282 m_scratch_space, m_scratch_flags)) { in operator ()() 290 : m_scratch_space; in operator ()() 349 m_scratch_space = in execute() 382 m_scratch_space, size); in execute() 405 m_scratch_space(nullptr), in ParallelReduce() 418 m_scratch_space(nullptr), in ParallelReduce()
|
H A D | Kokkos_HIP_Parallel_Team.hpp | 639 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 750 m_scratch_space, m_scratch_flags); in run() 760 : m_scratch_space; in run() 787 : reinterpret_cast<pointer_type>(m_scratch_space); in run() 801 m_scratch_space, result, m_scratch_flags, blockDim.y)) { in run() 827 m_scratch_space = Kokkos::Experimental::Impl::hip_internal_scratch_space( in execute() 855 m_result_ptr, m_scratch_space, size); in execute() 881 m_scratch_space(nullptr), in ParallelReduce() 983 m_scratch_space(nullptr), in ParallelReduce()
|
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/HIP/ |
H A D | Kokkos_HIP_Parallel_Range.hpp | 174 size_type* m_scratch_space = nullptr; member in Kokkos::Impl::ParallelReduce 249 m_scratch_space, m_scratch_flags); in run() 259 : m_scratch_space; in run() 292 pointer_type const result = reinterpret_cast<pointer_type>(m_scratch_space); in run() 315 m_scratch_space, result, m_scratch_flags, in run() 367 m_scratch_space = in execute() 407 m_result_ptr, m_scratch_space, size); in execute() 481 size_type* m_scratch_space = nullptr; member in Kokkos::Impl::ParallelScanHIPBase 536 m_scratch_space, m_scratch_flags); in initial() 558 m_scratch_space + word_count.value * (blockIdx.x - 1); in final() [all …]
|
H A D | Kokkos_HIP_Shuffle_Reduce.hpp | 140 Kokkos::Experimental::HIP::size_type* const m_scratch_space, in hip_inter_block_shuffle_reduction() argument 159 reinterpret_cast<pointer_type>(m_scratch_space) + blockIdx.x; in hip_inter_block_shuffle_reduction() 184 reinterpret_cast<pointer_type>(m_scratch_space); in hip_inter_block_shuffle_reduction() 264 Kokkos::Experimental::HIP::size_type* const m_scratch_space, in hip_inter_block_shuffle_reduction() argument 281 reinterpret_cast<pointer_type>(m_scratch_space) + blockIdx.x; in hip_inter_block_shuffle_reduction() 307 reinterpret_cast<pointer_type>(m_scratch_space); in hip_inter_block_shuffle_reduction()
|
H A D | Kokkos_HIP_Parallel_MDRange.hpp | 243 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 282 m_scratch_space, m_scratch_flags)) { in operator ()() 290 : m_scratch_space; in operator ()() 349 m_scratch_space = in execute() 382 m_scratch_space, size); in execute() 405 m_scratch_space(nullptr), in ParallelReduce() 418 m_scratch_space(nullptr), in ParallelReduce()
|
H A D | Kokkos_HIP_Parallel_Team.hpp | 639 size_type* m_scratch_space; member in Kokkos::Impl::ParallelReduce 750 m_scratch_space, m_scratch_flags); in run() 760 : m_scratch_space; in run() 787 : reinterpret_cast<pointer_type>(m_scratch_space); in run() 801 m_scratch_space, result, m_scratch_flags, blockDim.y)) { in run() 827 m_scratch_space = Kokkos::Experimental::Impl::hip_internal_scratch_space( in execute() 855 m_result_ptr, m_scratch_space, size); in execute() 881 m_scratch_space(nullptr), in ParallelReduce() 983 m_scratch_space(nullptr), in ParallelReduce()
|
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/SYCL/ |
H A D | Kokkos_SYCL_Parallel_Scan.hpp | 82 pointer_type m_scratch_space = nullptr; member in Kokkos::Impl::ParallelScanSYCLBase 189 auto global_mem = m_scratch_space; in sycl_direct_launch() 206 scan_internal(q, functor, m_scratch_space, len); in sycl_direct_launch() 210 auto global_mem = m_scratch_space; in sycl_direct_launch() 241 m_scratch_space = result_memory.get(); in impl_execute() 292 &m_returnvalue, Base::m_scratch_space + nwork - 1, size); in execute()
|
/dports/devel/kokkos/kokkos-3.4.01/core/src/SYCL/ |
H A D | Kokkos_SYCL_Parallel_Scan.hpp | 82 pointer_type m_scratch_space = nullptr; member in Kokkos::Impl::ParallelScanSYCLBase 189 auto global_mem = m_scratch_space; in sycl_direct_launch() 206 scan_internal(q, functor, m_scratch_space, len); in sycl_direct_launch() 210 auto global_mem = m_scratch_space; in sycl_direct_launch() 241 m_scratch_space = result_memory.get(); in impl_execute() 292 &m_returnvalue, Base::m_scratch_space + nwork - 1, size); in execute()
|
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/SYCL/ |
H A D | Kokkos_SYCL_Parallel_Scan.hpp | 82 pointer_type m_scratch_space = nullptr; member in Kokkos::Impl::ParallelScanSYCLBase 189 auto global_mem = m_scratch_space; in sycl_direct_launch() 206 scan_internal(q, functor, m_scratch_space, len); in sycl_direct_launch() 210 auto global_mem = m_scratch_space; in sycl_direct_launch() 241 m_scratch_space = result_memory.get(); in impl_execute() 292 &m_returnvalue, Base::m_scratch_space + nwork - 1, size); in execute()
|