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