fml  0.1-0
Fused Matrix Library
fml::parmat_gpu< REAL > Class Template Reference
Inheritance diagram for fml::parmat_gpu< REAL >:
fml::parmat< gpumat< REAL >, gpuvec< REAL >, REAL >

Public Member Functions

 parmat_gpu (comm mpi_comm, card_sp_t gpu_card, const len_global_t nrows, const len_t ncols)
 
 parmat_gpu (comm mpi_comm, card_sp_t gpu_card, const len_global_t nrows, const len_t ncols, const len_global_t nb4_)
 
void print (uint8_t ndigits=4, bool add_final_blank=true)
 
void fill_linspace (const REAL start, const REAL stop)
 
void fill_eye ()
 
void fill_diag (const gpuvec< REAL > &d)
 
card_sp_t get_card () const
 
- Public Member Functions inherited from fml::parmat< gpumat< REAL >, gpuvec< REAL >, REAL >
 parmat (comm mpi_comm, gpumat< REAL > &data_)
 
 parmat (parmat< gpumat< REAL >, gpuvec< REAL >, REAL > &&x)
 
void resize (len_global_t nrows, len_t ncols)
 
void print (uint8_t ndigits=4, bool add_final_blank=true)
 
void info ()
 
void fill_zero ()
 
void fill_val (const REAL v)
 
void fill_linspace (const REAL start, const REAL stop)
 
void fill_eye ()
 
void fill_diag (const gpuvec< REAL > &d)
 
void fill_runif (const uint32_t seed, const REAL min=0, const REAL max=1)
 
void fill_runif (const REAL min=0, const REAL max=1)
 
void fill_rnorm (const uint32_t seed, const REAL mean=0, const REAL sd=1)
 
void fill_rnorm (const REAL mean=0, const REAL sd=1)
 
void scale (const REAL s)
 
void rev_cols ()
 
bool any_inf () const
 
bool any_nan () const
 
REAL get (const len_global_t i) const
 
REAL get (const len_global_t i, const len_t j) const
 
void set (const len_global_t i, const REAL v)
 
void set (const len_global_t i, const len_t j, const REAL v)
 
bool operator== (const parmat< gpumat< REAL >, gpuvec< REAL >, REAL > &x) const
 
bool operator!= (const parmat< gpumat< REAL >, gpuvec< REAL >, REAL > &x) const
 
parmat< gpumat< REAL >, gpuvec< REAL >, REAL > & operator= (const parmat< gpumat< REAL >, gpuvec< REAL >, REAL > &x)
 
len_global_t nrows () const
 
len_local_t nrows_local () const
 
len_local_t ncols () const
 
len_global_t nrows_before () const
 
comm get_comm () const
 
const gpumat< REAL > & data_obj () const
 
gpumat< REAL > & data_obj ()
 

Additional Inherited Members

- Protected Member Functions inherited from fml::parmat< gpumat< REAL >, gpuvec< REAL >, REAL >
void num_preceding_rows ()
 
len_t get_local_dim ()
 
void check_index (const len_global_t i) const
 
void check_index (const len_global_t i, const len_t j) const
 
- Protected Attributes inherited from fml::parmat< gpumat< REAL >, gpuvec< REAL >, REAL >
gpumat< REAL > data
 
len_global_t m_global
 
comm r
 
len_global_t nb4
 

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