Moose::Kokkos::VariableValueTempl< is_ad > Class Template Reference

The Kokkos wrapper classes for MOOSE-like variable value access. More...

#include <KokkosVariableValue.h>

Public Member Functions
	VariableValueTempl ()=default
	Default constructor.
	VariableValueTempl (Variable var, bool dof=false)
	Constructor.
	VariableValueTempl (const MooseVariableBase &var, const TagName &tag=Moose::SOLUTION_TAG, bool dof=false)
	Constructor.
KOKKOS_FUNCTION	operator bool () const
	Get whether the variable was coupled.
KOKKOS_FUNCTION auto	operator() (Datum &datum, unsigned int idx, unsigned int comp=0) const
	Get the current variable value.
KOKKOS_FUNCTION const Variable &	variable () const
	Get the Kokkos variable.

Private Types
using	real_type = std::conditional_t< is_ad, ADReal, Real >

Private Attributes
Variable	_var
	Coupled Kokkos variable.
Array< Real >	_seed
	Derivative seed of each component for AD.
bool	_dof = false
	Flag whether DOF values are requested.
	VariableValueTempl (const VariableValueTempl< is_ad > &object)
	Copy constructor for parallel dispatch.
VariableValueTempl< is_ad > &	operator= (const VariableValueTempl< is_ad > &object)
	Copy assignment operator.
KOKKOS_FUNCTION auto	operator() (AssemblyDatum &datum, unsigned int idx, unsigned int comp=0) const
	Get the current variable value.
KOKKOS_FUNCTION auto	get (Datum &datum, unsigned int idx, unsigned int comp=0, Real seed=0) const
	Get the current variable value.

Detailed Description

template<bool is_ad>
class Moose::Kokkos::VariableValueTempl< is_ad >

The Kokkos wrapper classes for MOOSE-like variable value access.

Definition at line 217 of file KokkosVariableValue.h.

Member Typedef Documentation

real_type

template<bool is_ad>

using Moose::Kokkos::VariableValueTempl< is_ad >::real_type = std::conditional_t<is_ad, ADReal, Real>

private

Definition at line 219 of file KokkosVariableValue.h.

Constructor & Destructor Documentation

VariableValueTempl() [1/4]

template<bool is_ad>

Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl ( )

default

Default constructor.

VariableValueTempl() [2/4]

template<bool is_ad>

Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl ( Variable  var, bool  dof = false  )

inline

Constructor.

Parameters

var	The Kokkos variable
dof	Whether to get DOF values

Definition at line 231 of file KokkosVariableValue.h.

                                                     : _var(var), _dof(dof || _var.scalar())
  {
    checkVariable(_var, false, is_ad ? "ADVariableValue" : "VariableValue");
  }

VariableValueTempl() [3/4]

template<bool is_ad>

Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl ( const MooseVariableBase &  var, const TagName &  tag = Moose::SOLUTION_TAG, bool  dof = false  )

inline

Constructor.

Parameters

var	The MOOSE variable
tag	The vector tag name
dof	Whether to get DOF values

Definition at line 241 of file KokkosVariableValue.h.

    : _var(var, tag), _dof(dof || _var.scalar())
  {
    checkVariable(_var, false, is_ad ? "ADVariableValue" : "VariableValue");
  }

VariableValueTempl() [4/4]

template<bool is_ad>

Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl

(

const VariableValueTempl< is_ad > &

object

)

Copy constructor for parallel dispatch.

Definition at line 319 of file KokkosVariableValue.h.

  : _var(object._var), _seed(object._seed), _dof(object._dof)
{
  if constexpr (is_ad)
    if (_var.coupled())
    {
      if (!_seed.isAlloc())
        _seed.create(_var.components());
 
      for (unsigned int comp = 0; comp < _var.components(); ++comp)
        _seed[comp] =
            _var.dot() ? _var.mooseVar(comp)->sys().duDotDu(_var.var(comp)) : (_var.old() ? 0 : 1);
 
      _seed.copyToDevice();
    }
}

Member Function Documentation

get()

template<bool is_ad>

KOKKOS_FUNCTION auto Moose::Kokkos::VariableValueTempl< is_ad >::get ( Datum &  datum, unsigned int  idx, unsigned int  comp = 0, Real  seed = 0  ) const

private

Get the current variable value.

Parameters

datum	The Datum object of the current thread
idx	The local quadrature point or DOF index
comp	The variable component
seed	The derivative seed (only meaningful for AD)

Returns

The variable value

Definition at line 365 of file KokkosVariableValue.h.

