KokkosVectorKernelGrad.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 "KokkosVectorKernel.h"

namespace Moose::Kokkos
{

class VectorKernelGrad : public VectorKernel
{
public:
  static InputParameters validParams();

  static constexpr bool use_precompute_hooks = true;

  VectorKernelGrad(const InputParameters & parameters);

  template <typename Derived>
  KOKKOS_FUNCTION Real33 precomputeQpJacobian(const unsigned int /* j */,
                                              const unsigned int /* qp */,
                                              AssemblyDatum & /* datum */) const
  {
    ::Kokkos::abort("Default precomputeQpJacobian() should never be called. Make sure you properly "
                    "redefined this method in your class without typos.");

    return Real33(0);
  }
  template <typename Derived>
  KOKKOS_FUNCTION Real33 precomputeQpOffDiagJacobian(const unsigned int /* j */,
                                                     const unsigned int /* jvar */,
                                                     const unsigned int /* qp */,
                                                     AssemblyDatum & /* datum */) const
  {
    ::Kokkos::abort(
        "Default precomputeQpOffDiagJacobian() should never be called. Make sure you properly "
        "redefined this method in your class without typos.");

    return Real33(0);
  }

  template <typename Derived>
  static auto defaultJacobian()
  {
    return &VectorKernelGrad::precomputeQpJacobian<Derived>;
  }
  template <typename Derived>
  static auto defaultOffDiagJacobian()
  {
    return &VectorKernelGrad::precomputeQpOffDiagJacobian<Derived>;
  }

  template <typename Derived>
  KOKKOS_FUNCTION void computeResidualInternal(const Derived & kernel, AssemblyDatum & datum) const;
  template <typename Derived>
  KOKKOS_FUNCTION void computeJacobianInternal(const Derived & kernel, AssemblyDatum & datum) const;
  template <typename Derived>
  KOKKOS_FUNCTION void computeOffDiagJacobianInternal(const Derived & kernel,
                                                      AssemblyDatum & datum) const;
};

template <typename Derived>
KOKKOS_FUNCTION void
VectorKernelGrad::computeResidualInternal(const Derived & kernel, AssemblyDatum & datum) const
{
  ResidualObject::computeResidualInternal(
      datum,
      [&](Real * local_re, const unsigned int ib, const unsigned int ie)
      {
        for (unsigned int qp = 0; qp < datum.n_qps(); ++qp)
        {
          Real33 value =
              (datum.JxW(qp) * kernel.template precomputeQpResidual<Derived>(qp, datum)) *
              datum.J(qp);

          for (unsigned int i = ib; i < ie; ++i)
            local_re[i] += value.contract(_grad_test.reference(datum, i, qp));
        }
      });
}

template <typename Derived>
KOKKOS_FUNCTION void
VectorKernelGrad::computeJacobianInternal(const Derived & kernel, AssemblyDatum & datum) const
{
  ResidualObject::computeJacobianInternal(
      datum,
      [&](Real * local_ke, const unsigned int ib, const unsigned int ie, const unsigned int j)
      {
        for (unsigned int qp = 0; qp < datum.n_qps(); ++qp)
        {
          Real33 value =
              (datum.JxW(qp) * kernel.template precomputeQpJacobian<Derived>(j, qp, datum)) *
              datum.J(qp);

          for (unsigned int i = ib; i < ie; ++i)
            local_ke[i] += value.contract(_grad_test.reference(datum, i, qp));
        }
      });
}

template <typename Derived>
KOKKOS_FUNCTION void
VectorKernelGrad::computeOffDiagJacobianInternal(const Derived & kernel,
                                                 AssemblyDatum & datum) const
{
  ResidualObject::computeJacobianInternal(
      datum,
      [&](Real * local_ke, const unsigned int ib, const unsigned int ie, const unsigned int j)
      {
        for (unsigned int qp = 0; qp < datum.n_qps(); ++qp)
        {
          Real33 value = (datum.JxW(qp) * kernel.template precomputeQpOffDiagJacobian<Derived>(
                                              j, datum.jvar(), qp, datum)) *
                         datum.J(qp);

          for (unsigned int i = ib; i < ie; ++i)
            local_ke[i] += value.contract(_grad_test.reference(datum, i, qp));
        }
      });
}

} // namespace Moose::Kokkos