CudaAllocator Class Reference

#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 BaselineJoinHashTable::approximateTupleCount(), OverlapsJoinHashTable::approximateTupleCount(), anonymous_namespace{TableFunctionExecutionContext.cpp}::create_literal_buffer(), BaselineJoinHashTableBuilder::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(), PerfectHashTable::allocateGpuMemory(), ThrustAllocator::allocateScopedBuffer(), BaselineHashTable::BaselineHashTable(), 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 anonymous_namespace{OverlapsJoinHashTable.cpp}::compute_bucket_sizes().

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

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

Here is the call 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