Line data    Source code

       1             : //* This file is part of the MOOSE framework
       2             : //* https://mooseframework.inl.gov
       3             : //*
       4             : //* All rights reserved, see COPYRIGHT for full restrictions
       5             : //* https://github.com/idaholab/moose/blob/master/COPYRIGHT
       6             : //*
       7             : //* Licensed under LGPL 2.1, please see LICENSE for details
       8             : //* https://www.gnu.org/licenses/lgpl-2.1.html
       9             : 
      10             : #pragma once
      11             : 
      12             : #include "KokkosReducerBase.h"
      13             : 
      14             : #include "BoundaryRestrictableRequired.h"
      15             : 
      16             : namespace Moose::Kokkos
      17             : {
      18             : 
      19             : class SideReducer : public ReducerBase, public ::BoundaryRestrictableRequired
      20             : {
      21             : public:
      22             :   static InputParameters validParams();
      23             : 
      24             :   SideReducer(const MooseObject * moose_object);
      25             : 
      26             :   /**
      27             :    * Copy constructor for parallel dispatch
      28             :    */
      29             :   SideReducer(const SideReducer & object);
      30             : 
      31         182 :   virtual ThreadID numReducerThreads() const override { return numKokkosBoundarySides(); }
      32             : 
      33             :   /**
      34             :    * The parallel computation entry function called by Kokkos
      35             :    */
      36             :   template <typename Derived>
      37             :   KOKKOS_FUNCTION void
      38             :   operator()(ReducerLoop, const ThreadID tid, const Derived & reducer, Real * result) const;
      39             : };
      40             : 
      41             : template <typename Derived>
      42             : KOKKOS_FUNCTION void
      43         740 : SideReducer::operator()(ReducerLoop,
      44             :                         const ThreadID tid,
      45             :                         const Derived & reducer,
      46             :                         Real * result) const
      47             : {
      48         740 :   auto [elem, side] = kokkosBoundaryElementSideID(tid);
      49             : 
      50         740 :   Datum datum(elem, side, kokkosAssembly(), kokkosSystems());
      51             : 
      52         740 :   reducer.template reduce<Derived>(datum, result);
      53         740 : }
      54             : 
      55             : } // namespace Moose::Kokkos