Matrix< Device > Class Template Reference

#include <matrix.hpp>

Collaboration diagram for Matrix< Device >:

Public Types
enum	SetValueOpt { Replace =0, Add =1 }
enum	Type { Identity =0 }

Public Member Functions
	Matrix ()
	Matrix (int m_in, int n_in, int w_in)
	Matrix (Type matrix_type, int m_in)
	Matrix (int m_in, int n_in, int w_in, int nnz_in, bool is_csr_in, int *csr_ridx_or_eindex, int *csr_cidx_or_nelement, double *values)
template<class Device2 >
Matrix< Device2 >	mirror () const
void	convert_org_to_csr ()
KOKKOS_INLINE_FUNCTION void	set_value (int i, int j, double value_in, SetValueOpt flag) const
void	mult_org (const View< double *, CLayout, Device > &x, const View< double *, Kokkos::LayoutRight, Device > &y) const
void	mult_csr (const View< double *, CLayout, Device > &x, const View< double *, Kokkos::LayoutRight, Device > &y) const
void	mult_tensor_org (const View< double **, CLayout, Device > &x, const View< double **, Kokkos::LayoutRight, Device > &y) const
void	mult_tensor_csr (const View< double **, CLayout, Device > &x, const View< double **, Kokkos::LayoutRight, Device > &y) const
void	mult (const View< double *, CLayout, Device > &x, const View< double *, Kokkos::LayoutRight, Device > &y) const
void	mult_tensor (const View< double **, CLayout, Device > &x, const View< double **, Kokkos::LayoutRight, Device > &y) const

Public Attributes
int	m
	of rows (size of each column) More...
int	n
	of columns (size of each row) More...
int	width
View< double **, CLayout, Device >	value
	matrix value More...
View< int **, CLayout, Device >	eindex
	column index - 1-indexed!! More...
View< int *, CLayout, Device >	nelement
	of non-zero element of each row More...
bool	is_csr
	Whether the matrix is in CSR format. More...
int	nnz
	if in CSR format, number of (nonzero) values More...
View< int *, CLayout, Device >	csr_ridx
	row index for CSR - m+1 More...
View< double *, CLayout, Device >	csr_v
	value of CSR - nnz More...
View< int *, CLayout, Device >	csr_cidx
	columun index - nnz More...

Private Types
using	exspace = typename Device::execution_space
	Use execution space where matrix views are allocated. More...

Member Typedef Documentation

template<class Device>

using Matrix< Device >::exspace = typename Device::execution_space

private

Use execution space where matrix views are allocated.

Member Enumeration Documentation

template<class Device>

enum Matrix::SetValueOpt

Enumerator
Replace
Add

template<class Device>

enum Matrix::Type

Enumerator
Identity

Constructor & Destructor Documentation

template<class Device>

Matrix< Device >::Matrix ( )

inline

template<class Device>

Matrix< Device >::Matrix ( int  m_in, int  n_in, int  w_in  )

inline

Constructor that allocates the original format

Parameters

[in]	m_in	is the # of rows
[in]	n_in	is the # of columns
[in]	w_in	is the width you think you will need (youll get an error later if its too small)

template<class Device>

Matrix< Device >::Matrix ( Type  matrix_type, int  m_in  )

inline

Constructor of identity matrix

Parameters

[in]

m_in

is the # of rows and columns

template<class Device>

Matrix< Device >::Matrix ( int  m_in, int  n_in, int  w_in, int  nnz_in, bool  is_csr_in, int *  csr_ridx_or_eindex, int *  csr_cidx_or_nelement, double *  values  )

inline

Constructor that deep copies the matrix from pointers (to copy from Fortran mat_type)

Parameters

[in]	m_in	is the # of rows
[in]	n_in	is the # of columns
[in]	w_in	is the width you think you will need (youll get an error later if its too small)
[in]	nnz_in	is the number of nonzero entries
[in]	is_csr_in	is whether the matrix is in CSR format
[in]	csr_ridx_or_eindex	points to csr_ridx if in CSR, otherwise to eindex
[in]	csr_cidx_or_nelement	points to csr_cidx if in CSR, otherwise to nelement
[in]	values	is the pointer to the matrix entries

Member Function Documentation

template<class Device>

void Matrix< Device >::convert_org_to_csr ( )

inline

Converts matrix from original format to CSR.

Returns

