Commit b28ef468 authored by Carter Edwards's avatar Carter Edwards
Browse files

Kokkos: Example development.

parent fb5a2d6d
......@@ -88,6 +88,12 @@ public:
typedef Kokkos::View< const unsigned * [2] , Device , Kokkos::MemoryUnmanaged > comm_list_type ;
typedef Kokkos::View< const unsigned * , Device , Kokkos::MemoryUnmanaged > send_nodeid_type ;
KOKKOS_INLINE_FUNCTION
unsigned node_count() const { return m_node_grid.dimension_0(); }
KOKKOS_INLINE_FUNCTION
unsigned elem_count() const { return m_elem_node.dimension_0(); }
KOKKOS_INLINE_FUNCTION
unsigned elem_node_local( unsigned inode , unsigned k ) const
{ return m_elem_node_local[inode][k] ; }
......@@ -95,6 +101,9 @@ public:
KOKKOS_INLINE_FUNCTION
unsigned node_grid( unsigned inode , unsigned iaxis ) const { return m_node_grid(inode,iaxis); }
KOKKOS_INLINE_FUNCTION
unsigned node_grid_max( unsigned iaxis ) const { return m_box_part.global_coord_max(iaxis); }
KOKKOS_INLINE_FUNCTION
unsigned elem_node( unsigned ielem , unsigned inode ) const { return m_elem_node(ielem,inode); }
......
......@@ -146,6 +146,10 @@ public:
//----------------------------------------
KOKKOS_INLINE_FUNCTION
unsigned global_coord_max( unsigned axis ) const
{ return m_global_node_box[axis][1]; }
KOKKOS_INLINE_FUNCTION
void local_node_coord( size_t lid , unsigned coord[] ) const
{
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment