KokkosHeatConduction.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 "KokkosKernelGrad.h"
 
class KokkosHeatConduction : public Moose::Kokkos::KernelGrad
{
  using Real3 = Moose::Kokkos::Real3;
 
public:
  static InputParameters validParams();
 
  KokkosHeatConduction(const InputParameters & parameters);
 
  template <typename Derived>
  KOKKOS_FUNCTION Real3 precomputeQpResidual(const unsigned int qp, AssemblyDatum & datum) const;
  template <typename Derived>
  KOKKOS_FUNCTION Real3 precomputeQpJacobian(const unsigned int j,
                                             const unsigned int qp,
                                             AssemblyDatum & datum) const;
 
private:
  Moose::Kokkos::MaterialProperty<Real> _thermal_conductivity;
  Moose::Kokkos::MaterialProperty<Real> _d_thermal_conductivity_dT;
};
 
template <typename Derived>
KOKKOS_FUNCTION Moose::Kokkos::Real3
KokkosHeatConduction::precomputeQpResidual(const unsigned int qp, AssemblyDatum & datum) const
{
  return _thermal_conductivity(datum, qp) * _grad_u(datum, qp);
}
 
template <typename Derived>
KOKKOS_FUNCTION Moose::Kokkos::Real3
KokkosHeatConduction::precomputeQpJacobian(const unsigned int j,
                                           const unsigned int qp,
                                           AssemblyDatum & datum) const
{
  Real3 jac = _thermal_conductivity(datum, qp) * _grad_phi(datum, j, qp);
  if (_d_thermal_conductivity_dT)
    jac += _d_thermal_conductivity_dT(datum, qp) * _phi(datum, j, qp) * _grad_u(datum, qp);
  return jac;
}