OmniSciDB  1dac507f6e
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Buffer_Namespace::GpuCudaBufferMgr Class Reference

#include <GpuCudaBufferMgr.h>

+ Inheritance diagram for Buffer_Namespace::GpuCudaBufferMgr:
+ Collaboration diagram for Buffer_Namespace::GpuCudaBufferMgr:

Public Member Functions

 GpuCudaBufferMgr (const int device_id, const size_t max_buffer_size, CudaMgr_Namespace::CudaMgr *cuda_mgr, const size_t buffer_alloc_increment=1073741824, const size_t page_size=512, AbstractBufferMgr *parent_mgr=0)
 
MgrType getMgrType () override
 
std::string getStringMgrType () override
 
 ~GpuCudaBufferMgr () override
 
- Public Member Functions inherited from Buffer_Namespace::BufferMgr
 BufferMgr (const int device_id, const size_t max_buffer_size, const size_t max_slab_size=2147483648, const size_t page_size=512, AbstractBufferMgr *parent_mgr=0)
 Constructs a BufferMgr object that allocates memSize bytes. More...
 
 ~BufferMgr () override
 Destructor. More...
 
void reinit ()
 
void clear ()
 
std::string printSlab (size_t slab_num)
 
std::string printSlabs () override
 
void clearSlabs () override
 
std::string printMap ()
 
void printSegs ()
 
std::string printSeg (BufferList::iterator &seg_it)
 
std::string keyToString (const ChunkKey &key)
 
size_t getInUseSize () override
 
size_t getMaxSize () override
 
size_t getAllocated () override
 
size_t getMaxBufferSize ()
 
size_t getMaxSlabSize ()
 
size_t getPageSize ()
 
bool isAllocationCapped () override
 
const std::vector< BufferList > & getSlabSegments ()
 
AbstractBuffercreateBuffer (const ChunkKey &key, const size_t page_size=0, const size_t initial_size=0) override
 Creates a chunk with the specified key and page size. More...
 
void deleteBuffer (const ChunkKey &key, const bool purge=true) override
 Deletes the chunk with the specified key. More...
 
void deleteBuffersWithPrefix (const ChunkKey &key_prefix, const bool purge=true) override
 
AbstractBuffergetBuffer (const ChunkKey &key, const size_t num_bytes=0) override
 Returns the a pointer to the chunk with the specified key. More...
 
bool isBufferOnDevice (const ChunkKey &key) override
 Puts the contents of d into the Buffer with ChunkKey key. More...
 
void fetchBuffer (const ChunkKey &key, AbstractBuffer *dest_buffer, const size_t num_bytes=0) override
 
AbstractBufferputBuffer (const ChunkKey &key, AbstractBuffer *d, const size_t num_bytes=0) override
 
void checkpoint () override
 
void checkpoint (const int db_id, const int tb_id) override
 
AbstractBufferalloc (const size_t num_bytes=0) override
 client is responsible for deleting memory allocated for b->mem_ More...
 
void free (AbstractBuffer *buffer) override
 
size_t size ()
 Returns the total number of bytes allocated. More...
 
size_t getNumChunks () override
 
BufferList::iterator reserveBuffer (BufferList::iterator &seg_it, const size_t num_bytes)
 
void getChunkMetadataVec (std::vector< std::pair< ChunkKey, ChunkMetadata >> &chunk_metadata_vec) override
 
void getChunkMetadataVecForKeyPrefix (std::vector< std::pair< ChunkKey, ChunkMetadata >> &chunk_metadata_vec, const ChunkKey &key_prefix) override
 

Private Member Functions

void addSlab (const size_t slab_size) override
 
void freeAllMem () override
 
void allocateBuffer (BufferList::iterator seg_it, const size_t page_size, const size_t initial_size) override
 

Private Attributes

CudaMgr_Namespace::CudaMgrcuda_mgr_
 

Additional Inherited Members

- Protected Attributes inherited from Buffer_Namespace::BufferMgr
std::vector< int8_t * > slabs_
 
std::vector< BufferListslab_segments_
 
size_t page_size_
 

Detailed Description

Definition at line 27 of file GpuCudaBufferMgr.h.

Constructor & Destructor Documentation

Buffer_Namespace::GpuCudaBufferMgr::GpuCudaBufferMgr ( const int  device_id,
const size_t  max_buffer_size,
CudaMgr_Namespace::CudaMgr cuda_mgr,
const size_t  buffer_alloc_increment = 1073741824,
const size_t  page_size = 512,
AbstractBufferMgr parent_mgr = 0 
)

Definition at line 25 of file GpuCudaBufferMgr.cpp.

31  : BufferMgr(device_id, max_buffer_size, buffer_alloc_increment, page_size, parent_mgr)
32  , cuda_mgr_(cuda_mgr) {}
BufferMgr(const int device_id, const size_t max_buffer_size, const size_t max_slab_size=2147483648, const size_t page_size=512, AbstractBufferMgr *parent_mgr=0)
Constructs a BufferMgr object that allocates memSize bytes.
Definition: BufferMgr.cpp:47
CudaMgr_Namespace::CudaMgr * cuda_mgr_
Buffer_Namespace::GpuCudaBufferMgr::~GpuCudaBufferMgr ( )
override

