Moose::Kokkos::Function Class Reference

The abstract class that provides polymorphic interfaces for a function. More...

#include <KokkosFunction.h>

Public Member Functions
	Function ()=default
	Default constructor. More...
	Function (std::shared_ptr< FunctionWrapperHostBase > wrapper)
	Constructor. More...
	Function (const Function &function)
	Copy constructor for parallel dispatch. More...
	~Function ()
	Destructor. More...
KOKKOS_FUNCTION	operator bool () const
	Get whether the function wrapper is valid. More...
KOKKOS_FUNCTION Real	value (Real t, Real3 p) const
KOKKOS_FUNCTION Real3	vectorValue (Real t, Real3 p) const
KOKKOS_FUNCTION Real3	gradient (Real t, Real3 p) const
KOKKOS_FUNCTION Real3	curl (Real t, Real3 p) const
KOKKOS_FUNCTION Real	div (Real t, Real3 p) const
KOKKOS_FUNCTION Real	timeDerivative (Real t, Real3 p) const
KOKKOS_FUNCTION Real	timeIntegral (Real t1, Real t2, Real3 p) const
KOKKOS_FUNCTION Real	integral () const
KOKKOS_FUNCTION Real	average () const

Private Attributes
std::shared_ptr< FunctionWrapperHostBase >	_wrapper_host
	Pointer to the host function wrapper. More...
FunctionWrapperDeviceBase *	_wrapper_device = nullptr
	Pointer to the device function wrapper. More...

Detailed Description

The abstract class that provides polymorphic interfaces for a function.

NOTE: This class is not the base class for a Kokkos function derivation. The user should derive their own function from Moose::Kokkos::FunctionBase.

Definition at line 147 of file KokkosFunction.h.

Constructor & Destructor Documentation

Function() [1/3]

Moose::Kokkos::Function::Function ( )

default

Default constructor.

Function() [2/3]

Moose::Kokkos::Function::Function

(

std::shared_ptr< FunctionWrapperHostBase >

wrapper

)

Constructor.

Parameters

wrapper

The host function wrapper

Function() [3/3]

Moose::Kokkos::Function::Function

(

const Function &

function

)

Copy constructor for parallel dispatch.

~Function()

Moose::Kokkos::Function::~Function

(

)

Destructor.

Member Function Documentation

average()

KOKKOS_FUNCTION Real Moose::Kokkos::Function::average ( ) const

inline

Definition at line 227 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->average();
  }

curl()

KOKKOS_FUNCTION Real3 Moose::Kokkos::Function::curl ( Real  t, Real3  p  ) const

inline

Definition at line 197 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->curl(t, p);
  }

div()

KOKKOS_FUNCTION Real Moose::Kokkos::Function::div ( Real  t, Real3  p  ) const

inline

Definition at line 203 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->div(t, p);
  }

gradient()

KOKKOS_FUNCTION Real3 Moose::Kokkos::Function::gradient ( Real  t, Real3  p  ) const

inline

Definition at line 191 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->gradient(t, p);
  }

integral()

KOKKOS_FUNCTION Real Moose::Kokkos::Function::integral ( ) const

inline

Definition at line 221 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->integral();
  }

operator bool()

KOKKOS_FUNCTION Moose::Kokkos::Function::operator bool ( ) const

inline

Get whether the function wrapper is valid.

Returns

Whether the function wrapper is valid

Definition at line 172 of file KokkosFunction.h.

  {
    KOKKOS_IF_ON_HOST(return static_cast<bool>(_wrapper_host);)

    return _wrapper_device != nullptr;
  }

timeDerivative()

KOKKOS_FUNCTION Real Moose::Kokkos::Function::timeDerivative ( Real  t, Real3  p  ) const

inline

Definition at line 209 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->timeDerivative(t, p);
  }

timeIntegral()

KOKKOS_FUNCTION Real Moose::Kokkos::Function::timeIntegral ( Real  t1, Real  t2, Real3  p  ) const

inline

Definition at line 215 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->timeIntegral(t1, t2, p);
  }

value()

KOKKOS_FUNCTION Real Moose::Kokkos::Function::value ( Real  t, Real3  p  ) const

inline

Definition at line 179 of file KokkosFunction.h.

Referenced by KokkosFunctionAux::computeValue().

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->value(t, p);
  }

vectorValue()

KOKKOS_FUNCTION Real3 Moose::Kokkos::Function::vectorValue ( Real  t, Real3  p  ) const

inline

Definition at line 185 of file KokkosFunction.h.

  {
    KOKKOS_ASSERT(_wrapper_device);

    return _wrapper_device->vectorValue(t, p);
  }

Member Data Documentation

_wrapper_device

FunctionWrapperDeviceBase* Moose::Kokkos::Function::_wrapper_device = nullptr

private

Pointer to the device function wrapper.

Definition at line 242 of file KokkosFunction.h.

Referenced by average(), curl(), div(), gradient(), integral(), operator bool(), timeDerivative(), timeIntegral(), value(), and vectorValue().

_wrapper_host

std::shared_ptr<FunctionWrapperHostBase> Moose::Kokkos::Function::_wrapper_host

private

Pointer to the host function wrapper.

Definition at line 238 of file KokkosFunction.h.

Referenced by operator bool().

---

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

• include/kokkos/functions/KokkosFunction.h