#include <CudaAllocator.h>

Inheritance diagram for CudaAllocator:

Collaboration diagram for CudaAllocator:

Public Member Functions
	CudaAllocator (Data_Namespace::DataMgr *data_mgr, const int device_id)

	~CudaAllocator () override

int8_t *	alloc (const size_t num_bytes) override

void	free (Data_Namespace::AbstractBuffer *ab) const override

void	copyToDevice (int8_t device_dst, const int8_t host_src, const size_t num_bytes) const override

void	copyFromDevice (int8_t host_dst, const int8_t device_src, const size_t num_bytes) const override

void	zeroDeviceMem (int8_t *device_ptr, const size_t num_bytes) const override

void	setDeviceMem (int8_t *device_ptr, unsigned char uc, const size_t num_bytes) const override

Public Member Functions inherited from Allocator
	Allocator ()

virtual	~Allocator ()

Static Public Member Functions
static int8_t *	alloc (Data_Namespace::DataMgr *data_mgr, const size_t num_bytes, const int device_id)

static Data_Namespace::AbstractBuffer *	allocGpuAbstractBuffer (Data_Namespace::DataMgr *data_mgr, const size_t num_bytes, const int device_id)

static void	freeGpuAbstractBuffer (Data_Namespace::DataMgr data_mgr, Data_Namespace::AbstractBuffer ab)

Private Attributes
Data_Namespace::DataMgr *	data_mgr_

int	device_id_

Detailed Description

Definition at line 43 of file CudaAllocator.h.

Constructor & Destructor Documentation

CudaAllocator::CudaAllocator	(	Data_Namespace::DataMgr *	data_mgr,
		const int	device_id
	)

Definition at line 25 of file CudaAllocator.cpp.

References CHECK(), data_mgr_, and Data_Namespace::DataMgr::getCudaMgr().

     : data_mgr_(data_mgr), device_id_(device_id) {
   CHECK(data_mgr_);
 #ifdef HAVE_CUDA
   const auto cuda_mgr = data_mgr_->getCudaMgr();
   CHECK(cuda_mgr);
   cuda_mgr->setContext(device_id);
 #endif  // HAVE_CUDA
 }

Here is the call graph for this function:

CudaAllocator::~CudaAllocator ( )

inlineoverride

Definition at line 47 of file CudaAllocator.h.

47 {}

Member Function Documentation

int8_t * CudaAllocator::alloc	(	Data_Namespace::DataMgr *	data_mgr,
		const size_t	num_bytes,
		const int	device_id
	)

static

Definition at line 35 of file CudaAllocator.cpp.

References allocGpuAbstractBuffer(), and CHECK().

Referenced by alloc(), anonymous_namespace{TableFunctionExecutionContext.cpp}::create_literal_buffer(), ColumnFetcher::getOneTableColumnFragment(), InValuesBitmap::InValuesBitmap(), and ColumnFetcher::transferColumnIfNeeded().

                                                   {
   CHECK(data_mgr);
   auto ab = CudaAllocator::allocGpuAbstractBuffer(data_mgr, num_bytes, device_id);
   return ab->getMemoryPtr();
 }

Here is the call graph for this function:

Here is the caller graph for this function:

int8_t * CudaAllocator::alloc ( const size_t num_bytes )

overridevirtual

Implements Allocator.

Definition at line 59 of file CudaAllocator.cpp.

References alloc(), data_mgr_, and device_id_.

                                                    {
   return CudaAllocator::alloc(data_mgr_, num_bytes, device_id_);
 }

Here is the call graph for this function:

Data_Namespace::AbstractBuffer * CudaAllocator::allocGpuAbstractBuffer	(	Data_Namespace::DataMgr *	data_mgr,
		const size_t	num_bytes,
		const int	device_id
	)

static

Definition at line 43 of file CudaAllocator.cpp.

References Data_Namespace::DataMgr::alloc(), CHECK(), CHECK_EQ, and Data_Namespace::GPU_LEVEL.

Referenced by alloc(), ThrustAllocator::allocate(), ThrustAllocator::allocateScopedBuffer(), BaselineJoinHashTable::initHashTableForDevice(), JoinHashTable::initOneToManyHashTable(), and JoinHashTable::initOneToOneHashTable().

                          {
   CHECK(data_mgr);
   auto ab = data_mgr->alloc(Data_Namespace::GPU_LEVEL, device_id, num_bytes);
   CHECK_EQ(ab->getPinCount(), 1);
   return ab;
 }

