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