Home
last modified time | relevance | path

Searched refs:m_scratch_size (Results 1 – 25 of 42) sorted by relevance

12

/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/OpenMPTarget/
H A DKokkos_OpenMPTarget_Exec.cpp96 int64_t OpenMPTargetExec::m_scratch_size = 0; member in Kokkos::Impl::OpenMPTargetExec
103 space.deallocate(m_scratch_ptr, m_scratch_size); in clear_scratch()
105 m_scratch_size = 0; in clear_scratch()
131 if (total_size > m_scratch_size) { in resize_scratch()
132 space.deallocate(m_scratch_ptr, m_scratch_size); in resize_scratch()
133 m_scratch_size = total_size; in resize_scratch()
/dports/devel/kokkos/kokkos-3.4.01/core/src/OpenMPTarget/
H A DKokkos_OpenMPTarget_Exec.cpp96 int64_t OpenMPTargetExec::m_scratch_size = 0; member in Kokkos::Impl::OpenMPTargetExec
103 space.deallocate(m_scratch_ptr, m_scratch_size); in clear_scratch()
105 m_scratch_size = 0; in clear_scratch()
131 if (total_size > m_scratch_size) { in resize_scratch()
132 space.deallocate(m_scratch_ptr, m_scratch_size); in resize_scratch()
133 m_scratch_size = total_size; in resize_scratch()
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/OpenMPTarget/
H A DKokkos_OpenMPTarget_Exec.cpp96 int64_t OpenMPTargetExec::m_scratch_size = 0; member in Kokkos::Impl::OpenMPTargetExec
103 space.deallocate(m_scratch_ptr, m_scratch_size); in clear_scratch()
105 m_scratch_size = 0; in clear_scratch()
131 if (total_size > m_scratch_size) { in resize_scratch()
132 space.deallocate(m_scratch_ptr, m_scratch_size); in resize_scratch()
133 m_scratch_size = total_size; in resize_scratch()
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/SYCL/
H A DKokkos_SYCL_Parallel_Team.hpp376 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
397 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
448 m_scratch_size[0] = m_shmem_size; in ParallelFor()
449 m_scratch_size[1] = m_policy.scratch_size(1, m_team_size); in ParallelFor()
458 sizeof(char) * m_scratch_size[1] * m_league_size, q); in ParallelFor()
526 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
585 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
636 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
776 m_scratch_size[0] = m_shmem_size; in initialize()
777 m_scratch_size[1] = m_policy.scratch_size(1, m_team_size); in initialize()
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/SYCL/
H A DKokkos_SYCL_Parallel_Team.hpp376 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
397 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
448 m_scratch_size[0] = m_shmem_size; in ParallelFor()
449 m_scratch_size[1] = m_policy.scratch_size(1, m_team_size); in ParallelFor()
458 sizeof(char) * m_scratch_size[1] * m_league_size, q); in ParallelFor()
526 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
585 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
636 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
776 m_scratch_size[0] = m_shmem_size; in initialize()
777 m_scratch_size[1] = m_policy.scratch_size(1, m_team_size); in initialize()
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/SYCL/
H A DKokkos_SYCL_Parallel_Team.hpp376 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
397 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
448 m_scratch_size[0] = m_shmem_size; in ParallelFor()
449 m_scratch_size[1] = m_policy.scratch_size(1, m_team_size); in ParallelFor()
458 sizeof(char) * m_scratch_size[1] * m_league_size, q); in ParallelFor()
526 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
585 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
636 const int scratch_size[2] = {m_scratch_size[0], m_scratch_size[1]}; in sycl_direct_launch()
776 m_scratch_size[0] = m_shmem_size; in initialize()
777 m_scratch_size[1] = m_policy.scratch_size(1, m_team_size); in initialize()
[all …]
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/HIP/
H A DKokkos_HIP_Parallel_Team.hpp435 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
458 if (m_scratch_size[1] > 0) { in operator ()()
492 m_scratch_size[1]), in operator ()()
493 m_scratch_size[1], league_rank, m_league_size)); in operator ()()
495 if (m_scratch_size[1] > 0) { in operator ()()
645 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
680 m_scratch_size[1]), in iterate_through_league()
689 if (m_scratch_size[1] > 0) { in operator ()()
719 if (m_scratch_size[1] > 0) { in operator ()()
915 m_scratch_size[0] = m_shmem_size;
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/HIP/
H A DKokkos_HIP_Parallel_Team.hpp435 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
458 if (m_scratch_size[1] > 0) { in operator ()()
492 m_scratch_size[1]), in operator ()()
493 m_scratch_size[1], league_rank, m_league_size)); in operator ()()
495 if (m_scratch_size[1] > 0) { in operator ()()
645 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
680 m_scratch_size[1]), in iterate_through_league()
689 if (m_scratch_size[1] > 0) { in operator ()()
719 if (m_scratch_size[1] > 0) { in operator ()()
915 m_scratch_size[0] = m_shmem_size;
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/HIP/
H A DKokkos_HIP_Parallel_Team.hpp435 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
458 if (m_scratch_size[1] > 0) { in operator ()()
492 m_scratch_size[1]), in operator ()()
493 m_scratch_size[1], league_rank, m_league_size)); in operator ()()
495 if (m_scratch_size[1] > 0) { in operator ()()
645 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
680 m_scratch_size[1]), in iterate_through_league()
689 if (m_scratch_size[1] > 0) { in operator ()()
719 if (m_scratch_size[1] > 0) { in operator ()()
915 m_scratch_size[0] = m_shmem_size;
[all …]
/dports/science/chrono/chrono-7.0.1/src/chrono_sensor/optix/
H A DChFilterOptixRender.cpp214 m_scratch_size = static_cast<uint32_t>(denoiser_sizes.withoutOverlapScratchSizeInBytes); in Initialize()
215 CUDA_ERROR_CHECK(cudaMalloc(reinterpret_cast<void**>(&md_scratch), m_scratch_size)); in Initialize()
251 …enoiserSetup(m_denoiser, m_cuda_stream, w, h, md_state, m_state_size, md_scratch, m_scratch_size)); in Initialize()
261 &md_output, md_scratch, m_scratch_size)); in Execute()
H A DChFilterOptixRender.h56 uint32_t m_scratch_size = 0; variable
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/Cuda/
H A DKokkos_Cuda_Parallel.hpp689 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
711 if (m_scratch_size[1] > 0) { in operator ()()
745 m_scratch_size[1]), in operator ()()
748 if (m_scratch_size[1] > 0) { in operator ()()
1582 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
1606 if (m_scratch_size[1] > 0) { in operator ()()
1636 if (m_scratch_size[1] > 0) { in operator ()()
1665 m_scratch_size[1]), in run()
1743 m_scratch_size[1]), in run()
1896 m_scratch_size[0] = m_shmem_size;
[all …]
/dports/devel/kokkos/kokkos-3.4.01/core/src/Cuda/
H A DKokkos_Cuda_Parallel.hpp689 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
711 if (m_scratch_size[1] > 0) { in operator ()()
745 m_scratch_size[1]), in operator ()()
748 if (m_scratch_size[1] > 0) { in operator ()()
1582 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
1606 if (m_scratch_size[1] > 0) { in operator ()()
1636 if (m_scratch_size[1] > 0) { in operator ()()
1665 m_scratch_size[1]), in run()
1743 m_scratch_size[1]), in run()
1896 m_scratch_size[0] = m_shmem_size;
[all …]
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/Cuda/
H A DKokkos_Cuda_Parallel.hpp689 int m_scratch_size[2]; member in Kokkos::Impl::ParallelFor
711 if (m_scratch_size[1] > 0) { in operator ()()
745 m_scratch_size[1]), in operator ()()
748 if (m_scratch_size[1] > 0) { in operator ()()
1582 int m_scratch_size[2]; member in Kokkos::Impl::ParallelReduce
1606 if (m_scratch_size[1] > 0) { in operator ()()
1636 if (m_scratch_size[1] > 0) { in operator ()()
1665 m_scratch_size[1]), in run()
1743 m_scratch_size[1]), in run()
1896 m_scratch_size[0] = m_shmem_size;
[all …]
/dports/science/lammps/lammps-stable_29Sep2021/lib/kokkos/core/src/impl/
H A DKokkos_HostThreadTeam.hpp110 int m_scratch_size; member in Kokkos::Impl::HostThreadTeamData
187 m_scratch_size(0), in HostThreadTeamData()
248 return m_scratch_size ? sizeof(int64_t) * (m_team_reduce - m_pool_reduce) in pool_reduce_bytes()
261 return sizeof(int64_t) * (m_scratch_size - m_thread_local); in thread_local_bytes()
265 return sizeof(int64_t) * m_scratch_size; in scratch_bytes()
337 m_scratch_size = align_to_int64(alloc_size); in scratch_assign()
347 , int(m_scratch_size) in scratch_assign()
/dports/devel/kokkos/kokkos-3.4.01/core/src/impl/
H A DKokkos_HostThreadTeam.hpp110 int m_scratch_size; member in Kokkos::Impl::HostThreadTeamData
187 m_scratch_size(0), in HostThreadTeamData()
248 return m_scratch_size ? sizeof(int64_t) * (m_team_reduce - m_pool_reduce) in pool_reduce_bytes()
261 return sizeof(int64_t) * (m_scratch_size - m_thread_local); in thread_local_bytes()
265 return sizeof(int64_t) * m_scratch_size; in scratch_bytes()
337 m_scratch_size = align_to_int64(alloc_size); in scratch_assign()
347 , int(m_scratch_size) in scratch_assign()
/dports/science/sparta/sparta-20Oct2021/lib/kokkos/core/src/impl/
H A DKokkos_HostThreadTeam.hpp110 int m_scratch_size; member in Kokkos::Impl::HostThreadTeamData
187 m_scratch_size(0), in HostThreadTeamData()
248 return m_scratch_size ? sizeof(int64_t) * (m_team_reduce - m_pool_reduce) in pool_reduce_bytes()
261 return sizeof(int64_t) * (m_scratch_size - m_thread_local); in thread_local_bytes()
265 return sizeof(int64_t) * m_scratch_size; in scratch_bytes()
337 m_scratch_size = align_to_int64(alloc_size); in scratch_assign()
347 , int(m_scratch_size) in scratch_assign()
/dports/lang/clover/mesa-21.3.6/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h214 int m_scratch_size; variable
/dports/graphics/libosmesa/mesa-21.3.6/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h214 int m_scratch_size; variable
/dports/graphics/libosmesa-gallium/mesa-21.3.6/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h214 int m_scratch_size; variable
/dports/graphics/mesa-libs/mesa-21.3.6/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h214 int m_scratch_size; variable
/dports/graphics/mesa-dri-classic/mesa-20.2.3/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h207 int m_scratch_size; variable
/dports/graphics/mesa-gallium-va/mesa-21.3.6/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h214 int m_scratch_size; variable
/dports/graphics/mesa-gallium-vdpau/mesa-21.3.6/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h214 int m_scratch_size; variable
/dports/graphics/mesa-dri-gallium/mesa-21.3.6/src/gallium/drivers/r600/sfn/
H A Dsfn_shader_base.h214 int m_scratch_size; variable

12