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

Allow VexCL to use unblocked matrices

Browse Source
This commit is contained in:
Tong Dong Qiu 2021-07-05 10:42:49 +02:00
parent 47ffeaf24a
commit 5306ae6a60
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
opm/simulators/linalg/bda/amgclSolverBackend.cpp
View File

@ -262,6 +262,28 @@ void amgclSolverBackend<block_size>::solve_system(double *b, WellContributions &
std::tie(iters, error) = solve(B, X);
vex::copy(X, x_ptr);
} else {
vex::Context ctx(vex::Filter::Env && vex::Filter::Count(1));
std::cout << ctx << std::endl;
typedef amgcl::backend::vexcl<double> Backend;
typedef amgcl::make_solver<amgcl::runtime::preconditioner<Backend>, amgcl::runtime::solver::wrapper<Backend> > Solver;
typename Backend::params bprm;
bprm.q = ctx; // set vexcl context
auto A = std::tie(N, A_rows, A_cols, A_vals);
Solver solve(A, prm, bprm);
auto b_ptr = reinterpret_cast<double*>(b);
auto x_ptr = reinterpret_cast<double*>(x.data());
vex::vector<double> B(ctx, N / block_size, b_ptr);
vex::vector<double> X(ctx, N / block_size, x_ptr);
std::tie(iters, error) = solve(B, X);
vex::copy(X, x_ptr);
}
#endif