void

Here is the caller graph for this function:

template<class Device>

template<class Device2 >

Matrix<Device2> Matrix< Device >::mirror ( ) const

inline

template<class Device>

void Matrix< Device >::mult ( const View< double *, CLayout, Device > &  x, const View< double *, Kokkos::LayoutRight, Device > &  y  ) const

inline

Matrix multiplication of the form: y = Ax. Dispatch for the two matrix formats

Parameters

[in]	x	is the input vector
[out]	y	is the result

Returns

void

Here is the caller graph for this function:

template<class Device>

void Matrix< Device >::mult_csr ( const View< double *, CLayout, Device > &  x, const View< double *, Kokkos::LayoutRight, Device > &  y  ) const

inline

Matrix multiplication of the form: y = Ax. Matrix is in CSR format.

Parameters

[in]	x	is the input vector
[out]	y	is the result

Returns

void

Here is the caller graph for this function:

template<class Device>

void Matrix< Device >::mult_org ( const View< double *, CLayout, Device > &  x, const View< double *, Kokkos::LayoutRight, Device > &  y  ) const

inline

Matrix multiplication of the form: y = Ax. Matrix is in original data format.

Parameters

[in]	x	is the input vector
[out]	y	is the result

Returns

void

Here is the caller graph for this function:

template<class Device>

void Matrix< Device >::mult_tensor ( const View< double **, CLayout, Device > &  x, const View< double **, Kokkos::LayoutRight, Device > &  y  ) const

inline

Matrix multiplication using the transpose, i.e.: y = A^T x. Dispatch for the two matrix formats

Parameters

[in]	x	is the input vector
[out]	y	is the result

Returns

void Matrix multiplication of tensor of the form: y = Ax. Dispatch for the two matrix formats

Parameters

[in]	x	is the input tensor
[out]	y	is the result

Returns

void

template<class Device>

void Matrix< Device >::mult_tensor_csr ( const View< double **, CLayout, Device > &  x, const View< double **, Kokkos::LayoutRight, Device > &  y  ) const

inline

Matrix multiplication of tensor of the form: y = Ax. Matrix is in CSR format.

Parameters

[in]	x	is the input tensor
[in]	y	is the result

Returns

void

Here is the caller graph for this function:

template<class Device>

void Matrix< Device >::mult_tensor_org ( const View< double **, CLayout, Device > &  x, const View< double **, Kokkos::LayoutRight, Device > &  y  ) const

inline

Matrix multiplication of tensor of the form: y = Ax. Matrix is in original data format.

Parameters

[in]	x	is the input tensor
[in]	y	is the result

Returns

void

Here is the caller graph for this function:

template<class Device>

KOKKOS_INLINE_FUNCTION void Matrix< Device >::set_value ( int  i, int  j, double  value_in, SetValueOpt  flag  ) const

inline

Set value of matrix component. Inlined since this will presumably be called a lot in a loop

Parameters

[in]	i	is the row index
[in]	j	is the column index
[in]	value_in	is the value to set
[in]	flag	is whether to add or replace if there is an existing value in the chosen matrix element

Returns

void

Here is the caller graph for this function:

Member Data Documentation

template<class Device>

View<int*,CLayout, Device> Matrix< Device >::csr_cidx

columun index - nnz

template<class Device>

View<int*,CLayout, Device> Matrix< Device >::csr_ridx

row index for CSR - m+1

template<class Device>

View<double*,CLayout, Device> Matrix< Device >::csr_v

value of CSR - nnz

template<class Device>

View<int**,CLayout, Device> Matrix< Device >::eindex

column index - 1-indexed!!

template<class Device>

bool Matrix< Device >::is_csr

Whether the matrix is in CSR format.

template<class Device>

int Matrix< Device >::m

of rows (size of each column)

template<class Device>

int Matrix< Device >::n

of columns (size of each row)

template<class Device>

View<int*,CLayout, Device> Matrix< Device >::nelement

of non-zero element of each row

template<class Device>

int Matrix< Device >::nnz

if in CSR format, number of (nonzero) values

template<class Device>

View<double**,CLayout, Device> Matrix< Device >::value

matrix value

template<class Device>

int Matrix< Device >::width

---

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

• /u/gitlab-xgc/builds/YGMz2TJ8/0/xgc/XGC-Devel/XGC_core/cpp/matrix.hpp