matrix_batch.hpp

#pragma once
 
#include <Kokkos_Core.hpp>
 
#include "sll/view.hpp"
 
 
template <typename ExecSpace>
class MatrixBatch
{
public:
    using BatchedRHS = Kokkos::View<double**, Kokkos::LayoutRight, ExecSpace>;
 
 
private:
    std::size_t m_size;
    std::size_t m_batch_size;
 
protected:
    explicit MatrixBatch(const std::size_t batch_size, const std::size_t mat_size)
        : m_size(mat_size)
        , m_batch_size(batch_size)
    {
    }
 
public:
    virtual ~MatrixBatch() = default;
 
    virtual void setup_solver() = 0;
 
    virtual void solve(BatchedRHS b) const = 0;
 
    std::size_t size() const
    {
        return m_size;
    }
 
    std::size_t batch_size() const
    {
        return m_batch_size;
    }
};