OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-01-19 16:32:56 -06:00
Code Issues Packages Projects Releases Wiki Activity

Enable the VexCL backend, disabled for blocksize=1

Browse Source
This commit is contained in:
Tong Dong Qiu 2021-07-02 16:30:52 +02:00
parent a3d8d6655f
commit b4eb711c71
2 changed files with 30 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -108,6 +108,11 @@ void amgclSolverBackend<block_size>::initialize(int N_, int nnz_, int dim, doubl
cusparseCreate(&CUDA_bprm.cusparse_handle);
#else
OPM_THROW(std::logic_error, "Error amgcl is trying to use CUDA, but CUDA was not found by CMake");
#endif
}
if (backend_type == Amgcl_backend_type::vexcl) {
#if !HAVE_VEXCL
OPM_THROW(std::logic_error, "Error amgcl is trying to use VexCL, but VexCL was not found by CMake");
#endif
}
OpmLog::info(out.str());
@ -231,43 +236,38 @@ void amgclSolverBackend<block_size>::solve_system(double *b, WellContributions &
// actually solve
std::tie(iters, error) = solve(B, X);
} else if (backend_type == Amgcl_backend_type::vexcl) {
// vex::Context ctx(vex::Filter::Env && vex::Filter::Count(1));
// std::cout << ctx << std::endl;
#if HAVE_VEXCL
if constexpr(block_size != 1){
vex::Context ctx(vex::Filter::Env && vex::Filter::Count(1));
std::cout << ctx << std::endl;
// // Enable support for block-valued matrices in the VexCL kernels:
// vex::scoped_program_header h1(ctx, amgcl::backend::vexcl_static_matrix_declaration<double, block_size>());
// allow vexcl to use blocked matrices
vex::scoped_program_header h1(ctx, amgcl::backend::vexcl_static_matrix_declaration<double, block_size>());
// // typedef amgcl::static_matrix<double, block_size, block_size> dmat_type; // matrix value type in double precision
// // typedef amgcl::static_matrix<double, block_size, 1> dvec_type; // the corresponding vector value type
// typedef amgcl::backend::vexcl<dmat_type> Backend;
typedef amgcl::backend::vexcl<dmat_type> Backend;
// typedef amgcl::make_block_solver<
// amgcl::relaxation::as_preconditioner<Backend, amgcl::relaxation::ilu0>,
// amgcl::solver::bicgstab<Backend>
// > Solver;
typedef amgcl::make_block_solver<
amgcl::relaxation::as_preconditioner<Backend, amgcl::relaxation::ilu0>,
amgcl::solver::bicgstab<Backend>
> Solver;
// typename Backend::params bprm;
// bprm.q = ctx; // set vexcl context
typename Backend::params bprm;
bprm.q = ctx; // set vexcl context
// typename Solver::params prm;
// prm.precond.damping = 0.9;
// prm.solver.maxiter = maxit;
// prm.solver.tol = tolerance;
// prm.solver.verbose = (verbosity >= 2);
auto A = std::tie(N, A_rows, A_cols, A_vals);
// auto A = std::tie(N, A_rows, A_cols, A_vals);
Solver solve(A, prm, bprm);
// Solver solve(A, prm, bprm);
auto b_ptr = reinterpret_cast<dvec_type*>(b);
auto x_ptr = reinterpret_cast<dvec_type*>(x.data());
vex::vector<dvec_type> B(ctx, N / block_size, b_ptr);
vex::vector<dvec_type> X(ctx, N / block_size, x_ptr);
// auto f_ptr = reinterpret_cast<dvec_type*>(rhs.data());
// auto x_ptr = reinterpret_cast<dvec_type*>(x.data());
// vex::vector<dvec_type> F(ctx, N / block_size, f_ptr);
// vex::vector<dvec_type> X(ctx, N / block_size, x_ptr);
// std::tie(iters, error) = solve(F, X);
// vex::copy(X, x_ptr);
std::tie(iters, error) = solve(B, X);
vex::copy(X, x_ptr);
}
#endif
}
} catch (const std::exception& ex) {
std::cerr << "Caught exception: " << ex.what() << std::endl;

2
opm/simulators/linalg/bda/amgclSolverBackend.hpp
View File

@ -45,7 +45,9 @@
#if HAVE_VEXCL
#include <amgcl/backend/vexcl.hpp>
#include <amgcl/backend/vexcl_static_matrix.hpp>
#endif
namespace bda
{