Here is the call graph for this function:

Here is the caller graph for this function:

void CudaAllocator::copyFromDevice	(	int8_t *	host_dst,
		const int8_t *	device_src,
		const size_t	num_bytes
	)		const

overridevirtual

Implements DeviceAllocator.

Definition at line 75 of file CudaAllocator.cpp.

References CHECK(), data_mgr_, device_id_, and Data_Namespace::DataMgr::getCudaMgr().

                                                                  {
   const auto cuda_mgr = data_mgr_->getCudaMgr();
   CHECK(cuda_mgr);
   cuda_mgr->copyDeviceToHost(host_dst, device_src, num_bytes, device_id_);
 }

Here is the call graph for this function:

void CudaAllocator::copyToDevice	(	int8_t *	device_dst,
		const int8_t *	host_src,
		const size_t	num_bytes
	)		const

overridevirtual

Implements DeviceAllocator.

Definition at line 67 of file CudaAllocator.cpp.

References CHECK(), data_mgr_, device_id_, and Data_Namespace::DataMgr::getCudaMgr().

Referenced by anonymous_namespace{TableFunctionExecutionContext.cpp}::create_literal_buffer().

                                                                {
   const auto cuda_mgr = data_mgr_->getCudaMgr();
   CHECK(cuda_mgr);
   cuda_mgr->copyHostToDevice(device_dst, host_src, num_bytes, device_id_);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

void CudaAllocator::free ( Data_Namespace::AbstractBuffer * ab ) const

overridevirtual

Implements DeviceAllocator.

Definition at line 63 of file CudaAllocator.cpp.

References data_mgr_, and Data_Namespace::DataMgr::free().

                                                                {
   data_mgr_->free(ab);
 }

Here is the call graph for this function:

void CudaAllocator::freeGpuAbstractBuffer	(	Data_Namespace::DataMgr *	data_mgr,
		Data_Namespace::AbstractBuffer *	ab
	)

static

Definition at line 53 of file CudaAllocator.cpp.

References CHECK(), and Data_Namespace::DataMgr::free().

Referenced by JoinHashTable::freeHashBufferGpuMemory(), and BaselineJoinHashTable::freeHashBufferGpuMemory().

                                                                             {
   CHECK(data_mgr);
   data_mgr->free(ab);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

void CudaAllocator::setDeviceMem	(	int8_t *	device_ptr,
		unsigned char	uc,
		const size_t	num_bytes
	)		const

overridevirtual

Implements DeviceAllocator.

Definition at line 89 of file CudaAllocator.cpp.

References CHECK(), data_mgr_, device_id_, and Data_Namespace::DataMgr::getCudaMgr().

                                                                {
   const auto cuda_mgr = data_mgr_->getCudaMgr();
   CHECK(cuda_mgr);
   cuda_mgr->setDeviceMem(device_ptr, uc, num_bytes, device_id_);
 }

Here is the call graph for this function:

void CudaAllocator::zeroDeviceMem	(	int8_t *	device_ptr,
		const size_t	num_bytes
	)		const

overridevirtual

Implements DeviceAllocator.

Definition at line 83 of file CudaAllocator.cpp.

References CHECK(), data_mgr_, device_id_, and Data_Namespace::DataMgr::getCudaMgr().

                                                                                   {
   const auto cuda_mgr = data_mgr_->getCudaMgr();
   CHECK(cuda_mgr);
   cuda_mgr->zeroDeviceMem(device_ptr, num_bytes, device_id_);
 }

Here is the call graph for this function:

Member Data Documentation

Data_Namespace::DataMgr* CudaAllocator::data_mgr_

private

Definition at line 80 of file CudaAllocator.h.

Referenced by alloc(), copyFromDevice(), copyToDevice(), CudaAllocator(), free(), setDeviceMem(), and zeroDeviceMem().

int CudaAllocator::device_id_

private

Definition at line 81 of file CudaAllocator.h.

Referenced by alloc(), copyFromDevice(), copyToDevice(), setDeviceMem(), and zeroDeviceMem().

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

/home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/Allocators/CudaAllocator.h
/home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/Allocators/CudaAllocator.cpp

Public Member Functions

Static Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation