Moose::Kokkos::AssemblyDatum Class Reference

The Kokkos object that holds thread-private data in the parallel operations of Kokkos kernels. More...

#include <KokkosDatum.h>

Inheritance diagram for Moose::Kokkos::AssemblyDatum:

Public Member Functions
KOKKOS_FUNCTION	AssemblyDatum (const ContiguousElementID elem, const unsigned int side, const Assembly &assembly, const Array< System > &systems, const Variable &ivar, const unsigned int jvar, const unsigned int comp=0)
	Constructor for element and side data. More...
KOKKOS_FUNCTION	AssemblyDatum (const ContiguousNodeID node, const Assembly &assembly, const Array< System > &systems, const Variable &ivar, const unsigned int jvar, const unsigned int comp=0)
	Constructor for node data. More...
KOKKOS_FUNCTION unsigned int	n_dofs () const
	Get the number of local DOFs. More...
KOKKOS_FUNCTION unsigned int	n_idofs () const
	Get the number of local DOFs. More...
KOKKOS_FUNCTION unsigned int	n_jdofs () const
	Get the number of local DOFs for the coupled variable. More...
KOKKOS_FUNCTION unsigned int	sys () const
	Get the system number of variable. More...
KOKKOS_FUNCTION unsigned int	var () const
	Get the variable number. More...
KOKKOS_FUNCTION unsigned int	ivar () const
	Get the variable number. More...
KOKKOS_FUNCTION unsigned int	jvar () const
	Get the coupled variable number. More...
KOKKOS_FUNCTION unsigned int	fe () const
	Get the variable FE type ID. More...
KOKKOS_FUNCTION unsigned int	ife () const
	Get the variable FE type ID. More...
KOKKOS_FUNCTION unsigned int	jfe () const
	Get the coupled variable FE type ID. More...
KOKKOS_FUNCTION void	do_derivatives (const bool flag)
	Set whether to compute derivatives for automatic differentiation (AD) More...
KOKKOS_FUNCTION bool	do_derivatives () const
	Get whether to compute derivatives for automatic differentiation (AD) More...
KOKKOS_FUNCTION const Assembly &	assembly () const
	Get the Kokkos assembly. More...
KOKKOS_FUNCTION const System &	system (unsigned int sys) const
	Get the Kokkos system. More...
KOKKOS_FUNCTION const Mesh &	mesh () const
	Get the Kokkos mesh. More...
KOKKOS_FUNCTION const ElementInfo &	elem () const
	Get the element information object. More...
KOKKOS_FUNCTION ContiguousElementID	elemID () const
	Get the contiguous element ID. More...
KOKKOS_FUNCTION dof_id_type	extraElemID (unsigned int index) const
	Get the extra element ID. More...
KOKKOS_FUNCTION ContiguousSubdomainID	subdomain () const
	Get the contiguous subdomain ID. More...
KOKKOS_FUNCTION unsigned int	side () const
	Get the side index. More...
KOKKOS_FUNCTION ContiguousNodeID	node () const
	Get the contiguous node ID. More...
KOKKOS_FUNCTION unsigned int	n_qps () const
	Get the number of local quadrature points. More...
KOKKOS_FUNCTION dof_id_type	qpOffset () const
	Get the starting offset into the global quadrature point index. More...
KOKKOS_FUNCTION dof_id_type	propertyIdx (const PropertyConstantOption constant_option, const unsigned int qp) const
	Get the index into the property data storage. More...
KOKKOS_FUNCTION bool	hasNeighbor () const
	Get whether the current side has a neighbor. More...
KOKKOS_FUNCTION bool	isSide () const
	Get whether the current datum is on a side. More...
KOKKOS_FUNCTION bool	isNodal () const
	Get whether the current datum is on a node. More...
KOKKOS_FUNCTION bool	isNodalDefined (const Variable &var) const
	Get whether the a variable is defined on the current node. More...
KOKKOS_FUNCTION const Real33 &	J (const unsigned int qp)
	Get the inverse of Jacobian matrix | dxi/dx deta/dx dzeta/dx | | dxi/dy deta/dy dzeta/dy | | dxi/dz deta/dz dzeta/dz |. More...
KOKKOS_FUNCTION Real	JxW (const unsigned int qp)
	Get the transformed Jacobian weight. More...
KOKKOS_FUNCTION Real3	q_point (const unsigned int qp)
	Get the physical quadrature point coordinate. More...
KOKKOS_FUNCTION Real3	normals (const unsigned int qp)
	Get the normal vector on surface. More...
KOKKOS_FUNCTION void	set_local_parallel (const unsigned int local_thread_id, const unsigned int num_local_threads)
	Set local parallelization option. More...
KOKKOS_FUNCTION unsigned int	local_thread_id () const
	Get the current local thread ID. More...
KOKKOS_FUNCTION unsigned int	num_local_threads () const
	Get the number of local threads. More...

Protected Attributes
const TagID	_tag
	Solution tag ID. More...
const unsigned int	_sys
	System number. More...
const unsigned int	_ivar
	Variable numbers. More...
const unsigned int	_jvar
const unsigned int	_ife
	FE type IDs of variables. More...
const unsigned int	_jfe
const unsigned int	_n_idofs = 1
	Number of local DOFs. More...
const unsigned int	_n_jdofs = 1
bool	_do_derivatives = true
	Whether to compute derivatives for automatic differentiation (AD) More...
const Assembly &	_assembly
	Reference of the Kokkos assembly. More...
const Array< System > &	_systems
	Reference of the Kokkos systems. More...
const Mesh &	_mesh
	Reference of the Kokkos mesh. More...
const ElementInfo	_elem
	Current element information object. More...
const unsigned int	_side = libMesh::invalid_uint
	Current side index. More...
const ContiguousNodeID	_node = libMesh::DofObject::invalid_id
	Current contiguous node ID. More...
const ContiguousElementID	_neighbor = libMesh::DofObject::invalid_id
	Current contiguous element ID of neighbor. More...
const unsigned int	_n_qps = 1
	Number of local quadrature points. More...
const dof_id_type	_qp_offset = libMesh::DofObject::invalid_id
	Starting offset into the global quadrature point index. More...
const dof_id_type	_elem_property_idx = libMesh::DofObject::invalid_id
	Index for element-constant material properties. More...

Detailed Description

The Kokkos object that holds thread-private data in the parallel operations of Kokkos kernels.

Definition at line 364 of file KokkosDatum.h.

Constructor & Destructor Documentation

AssemblyDatum() [1/2]

KOKKOS_FUNCTION Moose::Kokkos::AssemblyDatum::AssemblyDatum ( const ContiguousElementID  elem, const unsigned int  side, const Assembly &  assembly, const Array< System > &  systems, const Variable &  ivar, const unsigned int  jvar, const unsigned int  comp = 0  )

inline

Constructor for element and side data.

Parameters

elem	The contiguous element ID of the current thread
side	The side index of the current thread
assembly	The Kokkos assembly
systems	The Kokkos systems
ivar	The Kokkos variable
jvar	The coupled variable number
comp	The variable component

Definition at line 378 of file KokkosDatum.h.

    : Datum(elem, side, assembly, systems),
      _tag(ivar.tag()),
      _sys(ivar.sys(comp)),
      _ivar(ivar.var(comp)),
      _jvar(jvar),
      _ife(systems[ivar.sys(comp)].getFETypeID(_ivar)),
      _jfe(systems[ivar.sys(comp)].getFETypeID(_jvar)),
      _n_idofs(assembly.getNumDofs(_elem.type, _ife)),
      _n_jdofs(assembly.getNumDofs(_elem.type, _jfe))
  {
  }

AssemblyDatum() [2/2]

KOKKOS_FUNCTION Moose::Kokkos::AssemblyDatum::AssemblyDatum ( const ContiguousNodeID  node, const Assembly &  assembly, const Array< System > &  systems, const Variable &  ivar, const unsigned int  jvar, const unsigned int  comp = 0  )

inline

Constructor for node data.

Parameters

elem	The contiguous element ID of the current thread
assembly	The Kokkos assembly
systems	The Kokkos systems
ivar	The Kokkos variable
jvar	The coupled variable number
comp	The variable component

Definition at line 406 of file KokkosDatum.h.

    : Datum(node, assembly, systems),
      _tag(ivar.tag()),
      _sys(ivar.sys(comp)),
      _ivar(ivar.var(comp)),
      _jvar(jvar),
      _ife(systems[ivar.sys(comp)].getFETypeID(_ivar)),
      _jfe(systems[ivar.sys(comp)].getFETypeID(_jvar))
  {
  }

Member Function Documentation

assembly()

KOKKOS_FUNCTION const Assembly& Moose::Kokkos::Datum::assembly ( ) const

inlineinherited

Get the Kokkos assembly.

Returns

The Kokkos assembly

Definition at line 67 of file KokkosDatum.h.

Referenced by Moose::Kokkos::VariablePhiValue::operator()(), Moose::Kokkos::VariablePhiGradient::operator()(), Moose::Kokkos::VariableTestValue::operator()(), and Moose::Kokkos::VariableTestGradient::operator()().

{ return _assembly; }

do_derivatives() [1/2]

KOKKOS_FUNCTION void Moose::Kokkos::AssemblyDatum::do_derivatives ( const bool  flag )

inline

Set whether to compute derivatives for automatic differentiation (AD)

Parameters

flag	Whether to compute derivatives

Definition at line 476 of file KokkosDatum.h.

Referenced by Moose::Kokkos::ADNodalBC::operator()(), Moose::Kokkos::VariableValueTempl< is_ad >::operator()(), and Moose::Kokkos::VariableGradientTempl< is_ad >::operator()().

{ _do_derivatives = flag; }

do_derivatives() [2/2]

KOKKOS_FUNCTION bool Moose::Kokkos::AssemblyDatum::do_derivatives ( ) const

inline

Get whether to compute derivatives for automatic differentiation (AD)

Returns

Whether to compute derivatives

Definition at line 481 of file KokkosDatum.h.

{ return _do_derivatives; }

elem()

KOKKOS_FUNCTION const ElementInfo& Moose::Kokkos::Datum::elem ( ) const

inlineinherited

Get the element information object.

Returns

The element information object

Definition at line 84 of file KokkosDatum.h.

Referenced by Moose::Kokkos::ResidualObject::accumulateTaggedElementalMatrix(), KokkosCopyValueAux::computeElementInternal(), KokkosTimeDerivative::computeJacobianInternal(), Moose::Kokkos::ResidualObject::computeJacobianInternal(), Moose::Kokkos::ADKernel::computeResidualInternal(), Moose::Kokkos::ADIntegratedBC::computeResidualInternal(), Moose::Kokkos::ResidualObject::computeResidualInternal(), Moose::Kokkos::VariableValueTempl< is_ad >::get(), Moose::Kokkos::VariableGradientTempl< is_ad >::get(), Moose::Kokkos::VariablePhiValue::operator()(), Moose::Kokkos::VariablePhiGradient::operator()(), Moose::Kokkos::VariableTestValue::operator()(), Moose::Kokkos::VariableTestGradient::operator()(), and Moose::Kokkos::AuxKernel::setElementSolution().

{ return _elem; }

elemID()

KOKKOS_FUNCTION ContiguousElementID Moose::Kokkos::Datum::elemID ( ) const

inlineinherited

Get the contiguous element ID.

Returns

The contiguous element ID

Definition at line 89 of file KokkosDatum.h.

Referenced by KokkosExtraIDIntegralVectorPostprocessor::execute(), and KokkosExtraIDIntegralVectorPostprocessor::reduce().

{ return _elem.id; }

extraElemID()

KOKKOS_FUNCTION dof_id_type Moose::Kokkos::Datum::extraElemID ( unsigned int  index ) const

inlineinherited

Get the extra element ID.

Parameters

index

The extra element ID index

Returns

The extra element ID

Definition at line 95 of file KokkosDatum.h.

Referenced by KokkosExtraElementIDAux::computeValue().

  {
    return isNodal() ? libMesh::DofObject::invalid_id : _mesh.getExtraElementID(_elem.id, index);
  }

fe()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::fe ( ) const

inline

Get the variable FE type ID.

Returns

The variable FE type ID

Definition at line 461 of file KokkosDatum.h.

{ return _ife; }

hasNeighbor()

KOKKOS_FUNCTION bool Moose::Kokkos::Datum::hasNeighbor ( ) const

inlineinherited

Get whether the current side has a neighbor.

Returns

Whether the current side has a neighbor

Definition at line 136 of file KokkosDatum.h.

