KokkosADNodalBC.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 "KokkosNodalBCBase.h"

namespace Moose::Kokkos
{

class ADNodalBC : public NodalBCBase
{
public:
  static InputParameters validParams();

  ADNodalBC(const InputParameters & parameters);

  virtual void computeResidual() override;
  virtual void computeJacobian() override;
  virtual void computeResidualAndJacobian() override;

  template <typename Derived>
  KOKKOS_FUNCTION void operator()(ResidualLoop, const ThreadID tid, const Derived & bc) const;

protected:
  virtual void dispatch();

  const ADVariableValue _u;
  bool _computing_residual = false;
  bool _computing_jacobian = false;
};

template <typename Derived>
KOKKOS_FUNCTION void
ADNodalBC::operator()(ResidualLoop, const ThreadID tid, const Derived & bc) const
{
  auto node = kokkosBoundaryNodeID(tid);
  auto & sys = kokkosSystem(_kokkos_var.sys());

  if (!sys.isNodalDefined(node, _kokkos_var.var()))
    return;

  AssemblyDatum datum(node, kokkosAssembly(), kokkosSystems(), _kokkos_var, _kokkos_var.var());

  datum.do_derivatives(_computing_jacobian);

  ADReal local_re = bc.template computeQpResidual<Derived>(0, datum);

  if (_computing_residual)
    accumulateTaggedNodalResidual(false, local_re.value(), node);
  if (_computing_jacobian)
    accumulateTaggedNodalMatrix(false, local_re.derivatives(), node);
}

} // namespace Moose::Kokkos