HMLP: High-performance Machine Learning Primitives
hmlp::gpu::Nvidia Class Reference
Inheritance diagram for hmlp::gpu::Nvidia:
hmlp::Device

Public Member Functions

 Nvidia (int device_id)
 
void prefetchd2h (void *ptr_h, void *ptr_d, size_t size, int stream_id)
 
void prefetchh2d (void *ptr_d, void *ptr_h, size_t size, int stream_id)
 
void waitexecute ()
 
void wait (int stream_id)
 
size_t get_memory_left ()
 
void * malloc (size_t size)
 
void malloc (void *ptr_d, size_t size)
 
char * workspace ()
 
void free (void *ptr_d, size_t size)
 
cudaStream_t & getstream (int stream_id)
 
cublasHandle_t & gethandle (int stream_id)
 
- Public Member Functions inherited from hmlp::Device
 Device ()
 Device implementation.
 
virtual class CacheLinegetline (size_t size)
 

Additional Inherited Members

- Public Attributes inherited from hmlp::Device
DeviceType devicetype
 
std::string name
 
class Cache cache
 

Constructor & Destructor Documentation

hmlp::gpu::Nvidia::Nvidia ( int  device_id)
inline

allocate workspace

cache


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