Moose::Kokkos::Map< T1, T2 > Class Template Reference

The Kokkos wrapper class for standard map. More...

#include <KokkosMap.h>

Public Member Functions
auto	begin ()
	Get the beginning writeable iterator of the host map. More...
auto	begin () const
	Get the beginning const iterator of the host map. More...
auto	end ()
	Get the end writeable iterator of the host map. More...
auto	end () const
	Get the end const iterator of the host map. More...
auto &	get ()
	Get the underlying writeable host map. More...
const auto &	get () const
	Get the underlying const host map. More...
T2 &	operator[] (const T1 &key)
	Call host map's operator[]. More...
void	copyToDevice ()
	Copy host map to device. More...
void	copyToDeviceNested ()
	Copy host map to device, perform nested copy for Kokkos arrays. More...
void	swap (Map< T1, T2 > &map)
	Swap with another Kokkos map. More...
KOKKOS_FUNCTION dof_id_type	size () const
	Get the size of map. More...
KOKKOS_FUNCTION dof_id_type	find (const T1 &key) const
	Find the index of a key. More...
KOKKOS_FUNCTION bool	exists (const T1 &key) const
	Get whether a key exists. More...
KOKKOS_FUNCTION const T1 &	key (dof_id_type idx) const
	Get the key of an index. More...
KOKKOS_FUNCTION const T2 &	value (dof_id_type idx) const
	Get the value of an index. More...
KOKKOS_FUNCTION const T2 &	operator[] (const T1 &key) const
	Get the value corresponding to a key. More...
KOKKOS_FUNCTION const T2 &	operator() (const T1 &key) const

Static Public Attributes
static const dof_id_type	invalid_id = libMesh::DofObject::invalid_id

Private Member Functions
void	copy ()
	Internal method to copy host map to device. More...

Private Attributes
std::shared_ptr< std::map< T1, T2 > >	_map_host
	Standard map on host Stored as a shared pointer to avoid deep copy. More...
Array< T1 >	_keys
	Keys on device. More...
Array< T2 >	_values
	Values on device. More...
Array< dof_id_type >	_offset
	Beginning offset into device arrays of each bucket. More...

Friends
void	dataStore (std::ostream &, Map< T1, T2 > &, void *)
void	dataLoad (std::istream &, Map< T1, T2 > &, void *)

Detailed Description

template<typename T1, typename T2>
class Moose::Kokkos::Map< T1, T2 >

The Kokkos wrapper class for standard map.

The map can only be populated on host. Make sure to call copyToDevice() or copyToDeviceNested() after populating the map on host.

Definition at line 60 of file KokkosMap.h.

Member Function Documentation

begin() [1/2]

template<typename T1, typename T2>

auto Moose::Kokkos::Map< T1, T2 >::begin ( )

inline

Get the beginning writeable iterator of the host map.

Returns

The beginning iterator

Definition at line 80 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::begin().

{ return get().begin(); }

begin() [2/2]

template<typename T1, typename T2>

auto Moose::Kokkos::Map< T1, T2 >::begin ( ) const

inline

Get the beginning const iterator of the host map.

Returns

The beginning iterator

Definition at line 85 of file KokkosMap.h.

{ return get().cbegin(); }

copy()

template<typename T1 , typename T2 >

void Moose::Kokkos::Map< T1, T2 >::copy ( )

private

Internal method to copy host map to device.

Definition at line 243 of file KokkosMap.h.

{
  _keys.create(size());
  _values.create(size());
  _offset.create(size() + 1);
  _offset = 0;

  for (const auto & [key, value] : get())
  {
    auto bucket = fnv1aHash(key) % size();

    _offset[bucket]++;
  }

  std::exclusive_scan(_offset.begin(), _offset.end(), _offset.begin(), 0);

  _offset.copyToDevice();

  std::vector<dof_id_type> idx(size(), 0);

  for (const auto & [key, value] : get())
  {
    auto bucket = fnv1aHash(key) % size();

    _keys[_offset[bucket] + idx[bucket]] = key;
    _values[_offset[bucket] + idx[bucket]] = value;
    idx[bucket]++;
  }
}

copyToDevice()

template<typename T1 , typename T2 >

void Moose::Kokkos::Map< T1, T2 >::copyToDevice

(

)

Copy host map to device.

Definition at line 275 of file KokkosMap.h.

{
  copy();

  _keys.copyToDevice();
  _values.copyToDevice();
}

copyToDeviceNested()

template<typename T1 , typename T2 >

void Moose::Kokkos::Map< T1, T2 >::copyToDeviceNested

(

)

Copy host map to device, perform nested copy for Kokkos arrays.

Definition at line 285 of file KokkosMap.h.

{
  copy();

  _keys.copyToDeviceNested();
  _values.copyToDeviceNested();
}

end() [1/2]

template<typename T1, typename T2>

auto Moose::Kokkos::Map< T1, T2 >::end ( )

inline

Get the end writeable iterator of the host map.

Returns

The end iterator

Definition at line 90 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::end().

{ return get().end(); }

end() [2/2]

template<typename T1, typename T2>

auto Moose::Kokkos::Map< T1, T2 >::end ( ) const

inline

Get the end const iterator of the host map.

Returns

The end iterator

Definition at line 95 of file KokkosMap.h.

{ return get().cend(); }

exists()

template<typename T1, typename T2>

KOKKOS_FUNCTION bool Moose::Kokkos::Map< T1, T2 >::exists ( const T1 &  key ) const

inline

Get whether a key exists.

Parameters

key

The key

Returns

Whether the key exists

Definition at line 161 of file KokkosMap.h.

Referenced by KokkosMatCoupledForce::computeQpOffDiagJacobian().

{ return find(key) != invalid_id; }

find()

template<typename T1, typename T2 >

KOKKOS_FUNCTION dof_id_type Moose::Kokkos::Map< T1, T2 >::find

(

const T1 &

key

)

const

Find the index of a key.

Parameters

key

The key

Returns

The index of the key, invalid_id if the key does not exist

Definition at line 305 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::exists().

{
  if (!size())
    return invalid_id;

  auto bucket = fnv1aHash(key) % size();
  auto begin = _offset[bucket];
  auto end = _offset[bucket + 1];

  for (dof_id_type i = begin; i < end; ++i)
    if (_keys[i] == key)
      return i;

  return invalid_id;
}

get() [1/2]

template<typename T1, typename T2>

auto& Moose::Kokkos::Map< T1, T2 >::get ( )

inline

Get the underlying writeable host map.

Returns

The writeable host map

Definition at line 100 of file KokkosMap.h.

Referenced by Moose::Kokkos::dataLoad(), Moose::Kokkos::dataStore(), and Moose::Kokkos::Map< unsigned int, unsigned int >::swap().

  {
    if (!_map_host)
      _map_host = std::make_shared<std::map<T1, T2>>();

    return *_map_host;
  }

get() [2/2]

template<typename T1, typename T2>

const auto& Moose::Kokkos::Map< T1, T2 >::get ( ) const

inline

Get the underlying const host map.

Returns

The const host map

Definition at line 111 of file KokkosMap.h.

  {
    if (!_map_host)
      _map_host = std::make_shared<std::map<T1, T2>>();

    return *_map_host;
  }

key()

template<typename T1, typename T2>

KOKKOS_FUNCTION const T1& Moose::Kokkos::Map< T1, T2 >::key ( dof_id_type  idx ) const

inline

Get the key of an index.

Parameters

idx	The index returned by find()

Returns

The const reference of the key

Definition at line 167 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::exists(), Moose::Kokkos::Map< unsigned int, unsigned int >::operator()(), and Moose::Kokkos::Map< unsigned int, unsigned int >::operator[]().

  {
    KOKKOS_ASSERT(idx != invalid_id);

    return _keys[idx];
  }

operator()()

template<typename T1, typename T2>

KOKKOS_FUNCTION const T2& Moose::Kokkos::Map< T1, T2 >::operator() ( const T1 &  key ) const

inline

Definition at line 204 of file KokkosMap.h.

{ return operator[](key); }

operator[]() [1/2]

template<typename T1, typename T2>

T2& Moose::Kokkos::Map< T1, T2 >::operator[] ( const T1 &  key )

inline

Call host map's operator[].

Parameters

key

The key

Returns

The writeable reference of the value

Definition at line 123 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::operator()().

{ return get()[key]; }

operator[]() [2/2]

template<typename T1, typename T2>

KOKKOS_FUNCTION const T2& Moose::Kokkos::Map< T1, T2 >::operator[] ( const T1 &  key ) const

inline

Get the value corresponding to a key.

Parameters

key

The key

Returns

The const reference of the value