{ return _neighbor != libMesh::DofObject::invalid_id; }

ife()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::ife ( ) const

inline

Get the variable FE type ID.

Returns

The variable FE type ID

Definition at line 466 of file KokkosDatum.h.

Referenced by Moose::Kokkos::VariableTestValue::operator()(), and Moose::Kokkos::VariableTestGradient::operator()().

{ return _ife; }

isNodal()

KOKKOS_FUNCTION bool Moose::Kokkos::Datum::isNodal ( ) const

inlineinherited

Get whether the current datum is on a node.

Returns

Whether the current datum is on a node

Definition at line 146 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::extraElemID(), Moose::Kokkos::VariableValueTempl< is_ad >::get(), Moose::Kokkos::VariableGradientTempl< is_ad >::get(), Moose::Kokkos::Datum::isNodalDefined(), Moose::Kokkos::Datum::J(), Moose::Kokkos::Datum::JxW(), and Moose::Kokkos::Datum::q_point().

{ return _node != libMesh::DofObject::invalid_id; }

isNodalDefined()

KOKKOS_FUNCTION bool Moose::Kokkos::Datum::isNodalDefined ( const Variable &  var ) const

inlineinherited

Get whether the a variable is defined on the current node.

Parameters

var	The variable

Returns

Whether the variable is defined on the current node

Definition at line 291 of file KokkosDatum.h.

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

{
  if (!isNodal() || !var.nodal())
    return false;

  return _systems[var.sys()].isNodalDefined(_node, var.var());
}

isSide()

KOKKOS_FUNCTION bool Moose::Kokkos::Datum::isSide ( ) const

inlineinherited

Get whether the current datum is on a side.

Returns

Whether the current datum is on a side

Definition at line 141 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::normals(), and Moose::Kokkos::Datum::reinitTransform().

{ return _side != libMesh::invalid_uint; }

ivar()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::ivar ( ) const

inline

Get the variable number.

Returns

The variable number

Definition at line 451 of file KokkosDatum.h.

{ return _ivar; }

J()

KOKKOS_FUNCTION const Real33 & Moose::Kokkos::Datum::J ( const unsigned int  qp )

inlineinherited

Get the inverse of Jacobian matrix | dxi/dx deta/dx dzeta/dx | | dxi/dy deta/dy dzeta/dy | | dxi/dz deta/dz dzeta/dz |.

Parameters

qp	The local quadrature point index

Returns

The Jacobian matrix

Definition at line 300 of file KokkosDatum.h.

Referenced by Moose::Kokkos::VariableGradientTempl< is_ad >::get(), Moose::Kokkos::VariablePhiGradient::operator()(), and Moose::Kokkos::VariableTestGradient::operator()().

{
  if (!isNodal())
    reinitTransform(qp);
  else
    _J.identity(_assembly.getDimension());

  return _J;
}

jfe()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::jfe ( ) const

inline

Get the coupled variable FE type ID.

Returns

The variable FE type ID

Definition at line 471 of file KokkosDatum.h.

Referenced by Moose::Kokkos::VariablePhiValue::operator()(), and Moose::Kokkos::VariablePhiGradient::operator()().

{ return _jfe; }

jvar()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::jvar ( ) const

inline

Get the coupled variable number.

Returns

The variable number

Definition at line 456 of file KokkosDatum.h.

Referenced by KokkosTimeDerivative::computeJacobianInternal(), Moose::Kokkos::ResidualObject::computeJacobianInternal(), Moose::Kokkos::KernelValue::computeOffDiagJacobianInternal(), Moose::Kokkos::KernelGrad::computeOffDiagJacobianInternal(), Moose::Kokkos::IntegratedBCValue::computeOffDiagJacobianInternal(), Moose::Kokkos::Kernel::computeOffDiagJacobianInternal(), and Moose::Kokkos::IntegratedBC::computeOffDiagJacobianInternal().

{ return _jvar; }

JxW()

KOKKOS_FUNCTION Real Moose::Kokkos::Datum::JxW ( const unsigned int  qp )

inlineinherited

Get the transformed Jacobian weight.

Parameters

qp	The local quadrature point index

Returns

The transformed Jacobian weights

Definition at line 311 of file KokkosDatum.h.

