cpr_amg_operations.hpp

/*
  Copyright 2025 Equinor ASA

  This file is part of the Open Porous Media project (OPM).

  OPM is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  OPM is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef OPM_GPU_CPR_AMG_OPERATIONS_HPP
#define OPM_GPU_CPR_AMG_OPERATIONS_HPP

#include <opm/simulators/linalg/gpuistl/GpuSparseMatrixWrapper.hpp>
#include <opm/simulators/linalg/gpuistl/GpuVector.hpp>

#include <cstddef>

namespace Opm::gpuistl::detail
{

template <typename T, bool transpose>
void getQuasiImpesWeights(const GpuSparseMatrixWrapper<T>& matrix,
                          std::size_t pressureVarIndex,
                          GpuVector<T>& weights,
                          const GpuVector<int>& diagonalIndices);

template <typename T, bool transpose>
void calculateCoarseEntries(const GpuSparseMatrixWrapper<T>& fineMatrix,
                            GpuSparseMatrixWrapper<T>& coarseMatrix,
                            const GpuVector<T>& weights,
                            std::size_t pressureVarIndex);


template <typename T, bool transpose>
void restrictVector(const GpuVector<T>& fine,
                    GpuVector<T>& coarse,
                    const GpuVector<T>& weights,
                    std::size_t pressureVarIndex);

template <typename T, bool transpose>
void prolongateVector(const GpuVector<T>& coarse,
                      GpuVector<T>& fine,
                      const GpuVector<T>& weights,
                      std::size_t pressureVarIndex);

} // namespace Opm::gpuistl::detail

#endif // OPM_GPU_CPR_AMG_OPERATIONS_HPP