Public Member Functions | Public Attributes | List of all members
ML::SimpleMat< T > Struct Template Referenceabstract

#include <base.hpp>

Inheritance diagram for ML::SimpleMat< T >:
Inheritance graph
Collaboration diagram for ML::SimpleMat< T >:
Collaboration graph

Public Member Functions

 SimpleMat (int m, int n)
 
void operator= (const SimpleMat< T > &other)=delete
 
virtual void print (std::ostream &oss) const =0
 
virtual 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 =0
 

Public Attributes

int m
 
int n
 

Constructor & Destructor Documentation

◆ SimpleMat()

template<typename T >
ML::SimpleMat< T >::SimpleMat ( int  m,
int  n 
)
inline

Member Function Documentation

◆ gemmb()

template<typename T >
virtual void ML::SimpleMat< T >::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
pure virtual

GEMM assigning to C where this refers to B.

C <- alpha * A^transA * (*this)^transB + beta * C

Implemented in ML::SimpleSparseMat< T, I >, and ML::SimpleDenseMat< T >.

◆ operator=()

template<typename T >
void ML::SimpleMat< T >::operator= ( const SimpleMat< T > &  other)
delete

◆ print()

template<typename T >
virtual void ML::SimpleMat< T >::print ( std::ostream &  oss) const
pure virtual

Member Data Documentation

◆ 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: