Class MemoryPool

Defined in File memorypool.cuh

Class Documentation

class MemoryPool

Public Functions

void initialize()

void initialize(const MemoryPoolConfig &config)

void use_memory_pool(bool use)

void *allocate(size_t size, cudaStream_t stream = cudaStreamDefault)

void deallocate(void *ptr, size_t size, cudaStream_t stream = cudaStreamDefault)

rmm::mr::device_memory_resource *get_device_resource() const

HostStatsAdaptor *get_host_resource() const

void *host_allocate(size_t size)

void host_deallocate(void *ptr, size_t size)

void print_memory_pool_status() const

size_t get_current_device_pool_memory_usage() const

size_t get_free_device_pool_memory() const

size_t get_current_host_pool_memory_usage() const

size_t get_free_host_pool_memory() const

~MemoryPool()

Public Static Functions

static MemoryPool &instance()