This class implements a Blocked ILU0 preconditioner The decomposition is done on GPU, using exact decomposition, or ChowPatel decomposition The preconditioner is applied via two exact triangular solves. More...

#include <rocsparseBILU0.hpp>

Inheritance diagram for Opm::Accelerator::rocsparseBILU0< Scalar, block_size >:

Public Member Functions
	rocsparseBILU0 (int verbosity_)

bool	initialize (std::shared_ptr< BlockedMatrix< Scalar >> matrix, std::shared_ptr< BlockedMatrix< Scalar >> jacMatrix, rocsparse_int d_Arows, rocsparse_int d_Acols) override
	Initialize GPU and allocate memory. More...

bool	analyze_matrix ()
	Analysis, extract parallelism if specified.

bool	analyze_matrix (BlockedMatrix< Scalar > *mat) override
	Analysis, extract parallelism if specified. More...

bool	analyze_matrix (BlockedMatrix< Scalar > mat, BlockedMatrix< Scalar > jacMat) override
	Analysis, extract parallelism if specified. More...

bool	create_preconditioner (BlockedMatrix< Scalar > *mat) override
	ILU decomposition. More...

bool	create_preconditioner (BlockedMatrix< Scalar > mat, BlockedMatrix< Scalar > jacMat) override
	ILU decomposition. More...

void	apply (const Scalar &y, Scalar &x, WellContributions< Scalar > &wellContribs) override
	Apply preconditioner, x = prec(y) via Lz = y and Ux = z. More...

void	copy_system_to_gpu (Scalar *mVals) override
	Copy matrix A values to GPU. More...

void	copy_values_to_gpu (Scalar mVals, int mRows, int *mCols, bool reuse)
	Copy matrix A values to GPU. More...

void	update_system_on_gpu (Scalar , Scalar b) override
	Update GPU values after a new assembly is done. More...

Public Member Functions inherited from Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >
void	set_matrix_analysis (rocsparse_mat_descr descr_L, rocsparse_mat_descr descr_U)

void	set_context (rocsparse_handle handle, rocblas_handle blas_handle, rocsparse_direction dir, rocsparse_operation operation, hipStream_t stream)

void	setJacMat (const BlockedMatrix< Scalar > &jacMat)

Additional Inherited Members
Static Public Member Functions inherited from Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >
static std::unique_ptr< rocsparsePreconditioner< Scalar, block_size > >	create (PreconditionerType type, int verbosity)

Static Public Member Functions inherited from Opm::Accelerator::Preconditioner< Scalar, block_size >
static std::unique_ptr< Preconditioner >	create (PreconditionerType type, bool opencl_ilu_parallel, int verbosity)

Public Attributes inherited from Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >
int	nnzbs_prec = 0

bool	useJacMatrix = false

std::shared_ptr< BlockedMatrix< Scalar > >	jacMat {}

Protected Member Functions inherited from Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >
	rocsparsePreconditioner (int verbosity_)

Protected Member Functions inherited from Opm::Accelerator::Preconditioner< Scalar, block_size >
	Preconditioner (int verbosity_)

Protected Attributes inherited from Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >
rocsparse_handle	handle

rocblas_handle	blas_handle

rocsparse_direction	dir = rocsparse_direction_row

rocsparse_operation	operation = rocsparse_operation_none

rocsparse_mat_descr	descr_L

rocsparse_mat_descr	descr_U

hipStream_t	stream

Protected Attributes inherited from Opm::Accelerator::Preconditioner< Scalar, block_size >
int	N

int	Nb

int	nnz

int	nnzb

int	verbosity

Detailed Description

template<class Scalar, unsigned int block_size>
class Opm::Accelerator::rocsparseBILU0< Scalar, block_size >

This class implements a Blocked ILU0 preconditioner The decomposition is done on GPU, using exact decomposition, or ChowPatel decomposition The preconditioner is applied via two exact triangular solves.

Member Function Documentation

◆ analyze_matrix() [1/2]

template<class Scalar , unsigned int block_size>

bool Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::analyze_matrix ( BlockedMatrix< Scalar > * mat )

overridevirtual

Analysis, extract parallelism if specified.

Parameters

[in] mat matrix A

Implements Opm::Accelerator::Preconditioner< Scalar, block_size >.

◆ analyze_matrix() [2/2]