Definition at line 34 of file GpuCudaBufferMgr.cpp.

References cuda_mgr_, logger::ERROR, freeAllMem(), LOG, and CudaMgr_Namespace::CudaMgr::synchronizeDevices().

34  {
35  try {
37  freeAllMem();
38 #ifdef HAVE_CUDA
39  } catch (const CudaMgr_Namespace::CudaErrorException& e) {
40  if (e.getStatus() == CUDA_ERROR_DEINITIALIZED) {
41  // TODO(adb / asuhan): Verify cuModuleUnload removes the context
42  return;
43  }
44 #endif
45  } catch (const std::runtime_error& e) {
46  LOG(ERROR) << "CUDA Error: " << e.what();
47  }
48 }
#define LOG(tag)
Definition: Logger.h:185
CudaMgr_Namespace::CudaMgr * cuda_mgr_
void synchronizeDevices() const
Definition: CudaMgr.cpp:74

+ Here is the call graph for this function:

Member Function Documentation

void Buffer_Namespace::GpuCudaBufferMgr::addSlab ( const size_t  slab_size)
overrideprivatevirtual

Implements Buffer_Namespace::BufferMgr.

Definition at line 50 of file GpuCudaBufferMgr.cpp.

References CudaMgr_Namespace::CudaMgr::allocateDeviceMem(), cuda_mgr_, Buffer_Namespace::BufferMgr::page_size_, Buffer_Namespace::BufferMgr::slab_segments_, and Buffer_Namespace::BufferMgr::slabs_.

50  {
51  slabs_.resize(slabs_.size() + 1);
52  try {
53  slabs_.back() = cuda_mgr_->allocateDeviceMem(slab_size, device_id_);
54  } catch (std::runtime_error& error) {
55  slabs_.resize(slabs_.size() - 1);
56  throw FailedToCreateSlab(slab_size);
57  }
58  slab_segments_.resize(slab_segments_.size() + 1);
59  slab_segments_[slab_segments_.size() - 1].push_back(
60  BufferSeg(0, slab_size / page_size_));
61 }
std::vector< BufferList > slab_segments_
Definition: BufferMgr.h:177
CudaMgr_Namespace::CudaMgr * cuda_mgr_
std::vector< int8_t * > slabs_
Definition: BufferMgr.h:175
int8_t * allocateDeviceMem(const size_t num_bytes, const int device_num)
Definition: CudaMgr.cpp:217

+ Here is the call graph for this function:

void Buffer_Namespace::GpuCudaBufferMgr::allocateBuffer ( BufferList::iterator  seg_it,
const size_t  page_size,
const size_t  initial_size 
)
overrideprivatevirtual

Implements Buffer_Namespace::BufferMgr.

Definition at line 69 of file GpuCudaBufferMgr.cpp.

References cuda_mgr_.

71  {
72  new GpuCudaBuffer(this,
73  seg_it,
74  device_id_,
75  cuda_mgr_,
76  page_size,
77  initial_size); // this line is admittedly a bit weird
78  // but the segment iterator passed into
79  // buffer takes the address of the new
80  // Buffer in its buffer member
81 }
CudaMgr_Namespace::CudaMgr * cuda_mgr_
void Buffer_Namespace::GpuCudaBufferMgr::freeAllMem ( )
overrideprivatevirtual

Implements Buffer_Namespace::BufferMgr.

Definition at line 63 of file GpuCudaBufferMgr.cpp.

References cuda_mgr_, CudaMgr_Namespace::CudaMgr::freeDeviceMem(), and Buffer_Namespace::BufferMgr::slabs_.

Referenced by ~GpuCudaBufferMgr().

63  {
64  for (auto buf_it = slabs_.begin(); buf_it != slabs_.end(); ++buf_it) {
65  cuda_mgr_->freeDeviceMem(*buf_it);
66  }
67 }
void freeDeviceMem(int8_t *device_ptr)
Definition: CudaMgr.cpp:228
CudaMgr_Namespace::CudaMgr * cuda_mgr_
std::vector< int8_t * > slabs_
Definition: BufferMgr.h:175

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

MgrType Buffer_Namespace::GpuCudaBufferMgr::getMgrType ( )
inlineoverride

Definition at line 35 of file GpuCudaBufferMgr.h.

35 { return GPU_MGR; }
std::string Buffer_Namespace::GpuCudaBufferMgr::getStringMgrType ( )
inlineoverride

Definition at line 36 of file GpuCudaBufferMgr.h.

36 { return ToString(GPU_MGR); }

Member Data Documentation

CudaMgr_Namespace::CudaMgr* Buffer_Namespace::GpuCudaBufferMgr::cuda_mgr_
private

Definition at line 45 of file GpuCudaBufferMgr.h.

Referenced by addSlab(), allocateBuffer(), freeAllMem(), and ~GpuCudaBufferMgr().


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