Moose::Kokkos::MemoryPool Class Reference

The Kokkos class that manages memory pool for dynamically-sized temporary arrays in Kokkos parallel functions. More...

#include <KokkosMemoryPool.h>

Public Member Functions
	MemoryPool (dof_id_type size, unsigned int ways)
	Constructor.
template<typename T >
KOKKOS_FUNCTION MemoryChunk< T >	allocate (dof_id_type idx, unsigned int size) const
	Allocate a memory chunk The returned object should be alive while the memory chunk is used.

Private Attributes
Array<::Kokkos::MemoryPool< MemSpace > >	_pools
	Kokkos memory pools.

Detailed Description

The Kokkos class that manages memory pool for dynamically-sized temporary arrays in Kokkos parallel functions.

Definition at line 75 of file KokkosMemoryPool.h.

Constructor & Destructor Documentation

MemoryPool()

Moose::Kokkos::MemoryPool::MemoryPool	(	dof_id_type	size,
		unsigned int	ways
	)

Constructor.

Parameters

size	The memory pool size in the number of bytes
ways	The number of parallel ways

Member Function Documentation

allocate()

template<typename T >

KOKKOS_FUNCTION MemoryChunk< T > Moose::Kokkos::MemoryPool::allocate ( dof_id_type  idx, unsigned int  size  ) const

inline

Allocate a memory chunk The returned object should be alive while the memory chunk is used.

Parameters

indx	The index to select pool
size	The memory chunk size in the number of elements

Returns

The memory chunk object containing the pointer to the memory chunk

Definition at line 93 of file KokkosMemoryPool.h.

  {
    auto pool = idx % _pools.size();
 
    return MemoryChunk<T>(_pools[pool], size * sizeof(T));
  }

Referenced by Moose::Kokkos::MemoryChunk< T >::MemoryChunk().

Member Data Documentation

_pools

Array<::Kokkos::MemoryPool<MemSpace> > Moose::Kokkos::MemoryPool::_pools

private

Kokkos memory pools.

Definition at line 104 of file KokkosMemoryPool.h.

Referenced by allocate().

---

The documentation for this class was generated from the following file:

• include/kokkos/base/KokkosMemoryPool.h