{
  KOKKOS_ASSERT(_var.initialized());
 
  real_type value;
 
  if (_var.coupled())
  {
    auto & sys = datum.system(_var.sys(comp));
    auto var = _var.var(comp);
    auto tag = _var.tag();
 
    if (_dof)
    {
      unsigned int dof;
 
      if (_var.scalar())
        dof = sys.getScalarLocalDofIndex(idx, var);
      else if (datum.isNodal())
      {
        auto node = datum.node();
        dof = sys.getNodeLocalDofIndex(node, 0, var);
      }
      else
      {
        auto elem = datum.elem().id;
        dof = sys.getElemLocalDofIndex(elem, idx, var);
      }
 
      if constexpr (is_ad)
        value = sys.getVectorDofADValue(dof, tag, seed);
      else
        value = sys.getVectorDofValue(dof, tag);
    }
    else
    {
      auto & elem = datum.elem();
      auto side = datum.side();
 
      if constexpr (is_ad)
        value = side == libMesh::invalid_uint
                    ? sys.getVectorQpADValue(elem, datum.qpOffset(), idx, var, tag, seed)
                    : sys.getVectorQpADValueFace(elem, side, idx, var, tag, seed);
      else
        value = side == libMesh::invalid_uint
                    ? sys.getVectorQpValue(elem, datum.qpOffset() + idx, var, tag)
                    : sys.getVectorQpValueFace(elem, side, idx, var, tag);
    }
  }
  else
    value = _var.value(_var.scalar() ? idx : comp);
 
  return value;
}

Referenced by Moose::Kokkos::VariableValueTempl< is_ad >::operator()().

operator bool()

template<bool is_ad>

KOKKOS_FUNCTION Moose::Kokkos::VariableValueTempl< is_ad >::operator bool ( ) const

inline

Get whether the variable was coupled.

Returns

Whether the variable was coupled

Definition at line 262 of file KokkosVariableValue.h.

{ return _var.coupled(); }

operator()() [1/2]

template<bool is_ad>

KOKKOS_FUNCTION auto Moose::Kokkos::VariableValueTempl< is_ad >::operator()	(	AssemblyDatum &	datum,
		unsigned int	idx,
		unsigned int	comp = 0
	)		const

Get the current variable value.

Parameters

datum	The AssemblyDatum object of the current thread
idx	The local quadrature point or DOF index
comp	The variable component

Returns

The variable value

Definition at line 348 of file KokkosVariableValue.h.

{
  if constexpr (is_ad)
  {
    Real seed =
        datum.do_derivatives() && _var.coupled() && _var.sys(comp) == datum.sys() ? _seed[comp] : 0;
 
    return get(datum, idx, comp, seed);
  }
  else
    return get(datum, idx, comp);
}

operator()() [2/2]

template<bool is_ad>

KOKKOS_FUNCTION auto Moose::Kokkos::VariableValueTempl< is_ad >::operator() ( Datum &  datum, unsigned int  idx, unsigned int  comp = 0  ) const

inline

Get the current variable value.

Parameters

datum	The Datum object of the current thread
idx	The local quadrature point or DOF index
comp	The variable component

Returns

The variable value

Definition at line 271 of file KokkosVariableValue.h.

  {
    return get(datum, idx, comp);
  }

operator=()

template<bool is_ad>

VariableValueTempl< is_ad > & Moose::Kokkos::VariableValueTempl< is_ad >::operator=

(

const VariableValueTempl< is_ad > &

object

)

Copy assignment operator.

Definition at line 338 of file KokkosVariableValue.h.

{
  _var = object._var;
  _dof = object._dof;
 
  return *this;
}

variable()

template<bool is_ad>

KOKKOS_FUNCTION const Variable & Moose::Kokkos::VariableValueTempl< is_ad >::variable ( ) const

inline

Get the Kokkos variable.

Returns

The Kokkos variable

Definition at line 290 of file KokkosVariableValue.h.

{ return _var; }

Referenced by KokkosExtraIDIntegralVectorPostprocessor::execute(), KokkosNodalExtremeValue::reduce(), KokkosNodalMaxValueId::reduce(), KokkosNodalSum::reduce(), and KokkosExtraIDIntegralVectorPostprocessor::reduce().

Member Data Documentation

_dof

template<bool is_ad>

bool Moose::Kokkos::VariableValueTempl< is_ad >::_dof = false

private

Flag whether DOF values are requested.

Definition at line 315 of file KokkosVariableValue.h.

_seed

template<bool is_ad>

Array<Real> Moose::Kokkos::VariableValueTempl< is_ad >::_seed

private

Derivative seed of each component for AD.

Definition at line 311 of file KokkosVariableValue.h.

Referenced by Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl().

_var

template<bool is_ad>

Variable Moose::Kokkos::VariableValueTempl< is_ad >::_var

private

Coupled Kokkos variable.

Definition at line 307 of file KokkosVariableValue.h.

Referenced by Moose::Kokkos::VariableValueTempl< is_ad >::operator bool(), Moose::Kokkos::VariableValueTempl< is_ad >::operator=(), Moose::Kokkos::VariableValueTempl< is_ad >::variable(), Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl(), Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl(), and Moose::Kokkos::VariableValueTempl< is_ad >::VariableValueTempl().

---

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

• include/kokkos/base/KokkosVariableValue.h