#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 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
std::vector < Data_Namespace::AbstractBuffer * >	owned_buffers_

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

override

Definition at line 34 of file CudaAllocator.cpp.

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

                               {
   CHECK(data_mgr_);
   for (auto& buffer_ptr : owned_buffers_) {
     data_mgr_->free(buffer_ptr);
   }
 }

Here is the call graph for this function:

Member Function Documentation

int8_t * CudaAllocator::alloc ( const size_t num_bytes )

overridevirtual

Implements Allocator.

Definition at line 57 of file CudaAllocator.cpp.

References allocGpuAbstractBuffer(), CHECK, data_mgr_, device_id_, and owned_buffers_.

Referenced by OverlapsJoinHashTable::approximateTupleCount(), BaselineJoinHashTable::approximateTupleCount(), anonymous_namespace{TableFunctionExecutionContext.cpp}::create_literal_buffer(), OverlapsJoinHashTable::initHashTableOnGpu(), BaselineJoinHashTable::initHashTableOnGpu(), transfer_flat_object_to_gpu(), and transfer_vector_of_flat_objects_to_gpu().

                                                    {
   CHECK(data_mgr_);
   owned_buffers_.emplace_back(
       CudaAllocator::allocGpuAbstractBuffer(data_mgr_, num_bytes, device_id_));
   return owned_buffers_.back()->getMemoryPtr();
 }

Here is the call graph for this function:

Here is the caller 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 41 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(), JoinHashTable::initOneToOneHashTable(), and InValuesBitmap::InValuesBitmap().

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

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

Referenced by OverlapsJoinHashTable::computeBucketSizes().

                                                                  {
   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:

Here is the caller 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 68 of file CudaAllocator.cpp.

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

Referenced by anonymous_namespace{TableFunctionExecutionContext.cpp}::create_literal_buffer(), transfer_flat_object_to_gpu(), and transfer_vector_of_flat_objects_to_gpu().

                                                                {
   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 64 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 51 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 90 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 84 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 78 of file CudaAllocator.h.

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

int CudaAllocator::device_id_

private

Definition at line 79 of file CudaAllocator.h.

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

std::vector<Data_Namespace::AbstractBuffer*> CudaAllocator::owned_buffers_

private

Definition at line 76 of file CudaAllocator.h.

Referenced by alloc(), and ~CudaAllocator().

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

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

Public Member Functions

Static Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation