KokkosUtils.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#pragma once
 
#include "KokkosHeader.h"
 
namespace Moose::Kokkos
{
namespace Utils
{
 
template <typename T>
KOKKOS_INLINE_FUNCTION T
sign(T x)
{
  return x >= 0.0 ? 1.0 : -1.0;
}
 
template <typename T>
KOKKOS_INLINE_FUNCTION const T *
find(const T & target, const T * const begin, const T * const end)
{
  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;
}
 
KOKKOS_INLINE_FUNCTION void
choleskySolve(Real * const A, Real * const x, Real * const b, const unsigned int n)
{
  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];
  }
}
 
} // namespace Utils
} // namespace Moose::Kokkos