KokkosReducerBase.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* 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 "KokkosHeader.h"
#include "KokkosDatum.h"
#include "KokkosDispatcher.h"
#include "KokkosFESystem.h"
 
#include "MooseObject.h"
 
class FEProblemBase;
 
namespace Moose::Kokkos
{
 
class ReducerBase : public MeshHolder, public AssemblyHolder, public FESystemHolder
{
public:
  ReducerBase(const MooseObject * object);
 
  ReducerBase(const ReducerBase & reducer);
 
  struct ReducerLoop
  {
  };
 
  template <typename Derived>
  KOKKOS_FUNCTION void reduce(Datum & /* datum */, Real * /* result */) const
  {
    ::Kokkos::abort("Default reduce() should never be called. Make sure you properly redefined "
                    "this method in your class without typos.");
  }
  template <typename Derived>
  KOKKOS_FUNCTION void join(Real * /* result */, const Real * /* source */) const
  {
    ::Kokkos::abort("Default join() should never be called. Make sure you properly redefined this "
                    "method in your class without typos.");
  }
  template <typename Derived>
  KOKKOS_FUNCTION void init(Real * /* result */) const
  {
    ::Kokkos::abort("Default init() should never be called. Make sure you properly redefined this "
                    "method in your class without typos.");
  }
 
  template <typename Derived>
  static auto defaultReduce()
  {
    return &ReducerBase::reduce<Derived>;
  }
 
protected:
  virtual void computeReducer();
  virtual ThreadID numReducerThreads() const = 0;
  void allocateReductionBuffer(const unsigned int size)
  {
    ::Kokkos::realloc(_reduction_buffer, size);
  }
 
  const MooseObject * _reducer_object;
  std::unique_ptr<DispatcherBase> _reducer_dispatcher;
  ::Kokkos::View<Real *, ::Kokkos::HostSpace> _reduction_buffer;
};
 
} // namespace Moose::Kokkos