KokkosMaterialPropertyStorage.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 "MaterialPropertyStorage.h"

class FEProblemBase;

void
dataStore(std::ostream & stream, Moose::Kokkos::MaterialPropertyStorage & storage, void * context);
void
dataLoad(std::istream & stream, Moose::Kokkos::MaterialPropertyStorage & storage, void * context);

namespace Moose::Kokkos
{

class MaterialPropertyStorage : protected ::MaterialPropertyStorage
{
  friend void ::dataStore(std::ostream &, MaterialPropertyStorage &, void *);
  friend void ::dataLoad(std::istream &, MaterialPropertyStorage &, void *);

public:
  MaterialPropertyStorage(MaterialPropertyRegistry & registry, FEProblemBase & problem);

  static MaterialPropertyStorage & cast(::MaterialPropertyStorage & storage);

  using ::MaterialPropertyStorage::addConsumer;
  using ::MaterialPropertyStorage::getConsumers;
  using ::MaterialPropertyStorage::getMaterialData;
  using ::MaterialPropertyStorage::getMaterialPropertyRegistry;
  using ::MaterialPropertyStorage::hasStatefulProperties;
  using ::MaterialPropertyStorage::setRecovering;

#ifdef MOOSE_KOKKOS_SCOPE

  MaterialPropertyBase & addKokkosProperty(const std::string & prop_name,
                                           const std::type_info & type,
                                           const unsigned int state,
                                           const ::MaterialBase * declarer,
                                           std::shared_ptr<MaterialPropertyBase> shell);
  MaterialPropertyBase & addKokkosPropertyState(const std::string & prop_name,
                                                const unsigned int state,
                                                std::shared_ptr<MaterialPropertyBase> shell);
  MaterialPropertyBase & declareKokkosProperty(const std::string & prop_name,
                                               const std::type_info & type,
                                               const ::MaterialBase * declarer,
                                               const std::vector<unsigned int> & dims,
                                               const bool bnd,
                                               const bool on_demand,
                                               const PropertyConstantOption constant_option,
                                               std::shared_ptr<MaterialPropertyBase> shell);
  MaterialPropertyBase & getKokkosProperty(std::string prop_name, unsigned int state = 0);
  bool haveKokkosProperty(std::string prop_name, unsigned int state = 0);
  void registerLoadStore(std::type_index type, PropertyStore store, PropertyLoad load);
  void allocateKokkosProperties();
#endif

  void shift();
  void copy();

private:
  std::unordered_map<std::string, std::shared_ptr<MaterialPropertyBase>>
      _kokkos_props[MaterialData::max_state + 1];
  std::unordered_map<std::string, Moose::Kokkos::PropRecord> _kokkos_prop_records;
  static std::unordered_map<std::type_index, PropertyStore> _store_functions;
  static std::unordered_map<std::type_index, PropertyLoad> _load_functions;
};

} // namespace Moose::Kokkos