template<class Scalar , unsigned int block_size>

bool Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::analyze_matrix	(	BlockedMatrix< Scalar > *	mat,
		BlockedMatrix< Scalar > *	jacMat
	)

overridevirtual

Analysis, extract parallelism if specified.

Parameters

[in]	mat	matrix A
[in]	jacMat	matrix for preconditioner, analyze this as well

Implements Opm::Accelerator::Preconditioner< Scalar, block_size >.

◆ apply()

template<class Scalar , unsigned int block_size>

void Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::apply	(	const Scalar &	y,
		Scalar &	x,
		WellContributions< Scalar > &	wellContribs
	)

overridevirtual

Apply preconditioner, x = prec(y) via Lz = y and Ux = z.

Parameters

[in]	y	Input y vector
[out]	x	Output x vector
	wellContribs	Well contributions

Implements Opm::Accelerator::Preconditioner< Scalar, block_size >.

◆ copy_system_to_gpu()

template<class Scalar , unsigned int block_size>

void Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::copy_system_to_gpu ( Scalar * mVals )

overridevirtual

Copy matrix A values to GPU.

Parameters

[in] mVals Input values

Implements Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >.

◆ copy_values_to_gpu()

template<class Scalar , unsigned int block_size>

void Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::copy_values_to_gpu	(	Scalar *	mVals,
		int *	mRows,
		int *	mCols,
		bool	reuse
	)

Copy matrix A values to GPU.

Parameters

[in]	mVals	Input values
[in]	mRows	Array of matrix row indices
[in]	mCols	Array of matrix column indices
[in]	reuse	True to reuse old matrix

◆ create_preconditioner() [1/2]

template<class Scalar , unsigned int block_size>

bool Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::create_preconditioner ( BlockedMatrix< Scalar > * mat )

overridevirtual

ILU decomposition.

Parameters

[in] mat matrix A to decompose

Implements Opm::Accelerator::Preconditioner< Scalar, block_size >.

◆ create_preconditioner() [2/2]

template<class Scalar , unsigned int block_size>

bool Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::create_preconditioner	(	BlockedMatrix< Scalar > *	mat,
		BlockedMatrix< Scalar > *	jacMat
	)

overridevirtual

ILU decomposition.

Parameters

[in]	mat	matrix A
[in]	jacMat	matrix for preconditioner, decompose this one if used

Implements Opm::Accelerator::Preconditioner< Scalar, block_size >.

◆ initialize()

template<class Scalar , unsigned int block_size>

bool Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::initialize	(	std::shared_ptr< BlockedMatrix< Scalar >>	matrix,
		std::shared_ptr< BlockedMatrix< Scalar >>	jacMatrix,
		rocsparse_int *	d_Arows,
		rocsparse_int *	d_Acols
	)

overridevirtual

Initialize GPU and allocate memory.

Parameters

[in]	matrix	matrix A
[in]	jacMatrix	matrix for preconditioner
[in]	d_Arows	Array of row indices
[in]	d_Acols	Array of column indices

Implements Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >.

◆ update_system_on_gpu()

template<class Scalar , unsigned int block_size>

void Opm::Accelerator::rocsparseBILU0< Scalar, block_size >::update_system_on_gpu	(	Scalar *	,
		Scalar *	b
	)

overridevirtual

Update GPU values after a new assembly is done.

Parameters

[in] b New b vector

Implements Opm::Accelerator::rocsparsePreconditioner< Scalar, block_size >.

The documentation for this class was generated from the following files:

opm/simulators/linalg/gpubridge/rocm/rocsparseBILU0.hpp
opm/simulators/linalg/gpubridge/rocm/rocsparseBILU0.cpp

Public Member Functions

Additional Inherited Members

Detailed Description

template<class Scalar, unsigned int block_size> class Opm::Accelerator::rocsparseBILU0< Scalar, block_size >

Member Function Documentation

◆ analyze_matrix() [1/2]

◆ analyze_matrix() [2/2]

◆ apply()

◆ copy_system_to_gpu()

◆ copy_values_to_gpu()

◆ create_preconditioner() [1/2]

◆ create_preconditioner() [2/2]

◆ initialize()

◆ update_system_on_gpu()

template<class Scalar, unsigned int block_size>
class Opm::Accelerator::rocsparseBILU0< Scalar, block_size >