#include <CSRMatrixSpGEMMDevice.h>

Classes
struct	SpGEMMCommInfo

Public Types
using	allocator_type = typename Config::allocator_type

using	config_type = Config

using	gidx_t = typename Config::gidx_t

using	lidx_t = typename Config::lidx_t

using	localmatrixdata_t = typename matrixdata_t::localmatrixdata_t

using	matrixdata_t = CSRMatrixData< Config >

using	scalar_t = typename Config::scalar_t

Public Member Functions
	CSRMatrixSpGEMMDevice ()=default

	CSRMatrixSpGEMMDevice (std::shared_ptr< matrixdata_t > A_, std::shared_ptr< matrixdata_t > B_, std::shared_ptr< matrixdata_t > C_)

void	multiply ()

void	multiply (std::shared_ptr< localmatrixdata_t > A_data, std::shared_ptr< localmatrixdata_t > B_data, std::shared_ptr< localmatrixdata_t > C_data)

	~CSRMatrixSpGEMMDevice ()=default

Protected Member Functions
void	endBRemoteComm ()

void	merge (std::shared_ptr< localmatrixdata_t > inL, std::shared_ptr< localmatrixdata_t > inR, std::shared_ptr< localmatrixdata_t > out)

void	setupBRemoteComm ()

void	startBRemoteComm ()

Protected Attributes
std::shared_ptr< matrixdata_t >	A

std::shared_ptr< localmatrixdata_t >	A_diag

std::shared_ptr< localmatrixdata_t >	A_offd

std::shared_ptr< matrixdata_t >	B

std::shared_ptr< localmatrixdata_t >	B_diag

std::shared_ptr< localmatrixdata_t >	B_offd

std::shared_ptr< localmatrixdata_t >	BR_diag

std::shared_ptr< localmatrixdata_t >	BR_offd

std::shared_ptr< matrixdata_t >	C

std::shared_ptr< localmatrixdata_t >	C_diag

std::shared_ptr< localmatrixdata_t >	C_diag_diag

std::shared_ptr< localmatrixdata_t >	C_diag_offd

std::shared_ptr< localmatrixdata_t >	C_offd

std::shared_ptr< localmatrixdata_t >	C_offd_diag

std::shared_ptr< localmatrixdata_t >	C_offd_offd

AMP_MPI	comm

CSRMatrixCommunicator< Config >	d_csr_comm

std::map< int, SpGEMMCommInfo >	d_dest_info

lidx_t	d_num_rows

std::map< int, std::shared_ptr< localmatrixdata_t > >	d_recv_matrices

std::map< int, std::shared_ptr< localmatrixdata_t > >	d_send_matrices

std::map< int, SpGEMMCommInfo >	d_src_info

Detailed Description

template<typename Config>
class AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >

Definition at line 26 of file CSRMatrixSpGEMMDevice.h.

Member Typedef Documentation

◆ allocator_type

template<typename Config >

using AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::allocator_type = typename Config::allocator_type

Definition at line 29 of file CSRMatrixSpGEMMDevice.h.

◆ config_type

template<typename Config >

using AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::config_type = Config

Definition at line 30 of file CSRMatrixSpGEMMDevice.h.

◆ gidx_t

template<typename Config >

using AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::gidx_t = typename Config::gidx_t

Definition at line 34 of file CSRMatrixSpGEMMDevice.h.

◆ lidx_t

template<typename Config >

using AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::lidx_t = typename Config::lidx_t

Definition at line 33 of file CSRMatrixSpGEMMDevice.h.

◆ localmatrixdata_t

template<typename Config >

using AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::localmatrixdata_t = typename matrixdata_t::localmatrixdata_t

Definition at line 32 of file CSRMatrixSpGEMMDevice.h.

◆ matrixdata_t

template<typename Config >

using AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::matrixdata_t = CSRMatrixData<Config>

Definition at line 31 of file CSRMatrixSpGEMMDevice.h.

◆ scalar_t

template<typename Config >

using AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::scalar_t = typename Config::scalar_t

Definition at line 35 of file CSRMatrixSpGEMMDevice.h.

Constructor & Destructor Documentation

◆ CSRMatrixSpGEMMDevice() [1/2]

template<typename Config >

AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::CSRMatrixSpGEMMDevice ( )

default

◆ CSRMatrixSpGEMMDevice() [2/2]

template<typename Config >

AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::CSRMatrixSpGEMMDevice	(	std::shared_ptr< matrixdata_t >	A_,
		std::shared_ptr< matrixdata_t >	B_,
		std::shared_ptr< matrixdata_t >	C_
	)

inline

Definition at line 40 of file CSRMatrixSpGEMMDevice.h.

References AMP_DEBUG_INSIST, AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::B, AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::C, and AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::comm.

◆ ~CSRMatrixSpGEMMDevice()

template<typename Config >

AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::~CSRMatrixSpGEMMDevice ( )

default

Member Function Documentation

◆ endBRemoteComm()

template<typename Config >

void AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::endBRemoteComm ( )

protected

◆ merge()

template<typename Config >

void AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::merge	(	std::shared_ptr< localmatrixdata_t >	inL,
		std::shared_ptr< localmatrixdata_t >	inR,
		std::shared_ptr< localmatrixdata_t >	out
	)

protected

◆ multiply() [1/2]

template<typename Config >

void AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::multiply ( )

◆ multiply() [2/2]

template<typename Config >

void AMP::LinearAlgebra::CSRMatrixSpGEMMDevice< Config >::multiply	(	std::shared_ptr< localmatrixdata_t >	A_data,
		std::shared_ptr< localmatrixdata_t >	B_data,
		std::shared_ptr< localmatrixdata_t >	C_data
	)