#include <GpuCudaBuffer.h>

Inheritance diagram for Buffer_Namespace::GpuCudaBuffer:

Collaboration diagram for Buffer_Namespace::GpuCudaBuffer:

Public Member Functions
	GpuCudaBuffer (BufferMgr bm, BufferList::iterator seg_it, const int device_id, CudaMgr_Namespace::CudaMgr cuda_mgr, const size_t page_size=512, const size_t num_bytes=0)

Data_Namespace::MemoryLevel	getType () const override

Public Member Functions inherited from Buffer_Namespace::Buffer
	Buffer (BufferMgr *bm, BufferList::iterator seg_it, const int device_id, const size_t page_size=512, const size_t num_bytes=0)
	Constructs a Buffer object. The constructor requires a memory address (provided by BufferMgr), number of pages, and the size in bytes of each page. Additionally, the Buffer can be initialized with an epoch. More...

	~Buffer () override
	Destructor. More...

void	read (int8_t *const dst, const size_t num_bytes, const size_t offset=0, const MemoryLevel dst_buffer_type=CPU_LEVEL, const int device_id=-1) override
	Reads (copies) data from the buffer to the destination (dst) memory location. Reads (copies) nbytes of data from the buffer, beginning at the specified byte offset, into the destination (dst) memory location. More...

void	reserve (const size_t num_bytes) override

void	write (int8_t *src, const size_t num_bytes, const size_t offset=0, const MemoryLevel src_buffer_type=CPU_LEVEL, const int device_id=-1) override
	Writes (copies) data from src into the buffer. Writes (copies) nbytes of data into the buffer at the specified byte offset, from the source (src) memory location. More...

void	append (int8_t *src, const size_t num_bytes, const MemoryLevel src_buffer_type=CPU_LEVEL, const int deviceId=-1) override

int8_t *	getMemoryPtr () override
	Returns a raw, constant (read-only) pointer to the underlying buffer. More...

void	setMemoryPtr (int8_t *new_ptr) override

size_t	reservedSize () const override
	Returns the total number of bytes allocated for the buffer. More...

size_t	pageCount () const override
	Returns the number of pages in the buffer. More...

size_t	pageSize () const override
	Returns the size in bytes of each page in the buffer. More...

int	pin () override

int	unPin () override

int	getPinCount () override

int32_t	getSlabNum () const

Public Member Functions inherited from Data_Namespace::AbstractBuffer
	AbstractBuffer (const int device_id)

	AbstractBuffer (const int device_id, const SQLTypeInfo sql_type)

virtual	~AbstractBuffer ()

size_t	size () const

int	getDeviceId () const

bool	isDirty () const

bool	isAppended () const

bool	isUpdated () const

bool	hasEncoder () const

SQLTypeInfo	getSqlType () const

void	setSqlType (const SQLTypeInfo &sql_type)

Encoder *	getEncoder () const

void	setDirty ()

void	setUpdated ()

void	setAppended ()

void	setSize (const size_t size)

void	clearDirtyBits ()

void	initEncoder (const SQLTypeInfo &tmp_sql_type)

void	syncEncoder (const AbstractBuffer *src_buffer)

void	copyTo (AbstractBuffer *destination_buffer, const size_t num_bytes=0)

void	resetToEmpty ()

Private Member Functions
void	readData (int8_t *const dst, const size_t num_bytes, const size_t offset=0, const MemoryLevel dst_buffer_type=CPU_LEVEL, const int dst_devic_id=-1) override

void	writeData (int8_t *const src, const size_t num_bytes, const size_t offset=0, const MemoryLevel src_buffer_type=CPU_LEVEL, const int src_device_id=-1) override

Private Attributes
CudaMgr_Namespace::CudaMgr *	cuda_mgr_

Additional Inherited Members
Protected Attributes inherited from Buffer_Namespace::Buffer
int8_t *	mem_

Protected Attributes inherited from Data_Namespace::AbstractBuffer
std::unique_ptr< Encoder >	encoder_

SQLTypeInfo	sql_type_

size_t	size_

int	device_id_

Detailed Description

Definition at line 26 of file GpuCudaBuffer.h.

Constructor & Destructor Documentation

Buffer_Namespace::GpuCudaBuffer::GpuCudaBuffer	(	BufferMgr *	bm,
		BufferList::iterator	seg_it,
		const int	device_id,
		CudaMgr_Namespace::CudaMgr *	cuda_mgr,
		const size_t	page_size = `512`,
		const size_t	num_bytes = `0`
	)

