This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| import warnings | |
| class BiCGSTAB(): | |
| """ | |
| This is a pytorch implementation of BiCGSTAB or BCGSTAB, a stable version | |
| of the CGD method, published first by Van Der Vrost. | |
| For solving ``Ax = b`` system. |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # This is a modified version of the original file from the Modulus repository (NVIDIA) | |
| # The original file can be found at: | |
| # https://github.com/NVIDIA/modulus-sym/blob/main/modulus/sym/utils/sympy/torch_printer.py | |
| # The original file is licensed under the Apache License, Version 2.0 | |
| # The modified file is provided under the same license | |
| # The original file disclaimer is provided below: | |
| # SPDX-FileCopyrightText: Copyright (c) 2023 - 2024 NVIDIA CORPORATION & AFFILIATES. | |
| # SPDX-FileCopyrightText: All rights reserved. | |
| # SPDX-License-Identifier: Apache-2.0 |
OlderNewer