ML::SimpleMatOwning< T > Struct Template Reference

#include <dense.hpp>

Inheritance diagram for ML::SimpleMatOwning< T >:

Collaboration diagram for ML::SimpleMatOwning< T >:

Public Types
typedef SimpleDenseMat< T >	Super
typedef rmm::device_uvector< T >	Buffer
Public Types inherited from ML::SimpleDenseMat< T >
typedef SimpleMat< T >	Super

Public Member Functions
	SimpleMatOwning ()=delete
	SimpleMatOwning (int m, int n, cudaStream_t stream, STORAGE_ORDER order=COL_MAJOR)
void	operator= (const SimpleVec< T > &other)=delete
Public Member Functions inherited from ML::SimpleDenseMat< T >
	SimpleDenseMat (STORAGE_ORDER order=COL_MAJOR)
	SimpleDenseMat (T *data, int m, int n, STORAGE_ORDER order=COL_MAJOR)
void	reset (T *data_, int m_, int n_)
void	gemmb (const raft::handle_t &handle, const T alpha, const SimpleDenseMat< T > &A, const bool transA, const bool transB, const T beta, SimpleDenseMat< T > &C, cudaStream_t stream) const override
void	assign_gemm (const raft::handle_t &handle, const T alpha, const SimpleDenseMat< T > &A, const bool transA, const SimpleMat< T > &B, const bool transB, const T beta, cudaStream_t stream)
void	ax (const T a, const SimpleDenseMat< T > &x, cudaStream_t stream)
void	axpy (const T a, const SimpleDenseMat< T > &x, const SimpleDenseMat< T > &y, cudaStream_t stream)
template<typename Lambda >
void	assign_unary (const SimpleDenseMat< T > &other, Lambda f, cudaStream_t stream)
template<typename Lambda >
void	assign_binary (const SimpleDenseMat< T > &other1, const SimpleDenseMat< T > &other2, Lambda &f, cudaStream_t stream)
template<typename Lambda >
void	assign_ternary (const SimpleDenseMat< T > &other1, const SimpleDenseMat< T > &other2, const SimpleDenseMat< T > &other3, Lambda &f, cudaStream_t stream)
void	fill (const T val, cudaStream_t stream)
void	copy_async (const SimpleDenseMat< T > &other, cudaStream_t stream)
void	print (std::ostream &oss) const override
void	operator= (const SimpleDenseMat< T > &other)=delete
Public Member Functions inherited from ML::SimpleMat< T >
	SimpleMat (int m, int n)
void	operator= (const SimpleMat< T > &other)=delete

Public Attributes
Buffer	buf
int	m
int	n
Public Attributes inherited from ML::SimpleDenseMat< T >
int	len
T *	data
STORAGE_ORDER	ord
Public Attributes inherited from ML::SimpleMat< T >
int	m
int	n

Additional Inherited Members
Static Public Member Functions inherited from ML::SimpleDenseMat< T >
static void	gemm (const raft::handle_t &handle, const T alpha, const SimpleDenseMat< T > &A, const bool transA, const SimpleDenseMat< T > &B, const bool transB, const T beta, SimpleDenseMat< T > &C, cudaStream_t stream)

Member Typedef Documentation

Buffer

template<typename T >

typedef rmm::device_uvector<T> ML::SimpleMatOwning< T >::Buffer

Super

template<typename T >

typedef SimpleDenseMat<T> ML::SimpleMatOwning< T >::Super

Constructor & Destructor Documentation

SimpleMatOwning() [1/2]

template<typename T >

ML::SimpleMatOwning< T >::SimpleMatOwning ( )

delete

SimpleMatOwning() [2/2]

template<typename T >

ML::SimpleMatOwning< T >::SimpleMatOwning ( int  m, int  n, cudaStream_t  stream, STORAGE_ORDER  order = COL_MAJOR  )

inline

Member Function Documentation

operator=()

template<typename T >

void ML::SimpleMatOwning< T >::operator= ( const SimpleVec< T > &  other )

delete

Member Data Documentation

buf

template<typename T >

Buffer ML::SimpleMatOwning< T >::buf

m

template<typename T >

int ML::SimpleMat< T >::m

n

template<typename T >

int ML::SimpleMat< T >::n

---

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

• src/glm/qn/simple_mat/dense.hpp