KokkosCoupledForce.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 "KokkosKernel.h"

class KokkosCoupledForce final : public Moose::Kokkos::Kernel<KokkosCoupledForce>
{
public:
  static InputParameters validParams();

  KokkosCoupledForce(const InputParameters & parameters);

  KOKKOS_FUNCTION Real computeQpResidual(const unsigned int i,
                                         const unsigned int qp,
                                         ResidualDatum & datum) const;

private:
  const unsigned int _v_var;
  const Moose::Kokkos::VariableValue _v;
  const Real _coef;
};

KOKKOS_FUNCTION inline Real
KokkosCoupledForce::computeQpResidual(const unsigned int i,
                                      const unsigned int qp,
                                      ResidualDatum & datum) const
{
  return -_coef * _v(datum, qp) * _test(datum, i, qp);
}