ThrustAllocator Class Reference

#include <ThrustAllocator.h>

Collaboration diagram for ThrustAllocator:

Public Types
using	value_type = int8_t

Public Member Functions
	ThrustAllocator (Data_Namespace::DataMgr *mgr, const int id)
	~ThrustAllocator ()
int8_t *	allocate (std::ptrdiff_t num_bytes)
void	deallocate (int8_t *ptr, size_t num_bytes)
int8_t *	allocateScopedBuffer (std::ptrdiff_t num_bytes)
Data_Namespace::DataMgr *	getDataMgr () const
int	getDeviceId () const

Private Types
using	PtrMapperType = std::unordered_map< int8_t *, Data_Namespace::AbstractBuffer * >

Private Attributes
Data_Namespace::DataMgr *	data_mgr_
const int	device_id_
PtrMapperType	raw_to_ab_ptr_
std::vector< Data_Namespace::AbstractBuffer * >	scoped_buffers_
std::vector< int8_t * >	default_alloc_scoped_buffers_

Detailed Description

Definition at line 36 of file ThrustAllocator.h.

Member Typedef Documentation

PtrMapperType

using ThrustAllocator::PtrMapperType = std::unordered_map<int8_t*, Data_Namespace::AbstractBuffer*>

private

Definition at line 55 of file ThrustAllocator.h.

value_type

using ThrustAllocator::value_type = int8_t

Definition at line 38 of file ThrustAllocator.h.

Constructor & Destructor Documentation

ThrustAllocator()

ThrustAllocator::ThrustAllocator ( Data_Namespace::DataMgr *  mgr, const int  id  )

inline

Definition at line 39 of file ThrustAllocator.h.

      : data_mgr_(mgr), device_id_(id) {}

~ThrustAllocator()

ThrustAllocator::~ThrustAllocator

(

)

Definition at line 71 of file ThrustAllocator.cpp.

References CHECK_EQ, data_mgr_, default_alloc_scoped_buffers_, Data_Namespace::DataMgr::free(), and scoped_buffers_.

                                  {
  for (auto ab : scoped_buffers_) {
    data_mgr_->free(ab);
  }
#ifdef HAVE_CUDA
  for (auto ptr : default_alloc_scoped_buffers_) {
    const auto err = cuMemFree(reinterpret_cast<CUdeviceptr>(ptr));
    CHECK_EQ(CUDA_SUCCESS, err);
  }
#endif  // HAVE_CUDA
}

Here is the call graph for this function:

Member Function Documentation

allocate()

int8_t * ThrustAllocator::allocate

(

std::ptrdiff_t

num_bytes

)

Definition at line 24 of file ThrustAllocator.cpp.

References CudaAllocator::allocGpuAbstractBuffer(), CHECK, CHECK_EQ, data_mgr_, device_id_, Data_Namespace::AbstractBuffer::getMemoryPtr(), and raw_to_ab_ptr_.

Referenced by BaselineJoinHashTable::fetchColumn(), and JoinHashTable::fetchFragments().

                                                        {
#ifdef HAVE_CUDA
  if (!data_mgr_) {  // only for unit tests
    CUdeviceptr ptr;
    const auto err = cuMemAlloc(&ptr, num_bytes);
    CHECK_EQ(CUDA_SUCCESS, err);
    return reinterpret_cast<int8_t*>(ptr);
  }
#endif  // HAVE_CUDA
  Data_Namespace::AbstractBuffer* ab =
      CudaAllocator::allocGpuAbstractBuffer(data_mgr_, num_bytes, device_id_);
  int8_t* raw_ptr = reinterpret_cast<int8_t*>(ab->getMemoryPtr());
  CHECK(!raw_to_ab_ptr_.count(raw_ptr));
  raw_to_ab_ptr_.insert(std::make_pair(raw_ptr, ab));
  return raw_ptr;
}

Here is the call graph for this function:

Here is the caller graph for this function:

allocateScopedBuffer()

int8_t * ThrustAllocator::allocateScopedBuffer

(

std::ptrdiff_t

num_bytes

)

