dss_solve_klu.hpp

Go to the documentation of this file.

#pragma once
 
#include <Kokkos_Core.hpp>
#include <klu.h>
 
#include "dss_algorithm.hpp"
#include "dss_handle_klu.hpp"
 
namespace kynema::dss {
template <typename CrsMatrixType, typename MultiVectorType>
struct SolveFunction<Handle<Algorithm::KLU>, CrsMatrixType, MultiVectorType> {
    static void solve(
        Handle<Algorithm::KLU>& dss_handle, CrsMatrixType& A, MultiVectorType& b, MultiVectorType& x
    ) {
        const auto num_rows = A.numRows();
 
        auto*& symbolic = dss_handle.get_symbolic();
        auto*& numeric = dss_handle.get_numeric();
        auto& common = dss_handle.get_common();
 
        Kokkos::deep_copy(x, b);
        auto x_host = Kokkos::create_mirror_view_and_copy(Kokkos::HostSpace(), x);
 
        klu_tsolve(symbolic, numeric, num_rows, 1, x_host.data(), &common);
        Kokkos::deep_copy(x, x_host);
    }
};
 
}  // namespace kynema::dss