OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Autodetect if transposing is needed

Browse Source
This commit is contained in:
Tong Dong Qiu 2022-11-03 14:13:56 +01:00
parent 5dea01b9dc
commit f91154c165
1 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
opm/simulators/linalg/bda/rocalutionSolverBackend.cpp
View File

@ -28,6 +28,7 @@
#include <opm/simulators/linalg/bda/rocalutionSolverBackend.hpp> #include <opm/simulators/linalg/bda/rocalutionSolverBackend.hpp>
#include <rocalution.hpp> #include <rocalution.hpp>
#include <base/matrix_formats_ind.hpp> // check if blocks are interpreted as row-major or column-major
namespace Opm namespace Opm
{ {
@ -88,16 +89,20 @@ void rocalutionSolverBackend<block_size>::convert_matrix(std::shared_ptr<Blocked
// convert values inside block from row major to col major // convert values inside block from row major to col major
// this is the same as transposing a block // this is the same as transposing a block
// when compiling rocm from scratch, it is possible to choose the direction, making this transposing unnecessary // when compiling rocm from scratch, it is possible to choose the direction, making this transposing unnecessary
for(int i = 0; i < nnzb; ++i){ // BCSR_IND_BASE == 0: rocalution expects column-major
tmp_nnzvalues[i * block_size * block_size + 0] = matrix->nnzValues[i * block_size * block_size + 0]; // BCSR_IND_BASE == 1: rocalution expects row-major
tmp_nnzvalues[i * block_size * block_size + 1] = matrix->nnzValues[i * block_size * block_size + 3]; if (BCSR_IND_BASE == 0) {
tmp_nnzvalues[i * block_size * block_size + 2] = matrix->nnzValues[i * block_size * block_size + 6]; for(int i = 0; i < nnzb; ++i){
tmp_nnzvalues[i * block_size * block_size + 3] = matrix->nnzValues[i * block_size * block_size + 1]; tmp_nnzvalues[i * block_size * block_size + 0] = matrix->nnzValues[i * block_size * block_size + 0];
tmp_nnzvalues[i * block_size * block_size + 4] = matrix->nnzValues[i * block_size * block_size + 4]; tmp_nnzvalues[i * block_size * block_size + 1] = matrix->nnzValues[i * block_size * block_size + 3];
tmp_nnzvalues[i * block_size * block_size + 5] = matrix->nnzValues[i * block_size * block_size + 7]; tmp_nnzvalues[i * block_size * block_size + 2] = matrix->nnzValues[i * block_size * block_size + 6];
tmp_nnzvalues[i * block_size * block_size + 6] = matrix->nnzValues[i * block_size * block_size + 2]; tmp_nnzvalues[i * block_size * block_size + 3] = matrix->nnzValues[i * block_size * block_size + 1];
tmp_nnzvalues[i * block_size * block_size + 7] = matrix->nnzValues[i * block_size * block_size + 5]; tmp_nnzvalues[i * block_size * block_size + 4] = matrix->nnzValues[i * block_size * block_size + 4];
tmp_nnzvalues[i * block_size * block_size + 8] = matrix->nnzValues[i * block_size * block_size + 8]; tmp_nnzvalues[i * block_size * block_size + 5] = matrix->nnzValues[i * block_size * block_size + 7];
tmp_nnzvalues[i * block_size * block_size + 6] = matrix->nnzValues[i * block_size * block_size + 2];
tmp_nnzvalues[i * block_size * block_size + 7] = matrix->nnzValues[i * block_size * block_size + 5];
tmp_nnzvalues[i * block_size * block_size + 8] = matrix->nnzValues[i * block_size * block_size + 8];
}
} }
if (verbosity >= 3) { if (verbosity >= 3) {
std::ostringstream out; std::ostringstream out;