#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_pool_size, CudaMgr_Namespace::CudaMgr cuda_mgr, const size_t min_slab_size, const size_t max_slab_size, const size_t page_size, 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 min_slab_size, const size_t max_slab_size, const size_t page_size, AbstractBufferMgr *parent_mgr=0)
	Constructs a BufferMgr object that allocates memSize bytes. More...

	~BufferMgr () override
	Destructor. More...

std::string	printSlab (size_t slab_num)

std::string	printSlabs () override

void	clearSlabs ()

std::string	printMap ()

void	printSegs ()

std::string	printSeg (BufferList::iterator &seg_it)

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 ()

AbstractBuffer *	createBuffer (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

AbstractBuffer *	getBuffer (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

AbstractBuffer *	putBuffer (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

void	removeTableRelatedDS (const int db_id, const int table_id) override

AbstractBuffer *	alloc (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	getChunkMetadataVecForKeyPrefix (ChunkMetadataVector &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::CudaMgr *	cuda_mgr_

Additional Inherited Members
Protected Attributes inherited from Buffer_Namespace::BufferMgr
const size_t	max_buffer_pool_size_

const size_t	min_slab_size_
	max number of bytes allocated for the buffer pool More...

const size_t	max_slab_size_

const size_t	page_size_

std::vector< int8_t * >	slabs_

std::vector< BufferList >	slab_segments_

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_pool_size,
		CudaMgr_Namespace::CudaMgr *	cuda_mgr,
		const size_t	min_slab_size,
		const size_t	max_slab_size,
		const size_t	page_size,
		AbstractBufferMgr *	parent_mgr = `0`
	)

Definition at line 25 of file GpuCudaBufferMgr.cpp.

     : BufferMgr(device_id,
                 max_buffer_pool_size,
                 min_slab_size,
                 max_slab_size,
                 page_size,
                 parent_mgr)
     , cuda_mgr_(cuda_mgr) {}

Buffer_Namespace::GpuCudaBufferMgr::~GpuCudaBufferMgr ( )

override

Definition at line 40 of file GpuCudaBufferMgr.cpp.

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

                                     {
   try {
     cuda_mgr_->synchronizeDevices();
     freeAllMem();
 #ifdef HAVE_CUDA
   } catch (const CudaMgr_Namespace::CudaErrorException& e) {
     if (e.getStatus() == CUDA_ERROR_DEINITIALIZED) {
       // TODO(adb / asuhan): Verify cuModuleUnload removes the context
       return;
     }
 #endif
   } catch (const std::runtime_error& e) {
     LOG(ERROR) << "CUDA Error: " << e.what();
   }
 }

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 56 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_.

                                                      {
   slabs_.resize(slabs_.size() + 1);
   try {
     slabs_.back() =
         cuda_mgr_->allocateDeviceMem(slab_size, device_id_, /* is_slab */ true);
   } catch (std::runtime_error& error) {
     slabs_.resize(slabs_.size() - 1);
     throw FailedToCreateSlab(slab_size);
   }
   slab_segments_.resize(slab_segments_.size() + 1);
   slab_segments_[slab_segments_.size() - 1].push_back(
       BufferSeg(0, slab_size / page_size_));
 }

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 76 of file GpuCudaBufferMgr.cpp.

References cuda_mgr_.

                                                                  {
   new GpuCudaBuffer(this,
                     seg_it,
                     device_id_,
                     cuda_mgr_,
                     page_size,
                     initial_size);  // this line is admittedly a bit weird
                                     // but the segment iterator passed into
                                     // buffer takes the address of the new
                                     // Buffer in its buffer member
 }

void Buffer_Namespace::GpuCudaBufferMgr::freeAllMem ( )

overrideprivatevirtual

Implements Buffer_Namespace::BufferMgr.

Definition at line 70 of file GpuCudaBufferMgr.cpp.

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

Referenced by ~GpuCudaBufferMgr().

                                   {
   for (auto buf_it = slabs_.begin(); buf_it != slabs_.end(); ++buf_it) {
     cuda_mgr_->freeDeviceMem(*buf_it);
   }
 }

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 36 of file GpuCudaBufferMgr.h.

36 { return GPU_MGR; }

std::string Buffer_Namespace::GpuCudaBufferMgr::getStringMgrType ( )

inlineoverride

Definition at line 37 of file GpuCudaBufferMgr.h.

37 { return ToString(GPU_MGR); }

Member Data Documentation

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

private

Definition at line 46 of file GpuCudaBufferMgr.h.

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

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

/home/jenkins-slave/workspace/core-os-doxygen/DataMgr/BufferMgr/GpuCudaBufferMgr/GpuCudaBufferMgr.h
/home/jenkins-slave/workspace/core-os-doxygen/DataMgr/BufferMgr/GpuCudaBufferMgr/GpuCudaBufferMgr.cpp

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation