{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "4e32fbb6", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:44.201733Z", "iopub.status.busy": "2026-10-02T14:42:44.201531Z", "iopub.status.idle": "2026-10-02T14:42:44.206181Z", "shell.execute_reply": "2026-10-02T14:42:44.205576Z" }, "papermill": { "duration": 0.007378, "end_time": "2026-10-02T14:42:44.206728+00:00", "exception": false, "start_time": "2026-10-02T14:42:44.199350+00:00", "status": "completed" }, "tags": [ "remove-input", "active-ipynb", "remove-output" ] }, "outputs": [], "source": [ "try:\n", " from openmdao.utils.notebook_utils import notebook_mode # noqa: F401\n", "except ImportError:\n", " !python -m pip install openmdao[notebooks]" ] }, { "cell_type": "markdown", "id": "ae877281", "metadata": { "papermill": { "duration": 0.001156, "end_time": "2026-10-02T14:42:44.209335+00:00", "exception": false, "start_time": "2026-10-02T14:42:44.208179+00:00", "status": "completed" }, "tags": [] }, "source": [ "# NonlinearBlockGS\n", "\n", "NonlinearBlockGS applies Block Gauss-Seidel (also known as fixed-point iteration) to the\n", "components and subsystems in the system. This is mainly used to solve cyclic connections. You\n", "should try this solver for systems that satisfy the following conditions:\n", "\n", "1. System (or subsystem) contains a cycle, though subsystems may.\n", "2. System does not contain any implicit states, though subsystems may.\n", "\n", "NonlinearBlockGS is a block solver, so you can specify different nonlinear solvers in the subsystems and they\n", "will be utilized to solve the subsystem nonlinear problem.\n", "\n", "Note that you may not know if you satisfy the second condition, so choosing a solver can be a trial-and-error proposition. If\n", "NonlinearBlockGS doesn't work, then you will need to use [NewtonSolver](../../../_srcdocs/packages/solvers.nonlinear/newton).\n", "\n", "Here, we choose NonlinearBlockGS to solve the Sellar problem, which has two components with a\n", "cyclic dependency, has no implicit states, and works very well with Gauss-Seidel." ] }, { "cell_type": "code", "execution_count": 2, "id": "bd51210d", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:44.212472Z", "iopub.status.busy": "2026-10-02T14:42:44.212329Z", "iopub.status.idle": "2026-10-02T14:42:46.150234Z", "shell.execute_reply": "2026-10-02T14:42:46.149725Z" }, "papermill": { "duration": 1.940282, "end_time": "2026-10-02T14:42:46.150818+00:00", "exception": false, "start_time": "2026-10-02T14:42:44.210536+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [ { "data": { "application/papermill.record/text/html": "
class SellarDis1withDerivatives(SellarDis1):\n    """\n    Component containing Discipline 1 -- derivatives version.\n    """\n\n    def setup_partials(self):\n        # Analytic Derivs\n        self.declare_partials(of='*', wrt='*')\n\n    def compute_partials(self, inputs, partials):\n        """\n        Jacobian for Sellar discipline 1.\n        """\n        partials['y1', 'y2'] = -0.2\n        partials['y1', 'z'] = np.array([[2.0 * inputs['z'][0], 1.0]])\n        partials['y1', 'x'] = 1.0\n
\n", "application/papermill.record/text/latex": "\\begin{Verbatim}[commandchars=\\\\\\{\\}]\n\\PY{k}{class}\\PY{+w}{ }\\PY{n+nc}{SellarDis1withDerivatives}\\PY{p}{(}\\PY{n}{SellarDis1}\\PY{p}{)}\\PY{p}{:}\n\\PY{+w}{ }\\PY{l+s+sd}{\\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\\PY{l+s+sd}{ Component containing Discipline 1 \\PYZhy{}\\PYZhy{} derivatives version.}\n\\PY{l+s+sd}{ \\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\n \\PY{k}{def}\\PY{+w}{ }\\PY{n+nf}{setup\\PYZus{}partials}\\PY{p}{(}\\PY{n+nb+bp}{self}\\PY{p}{)}\\PY{p}{:}\n \\PY{c+c1}{\\PYZsh{} Analytic Derivs}\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{declare\\PYZus{}partials}\\PY{p}{(}\\PY{n}{of}\\PY{o}{=}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{*}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{wrt}\\PY{o}{=}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{*}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{)}\n\n \\PY{k}{def}\\PY{+w}{ }\\PY{n+nf}{compute\\PYZus{}partials}\\PY{p}{(}\\PY{n+nb+bp}{self}\\PY{p}{,} \\PY{n}{inputs}\\PY{p}{,} \\PY{n}{partials}\\PY{p}{)}\\PY{p}{:}\n\\PY{+w}{ }\\PY{l+s+sd}{\\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\\PY{l+s+sd}{ Jacobian for Sellar discipline 1.}\n\\PY{l+s+sd}{ \\PYZdq{}\\PYZdq{}\\PYZdq{}}\n \\PY{n}{partials}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]} \\PY{o}{=} \\PY{o}{\\PYZhy{}}\\PY{l+m+mf}{0.2}\n \\PY{n}{partials}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{z}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]} \\PY{o}{=} \\PY{n}{np}\\PY{o}{.}\\PY{n}{array}\\PY{p}{(}\\PY{p}{[}\\PY{p}{[}\\PY{l+m+mf}{2.0} \\PY{o}{*} \\PY{n}{inputs}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{z}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]}\\PY{p}{[}\\PY{l+m+mi}{0}\\PY{p}{]}\\PY{p}{,} \\PY{l+m+mf}{1.0}\\PY{p}{]}\\PY{p}{]}\\PY{p}{)}\n \\PY{n}{partials}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{x}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]} \\PY{o}{=} \\PY{l+m+mf}{1.0}\n\\end{Verbatim}\n", "application/papermill.record/text/plain": "class SellarDis1withDerivatives(SellarDis1):\n \"\"\"\n Component containing Discipline 1 -- derivatives version.\n \"\"\"\n\n def setup_partials(self):\n # Analytic Derivs\n self.declare_partials(of='*', wrt='*')\n\n def compute_partials(self, inputs, partials):\n \"\"\"\n Jacobian for Sellar discipline 1.\n \"\"\"\n partials['y1', 'y2'] = -0.2\n partials['y1', 'z'] = np.array([[2.0 * inputs['z'][0], 1.0]])\n partials['y1', 'x'] = 1.0" }, "metadata": { "scrapbook": { "mime_prefix": "application/papermill.record/", "name": "code_src33" } }, "output_type": "display_data" } ], "source": [ "from openmdao.utils.notebook_utils import get_code\n", "from myst_nb import glue\n", "glue(\"code_src33\", get_code(\"openmdao.test_suite.components.sellar.SellarDis1withDerivatives\"), display=False)" ] }, { "cell_type": "markdown", "id": "98a10a81", "metadata": { "papermill": { "duration": 0.001827, "end_time": "2026-10-02T14:42:46.155202+00:00", "exception": false, "start_time": "2026-10-02T14:42:46.153375+00:00", "status": "completed" }, "tags": [] }, "source": [ ":::{dropdown} `SellarDis1withDerivatives` class definition \n", "\n", "{glue:}`code_src33`\n", ":::" ] }, { "cell_type": "code", "execution_count": 3, "id": "304d9bd2", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:46.159337Z", "iopub.status.busy": "2026-10-02T14:42:46.159176Z", "iopub.status.idle": "2026-10-02T14:42:46.164757Z", "shell.execute_reply": "2026-10-02T14:42:46.164202Z" }, "papermill": { "duration": 0.008464, "end_time": "2026-10-02T14:42:46.165374+00:00", "exception": false, "start_time": "2026-10-02T14:42:46.156910+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [ { "data": { "application/papermill.record/text/html": "
class SellarDis2withDerivatives(SellarDis2):\n    """\n    Component containing Discipline 2 -- derivatives version.\n    """\n\n    def setup_partials(self):\n        # Analytic Derivs\n        self.declare_partials(of='*', wrt='*')\n\n    def compute_partials(self, inputs, J):\n        """\n        Jacobian for Sellar discipline 2.\n        """\n        y1 = inputs['y1']\n        if y1.real < 0.0:\n            y1 *= -1\n        if y1.real < 1e-8:\n            y1 = 1e-8\n\n        J['y2', 'y1'] = .5*y1**-.5\n        J['y2', 'z'] = np.array([[1.0, 1.0]])\n
\n", "application/papermill.record/text/latex": "\\begin{Verbatim}[commandchars=\\\\\\{\\}]\n\\PY{k}{class}\\PY{+w}{ }\\PY{n+nc}{SellarDis2withDerivatives}\\PY{p}{(}\\PY{n}{SellarDis2}\\PY{p}{)}\\PY{p}{:}\n\\PY{+w}{ }\\PY{l+s+sd}{\\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\\PY{l+s+sd}{ Component containing Discipline 2 \\PYZhy{}\\PYZhy{} derivatives version.}\n\\PY{l+s+sd}{ \\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\n \\PY{k}{def}\\PY{+w}{ }\\PY{n+nf}{setup\\PYZus{}partials}\\PY{p}{(}\\PY{n+nb+bp}{self}\\PY{p}{)}\\PY{p}{:}\n \\PY{c+c1}{\\PYZsh{} Analytic Derivs}\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{declare\\PYZus{}partials}\\PY{p}{(}\\PY{n}{of}\\PY{o}{=}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{*}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{wrt}\\PY{o}{=}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{*}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{)}\n\n \\PY{k}{def}\\PY{+w}{ }\\PY{n+nf}{compute\\PYZus{}partials}\\PY{p}{(}\\PY{n+nb+bp}{self}\\PY{p}{,} \\PY{n}{inputs}\\PY{p}{,} \\PY{n}{J}\\PY{p}{)}\\PY{p}{:}\n\\PY{+w}{ }\\PY{l+s+sd}{\\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\\PY{l+s+sd}{ Jacobian for Sellar discipline 2.}\n\\PY{l+s+sd}{ \\PYZdq{}\\PYZdq{}\\PYZdq{}}\n \\PY{n}{y1} \\PY{o}{=} \\PY{n}{inputs}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]}\n \\PY{k}{if} \\PY{n}{y1}\\PY{o}{.}\\PY{n}{real} \\PY{o}{\\PYZlt{}} \\PY{l+m+mf}{0.0}\\PY{p}{:}\n \\PY{n}{y1} \\PY{o}{*}\\PY{o}{=} \\PY{o}{\\PYZhy{}}\\PY{l+m+mi}{1}\n \\PY{k}{if} \\PY{n}{y1}\\PY{o}{.}\\PY{n}{real} \\PY{o}{\\PYZlt{}} \\PY{l+m+mf}{1e\\PYZhy{}8}\\PY{p}{:}\n \\PY{n}{y1} \\PY{o}{=} \\PY{l+m+mf}{1e\\PYZhy{}8}\n\n \\PY{n}{J}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]} \\PY{o}{=} \\PY{l+m+mf}{.5}\\PY{o}{*}\\PY{n}{y1}\\PY{o}{*}\\PY{o}{*}\\PY{o}{\\PYZhy{}}\\PY{l+m+mf}{.5}\n \\PY{n}{J}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{z}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]} \\PY{o}{=} \\PY{n}{np}\\PY{o}{.}\\PY{n}{array}\\PY{p}{(}\\PY{p}{[}\\PY{p}{[}\\PY{l+m+mf}{1.0}\\PY{p}{,} \\PY{l+m+mf}{1.0}\\PY{p}{]}\\PY{p}{]}\\PY{p}{)}\n\\end{Verbatim}\n", "application/papermill.record/text/plain": "class SellarDis2withDerivatives(SellarDis2):\n \"\"\"\n Component containing Discipline 2 -- derivatives version.\n \"\"\"\n\n def setup_partials(self):\n # Analytic Derivs\n self.declare_partials(of='*', wrt='*')\n\n def compute_partials(self, inputs, J):\n \"\"\"\n Jacobian for Sellar discipline 2.\n \"\"\"\n y1 = inputs['y1']\n if y1.real < 0.0:\n y1 *= -1\n if y1.real < 1e-8:\n y1 = 1e-8\n\n J['y2', 'y1'] = .5*y1**-.5\n J['y2', 'z'] = np.array([[1.0, 1.0]])" }, "metadata": { "scrapbook": { "mime_prefix": "application/papermill.record/", "name": "code_src34" } }, "output_type": "display_data" } ], "source": [ "from openmdao.utils.notebook_utils import get_code\n", "from myst_nb import glue\n", "glue(\"code_src34\", get_code(\"openmdao.test_suite.components.sellar.SellarDis2withDerivatives\"), display=False)" ] }, { "cell_type": "markdown", "id": "0d20a0e0", "metadata": { "papermill": { "duration": 0.00126, "end_time": "2026-10-02T14:42:46.168042+00:00", "exception": false, "start_time": "2026-10-02T14:42:46.166782+00:00", "status": "completed" }, "tags": [] }, "source": [ ":::{dropdown} `SellarDis2withDerivatives` class definition \n", "\n", "{glue:}`code_src34`\n", ":::" ] }, { "cell_type": "code", "execution_count": 4, "id": "1bb93c51", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:46.171362Z", "iopub.status.busy": "2026-10-02T14:42:46.171188Z", "iopub.status.idle": "2026-10-02T14:42:46.181193Z", "shell.execute_reply": "2026-10-02T14:42:46.180677Z" }, "papermill": { "duration": 0.012405, "end_time": "2026-10-02T14:42:46.181732+00:00", "exception": false, "start_time": "2026-10-02T14:42:46.169327+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [ { "data": { "application/papermill.record/text/html": "
class SellarDerivatives(om.Group):\n    """\n    Group containing the Sellar MDA. This version uses the disciplines with derivatives.\n    """\n\n    def setup(self):\n        self.add_subsystem('d1', SellarDis1withDerivatives(), promotes=['x', 'z', 'y1', 'y2'])\n        self.add_subsystem('d2', SellarDis2withDerivatives(), promotes=['z', 'y1', 'y2'])\n\n        self.add_subsystem('obj_cmp', om.ExecComp('obj = x**2 + z[1] + y1 + exp(-y2)', obj=0.0,\n                                                  x=0.0, z=np.array([0.0, 0.0]), y1=0.0, y2=0.0),\n                           promotes=['obj', 'x', 'z', 'y1', 'y2'])\n\n        self.add_subsystem('con_cmp1', om.ExecComp('con1 = 3.16 - y1', con1=0.0, y1=0.0),\n                           promotes=['con1', 'y1'])\n        self.add_subsystem('con_cmp2', om.ExecComp('con2 = y2 - 24.0', con2=0.0, y2=0.0),\n                           promotes=['con2', 'y2'])\n\n        self.set_input_defaults('x', 1.0)\n        self.set_input_defaults('z', np.array([5.0, 2.0]))\n
\n", "application/papermill.record/text/latex": "\\begin{Verbatim}[commandchars=\\\\\\{\\}]\n\\PY{k}{class}\\PY{+w}{ }\\PY{n+nc}{SellarDerivatives}\\PY{p}{(}\\PY{n}{om}\\PY{o}{.}\\PY{n}{Group}\\PY{p}{)}\\PY{p}{:}\n\\PY{+w}{ }\\PY{l+s+sd}{\\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\\PY{l+s+sd}{ Group containing the Sellar MDA. This version uses the disciplines with derivatives.}\n\\PY{l+s+sd}{ \\PYZdq{}\\PYZdq{}\\PYZdq{}}\n\n \\PY{k}{def}\\PY{+w}{ }\\PY{n+nf}{setup}\\PY{p}{(}\\PY{n+nb+bp}{self}\\PY{p}{)}\\PY{p}{:}\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{add\\PYZus{}subsystem}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{d1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{SellarDis1withDerivatives}\\PY{p}{(}\\PY{p}{)}\\PY{p}{,} \\PY{n}{promotes}\\PY{o}{=}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{x}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{z}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]}\\PY{p}{)}\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{add\\PYZus{}subsystem}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{d2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{SellarDis2withDerivatives}\\PY{p}{(}\\PY{p}{)}\\PY{p}{,} \\PY{n}{promotes}\\PY{o}{=}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{z}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]}\\PY{p}{)}\n\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{add\\PYZus{}subsystem}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{obj\\PYZus{}cmp}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{om}\\PY{o}{.}\\PY{n}{ExecComp}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{obj = x**2 + z[1] + y1 + exp(\\PYZhy{}y2)}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{obj}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{,}\n \\PY{n}{x}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{,} \\PY{n}{z}\\PY{o}{=}\\PY{n}{np}\\PY{o}{.}\\PY{n}{array}\\PY{p}{(}\\PY{p}{[}\\PY{l+m+mf}{0.0}\\PY{p}{,} \\PY{l+m+mf}{0.0}\\PY{p}{]}\\PY{p}{)}\\PY{p}{,} \\PY{n}{y1}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{,} \\PY{n}{y2}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{)}\\PY{p}{,}\n \\PY{n}{promotes}\\PY{o}{=}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{obj}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{x}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{z}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]}\\PY{p}{)}\n\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{add\\PYZus{}subsystem}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{con\\PYZus{}cmp1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{om}\\PY{o}{.}\\PY{n}{ExecComp}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{con1 = 3.16 \\PYZhy{} y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{con1}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{,} \\PY{n}{y1}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{)}\\PY{p}{,}\n \\PY{n}{promotes}\\PY{o}{=}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{con1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y1}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]}\\PY{p}{)}\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{add\\PYZus{}subsystem}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{con\\PYZus{}cmp2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{om}\\PY{o}{.}\\PY{n}{ExecComp}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{con2 = y2 \\PYZhy{} 24.0}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{con2}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{,} \\PY{n}{y2}\\PY{o}{=}\\PY{l+m+mf}{0.0}\\PY{p}{)}\\PY{p}{,}\n \\PY{n}{promotes}\\PY{o}{=}\\PY{p}{[}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{con2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{y2}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{]}\\PY{p}{)}\n\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{set\\PYZus{}input\\PYZus{}defaults}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{x}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{l+m+mf}{1.0}\\PY{p}{)}\n \\PY{n+nb+bp}{self}\\PY{o}{.}\\PY{n}{set\\PYZus{}input\\PYZus{}defaults}\\PY{p}{(}\\PY{l+s+s1}{\\PYZsq{}}\\PY{l+s+s1}{z}\\PY{l+s+s1}{\\PYZsq{}}\\PY{p}{,} \\PY{n}{np}\\PY{o}{.}\\PY{n}{array}\\PY{p}{(}\\PY{p}{[}\\PY{l+m+mf}{5.0}\\PY{p}{,} \\PY{l+m+mf}{2.0}\\PY{p}{]}\\PY{p}{)}\\PY{p}{)}\n\\end{Verbatim}\n", "application/papermill.record/text/plain": "class SellarDerivatives(om.Group):\n \"\"\"\n Group containing the Sellar MDA. This version uses the disciplines with derivatives.\n \"\"\"\n\n def setup(self):\n self.add_subsystem('d1', SellarDis1withDerivatives(), promotes=['x', 'z', 'y1', 'y2'])\n self.add_subsystem('d2', SellarDis2withDerivatives(), promotes=['z', 'y1', 'y2'])\n\n self.add_subsystem('obj_cmp', om.ExecComp('obj = x**2 + z[1] + y1 + exp(-y2)', obj=0.0,\n x=0.0, z=np.array([0.0, 0.0]), y1=0.0, y2=0.0),\n promotes=['obj', 'x', 'z', 'y1', 'y2'])\n\n self.add_subsystem('con_cmp1', om.ExecComp('con1 = 3.16 - y1', con1=0.0, y1=0.0),\n promotes=['con1', 'y1'])\n self.add_subsystem('con_cmp2', om.ExecComp('con2 = y2 - 24.0', con2=0.0, y2=0.0),\n promotes=['con2', 'y2'])\n\n self.set_input_defaults('x', 1.0)\n self.set_input_defaults('z', np.array([5.0, 2.0]))" }, "metadata": { "scrapbook": { "mime_prefix": "application/papermill.record/", "name": "code_nbgs_sellar_derivs" } }, "output_type": "display_data" } ], "source": [ "from openmdao.utils.notebook_utils import get_code\n", "from myst_nb import glue\n", "glue(\"code_nbgs_sellar_derivs\", get_code(\"openmdao.test_suite.components.sellar_feature.SellarDerivatives\"), display=False)" ] }, { "cell_type": "markdown", "id": "d6d15fd5", "metadata": { "papermill": { "duration": 0.001457, "end_time": "2026-10-02T14:42:46.184816+00:00", "exception": false, "start_time": "2026-10-02T14:42:46.183359+00:00", "status": "completed" }, "tags": [] }, "source": [ ":::{dropdown} `SellarDerivatives` class definition \n", "\n", "{glue:}`code_nbgs_sellar_derivs`\n", ":::" ] }, { "cell_type": "code", "execution_count": 5, "id": "fb5af9cb", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:46.188465Z", "iopub.status.busy": "2026-10-02T14:42:46.188300Z", "iopub.status.idle": "2026-10-02T14:42:47.412294Z", "shell.execute_reply": "2026-10-02T14:42:47.411791Z" }, "papermill": { "duration": 1.226838, "end_time": "2026-10-02T14:42:47.413106+00:00", "exception": false, "start_time": "2026-10-02T14:42:46.186268+00:00", "status": "completed" }, "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[1790952167.399369] [runnervm8df0l:7295 :0] ib_iface.c:1269 UCX ERROR mana_0: iface 0x55b63d8ac590 failed to create UD QP TX wr:256 sge:6 inl:64 resp:0 RX wr:4096 sge:1 resp:0 failed: Operation not supported\n", "[1790952167.399606] [runnervm8df0l:7295 :0] ucp_worker.c:1412 UCX ERROR uct_iface_open(ud_verbs/mana_0:1) failed: Input/output error\n", "NL: NLBGS Converged in 8 iterations\n", "[25.58830237]\n", "[12.05848815]\n" ] }, { "name": "stderr", "output_type": "stream", "text": [ "[runnervm8df0l:07295] pml_ucx.c:313 Error: Failed to create UCP worker\n" ] } ], "source": [ "import numpy as np\n", "\n", "import openmdao.api as om\n", "from openmdao.test_suite.components.sellar_feature import SellarDerivatives\n", "\n", "prob = om.Problem(model=SellarDerivatives())\n", "prob.setup()\n", "\n", "prob.model.nonlinear_solver = om.NonlinearBlockGS()\n", "\n", "prob.set_val('x', 1.)\n", "prob.set_val('z', np.array([5.0, 2.0]))\n", "\n", "prob.run_model()\n", "\n", "print(prob.get_val('y1'))\n", "print(prob.get_val('y2'))" ] }, { "cell_type": "code", "execution_count": 6, "id": "b47e7153", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.419348Z", "iopub.status.busy": "2026-10-02T14:42:47.419201Z", "iopub.status.idle": "2026-10-02T14:42:47.423602Z", "shell.execute_reply": "2026-10-02T14:42:47.422883Z" }, "papermill": { "duration": 0.007981, "end_time": "2026-10-02T14:42:47.424078+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.416097+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [ { "data": { "text/plain": [ "np.float64(3.266469240658292e-09)" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "from openmdao.utils.assert_utils import assert_near_equal\n", "\n", "assert_near_equal(prob.get_val('y1'), 25.58830273, .00001)\n", "assert_near_equal(prob.get_val('y2'), 12.05848819, .00001)" ] }, { "cell_type": "markdown", "id": "a1860a0d", "metadata": { "papermill": { "duration": 0.00162, "end_time": "2026-10-02T14:42:47.439821+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.438201+00:00", "status": "completed" }, "tags": [] }, "source": [ "This solver runs all of the subsystems each iteration, passing data along all connections\n", "including the cyclic ones. After each iteration, the iteration count and the residual norm are\n", "checked to see if termination has been satisfied.\n", "\n", "You can control the termination criteria for the solver using the following options:\n", "\n", "# NonlinearBlockGS Options" ] }, { "cell_type": "code", "execution_count": 7, "id": "0e93b4d2", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.443891Z", "iopub.status.busy": "2026-10-02T14:42:47.443735Z", "iopub.status.idle": "2026-10-02T14:42:47.447774Z", "shell.execute_reply": "2026-10-02T14:42:47.446991Z" }, "papermill": { "duration": 0.006877, "end_time": "2026-10-02T14:42:47.448238+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.441361+00:00", "status": "completed" }, "tags": [ "remove-input" ] }, "outputs": [ { "data": { "text/html": [ "\n", "\n", "\n", "\n", " \n", "\n", "\n", "

\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
OptionDefaultAcceptable ValuesAcceptable TypesDescription
aitken_initial_factor1.0N/AN/Ainitial value for Aitken relaxation factor
aitken_max_factor1.5N/AN/Aupper limit for Aitken relaxation factor
aitken_min_factor0.1N/AN/Alower limit for Aitken relaxation factor
atol1e-10N/AN/Aabsolute error tolerance
cs_reconvergeTrue[True, False]['bool']When True, when this driver solves under a complex step, nudge the Solution vector by a small amount so that it reconverges.
debug_printFalse[True, False]['bool']If True, the values of input and output variables at the start of iteration are printed and written to a file after a failure to converge or when encountering aninvalid value in the residual.
err_on_non_convergeFalse[True, False]['bool']When True, AnalysisError will be raised if we don't converge.
iprint1N/A['int']whether to print output
maxiter10N/A['int']maximum number of iterations
reraise_child_analysiserrorFalse[True, False]['bool']When the option is true, a solver will reraise any AnalysisError that arises during subsolve; when false, it will continue solving.
restart_from_successfulFalse[True, False]['bool']If True, the states are cached after a successful solve and used to restart the solver in the case of a failed solve.
rtol1e-10N/AN/Arelative error tolerance
stall_limit0N/AN/ANumber of iterations after which, if the residual norms are identical within the stall_tol, then terminate as if max iterations were reached. Default is 0, which disables this feature.
stall_tol1e-12N/AN/AWhen stall checking is enabled, the threshold below which the residual norm is considered unchanged.
stall_tol_typerel['abs', 'rel']N/ASpecifies whether the absolute or relative norm of the residual is used for stall detection.
use_aitkenFalse[True, False]['bool']set to True to use Aitken relaxation
use_apply_nonlinearFalse[True, False]['bool']Set to True to always call apply_nonlinear on the solver's system after solve_nonlinear has been called.
\n", "\n", "\n" ], "text/plain": [ "" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "om.show_options_table(\"openmdao.solvers.nonlinear.nonlinear_block_gs.NonlinearBlockGS\")" ] }, { "cell_type": "markdown", "id": "b1a77d83", "metadata": { "papermill": { "duration": 0.001633, "end_time": "2026-10-02T14:42:47.451590+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.449957+00:00", "status": "completed" }, "tags": [] }, "source": [ "## NonlinearBlockGS Constructor\n", "\n", "The call signature for the `NonlinearBlockGS` constructor is:\n", "\n", "```{eval-rst}\n", " .. automethod:: openmdao.solvers.nonlinear.nonlinear_block_gs.NonlinearBlockGS.__init__\n", " :noindex:\n", "```\n", "\n", "## Aitken relaxation\n", "\n", "This solver implements Aitken relaxation, as described in Algorithm 1 of this paper on aerostructual design [optimization](http://www.umich.edu/~mdolaboratory/pdf/Kenway2014a.pdf).\n", "The relaxation is turned off by default, but it may help convergence for more tightly coupled models.\n", "\n", "## Residual Calculation\n", "\n", "The `Unified Derivatives Equations` are formulated so that explicit equations (via `ExplicitComponent`) are also expressed\n", "as implicit relationships, and their residual is also calculated in \"apply_nonlinear\", which runs the component a second time and\n", "saves the difference in the output vector as the residual. However, this would require an extra call to `compute`, which is\n", "inefficient for slower components. To eliminate the inefficiency of running the model twice every iteration the NonlinearBlockGS\n", "driver saves a copy of the output vector and uses that to calculate the residual without rerunning the model. This does require\n", "a little more memory, so if you are solving a model where memory is more of a concern than execution time, you can set the\n", "\"use_apply_nonlinear\" option to True to use the original formulation that calls \"apply_nonlinear\" on the subsystem.\n", "\n", "\n", "## NonlinearBlockGS Option Examples\n", "\n", "**maxiter**\n", "\n", " `maxiter` lets you specify the maximum number of Gauss-Seidel iterations to apply. In this example, we\n", " cut it back from the default, ten, down to two, so that it terminates a few iterations earlier and doesn't\n", " reach the specified absolute or relative tolerance." ] }, { "cell_type": "code", "execution_count": 8, "id": "a912e1ec", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.455544Z", "iopub.status.busy": "2026-10-02T14:42:47.455406Z", "iopub.status.idle": "2026-10-02T14:42:47.463036Z", "shell.execute_reply": "2026-10-02T14:42:47.461746Z" }, "papermill": { "duration": 0.010446, "end_time": "2026-10-02T14:42:47.463631+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.453185+00:00", "status": "completed" }, "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "NL: NLBGSSolver 'NL: NLBGS' on system '' failed to converge in 1 iterations.\n", "1\n" ] } ], "source": [ "import openmdao.api as om\n", "from openmdao.test_suite.components.sellar_feature import SellarDerivatives\n", "\n", "prob = om.Problem(model=SellarDerivatives())\n", "prob.setup()\n", "\n", "nlbgs = prob.model.nonlinear_solver = om.NonlinearBlockGS()\n", "\n", "# basic test of number of iterations\n", "nlbgs.options['maxiter'] = 1\n", "prob.run_model()\n", "\n", "print(prob.model.nonlinear_solver._iter_count)" ] }, { "cell_type": "code", "execution_count": 9, "id": "b46cee57", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.469293Z", "iopub.status.busy": "2026-10-02T14:42:47.469122Z", "iopub.status.idle": "2026-10-02T14:42:47.471369Z", "shell.execute_reply": "2026-10-02T14:42:47.470939Z" }, "papermill": { "duration": 0.005615, "end_time": "2026-10-02T14:42:47.471785+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.466170+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [], "source": [ "assert(prob.model.nonlinear_solver._iter_count == 1)" ] }, { "cell_type": "code", "execution_count": 10, "id": "ab9bf419", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.477215Z", "iopub.status.busy": "2026-10-02T14:42:47.477046Z", "iopub.status.idle": "2026-10-02T14:42:47.481616Z", "shell.execute_reply": "2026-10-02T14:42:47.480972Z" }, "papermill": { "duration": 0.007931, "end_time": "2026-10-02T14:42:47.482064+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.474133+00:00", "status": "completed" }, "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "NL: NLBGSSolver 'NL: NLBGS' on system '' failed to converge in 5 iterations.\n", "5\n" ] } ], "source": [ "nlbgs.options['maxiter'] = 5\n", "prob.run_model()\n", "print(prob.model.nonlinear_solver._iter_count)" ] }, { "cell_type": "code", "execution_count": 11, "id": "b2b2ca6b", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.486219Z", "iopub.status.busy": "2026-10-02T14:42:47.486101Z", "iopub.status.idle": "2026-10-02T14:42:47.488140Z", "shell.execute_reply": "2026-10-02T14:42:47.487612Z" }, "papermill": { "duration": 0.004583, "end_time": "2026-10-02T14:42:47.488509+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.483926+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [], "source": [ "assert(prob.model.nonlinear_solver._iter_count == 5)" ] }, { "cell_type": "code", "execution_count": 12, "id": "a6519ce8", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.492629Z", "iopub.status.busy": "2026-10-02T14:42:47.492501Z", "iopub.status.idle": "2026-10-02T14:42:47.496810Z", "shell.execute_reply": "2026-10-02T14:42:47.496097Z" }, "papermill": { "duration": 0.007005, "end_time": "2026-10-02T14:42:47.497209+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.490204+00:00", "status": "completed" }, "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "NL: NLBGS 1 ; 1.15925306e-08 1\n", "NL: NLBGS 2 ; 2.29166181e-10 0.0197684344\n", "NL: NLBGS 3 ; 4.53131952e-12 0.000390882687\n", "NL: NLBGS Converged\n", "[25.58830237]\n", "[12.05848815]\n", "3\n" ] } ], "source": [ "# test of number of iterations AND solution after exit at maxiter\n", "prob.set_val('x', 1.)\n", "prob.set_val('z', np.array([5.0, 2.0]))\n", "\n", "nlbgs.options['maxiter'] = 3\n", "prob.set_solver_print()\n", "prob.run_model()\n", "\n", "print(prob.get_val('y1'))\n", "print(prob.get_val('y2'))\n", "print(prob.model.nonlinear_solver._iter_count)" ] }, { "cell_type": "code", "execution_count": 13, "id": "8dc842b5", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.501307Z", "iopub.status.busy": "2026-10-02T14:42:47.501182Z", "iopub.status.idle": "2026-10-02T14:42:47.503580Z", "shell.execute_reply": "2026-10-02T14:42:47.503002Z" }, "papermill": { "duration": 0.005064, "end_time": "2026-10-02T14:42:47.504020+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.498956+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [], "source": [ "assert_near_equal(prob.get_val('y1'), 25.58914915, .0001)\n", "assert_near_equal(prob.get_val('y2'), 12.05857185, .0001)\n", "assert(prob.model.nonlinear_solver._iter_count == 3)" ] }, { "cell_type": "markdown", "id": "b9cbfeda", "metadata": { "papermill": { "duration": 0.001696, "end_time": "2026-10-02T14:42:47.507481+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.505785+00:00", "status": "completed" }, "tags": [] }, "source": [ "**atol**\n", "\n", " Here, we set the absolute tolerance to a looser value that will trigger an earlier termination. After\n", " each iteration, the norm of the residuals is calculated one of two ways. If the \"use_apply_nonlinear\" option\n", " is set to False (its default), then the norm is calculated by subtracting a cached previous value of the\n", " outputs from the current value. If \"use_apply_nonlinear\" is True, then the norm is calculated by calling\n", " apply_nonlinear on all of the subsystems. In this case, `ExplicitComponents` are executed a second time.\n", " If this norm value is lower than the absolute tolerance `atol`, the iteration will terminate." ] }, { "cell_type": "code", "execution_count": 14, "id": "5cd9d7ac", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.511503Z", "iopub.status.busy": "2026-10-02T14:42:47.511380Z", "iopub.status.idle": "2026-10-02T14:42:47.519276Z", "shell.execute_reply": "2026-10-02T14:42:47.518725Z" }, "papermill": { "duration": 0.01047, "end_time": "2026-10-02T14:42:47.519674+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.509204+00:00", "status": "completed" }, "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "NL: NLBGS Converged in 5 iterations\n", "[25.5883027]\n", "[12.05848818]\n" ] } ], "source": [ "import numpy as np\n", "\n", "import openmdao.api as om\n", "from openmdao.test_suite.components.sellar_feature import SellarDerivatives\n", "\n", "prob = om.Problem(model=SellarDerivatives())\n", "prob.setup()\n", "\n", "nlbgs = prob.model.nonlinear_solver = om.NonlinearBlockGS()\n", "nlbgs.options['atol'] = 1e-4\n", "\n", "prob.set_val('x', 1.)\n", "prob.set_val('z', np.array([5.0, 2.0]))\n", "\n", "prob.run_model()\n", "\n", "print(prob.get_val('y1'))\n", "print(prob.get_val('y2'))" ] }, { "cell_type": "code", "execution_count": 15, "id": "abf3740e", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.570408Z", "iopub.status.busy": "2026-10-02T14:42:47.570246Z", "iopub.status.idle": "2026-10-02T14:42:47.574113Z", "shell.execute_reply": "2026-10-02T14:42:47.573477Z" }, "papermill": { "duration": 0.006988, "end_time": "2026-10-02T14:42:47.574522+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.567534+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [ { "data": { "text/plain": [ "np.float64(5.538631102517653e-10)" ] }, "execution_count": 15, "metadata": {}, "output_type": "execute_result" } ], "source": [ "assert_near_equal(prob.get_val('y1'), 25.5882856302, .00001)\n", "assert_near_equal(prob.get_val('y2'), 12.05848819, .00001)" ] }, { "cell_type": "markdown", "id": "c046f1bd", "metadata": { "papermill": { "duration": 0.001717, "end_time": "2026-10-02T14:42:47.578049+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.576332+00:00", "status": "completed" }, "tags": [] }, "source": [ "**rtol**\n", "\n", " Here, we set the relative tolerance to a looser value that will trigger an earlier termination. After\n", " each iteration, the norm of the residuals is calculated one of two ways. If the \"use_apply_nonlinear\" option\n", " is set to False (its default), then the norm is calculated by subtracting a cached previous value of the\n", " outputs from the current value. If \"use_apply_nonlinear\" is True, then the norm is calculated by calling\n", " apply_nonlinear on all of the subsystems. In this case, `ExplicitComponents` are executed a second time.\n", " If the ratio of the currently calculated norm to the initial residual norm is lower than the relative tolerance\n", " `rtol`, the iteration will terminate." ] }, { "cell_type": "code", "execution_count": 16, "id": "88baae06", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.582182Z", "iopub.status.busy": "2026-10-02T14:42:47.582050Z", "iopub.status.idle": "2026-10-02T14:42:47.589234Z", "shell.execute_reply": "2026-10-02T14:42:47.588655Z" }, "papermill": { "duration": 0.009854, "end_time": "2026-10-02T14:42:47.589651+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.579797+00:00", "status": "completed" }, "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "NL: NLBGS Converged in 4 iterations\n", "[25.58828563] 25.5883027 1e-05\n", "[12.0584865] 12.05848819 1e-05\n" ] } ], "source": [ "import numpy as np\n", "\n", "import openmdao.api as om\n", "from openmdao.test_suite.components.sellar_feature import SellarDerivatives\n", "\n", "prob = om.Problem(model=SellarDerivatives())\n", "prob.setup()\n", "\n", "nlbgs = prob.model.nonlinear_solver = om.NonlinearBlockGS()\n", "nlbgs.options['rtol'] = 1e-3\n", "\n", "prob.set_val('x', 1.)\n", "prob.set_val('z', np.array([5.0, 2.0]))\n", "\n", "prob.run_model()\n", "\n", "print(prob.get_val('y1'), 25.5883027, .00001)\n", "print(prob.get_val('y2'), 12.05848819, .00001)" ] }, { "cell_type": "code", "execution_count": 17, "id": "57247c94", "metadata": { "execution": { "iopub.execute_input": "2026-10-02T14:42:47.593931Z", "iopub.status.busy": "2026-10-02T14:42:47.593805Z", "iopub.status.idle": "2026-10-02T14:42:47.596494Z", "shell.execute_reply": "2026-10-02T14:42:47.595980Z" }, "papermill": { "duration": 0.005376, "end_time": "2026-10-02T14:42:47.596871+00:00", "exception": false, "start_time": "2026-10-02T14:42:47.591495+00:00", "status": "completed" }, "tags": [ "remove-input", "remove-output" ] }, "outputs": [ { "data": { "text/plain": [ "np.float64(1.404822467603069e-07)" ] }, "execution_count": 17, "metadata": {}, "output_type": "execute_result" } ], "source": [ "assert_near_equal(prob.get_val('y1'), 25.5883027, .00001)\n", "assert_near_equal(prob.get_val('y2'), 12.05848819, .00001)" ] } ], "metadata": { "celltoolbar": "Tags", "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.13.14" }, "orphan": true, "papermill": { "default_parameters": {}, "duration": 4.97743, "end_time": "2026-10-02T14:42:48.514918+00:00", "environment_variables": {}, "exception": null, "input_path": "/home/runner/work/OpenMDAO/OpenMDAO/openmdao/docs/openmdao_book/features/building_blocks/solvers/nonlinear_block_gs.ipynb", "output_path": "/home/runner/work/OpenMDAO/OpenMDAO/openmdao/docs/_executed_book/features/building_blocks/solvers/nonlinear_block_gs.ipynb", "parameters": {}, "start_time": "2026-10-02T14:42:43.537488+00:00", "version": "2.7.0" }, "vscode": { "interpreter": { "hash": "f5b6148a529182228d9b1fece0411b5abe23db7750f72a75d2d2581d5362f10a" } } }, "nbformat": 4, "nbformat_minor": 5 }