KokkosNodalExtremeValue.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* 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 "KokkosNodalVariablePostprocessor.h"
#include "KokkosExtremeValueBase.h"

class KokkosNodalExtremeValue : public KokkosExtremeValueBase<KokkosNodalVariablePostprocessor>
{
public:
  static InputParameters validParams();

  KokkosNodalExtremeValue(const InputParameters & parameters);

  KOKKOS_FUNCTION Kokkos::pair<Real, Real> getProxyValuePair(const unsigned int qp,
                                                             Datum & datum) const;

  template <typename Derived>
  KOKKOS_FUNCTION void reduce(Datum & datum, Real * result) const;

protected:
  const Moose::Kokkos::VariableValue _proxy_variable;
};

KOKKOS_FUNCTION inline Kokkos::pair<Real, Real>
KokkosNodalExtremeValue::getProxyValuePair(const unsigned int qp, Datum & datum) const
{
  return Kokkos::make_pair(_proxy_variable(datum, qp), _u(datum, qp));
}

template <typename Derived>
KOKKOS_FUNCTION void
KokkosNodalExtremeValue::reduce(Datum & datum, Real * result) const
{
  if (datum.isNodalDefined(_u.variable()))
  {
    KOKKOS_ASSERT(datum.isNodalDefined(_proxy_variable.variable()));

    static_cast<const Derived *>(this)->template computeExtremeValue<Derived>(0, datum, result);
  }
}