Referenced by Moose::Kokkos::AuxKernel::computeElementInternal(), KokkosTimeDerivative::computeJacobianInternal(), Moose::Kokkos::KernelValue::computeJacobianInternal(), Moose::Kokkos::IntegratedBCValue::computeJacobianInternal(), Moose::Kokkos::KernelGrad::computeJacobianInternal(), Moose::Kokkos::IntegratedBC::computeJacobianInternal(), Moose::Kokkos::Kernel::computeJacobianInternal(), Moose::Kokkos::KernelValue::computeOffDiagJacobianInternal(), Moose::Kokkos::IntegratedBCValue::computeOffDiagJacobianInternal(), Moose::Kokkos::KernelGrad::computeOffDiagJacobianInternal(), Moose::Kokkos::IntegratedBC::computeOffDiagJacobianInternal(), Moose::Kokkos::Kernel::computeOffDiagJacobianInternal(), KokkosTimeDerivative::computeResidualInternal(), Moose::Kokkos::TimeKernel::computeResidualInternal(), Moose::Kokkos::ADKernel::computeResidualInternal(), Moose::Kokkos::ADIntegratedBC::computeResidualInternal(), Moose::Kokkos::KernelValue::computeResidualInternal(), Moose::Kokkos::KernelGrad::computeResidualInternal(), Moose::Kokkos::IntegratedBCValue::computeResidualInternal(), Moose::Kokkos::Kernel::computeResidualInternal(), Moose::Kokkos::IntegratedBC::computeResidualInternal(), KokkosElementVariableStatistics::computeValue(), KokkosExtraIDIntegralVectorPostprocessor::execute(), KokkosIntegralPostprocessor< Base >::reduce(), and KokkosExtraIDIntegralVectorPostprocessor::reduce().

{
  if (!isNodal())
    reinitTransform(qp);
  else
    _JxW = 1;

  return _JxW;
}

local_thread_id()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::Datum::local_thread_id ( ) const

inlineinherited

Get the current local thread ID.

Returns

The current local thread ID

Definition at line 197 of file KokkosDatum.h.

Referenced by KokkosTimeDerivative::computeJacobianInternal(), Moose::Kokkos::ResidualObject::computeJacobianInternal(), Moose::Kokkos::ADKernel::computeResidualInternal(), Moose::Kokkos::ADIntegratedBC::computeResidualInternal(), Moose::Kokkos::ResidualObject::computeResidualInternal(), and Moose::Kokkos::Datum::set_local_parallel().

{ return _local_thread_id; }

mesh()

KOKKOS_FUNCTION const Mesh& Moose::Kokkos::Datum::mesh ( ) const

inlineinherited

Get the Kokkos mesh.

Returns

The Kokkos mesh

Definition at line 78 of file KokkosDatum.h.

{ return _mesh; }

n_dofs()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::n_dofs ( ) const

inline

Get the number of local DOFs.

Returns

The number of local DOFs

Definition at line 426 of file KokkosDatum.h.

Referenced by KokkosCopyValueAux::computeElementInternal(), Moose::Kokkos::AuxKernel::computeElementInternal(), Moose::Kokkos::ADKernel::computeResidualInternal(), Moose::Kokkos::ADIntegratedBC::computeResidualInternal(), Moose::Kokkos::ResidualObject::computeResidualInternal(), and Moose::Kokkos::AuxKernel::setElementSolution().

{ return _n_idofs; }

n_idofs()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::n_idofs ( ) const

inline

Get the number of local DOFs.

Returns

The number of local DOFs

Definition at line 431 of file KokkosDatum.h.

Referenced by KokkosTimeDerivative::computeJacobianInternal(), and Moose::Kokkos::ResidualObject::computeJacobianInternal().

{ return _n_idofs; }

n_jdofs()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::n_jdofs ( ) const

inline

Get the number of local DOFs for the coupled variable.

Returns

The number of local DOFs

Definition at line 436 of file KokkosDatum.h.

Referenced by KokkosTimeDerivative::computeJacobianInternal(), and Moose::Kokkos::ResidualObject::computeJacobianInternal().

{ return _n_jdofs; }

n_qps()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::Datum::n_qps ( ) const

inlineinherited

Get the number of local quadrature points.

Returns

The number of local quadrature points

Definition at line 118 of file KokkosDatum.h.