Definition at line 190 of file KokkosMap.h.

  {
    KOKKOS_IF_ON_HOST(return get().at(key);)

    auto idx = find(key);

    KOKKOS_ASSERT(idx != invalid_id);

    return _values[idx];
  }

size()

template<typename T1, typename T2>

KOKKOS_FUNCTION dof_id_type Moose::Kokkos::Map< T1, T2 >::size ( ) const

inline

Get the size of map.

Returns

The size of map

Definition at line 144 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::size().

  {
    KOKKOS_IF_ON_HOST(return get().size();)

    return _keys.size();
  }

swap()

template<typename T1, typename T2>

void Moose::Kokkos::Map< T1, T2 >::swap

(

Map< T1, T2 > &

map

)

Swap with another Kokkos map.

Parameters

map	The Kokkos map to be swapped

Definition at line 295 of file KokkosMap.h.

{
  get().swap(map.get());
  _keys.swap(map._keys);
  _values.swap(map._values);
  _offset.swap(map._offset);
}

value()

template<typename T1, typename T2>

KOKKOS_FUNCTION const T2& Moose::Kokkos::Map< T1, T2 >::value ( dof_id_type  idx ) const

inline

Get the value of an index.

Parameters

idx	The index returned by find()

Returns

The const reference of the value

Definition at line 178 of file KokkosMap.h.

  {
    KOKKOS_ASSERT(idx != invalid_id);

    return _values[idx];
  }

Friends And Related Function Documentation

dataLoad

template<typename T1, typename T2>

void dataLoad ( std::istream &  stream, Map< T1, T2 > &  map, void *  context  )

friend

Definition at line 335 of file KokkosMap.h.

{
  using ::dataLoad;

  dataLoad(stream, map.get(), context);
  dataLoad(stream, map._keys, context);
  dataLoad(stream, map._values, context);
  dataLoad(stream, map._offset, context);
}

dataStore

template<typename T1, typename T2>

void dataStore ( std::ostream &  stream, Map< T1, T2 > &  map, void *  context  )

friend

Definition at line 323 of file KokkosMap.h.

{
  using ::dataStore;

  dataStore(stream, map.get(), context);
  dataStore(stream, map._keys, context);
  dataStore(stream, map._values, context);
  dataStore(stream, map._offset, context);
}

Member Data Documentation

_keys

template<typename T1, typename T2>

Array<T1> Moose::Kokkos::Map< T1, T2 >::_keys

private

Keys on device.

Definition at line 226 of file KokkosMap.h.

Referenced by Moose::Kokkos::dataLoad(), Moose::Kokkos::dataStore(), Moose::Kokkos::Map< unsigned int, unsigned int >::key(), Moose::Kokkos::Map< unsigned int, unsigned int >::size(), and Moose::Kokkos::Map< unsigned int, unsigned int >::swap().

_map_host

template<typename T1, typename T2>

std::shared_ptr<std::map<T1, T2> > Moose::Kokkos::Map< T1, T2 >::_map_host

mutableprivate

Standard map on host Stored as a shared pointer to avoid deep copy.

Definition at line 222 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::get().

_offset

template<typename T1, typename T2>

Array<dof_id_type> Moose::Kokkos::Map< T1, T2 >::_offset

private

Beginning offset into device arrays of each bucket.

Definition at line 234 of file KokkosMap.h.

Referenced by Moose::Kokkos::dataLoad(), Moose::Kokkos::dataStore(), and Moose::Kokkos::Map< unsigned int, unsigned int >::swap().

_values

template<typename T1, typename T2>

Array<T2> Moose::Kokkos::Map< T1, T2 >::_values

private

Values on device.

Definition at line 230 of file KokkosMap.h.

Referenced by Moose::Kokkos::dataLoad(), Moose::Kokkos::dataStore(), Moose::Kokkos::Map< unsigned int, unsigned int >::swap(), and Moose::Kokkos::Map< unsigned int, unsigned int >::value().

invalid_id

template<typename T1, typename T2>

const dof_id_type Moose::Kokkos::Map< T1, T2 >::invalid_id = libMesh::DofObject::invalid_id

static

Definition at line 208 of file KokkosMap.h.

Referenced by Moose::Kokkos::Map< unsigned int, unsigned int >::exists(), Moose::Kokkos::Map< unsigned int, unsigned int >::key(), and Moose::Kokkos::Map< unsigned int, unsigned int >::value().

---

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

• include/kokkos/base/KokkosMap.h