KokkosAmgPreconditioner.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_ENGINE_2D_GPU_KOKKOS_AMG_PRECONDITIONER_HPP
#define OPENSWMM_ENGINE_2D_GPU_KOKKOS_AMG_PRECONDITIONER_HPP
 
#include "KokkosSurfaceKernels.hpp"
 
namespace openswmm::twoD::gpu {

class KokkosAmgPreconditioner {
public:
    KokkosAmgPreconditioner() = default;
    ~KokkosAmgPreconditioner();
 
    KokkosAmgPreconditioner(const KokkosAmgPreconditioner&)            = delete;
    KokkosAmgPreconditioner& operator=(const KokkosAmgPreconditioner&) = delete;

    void initialize(const MeshViews& mesh);

    void setup(const MeshViews& mesh, const StateViews& state, double gamma,
               bool recompute);

    void solve(DView r, DView z, double gamma);
 
    void finalize();
    bool ready() const noexcept { return amg_ != nullptr; }

    void invalidate() noexcept { hierarchy_built_ = false; }
 
private:
    int n_ = 0;
    bool hierarchy_built_ = false;  
 
    // Static CSR sparsity, resident in the ExecSpace memory space.
    IView  row_ptr_;     
    IView  col_idx_;     
    IView  rows_;        
    IView  ncols_;       
    IView  diag_pos_;    
    IView  edge_pos_;    
    DView  values_;      
 
    // Opaque hypre handles (see HypreAmgPreconditioner.hpp rationale).
    void* A_   = nullptr;   
    void* b_   = nullptr;   
    void* x_   = nullptr;   
    void* amg_ = nullptr;   
};
 
} // namespace openswmm::twoD::gpu
 
#endif // OPENSWMM_ENGINE_2D_GPU_KOKKOS_AMG_PRECONDITIONER_HPP