Referenced by Moose::Kokkos::AuxKernel::computeElementInternal(), KokkosTimeDerivative::computeJacobianInternal(), Moose::Kokkos::KernelValue::computeJacobianInternal(), Moose::Kokkos::IntegratedBCValue::computeJacobianInternal(), Moose::Kokkos::KernelGrad::computeJacobianInternal(), Moose::Kokkos::IntegratedBC::computeJacobianInternal(), Moose::Kokkos::Kernel::computeJacobianInternal(), Moose::Kokkos::KernelValue::computeOffDiagJacobianInternal(), Moose::Kokkos::IntegratedBCValue::computeOffDiagJacobianInternal(), Moose::Kokkos::KernelGrad::computeOffDiagJacobianInternal(), Moose::Kokkos::IntegratedBC::computeOffDiagJacobianInternal(), Moose::Kokkos::Kernel::computeOffDiagJacobianInternal(), KokkosTimeDerivative::computeResidualInternal(), Moose::Kokkos::TimeKernel::computeResidualInternal(), Moose::Kokkos::ADKernel::computeResidualInternal(), Moose::Kokkos::ADIntegratedBC::computeResidualInternal(), Moose::Kokkos::KernelValue::computeResidualInternal(), Moose::Kokkos::KernelGrad::computeResidualInternal(), Moose::Kokkos::IntegratedBCValue::computeResidualInternal(), Moose::Kokkos::Kernel::computeResidualInternal(), Moose::Kokkos::IntegratedBC::computeResidualInternal(), KokkosElementVariableStatistics::computeValue(), KokkosExtraIDIntegralVectorPostprocessor::execute(), Moose::Kokkos::Material::operator()(), KokkosSideExtremeValue::reduce(), KokkosElementExtremeValue::reduce(), KokkosIntegralPostprocessor< Base >::reduce(), and KokkosExtraIDIntegralVectorPostprocessor::reduce().

{ return _n_qps; }

node()

KOKKOS_FUNCTION ContiguousNodeID Moose::Kokkos::Datum::node ( ) const

inlineinherited

Get the contiguous node ID.

Returns

The contiguous node ID

Definition at line 113 of file KokkosDatum.h.

Referenced by KokkosCopyValueAux::computeNodeInternal(), KokkosBoundNodalKernel::computeQpJacobian(), KokkosBoundNodalKernel::computeQpOffDiagJacobian(), KokkosBoundNodalKernel::computeQpResidual(), Moose::Kokkos::VariableValueTempl< is_ad >::get(), KokkosNodalMaxValueId::reduce(), and Moose::Kokkos::AuxKernel::setNodeSolution().

{ return _node; }

normals()

KOKKOS_FUNCTION Real3 Moose::Kokkos::Datum::normals ( const unsigned int  qp )

inlineinherited

Get the normal vector on surface.

Parameters

qp	The local quadrature point index

Returns

The normal vector

Definition at line 333 of file KokkosDatum.h.

Referenced by KokkosDirectionalNeumannBC::computeQpResidual().

{
  KOKKOS_ASSERT(isSide());

  if (isSide())
    reinitTransform(qp);

  return _normal;
}

num_local_threads()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::Datum::num_local_threads ( ) const

inlineinherited

Get the number of local threads.

Returns

The number of local threads

Definition at line 202 of file KokkosDatum.h.

Referenced by KokkosTimeDerivative::computeJacobianInternal(), Moose::Kokkos::ResidualObject::computeJacobianInternal(), Moose::Kokkos::ADKernel::computeResidualInternal(), Moose::Kokkos::ADIntegratedBC::computeResidualInternal(), Moose::Kokkos::ResidualObject::computeResidualInternal(), and Moose::Kokkos::Datum::set_local_parallel().

{ return _num_local_threads; }

propertyIdx()

KOKKOS_FUNCTION dof_id_type Moose::Kokkos::Datum::propertyIdx ( const PropertyConstantOption  constant_option, const unsigned int  qp  ) const

inlineinherited

Get the index into the property data storage.

Parameters

constant_option	The property constant option
qp	The local quadrature point index

Returns

The index

Definition at line 278 of file KokkosDatum.h.

