dss_numeric_cudss.hpp

Go to the documentation of this file.

#pragma once
 
#include <cudss.h>
 
#include "dss_algorithm.hpp"
#include "dss_handle_cudss.hpp"
 
namespace kynema::dss {
template <typename CrsMatrixType>
struct NumericFunction<Handle<Algorithm::CUDSS>, CrsMatrixType> {
    static void numeric(Handle<Algorithm::CUDSS>& dss_handle, CrsMatrixType& A) {
        const auto num_rows = A.numRows();
        const auto num_cols = A.numCols();
        const auto num_non_zero = A.nnz();
 
        auto* values = A.values.data();
        auto* row_ptrs = A.graph.row_map.data();
        auto* col_inds = A.graph.entries.data();
 
        auto& handle = dss_handle.get_handle();
        auto& config = dss_handle.get_config();
        auto& data = dss_handle.get_data();
 
        cudssMatrix_t A_cudss;
        cudssMatrix_t x_cudss;
        cudssMatrix_t b_cudss;
 
        cudssMatrixCreateCsr(
            &A_cudss, num_rows, num_cols, num_non_zero, const_cast<int*>(row_ptrs), nullptr,
            col_inds, values, CUDA_R_32I, CUDA_R_64F, CUDSS_MTYPE_GENERAL, CUDSS_MVIEW_FULL,
            CUDSS_BASE_ZERO
        );
        cudssMatrixCreateDn(
            &b_cudss, num_cols, 1, num_cols, nullptr, CUDA_R_64F, CUDSS_LAYOUT_COL_MAJOR
        );
        cudssMatrixCreateDn(
            &x_cudss, num_rows, 1, num_rows, nullptr, CUDA_R_64F, CUDSS_LAYOUT_COL_MAJOR
        );
 
        if (dss_handle.is_initial_factorization()) {
            cudssExecute(handle, CUDSS_PHASE_FACTORIZATION, config, data, A_cudss, x_cudss, b_cudss);
            dss_handle.set_initial_factorization(false);
        } else {
            cudssExecute(
                handle, CUDSS_PHASE_REFACTORIZATION, config, data, A_cudss, x_cudss, b_cudss
            );
        }
 
        cudssMatrixDestroy(A_cudss);
        cudssMatrixDestroy(b_cudss);
        cudssMatrixDestroy(x_cudss);
    }
};
}  // namespace kynema::dss