device_memory_resource derived class that uses cudaMalloc/Free for allocation/deallocation. More...

#include <cuda_memory_resource.hpp>

Inheritance diagram for rmm::mr::cuda_memory_resource:

Collaboration diagram for rmm::mr::cuda_memory_resource:

Public Member Functions
	cuda_memory_resource (cuda_memory_resource const &)=default
	Default copy constructor.

	cuda_memory_resource (cuda_memory_resource &&)=default
	Default move constructor.

cuda_memory_resource &	operator= (cuda_memory_resource const &)=default
	Default copy assignment operator. More...

cuda_memory_resource &	operator= (cuda_memory_resource &&)=default
	Default move assignment operator. More...

bool	supports_streams () const noexcept override
	Query whether the resource supports use of non-null CUDA streams for allocation/deallocation. `cuda_memory_resource` does not support streams. More...

bool	supports_get_mem_info () const noexcept override
	Query whether the resource supports the get_mem_info API. More...

Public Member Functions inherited from rmm::mr::device_memory_resource
	device_memory_resource (device_memory_resource const &)=default
	Default copy constructor.

	device_memory_resource (device_memory_resource &&) noexcept=default
	Default move constructor.

device_memory_resource &	operator= (device_memory_resource const &)=default
	Default copy assignment operator. More...

device_memory_resource &	operator= (device_memory_resource &&) noexcept=default
	Default move assignment operator. More...

void *	allocate (std::size_t bytes, cuda_stream_view stream=cuda_stream_view{})
	Allocates memory of size at least `bytes`. More...

void	deallocate (void *ptr, std::size_t bytes, cuda_stream_view stream=cuda_stream_view{})
	Deallocate memory pointed to by `p`. More...

bool	is_equal (device_memory_resource const &other) const noexcept
	Compare this resource to another. More...

void *	allocate (std::size_t bytes, std::size_t alignment)
	Allocates memory of size at least `bytes`. More...

void	deallocate (void *ptr, std::size_t bytes, std::size_t alignment)
	Deallocate memory pointed to by `p`. More...

void *	allocate_async (std::size_t bytes, std::size_t alignment, cuda_stream_view stream)
	Allocates memory of size at least `bytes`. More...

void *	allocate_async (std::size_t bytes, cuda_stream_view stream)
	Allocates memory of size at least `bytes`. More...

void	deallocate_async (void *ptr, std::size_t bytes, std::size_t alignment, cuda_stream_view stream)
	Deallocate memory pointed to by `p`. More...

void	deallocate_async (void *ptr, std::size_t bytes, cuda_stream_view stream)
	Deallocate memory pointed to by `p`. More...

bool	operator== (device_memory_resource const &other) const noexcept
	Comparison operator with another device_memory_resource. More...

bool	operator!= (device_memory_resource const &other) const noexcept
	Comparison operator with another device_memory_resource. More...

std::pair< std::size_t, std::size_t >	get_mem_info (cuda_stream_view stream) const
	Queries the amount of free and total memory for the resource. More...

Detailed Description

device_memory_resource derived class that uses cudaMalloc/Free for allocation/deallocation.

Member Function Documentation

◆ operator=() [1/2]

cuda_memory_resource& rmm::mr::cuda_memory_resource::operator= ( cuda_memory_resource && )

default

Default move assignment operator.

Returns: cuda_memory_resource& Reference to the assigned object

◆ operator=() [2/2]

cuda_memory_resource& rmm::mr::cuda_memory_resource::operator= ( cuda_memory_resource const & )

default

Default copy assignment operator.

Returns: cuda_memory_resource& Reference to the assigned object

◆ supports_get_mem_info()

bool rmm::mr::cuda_memory_resource::supports_get_mem_info ( ) const

inlineoverridevirtualnoexcept

Query whether the resource supports the get_mem_info API.

Returns: true

Implements rmm::mr::device_memory_resource.

◆ supports_streams()

bool rmm::mr::cuda_memory_resource::supports_streams ( ) const