{
  dof_id_type idx = 0;

  if (constant_option == PropertyConstantOption::NONE)
    idx = _qp_offset + qp;
  else if (constant_option == PropertyConstantOption::ELEMENT)
    idx = _elem_property_idx;

  return idx;
}

q_point()

KOKKOS_FUNCTION Real3 Moose::Kokkos::Datum::q_point ( const unsigned int  qp )

inlineinherited

Get the physical quadrature point coordinate.

Parameters

qp	The local quadrature point index

Returns

The physical quadrature point coordinate

Definition at line 322 of file KokkosDatum.h.

Referenced by KokkosParsedMaterial::computeQpProperties(), KokkosFunctionAux::computeValue(), and KokkosParsedAux::computeValue().

{
  if (!isNodal())
    reinitTransform(qp);
  else
    _xyz = _assembly.kokkosMesh().getNodePoint(_node);

  return _xyz;
}

qpOffset()

KOKKOS_FUNCTION dof_id_type Moose::Kokkos::Datum::qpOffset ( ) const

inlineinherited

Get the starting offset into the global quadrature point index.

Returns

The starting offset

Definition at line 123 of file KokkosDatum.h.

Referenced by Moose::Kokkos::VariableValueTempl< is_ad >::get(), and Moose::Kokkos::VariableGradientTempl< is_ad >::get().

{ return _qp_offset; }

set_local_parallel()

KOKKOS_FUNCTION void Moose::Kokkos::Datum::set_local_parallel ( const unsigned int  local_thread_id, const unsigned int  num_local_threads  )

inlineinherited

Set local parallelization option.

Parameters

local_thread_id	The current local thread ID
num_local_threads	The number of local threads

Definition at line 187 of file KokkosDatum.h.

Referenced by Moose::Kokkos::ADKernel::operator()(), Moose::Kokkos::ADIntegratedBC::operator()(), Moose::Kokkos::Kernel::operator()(), and Moose::Kokkos::IntegratedBC::operator()().

  {
    _local_thread_id = local_thread_id;
    _num_local_threads = num_local_threads;
  }

side()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::Datum::side ( ) const

inlineinherited

Get the side index.

Returns

The side index

Definition at line 108 of file KokkosDatum.h.

Referenced by Moose::Kokkos::VariableValueTempl< is_ad >::get(), Moose::Kokkos::VariableGradientTempl< is_ad >::get(), Moose::Kokkos::VariablePhiValue::operator()(), Moose::Kokkos::VariablePhiGradient::operator()(), Moose::Kokkos::VariableTestValue::operator()(), and Moose::Kokkos::VariableTestGradient::operator()().

{ return _side; }

subdomain()

KOKKOS_FUNCTION ContiguousSubdomainID Moose::Kokkos::Datum::subdomain ( ) const

inlineinherited

Get the contiguous subdomain ID.

Returns

The contiguous subdomain ID

Definition at line 103 of file KokkosDatum.h.

Referenced by KokkosExtraIDIntegralVectorPostprocessor::execute(), and KokkosExtraIDIntegralVectorPostprocessor::reduce().

{ return _elem.subdomain; }

sys()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::sys ( ) const

inline

Get the system number of variable.

Returns

The system number of variable

Definition at line 441 of file KokkosDatum.h.

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

{ return _sys; }

system()

KOKKOS_FUNCTION const System& Moose::Kokkos::Datum::system ( unsigned int  sys ) const

inlineinherited

Get the Kokkos system.

Parameters

sys	The system number

Returns

The Kokkos system

Definition at line 73 of file KokkosDatum.h.

Referenced by Moose::Kokkos::VariableValueTempl< is_ad >::get(), and Moose::Kokkos::VariableGradientTempl< is_ad >::get().

{ return _systems[sys]; }

var()

KOKKOS_FUNCTION unsigned int Moose::Kokkos::AssemblyDatum::var ( ) const

inline

Get the variable number.

Returns

The variable number

Definition at line 446 of file KokkosDatum.h.

{ return _ivar; }

Member Data Documentation

_assembly

const Assembly& Moose::Kokkos::Datum::_assembly

protectedinherited

Reference of the Kokkos assembly.

Definition at line 208 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::assembly(), Moose::Kokkos::Datum::J(), Moose::Kokkos::Datum::q_point(), and Moose::Kokkos::Datum::reinitTransform().