Definition at line 26 of file GpuCudaBuffer.cpp.

32 : Buffer(bm, seg_it, device_id, page_size, num_bytes), cuda_mgr_(cuda_mgr) {}

Buffer_Namespace::Buffer::Buffer

Buffer(BufferMgr *bm, BufferList::iterator seg_it, const int device_id, const size_t page_size=512, const size_t num_bytes=0)

Constructs a Buffer object. The constructor requires a memory address (provided by BufferMgr)...

Definition: Buffer.cpp:27

Buffer_Namespace::GpuCudaBuffer::cuda_mgr_

CudaMgr_Namespace::CudaMgr * cuda_mgr_

Definition: GpuCudaBuffer.h:48

Member Function Documentation

Data_Namespace::MemoryLevel Buffer_Namespace::GpuCudaBuffer::getType ( ) const

inlineoverridevirtual

Implements Data_Namespace::AbstractBuffer.

Definition at line 34 of file GpuCudaBuffer.h.

References Data_Namespace::GPU_LEVEL.

34 { return GPU_LEVEL; }

Data_Namespace::GPU_LEVEL

Definition: MemoryLevel.h:21

void Buffer_Namespace::GpuCudaBuffer::readData	(	int8_t *const	dst,
		const size_t	num_bytes,
		const size_t	offset = `0`,
		const MemoryLevel	dst_buffer_type = `CPU_LEVEL`,
		const int	dst_devic_id = `-1`
	)

overrideprivatevirtual

Implements Buffer_Namespace::Buffer.

Definition at line 34 of file GpuCudaBuffer.cpp.

References CudaMgr_Namespace::CudaMgr::copyDeviceToDevice(), CudaMgr_Namespace::CudaMgr::copyDeviceToHost(), Data_Namespace::CPU_LEVEL, cuda_mgr_, Data_Namespace::AbstractBuffer::device_id_, logger::FATAL, Data_Namespace::GPU_LEVEL, LOG, and Buffer_Namespace::Buffer::mem_.

                                                       {
   if (dst_buffer_type == CPU_LEVEL) {
     cuda_mgr_->copyDeviceToHost(
         dst, mem_ + offset, num_bytes);  // need to replace 0 with gpu num
   } else if (dst_buffer_type == GPU_LEVEL) {
     cuda_mgr_->copyDeviceToDevice(
         dst, mem_ + offset, num_bytes, dst_device_id, device_id_);
 
   } else {
     LOG(FATAL) << "Unsupported buffer type";
   }
 }

Here is the call graph for this function:

void Buffer_Namespace::GpuCudaBuffer::writeData	(	int8_t *const	src,
		const size_t	num_bytes,
		const size_t	offset = `0`,
		const MemoryLevel	src_buffer_type = `CPU_LEVEL`,
		const int	src_device_id = `-1`
	)

overrideprivatevirtual

Implements Buffer_Namespace::Buffer.

Definition at line 51 of file GpuCudaBuffer.cpp.

References CHECK_GE, CudaMgr_Namespace::CudaMgr::copyDeviceToDevice(), CudaMgr_Namespace::CudaMgr::copyHostToDevice(), Data_Namespace::CPU_LEVEL, cuda_mgr_, Data_Namespace::AbstractBuffer::device_id_, logger::FATAL, Data_Namespace::GPU_LEVEL, LOG, and Buffer_Namespace::Buffer::mem_.

                                                        {
   if (src_buffer_type == CPU_LEVEL) {
     // std::cout << "Writing to GPU from source CPU" << std::endl;
 
     cuda_mgr_->copyHostToDevice(
         mem_ + offset, src, num_bytes, device_id_);  // need to replace 0 with gpu num
 
   } else if (src_buffer_type == GPU_LEVEL) {
     // std::cout << "Writing to GPU from source GPU" << std::endl;
     CHECK_GE(src_device_id, 0);
     cuda_mgr_->copyDeviceToDevice(
         mem_ + offset, src, num_bytes, device_id_, src_device_id);
   } else {
     LOG(FATAL) << "Unsupported buffer type";
   }
 }

Here is the call graph for this function:

Member Data Documentation

CudaMgr_Namespace::CudaMgr* Buffer_Namespace::GpuCudaBuffer::cuda_mgr_

private

Definition at line 48 of file GpuCudaBuffer.h.

Referenced by readData(), and writeData().

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

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

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation