gpusparse_matrix_operations.hpp
Go to the documentation of this file.
1/*
2 Copyright 2022-2023 SINTEF AS
3
4 This file is part of the Open Porous Media project (OPM).
5
6 OPM is free software: you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation, either version 3 of the License, or
9 (at your option) any later version.
10
11 OPM is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with OPM. If not, see <http://www.gnu.org/licenses/>.
18*/
19#ifndef OPM_GPUISTL_GPUSPARSE_MATRIX_OPERATIONS_HPP
20#define OPM_GPUISTL_GPUSPARSE_MATRIX_OPERATIONS_HPP
21
22#include <cstddef>
23#include <opm/simulators/linalg/gpuistl/GpuBuffer.hpp>
24#include <opm/simulators/linalg/gpuistl/GpuSparseMatrixWrapper.hpp>
25#include <opm/simulators/linalg/gpuistl/detail/gpuThreadUtils.hpp>
26#include <cuda.h>
27#include <cuda_runtime.h>
28namespace Opm::gpuistl::detail
29{
30
42template <class T, int blocksize>
43void copyMatDataToReordered(const T* srcMatrix,
44 const int* srcRowIndices,
45 T* dstMatrix,
46 int* dstRowIndices,
47 int* naturalToReordered,
48 size_t numberOfRows,
49 int threadBlockSize);
50
67template <class T, int blocksize>
68void copyMatDataToReorderedSplit(const T* srcMatrix,
69 const int* srcRowIndices,
70 const int* srcColumnIndices,
71 T* dstLowerMatrix,
72 int* dstLowerRowIndices,
73 T* dstUpperMatrix,
74 int* dstUpperRowIndices,
75 T* dstDiag,
76 int* naturalToReordered,
77 size_t numberOfRows,
78 int threadBlockSize);
79
85template <class T>
86GpuBuffer<T*> getDiagPtrs(GpuSparseMatrixWrapper<T>& matrix);
87
88} // namespace Opm::gpuistl::detail
89
90#endif
Opm::gpuistl::GpuSparseMatrixWrapper
The GpuSparseMatrixWrapper Checks CUDA/HIP version and dispatches a version either using the old or t...
Definition: GpuSparseMatrixWrapper.hpp:61
Opm::gpuistl::detail
Definition: autotuner.hpp:30
Opm::gpuistl::detail::copyMatDataToReorderedSplit
void copyMatDataToReorderedSplit(const T *srcMatrix, const int *srcRowIndices, const int *srcColumnIndices, T *dstLowerMatrix, int *dstLowerRowIndices, T *dstUpperMatrix, int *dstUpperRowIndices, T *dstDiag, int *naturalToReordered, size_t numberOfRows, int threadBlockSize)
Reorders the elements of a matrix by copying them from one matrix to a split matrix using a permutati...
Opm::gpuistl::detail::copyMatDataToReordered
void copyMatDataToReordered(const T *srcMatrix, const int *srcRowIndices, T *dstMatrix, int *dstRowIndices, int *naturalToReordered, size_t numberOfRows, int threadBlockSize)
Reorders the elements of a matrix by copying them from one matrix to another using a permutation list...
Opm::gpuistl::detail::getDiagPtrs
GpuBuffer< T * > getDiagPtrs(GpuSparseMatrixWrapper< T > &matrix)
Return the pointers to diagonal elements in a GpuBuffer.