PHX::MDField< DataT, Props > Class Template Reference

#include <Phalanx_MDField.hpp>

Public Types
using	traits = FieldTraits< Scalar, Props...>
using	layout_type = typename traits::layout
using	device_type = typename traits::device
using	data_type = typename traits::data_type
using	array_type = typename traits::array_type
using	size_type = typename device_type::size_type
using	execution_space = typename array_type::execution_space
typedef Scalar	value_type
typedef Scalar &	reference_type

Public Member Functions
template<typename... Extents>
	MDField (const std::string name, const std::string layout_name, Extents...e)
	ONLY USE THIS CTOR FOR UNMANAGED FIELDS!!!! It will allocate memory unassociated with the DAG!
KOKKOS_FUNCTION	MDField (const MDField &source)
	Default empty constructor. More...
template<typename SourceScalar , typename... SourceProps>
KOKKOS_FUNCTION	MDField (const MDField< SourceScalar, SourceProps...> &source)
	Templated copy ctor. More...
constexpr bool	is_static () const
constexpr bool	is_dynamic () const
KOKKOS_INLINE_FUNCTION constexpr size_type	rank () const
KOKKOS_INLINE_FUNCTION constexpr size_t	size () const
KOKKOS_INLINE_FUNCTION constexpr size_t	span () const
const PHX::FieldTag &	fieldTag () const
Teuchos::RCP< const PHX::FieldTag >	fieldTagPtr () const
PHX::MDField< Scalar, Props...> &	operator= (const MDField< Scalar, Props...> &source)
template<typename SrcScalar , typename... SrcProps>
PHX::MDField< Scalar, Props...> &	operator= (const MDField< SrcScalar, SrcProps...> &source)
template<typename... index_pack>
KOKKOS_FORCEINLINE_FUNCTION PHX::MDFieldReturnType < array_type >::return_type	operator() (const index_pack &...indices) const
template<typename... index_pack>
KOKKOS_FORCEINLINE_FUNCTION PHX::MDFieldReturnType < array_type >::return_type	access (const index_pack &...indices) const
template<typename iType0 >
KOKKOS_FORCEINLINE_FUNCTION PHX::MDFieldReturnType < array_type >::return_type	operator[] (iType0 index0) const
template<typename iType >
KOKKOS_INLINE_FUNCTION constexpr std::enable_if < std::is_integral< iType > ::value, size_t >::type	extent (const iType &r) const
template<typename iType >
KOKKOS_INLINE_FUNCTION constexpr std::enable_if < std::is_integral< iType > ::value, int >::type	extent_int (const iType &r) const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_0 () const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_1 () const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_2 () const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_3 () const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_4 () const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_5 () const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_6 () const
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension_7 () const
template<typename iType >
KOKKOS_FORCEINLINE_FUNCTION constexpr size_type	dimension (const iType &ord) const
void	setFieldTag (const PHX::FieldTag &t)
void	setFieldTag (const Teuchos::RCP< const PHX::FieldTag > &t)
void	setFieldData (const std::any &a)
void	print (std::ostream &os, bool printValues=false) const
template<typename iType >
void	dimensions (std::vector< iType > &dims)
KOKKOS_FORCEINLINE_FUNCTION	operator array_type () const
KOKKOS_FORCEINLINE_FUNCTION Kokkos::DynRankView< Scalar, typename PHX::DevLayout < Scalar >::type, PHX::Device >	get_view ()
KOKKOS_FORCEINLINE_FUNCTION const Kokkos::DynRankView < Scalar, typename PHX::DevLayout< Scalar >::type, PHX::Device >	get_view () const
KOKKOS_FORCEINLINE_FUNCTION array_type	get_static_view ()
	Returns a static view of the underlying kokkos static view.
KOKKOS_FORCEINLINE_FUNCTION const array_type	get_static_view () const
	Returns a static view of the underlying kokkos static view.
template<typename SrcScalar , typename... SrcProps>
void	deep_copy (const PHX::MDField< SrcScalar, SrcProps...> &source)
void	deep_copy (const Scalar &source)
std::any	get_static_view_as_any () const
void	releaseFieldData ()
	Resets the underlying view ptr to null.

Public Attributes
m_host_data	m_tag = Teuchos::rcp(new PHX::Tag<Scalar>(name,dl))

Friends
template<typename FScalar , typename... FProps>
class	PHX::MDField

Detailed Description

template<typename DataT, typename... Props>
class PHX::MDField< DataT, Props >

A multidimensional array with optional compile time rank tags for self documentation.

This class currently wraps a Kokkos::View as the underlying data structure for performance portability. It also carries along a field tag with identifier and a data layout for sizing the multidimensional array.

Design Notes:

• There are essentially two versions of MDField, a runtime and compile time version that switched based on class template parameters.
• Tag dispatch is used to switch between the runtime rank and compile time rank implementations. The ViewSpecialization<traits::rank> is the tag. For the runtime rank version, the traits::rank is 0. For the compile time version, the rank is a positive integer greater than zero. Since c++17, we can now use "if consexpr" to replace some of the tag dispatch when appropriate.
• The private member m_host_data is a pointer to all data that is only accessed on host. This data is not marked with device ctors and generally creates warnings with cuda compilers if a copy constructor for an MDField is called on device. By using a pointer, the warnings are removed and these objects are not created on device.
• The runtime version has an extra data member in the m_host_data struct. The member is an any object to hold the true static type. The FieldManager always allocates all arrays using the static rank Kokkos::View. This ensures performance when we can use the static view. The runtime versions can always be created from the static versions, but tend to be less performant due to runtime indexing.
• We can assign runtime and static MDFields to each other. This means that the rank comparison can be a runtime check.

Constructor & Destructor Documentation

template<typename DataT, typename... Props>

KOKKOS_FUNCTION PHX::MDField< DataT, Props >::MDField ( const MDField< DataT, Props > &  source )

inline

Default empty constructor.

Copy ctor

template<typename DataT, typename... Props>

template<typename SourceScalar , typename... SourceProps>

KOKKOS_FUNCTION PHX::MDField< DataT, Props >::MDField ( const MDField< SourceScalar, SourceProps...> &  source )

inline

Templated copy ctor.

The templated version allows for different extra template parameters/dimension tags. For example, one evaluator could a use "Point" dim tag and be copied to an evaluator that uses a "QuadraturePoint" dim tag for the same field in a different evaluator. Another example is for assigning a compiletime MDFields to runtime MDFields and vice versa. Examples:

MDField<double,Cell,Point> a("a",data_layout); a.setFieldData(memory); MDField<double,Cell,QuadPoint> b; b = a;

MDField<double> c; // dynamic rank c = a;

Another example could be for atomic access flags.

References PHX::MDField< DataT, Props >::MDField().

Member Function Documentation

template<typename DataT, typename... Props>

template<typename iType >

void PHX::MDField< DataT, Props >::dimensions ( std::vector< iType > &  dims )

inline

WARNING: The vector data in this method should be a "size_type" to be consistent with Kokkos, but for backwards compatibility during the transition, needs to be templated in the index type.

void dimensions(std::vector<size_type>& dims);

---

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

• Phalanx_Evaluator_Utilities.hpp
• Phalanx_MDField.hpp