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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
/* ************************************************************************
* Copyright (C) 2023-2024 Advanced Micro Devices, Inc.
* ************************************************************************ */
#include "testing_csrlsvchol.hpp"
using ::testing::Combine;
using ::testing::TestWithParam;
using ::testing::Values;
using ::testing::ValuesIn;
using namespace std;
typedef std::tuple<int, vector<int>> csrlsvchol_tuple;
// each n_range vector is {n}
// each nnz_range vector is {nnzA, reorder, base1}
// case when n = 20 and nnz = 60 also execute the bad arguments test
// (null handle, null pointers and invalid values)
// for checkin_lapack tests
const vector<int> n_range = {
20,
50,
};
const vector<vector<int>> nnz_range = {
{60, 0, 1},
{60, 1, 0},
{100, 2, 0},
{140, 3, 1},
};
// for daily_lapack tests
const vector<int> large_n_range = {
// normal (valid) samples
100,
250,
};
const vector<vector<int>> large_nnz_range = {
// normal (valid) samples
{300, 0, 0},
{300, 1, 1},
{500, 2, 1},
{700, 3, 0},
};
Arguments csrlsvchol_setup_arguments(csrlsvchol_tuple tup)
{
int n_v = std::get<0>(tup);
vector<int> nnz_v = std::get<1>(tup);
Arguments arg;
arg.set<rocblas_int>("n", n_v);
arg.set<rocblas_int>("nnzA", nnz_v[0]);
arg.set<rocblas_int>("reorder", nnz_v[1]);
arg.set<rocblas_int>("base1", nnz_v[2]);
arg.timing = 0;
return arg;
}
template <bool HOST>
class CSRLSVCHOL_BASE : public ::TestWithParam<csrlsvchol_tuple>
{
protected:
void SetUp() override
{
if(hipsolverSpCreate(nullptr) == HIPSOLVER_STATUS_NOT_SUPPORTED)
GTEST_SKIP() << "Sparse dependencies could not be loaded";
}
void TearDown() override
{
EXPECT_EQ(hipGetLastError(), hipSuccess);
}
template <typename T>
void run_tests()
{
Arguments arg = csrlsvchol_setup_arguments(GetParam());
if(arg.peek<rocblas_int>("n") == 20 && arg.peek<rocblas_int>("nnzA") == 60)
testing_csrlsvchol_bad_arg<HOST, T>();
arg.batch_count = 1;
testing_csrlsvchol<HOST, T>(arg);
}
};
class CSRLSVCHOL : public CSRLSVCHOL_BASE<false>
{
};
class CSRLSVCHOLHOST : public CSRLSVCHOL_BASE<true>
{
};
// non-batch tests
TEST_P(CSRLSVCHOL, __float)
{
run_tests<float>();
}
TEST_P(CSRLSVCHOL, __double)
{
run_tests<double>();
}
// TEST_P(CSRLSVCHOL, __float_complex)
// {
// run_tests<rocblas_float_complex>();
// }
// TEST_P(CSRLSVCHOL, __double_complex)
// {
// run_tests<rocblas_double_complex>();
// }
TEST_P(CSRLSVCHOLHOST, __float)
{
run_tests<float>();
}
TEST_P(CSRLSVCHOLHOST, __double)
{
run_tests<double>();
}
// TEST_P(CSRLSVCHOLHOST, __float_complex)
// {
// run_tests<rocblas_float_complex>();
// }
// TEST_P(CSRLSVCHOLHOST, __double_complex)
// {
// run_tests<rocblas_double_complex>();
// }
INSTANTIATE_TEST_SUITE_P(daily_lapack,
CSRLSVCHOL,
Combine(ValuesIn(large_n_range), ValuesIn(large_nnz_range)));
INSTANTIATE_TEST_SUITE_P(checkin_lapack,
CSRLSVCHOL,
Combine(ValuesIn(n_range), ValuesIn(nnz_range)));
INSTANTIATE_TEST_SUITE_P(daily_lapack,
CSRLSVCHOLHOST,
Combine(ValuesIn(large_n_range), ValuesIn(large_nnz_range)));
INSTANTIATE_TEST_SUITE_P(checkin_lapack,
CSRLSVCHOLHOST,
Combine(ValuesIn(n_range), ValuesIn(nnz_range)));
|