_do_derivatives

bool Moose::Kokkos::AssemblyDatum::_do_derivatives = true

protected

Whether to compute derivatives for automatic differentiation (AD)

Definition at line 507 of file KokkosDatum.h.

Referenced by do_derivatives().

_elem

const ElementInfo Moose::Kokkos::Datum::_elem

protectedinherited

Current element information object.

Definition at line 220 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::elem(), Moose::Kokkos::Datum::elemID(), Moose::Kokkos::Datum::extraElemID(), Moose::Kokkos::Datum::reinitTransform(), and Moose::Kokkos::Datum::subdomain().

_elem_property_idx

const dof_id_type Moose::Kokkos::Datum::_elem_property_idx = libMesh::DofObject::invalid_id

protectedinherited

Index for element-constant material properties.

Definition at line 244 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::propertyIdx().

_ife

const unsigned int Moose::Kokkos::AssemblyDatum::_ife

protected

FE type IDs of variables.

Definition at line 499 of file KokkosDatum.h.

Referenced by fe(), and ife().

_ivar

const unsigned int Moose::Kokkos::AssemblyDatum::_ivar

protected

Variable numbers.

Definition at line 495 of file KokkosDatum.h.

Referenced by ivar(), and var().

_jfe

const unsigned int Moose::Kokkos::AssemblyDatum::_jfe

protected

Definition at line 499 of file KokkosDatum.h.

Referenced by jfe().

_jvar

const unsigned int Moose::Kokkos::AssemblyDatum::_jvar

protected

Definition at line 495 of file KokkosDatum.h.

Referenced by jvar().

_mesh

const Mesh& Moose::Kokkos::Datum::_mesh

protectedinherited

Reference of the Kokkos mesh.

Definition at line 216 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::extraElemID(), and Moose::Kokkos::Datum::mesh().

_n_idofs

const unsigned int Moose::Kokkos::AssemblyDatum::_n_idofs = 1

protected

Number of local DOFs.

Definition at line 503 of file KokkosDatum.h.

Referenced by n_dofs(), and n_idofs().

_n_jdofs

const unsigned int Moose::Kokkos::AssemblyDatum::_n_jdofs = 1

protected

Definition at line 503 of file KokkosDatum.h.

Referenced by n_jdofs().

_n_qps

const unsigned int Moose::Kokkos::Datum::_n_qps = 1

protectedinherited

Number of local quadrature points.

Definition at line 236 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::n_qps().

_neighbor

const ContiguousElementID Moose::Kokkos::Datum::_neighbor = libMesh::DofObject::invalid_id

protectedinherited

Current contiguous element ID of neighbor.

Definition at line 232 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::hasNeighbor().

_node

const ContiguousNodeID Moose::Kokkos::Datum::_node = libMesh::DofObject::invalid_id

protectedinherited

Current contiguous node ID.

Definition at line 228 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::isNodal(), Moose::Kokkos::Datum::isNodalDefined(), Moose::Kokkos::Datum::node(), and Moose::Kokkos::Datum::q_point().

_qp_offset

const dof_id_type Moose::Kokkos::Datum::_qp_offset = libMesh::DofObject::invalid_id

protectedinherited

Starting offset into the global quadrature point index.

Definition at line 240 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::propertyIdx(), and Moose::Kokkos::Datum::qpOffset().

_side

const unsigned int Moose::Kokkos::Datum::_side = libMesh::invalid_uint

protectedinherited

Current side index.

Definition at line 224 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::isSide(), Moose::Kokkos::Datum::reinitTransform(), and Moose::Kokkos::Datum::side().

_sys

const unsigned int Moose::Kokkos::AssemblyDatum::_sys

protected

System number.

Definition at line 491 of file KokkosDatum.h.

Referenced by sys().

_systems

const Array<System>& Moose::Kokkos::Datum::_systems

protectedinherited

Reference of the Kokkos systems.

Definition at line 212 of file KokkosDatum.h.

Referenced by Moose::Kokkos::Datum::isNodalDefined(), and Moose::Kokkos::Datum::system().

_tag

const TagID Moose::Kokkos::AssemblyDatum::_tag

protected

Solution tag ID.

Definition at line 487 of file KokkosDatum.h.

---

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

• include/kokkos/base/KokkosDatum.h