mirror of
https://github.com/OPM/opm-simulators.git
synced 2024-12-02 13:59:12 -06:00
67 lines
2.0 KiB
C++
67 lines
2.0 KiB
C++
#include <config.h>
|
|
#include <iostream>
|
|
|
|
#define BOOST_TEST_MODULE CsrToCscOffsetMap
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <boost/version.hpp>
|
|
|
|
#include <dune/common/version.hh>
|
|
#include <dune/common/fvector.hh>
|
|
#include <dune/istl/bcrsmatrix.hh>
|
|
#include <dune/istl/matrixmarket.hh>
|
|
|
|
#include <opm/simulators/linalg/bda/opencl/BISAI.hpp>
|
|
|
|
BOOST_AUTO_TEST_CASE(testcsrtocscoffsetmap){
|
|
|
|
#if DUNE_VERSION_NEWER(DUNE_ISTL, 2, 7)
|
|
using Matrix = Dune::BCRSMatrix<double>;
|
|
#else
|
|
using Matrix = Dune::BCRSMatrix<Dune::FieldMatrix<double,1,1>>;
|
|
#endif
|
|
|
|
Matrix matrix;
|
|
{
|
|
std::ifstream mfile("offset_map_matrix.txt");
|
|
if (!mfile) {
|
|
throw std::runtime_error("Could not read matrix file");
|
|
}
|
|
readMatrixMarket(matrix, mfile);
|
|
}
|
|
|
|
// a transposed version of the matrix is read because the transposed
|
|
// of a CSR representation is equivalente to CSC
|
|
Matrix matrix_transposed;
|
|
{
|
|
std::ifstream mfile("offset_map_matrix_transposed.txt");
|
|
if (!mfile) {
|
|
throw std::runtime_error("Could not read matrix file");
|
|
}
|
|
readMatrixMarket(matrix_transposed, mfile);
|
|
}
|
|
|
|
// has to make copy because the output of readMatrixMarket does not
|
|
// have contiguous non-zero values
|
|
Matrix matrix_copy(matrix);
|
|
Matrix matrix_transposed_copy(matrix_transposed);
|
|
|
|
std::vector<int> rowPointers, colIndices, map;
|
|
|
|
auto* nnzValues = &matrix_copy[0][0];
|
|
auto* nnzValues_transposed = &matrix_transposed_copy[0][0];
|
|
|
|
rowPointers.emplace_back(0);
|
|
for (Matrix::Iterator r = matrix.begin(); r != matrix.end(); ++r) {
|
|
for (auto c = r->begin(); c != r->end(); ++c) {
|
|
colIndices.emplace_back(c.index());
|
|
}
|
|
rowPointers.emplace_back(colIndices.size());
|
|
}
|
|
|
|
map = Opm::Accelerator::buildCsrToCscOffsetMap(rowPointers, colIndices);
|
|
|
|
for (unsigned int i = 0; i < colIndices.size(); i++){
|
|
BOOST_CHECK_EQUAL(nnzValues[i], nnzValues_transposed[map[i]]);
|
|
}
|
|
}
|