File: test_solvetransposed3x3.cpp

package info (click to toggle)
opm-simulators 2025.10%2Bds-5
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 21,552 kB
  • sloc: cpp: 193,037; sh: 1,807; python: 1,704; lisp: 1,108; makefile: 31; awk: 10
file content (89 lines) | stat: -rw-r--r-- 3,261 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/*
  Copyright 2021 Equinor ASA

  This file is part of the Open Porous Media project (OPM).

  OPM is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  OPM is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <config.h>

#define BOOST_TEST_MODULE SolveTransposed3x3
#include <boost/test/unit_test.hpp>
#include <boost/version.hpp>

#if BOOST_VERSION / 100000 == 1 && BOOST_VERSION / 100 % 1000 < 71
#include <boost/test/floating_point_comparison.hpp>
#else
#include <boost/test/tools/floating_point_comparison.hpp>
#endif

#include <dune/istl/bcrsmatrix.hh>

#include <opm/simulators/linalg/gpubridge/opencl/openclCPR.hpp>
#include <opm/simulators/linalg/gpubridge/Misc.hpp>

BOOST_AUTO_TEST_CASE(testsolvetransposed3x3)
{
    const unsigned numTests = 3;
    const unsigned blockSize = 3;

    std::vector<std::vector<double> > A = {{4, 2, 1,
                                            3, 4, 2,
                                            2, 4, 3},
                                           {1, 2, 4,
                                            1, 3, 2,
                                            2, 4, 2},
                                           {1, 2, 2,
                                            1, 3, 5,
                                            3, 2, 4}};

    std::vector<std::vector<double> > b = {{0, 1, 0},
                                           {1, 3, 5},
                                           {2, 4, 5}};

    std::vector<std::vector<double> > x_expected = {{-0.5,   1, -0.5},
                                                    {   1,   1, -0.5},
                                                    { 1.3, 0.4,  0.1}};

    for (unsigned testId = 0; testId < numTests; ++testId) {
        std::vector<double> x(blockSize);
        Opm::Accelerator::solve_transposed_3x3(A[testId].data(), b[testId].data(), x.data());

        for (unsigned i = 0; i < blockSize; ++i) {
            BOOST_CHECK_CLOSE(x[i], x_expected[testId][i], 1e-6);
        }
    }

    // retest cases using Dune methods
    using Mat3 = Dune::FieldMatrix<double, 3, 3>;
    using Vec3 = Dune::FieldVector<double, 3>;
    for (unsigned testId = 0; testId < numTests; ++testId) {
        Mat3 a3 = {{ A[testId][0], A[testId][1], A[testId][2] },
                   { A[testId][3], A[testId][4], A[testId][5] },
                   { A[testId][6], A[testId][7], A[testId][8] } };
        Vec3 y({b[testId][0], b[testId][1], b[testId][2]});
        Mat3 b3 = a3;

        // b3 = inv(a3^T)
        Dune::FMatrixHelp::invertMatrix_retTransposed(a3, b3);

        // x = b3 * y
        Vec3 x = Dune::FMatrixHelp::mult(b3, y);

        for (unsigned i = 0; i < blockSize; ++i) {
            BOOST_CHECK_CLOSE(x[i], x_expected[testId][i], 1e-6);
        }
    }
}