MOOSEToNEML2.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 "NEML2Utils.h"
#include "InputParameters.h"
 
// NEML2 v3 exchanges plain at::Tensor across its C++ API; at::Tensor is available via
// NEML2Utils.h (which includes ATen) when NEML2 is enabled.
 
class MOOSEToNEML2
{
public:
  static InputParameters validParams();
 
  MOOSEToNEML2(const InputParameters & params);
 
#ifdef NEML2_ENABLED
  const std::string & NEML2Name() const { return _neml2_name; }
 
  virtual at::Tensor gatheredData() const = 0;
 
  void insertInto(std::map<std::string, at::Tensor> &) const;
 
private:
  const std::string _neml2_name;
#endif
};