Moose::Kokkos::Utils Namespace Reference

Functions
template<typename T >
KOKKOS_INLINE_FUNCTION T	sign (T x)
	Returns the sign of a value.
template<typename T >
KOKKOS_INLINE_FUNCTION const T *	find (const T &target, const T *const begin, const T *const end)
	Find a value in an array.
KOKKOS_INLINE_FUNCTION void	choleskySolve (Real *const A, Real *const x, Real *const b, const unsigned int n)
	Perform an in-place linear solve using Cholesky decomposition Matrix and right-hand-side vector are modified after this call.

Function Documentation

choleskySolve()

KOKKOS_INLINE_FUNCTION void Moose::Kokkos::Utils::choleskySolve	(	Real *const	A,
		Real *const	x,
		Real *const	b,
		const unsigned int	n
	)

Perform an in-place linear solve using Cholesky decomposition Matrix and right-hand-side vector are modified after this call.

Parameters

A	The row-major matrix
x	The solution vector
b	The right-hand-side vector
n	The system size

Definition at line 72 of file KokkosUtils.h.

{
  for (unsigned int i = 0; i < n; ++i)
  {
    for (unsigned int j = 0; j <= i; ++j)
    {
      Real sum = A[j + n * i];
 
      for (unsigned int k = 0; k < j; ++k)
        sum -= A[k + n * i] * A[k + n * j];
 
      if (i == j)
        A[j + n * i] = ::Kokkos::sqrt(sum);
      else
        A[j + n * i] = sum / A[j + n * j];
    }
  }
 
  for (unsigned int i = 0; i < n; ++i)
  {
    Real sum = b[i];
 
    for (unsigned int j = 0; j < i; ++j)
      sum -= A[j + n * i] * b[j];
 
    b[i] = sum / A[i + n * i];
  }
 
  for (int i = n - 1; i >= 0; --i)
  {
    Real sum = b[i];
 
    for (unsigned int j = i + 1; j < n; ++j)
      sum -= A[i + n * j] * x[j];
 
    x[i] = sum / A[i + n * i];
  }
}

Referenced by Moose::Kokkos::AuxKernel::computeElementInternal().

find()

template<typename T >

KOKKOS_INLINE_FUNCTION const T * Moose::Kokkos::Utils::find	(	const T &	target,
		const T *const	begin,
		const T *const	end
	)

Find a value in an array.

Parameters

target	The target value to find
begin	The pointer to the first element of the array
end	The pointer next to the last element of the array

Returns

The pointer to the target element, end if the target element was not found

Definition at line 40 of file KokkosUtils.h.

{
  if (begin == end)
    return end;
 
  auto left = begin;
  auto right = end - 1;
 
  while (left <= right)
  {
    auto mid = left + (right - left) / 2;
 
    if (*mid == target)
      return mid;
    else if (*mid < target)
      left = mid + 1;
    else
      right = mid - 1;
  }
 
  return end;
}

Referenced by Moose::Kokkos::Matrix::find(), and Moose::Kokkos::Mesh::isBoundaryNode().

sign()

template<typename T >

KOKKOS_INLINE_FUNCTION T Moose::Kokkos::Utils::sign

(

T

x

)

Returns the sign of a value.

Parameters

x

The value

Returns

The sign of the value

Definition at line 26 of file KokkosUtils.h.

{
  return x >= 0.0 ? 1.0 : -1.0;
}

Referenced by KokkosPiecewiseConstant::value().