Definition at line 55 of file ThrustAllocator.cpp.

References CudaAllocator::allocGpuAbstractBuffer(), CHECK_EQ, data_mgr_, default_alloc_scoped_buffers_, device_id_, Data_Namespace::AbstractBuffer::getMemoryPtr(), and scoped_buffers_.

Referenced by OverlapsJoinHashTable::approximateTupleCount(), BaselineJoinHashTable::approximateTupleCount(), OverlapsJoinHashTable::initHashTableOnGpu(), BaselineJoinHashTable::initHashTableOnGpu(), transfer_object_to_gpu(), and transfer_pod_vector_to_gpu().

                                                                    {
#ifdef HAVE_CUDA
  if (!data_mgr_) {  // only for unit tests
    CUdeviceptr ptr;
    const auto err = cuMemAlloc(&ptr, num_bytes);
    CHECK_EQ(CUDA_SUCCESS, err);
    default_alloc_scoped_buffers_.push_back(reinterpret_cast<int8_t*>(ptr));
    return reinterpret_cast<int8_t*>(ptr);
  }
#endif  // HAVE_CUDA
  Data_Namespace::AbstractBuffer* ab =
      CudaAllocator::allocGpuAbstractBuffer(data_mgr_, num_bytes, device_id_);
  scoped_buffers_.push_back(ab);
  return reinterpret_cast<int8_t*>(ab->getMemoryPtr());
}

Here is the call graph for this function:

Here is the caller graph for this function:

deallocate()

void ThrustAllocator::deallocate	(	int8_t *	ptr,
		size_t	num_bytes
	)

Definition at line 41 of file ThrustAllocator.cpp.

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

                                                              {
#ifdef HAVE_CUDA
  if (!data_mgr_) {  // only for unit tests
    const auto err = cuMemFree(reinterpret_cast<CUdeviceptr>(ptr));
    CHECK_EQ(CUDA_SUCCESS, err);
    return;
  }
#endif  // HAVE_CUDA
  PtrMapperType::iterator ab_it = raw_to_ab_ptr_.find(ptr);
  CHECK(ab_it != raw_to_ab_ptr_.end());
  data_mgr_->free(ab_it->second);
  raw_to_ab_ptr_.erase(ab_it);
}

Here is the call graph for this function:

getDataMgr()

Data_Namespace::DataMgr* ThrustAllocator::getDataMgr ( ) const

inline

Definition at line 48 of file ThrustAllocator.h.

Referenced by transfer_object_to_gpu(), and transfer_pod_vector_to_gpu().

{ return data_mgr_; }

Here is the caller graph for this function:

getDeviceId()

int ThrustAllocator::getDeviceId ( ) const

inline

Definition at line 50 of file ThrustAllocator.h.

Referenced by transfer_object_to_gpu(), and transfer_pod_vector_to_gpu().

{ return device_id_; }

Here is the caller graph for this function:

Member Data Documentation

data_mgr_

Data_Namespace::DataMgr* ThrustAllocator::data_mgr_

private

Definition at line 53 of file ThrustAllocator.h.

Referenced by allocate(), allocateScopedBuffer(), deallocate(), and ~ThrustAllocator().

default_alloc_scoped_buffers_

std::vector<int8_t*> ThrustAllocator::default_alloc_scoped_buffers_

private

Definition at line 58 of file ThrustAllocator.h.

Referenced by allocateScopedBuffer(), and ~ThrustAllocator().

device_id_

const int ThrustAllocator::device_id_

private

Definition at line 54 of file ThrustAllocator.h.

Referenced by allocate(), and allocateScopedBuffer().

raw_to_ab_ptr_

PtrMapperType ThrustAllocator::raw_to_ab_ptr_

private

Definition at line 56 of file ThrustAllocator.h.

Referenced by allocate(), and deallocate().

scoped_buffers_

std::vector<Data_Namespace::AbstractBuffer*> ThrustAllocator::scoped_buffers_

private

Definition at line 57 of file ThrustAllocator.h.

Referenced by allocateScopedBuffer(), and ~ThrustAllocator().

---

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

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