NEML2 2.1.0
Loading...
Searching...
No Matches
ImplicitUpdate.h
1// Copyright 2024, UChicago Argonne, LLC
2// All Rights Reserved
3// Software Name: NEML2 -- the New Engineering material Model Library, version 2
4// By: Argonne National Laboratory
5// OPEN SOURCE LICENSE (MIT)
6//
7// Permission is hereby granted, free of charge, to any person obtaining a copy
8// of this software and associated documentation files (the "Software"), to deal
9// in the Software without restriction, including without limitation the rights
10// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11// copies of the Software, and to permit persons to whom the Software is
12// furnished to do so, subject to the following conditions:
13//
14// The above copyright notice and this permission notice shall be included in
15// all copies or substantial portions of the Software.
16//
17// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23// THE SOFTWARE.
24
25#pragma once
26
27#include "neml2/models/Model.h"
28#include "neml2/models/ModelNonlinearSystem.h"
29#include "neml2/solvers/NonlinearSolver.h"
30
31namespace neml2
32{
33class ImplicitUpdate : public Model
34{
35public:
37
38 ImplicitUpdate(const OptionSet & options);
39
40 void to(const TensorOptions & options) override;
41
42 std::size_t last_iterations() const { return _last_iterations; }
43
44protected:
45 void set_value(bool out, bool dout_din, bool d2out_din2) override;
46
48 std::shared_ptr<ModelNonlinearSystem> _sys;
49
51 std::shared_ptr<NonlinearSolver> _solver;
52
54 std::size_t _last_iterations = 0;
55};
56} // namespace neml2
std::shared_ptr< ModelNonlinearSystem > _sys
The underlying nonlinear system that wraps around a Model.
Definition ImplicitUpdate.h:48
std::size_t last_iterations() const
Definition ImplicitUpdate.h:42
ImplicitUpdate(const OptionSet &options)
void to(const TensorOptions &options) override
Send model to a different device or dtype.
std::size_t _last_iterations
Last solve result.
Definition ImplicitUpdate.h:54
static OptionSet expected_options()
void set_value(bool out, bool dout_din, bool d2out_din2) override
The map between input -> output, and optionally its derivatives.
std::shared_ptr< NonlinearSolver > _solver
The nonlinear solver used to solve the nonlinear system.
Definition ImplicitUpdate.h:51
Model(const OptionSet &options)
Construct a new Model object.
A custom map-like data structure. The keys are strings, and the values can be nonhomogeneously typed.
Definition OptionSet.h:52
Definition DiagnosticsInterface.h:31
c10::TensorOptions TensorOptions
Definition types.h:66