-
Notifications
You must be signed in to change notification settings - Fork 12
/
Copy pathCombination.C
126 lines (93 loc) · 3.62 KB
/
Combination.C
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// SPDX-FileCopyrightText: 2024 OGL authors
//
// SPDX-License-Identifier: GPL-3.0-or-later
#include <gtest/gtest.h>
#include "OGL/MatrixWrapper/Combination.H"
#include "fvCFD.H"
#include <ginkgo/ginkgo.hpp>
#include <fstream>
#include <string>
TEST(Combination, CanCreateEmptyCombination)
{
auto exec = gko::share(gko::ReferenceExecutor::create());
auto cmb = CombinationMatrix<gko::matrix::Coo<scalar, label>>::create(exec);
ASSERT_EQ(cmb->get_size(), gko::dim<2>(0, 0));
ASSERT_EQ(cmb->get_coefficients().size(), 0);
ASSERT_EQ(cmb->get_operators().size(), 0);
}
TEST(Combination, CanCreateCombinationWithLinOpVector)
{
auto dim = gko::dim<2>{5, 5};
gko::matrix_data<double, int> m1(dim, {{0, 0, 2}, {1, 1, 0}, {2, 3, 5}});
auto exec = gko::share(gko::ReferenceExecutor::create());
auto m1linop =
gko::share(gko::matrix::Csr<scalar, label>::create(exec, dim));
m1linop->read(m1);
std::vector<std::shared_ptr<const gko::LinOp>> linops{m1linop, m1linop};
auto cmb = CombinationMatrix<gko::matrix::Csr<scalar, label>>::create(
exec, dim, linops);
ASSERT_EQ(cmb->get_size(), gko::dim<2>(5, 5));
ASSERT_EQ(cmb->get_coefficients().size(), 2);
ASSERT_EQ(cmb->get_operators().size(), 2);
auto b = gko::matrix::Dense<scalar>::create(exec, gko::dim<2>{5, 1});
b->fill(1.0);
auto x = gko::matrix::Dense<scalar>::create(exec, gko::dim<2>{5, 1});
x->fill(0.0);
cmb->apply(b, x);
ASSERT_EQ(x->at(0, 0), 4.0);
ASSERT_EQ(x->at(1, 0), 0.0);
ASSERT_EQ(x->at(2, 0), 10.0);
ASSERT_EQ(x->at(3, 0), 0.0);
ASSERT_EQ(x->at(4, 0), 0.0);
}
TEST(Combination, CanCreateCombinationMatrixWithSparseRealMatrix)
{
// Arrange
using ValueType = scalar;
using IndexType = label;
using mtx = gko::matrix::Csr<ValueType, IndexType>;
using vec = gko::matrix::Dense<ValueType>;
int dim_size = 5;
auto dim = gko::dim<2>{dim_size, dim_size};
auto exec = gko::share(gko::ReferenceExecutor::create());
auto m1linop = gko::share(
gko::read<mtx>(std::ifstream("data/A_sparse.mtx"), exec));
auto b = gko::read<vec>(std::ifstream("data/b.mtx"), exec);
std::vector<std::shared_ptr<const gko::LinOp>> linops{m1linop, m1linop};
// Act
auto cmb = CombinationMatrix<gko::matrix::Csr<scalar, label>>::create(
exec, dim, linops);
// Assert
ASSERT_EQ(cmb->get_size(), gko::dim<2>(dim_size, dim_size));
ASSERT_EQ(cmb->get_coefficients().size(), 2);
ASSERT_EQ(cmb->get_operators().size(), 2);
// Act
auto x = gko::matrix::Dense<scalar>::create(exec, gko::dim<2>{dim_size, 1});
x->fill(0.0);
cmb->apply(b, x);
// Assert
ASSERT_EQ(x->at(0, 0), 38.0);
ASSERT_EQ(x->at(1, 0), 0.0);
ASSERT_EQ(x->at(2, 0), 2.0);
ASSERT_EQ(x->at(3, 0), -1414.4);
ASSERT_EQ(x->at(4, 0), 96.0);
}
TEST(Combination, CanConvertToCsr)
{
auto dim = gko::dim<2>{5, 5};
gko::matrix_data<double, int> m1(dim, {{0, 0, 2}, {1, 1, 0}, {2, 3, 5}});
gko::matrix_data<double, int> m2(dim, {{3, 3, 1}, {4, 4, 2}});
auto exec = gko::share(gko::ReferenceExecutor::create());
auto m1linop =
gko::share(gko::matrix::Csr<scalar, label>::create(exec, dim));
m1linop->read(m1);
auto m2linop =
gko::share(gko::matrix::Csr<scalar, label>::create(exec, dim));
m2linop->read(m2);
std::vector<std::shared_ptr<const gko::LinOp>> linops{m1linop, m2linop};
auto cmb = CombinationMatrix<gko::matrix::Csr<scalar, label>>::create(
exec, dim, linops);
auto out = CombinationMatrix<gko::matrix::Csr<scalar, label>>::create(
exec);
cmb->convert_to(out);
}