Apache Arrow (C++)
A columnar in-memory analytics layer designed to accelerate big data.
Public Member Functions | Static Public Member Functions | List of all members
arrow::gpu::CudaDeviceManager Class Reference

#include <arrow/gpu/cuda_context.h>

Public Member Functions

Status GetContext (int gpu_number, std::shared_ptr< CudaContext > *ctx)
 Get the shared CUDA driver context for a particular device. More...
 
Status CreateNewContext (int gpu_number, std::shared_ptr< CudaContext > *ctx)
 Create a new context for a given device number. More...
 
Status AllocateHost (int64_t nbytes, std::shared_ptr< CudaHostBuffer > *buffer)
 
Status FreeHost (uint8_t *data, int64_t nbytes)
 
int num_devices () const
 

Static Public Member Functions

static Status GetInstance (CudaDeviceManager **manager)
 

Member Function Documentation

◆ AllocateHost()

Status arrow::gpu::CudaDeviceManager::AllocateHost ( int64_t  nbytes,
std::shared_ptr< CudaHostBuffer > *  buffer 
)

◆ CreateNewContext()

Status arrow::gpu::CudaDeviceManager::CreateNewContext ( int  gpu_number,
std::shared_ptr< CudaContext > *  ctx 
)

Create a new context for a given device number.

In general code will use GetContext

◆ FreeHost()

Status arrow::gpu::CudaDeviceManager::FreeHost ( uint8_t *  data,
int64_t  nbytes 
)

◆ GetContext()

Status arrow::gpu::CudaDeviceManager::GetContext ( int  gpu_number,
std::shared_ptr< CudaContext > *  ctx 
)

Get the shared CUDA driver context for a particular device.

◆ GetInstance()

static Status arrow::gpu::CudaDeviceManager::GetInstance ( CudaDeviceManager **  manager)
static

◆ num_devices()

int arrow::gpu::CudaDeviceManager::num_devices ( ) const

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