Resource that uses Upstream to allocate memory and logs information about the requested allocation/deallocations. More...

#include <logging_resource_adaptor.hpp>

Inheritance diagram for rmm::mr::logging_resource_adaptor< Upstream >:

[legend]

Collaboration diagram for rmm::mr::logging_resource_adaptor< Upstream >:

[legend]

Public Member Functions
	logging_resource_adaptor (Upstream *upstream, std::string const &filename=get_default_filename(), bool auto_flush=false)
	Construct a new logging resource adaptor using `upstream` to satisfy allocation requests and logging information about each allocation/free to the file specified by `filename`. More...

	logging_resource_adaptor (Upstream *upstream, std::ostream &stream, bool auto_flush=false)
	Construct a new logging resource adaptor using `upstream` to satisfy allocation requests and logging information about each allocation/free to the ostream specified by `stream`. More...

	logging_resource_adaptor (Upstream *upstream, std::initializer_list< rapids_logger::sink_ptr > sinks, bool auto_flush=false)
	Construct a new logging resource adaptor using `upstream` to satisfy allocation requests and logging information about each allocation/free to the ostream specified by `stream`. More...

	logging_resource_adaptor (device_async_resource_ref upstream, std::string const &filename=get_default_filename(), bool auto_flush=false)
	Construct a new logging resource adaptor using `upstream` to satisfy allocation requests and logging information about each allocation/free to the file specified by `filename`. More...

	logging_resource_adaptor (device_async_resource_ref upstream, std::ostream &stream, bool auto_flush=false)
	Construct a new logging resource adaptor using `upstream` to satisfy allocation requests and logging information about each allocation/free to the ostream specified by `stream`. More...

	logging_resource_adaptor (device_async_resource_ref upstream, std::initializer_list< rapids_logger::sink_ptr > sinks, bool auto_flush=false)
	Construct a new logging resource adaptor using `upstream` to satisfy allocation requests and logging information about each allocation/free to the ostream specified by `stream`. More...

	logging_resource_adaptor (logging_resource_adaptor const &)=delete

logging_resource_adaptor &	operator= (logging_resource_adaptor const &)=delete

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

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

rmm::device_async_resource_ref	get_upstream_resource () const noexcept
	rmm::device_async_resource_ref to the upstream resource More...

void	flush ()
	Flush logger contents.

std::string	header () const
	Return the CSV header string. 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...

Static Public Member Functions
static std::string	get_default_filename ()
	Return the value of the environment variable RMM_LOG_FILE. More...

Detailed Description

template<typename Upstream>
class rmm::mr::logging_resource_adaptor< Upstream >

Resource that uses Upstream to allocate memory and logs information about the requested allocation/deallocations.

An instance of this resource can be constructed with an existing, upstream resource in order to satisfy allocation requests and log allocation/deallocation activity.

Template Parameters

Upstream Type of the upstream resource used for allocation/deallocation.

Constructor & Destructor Documentation

◆ logging_resource_adaptor() [1/6]

template<typename Upstream >

rmm::mr::logging_resource_adaptor< Upstream >::logging_resource_adaptor	(	Upstream *	upstream,
		std::string const &	filename = `get_default_filename()`,
		bool	auto_flush = `false`
	)

inline

Construct a new logging resource adaptor using upstream to satisfy allocation requests and logging information about each allocation/free to the file specified by filename.

The logfile will be written using CSV formatting.

Clears the contents of filename if it already exists.

Creating multiple logging_resource_adaptors with the same filename will result in undefined behavior.

Exceptions

rmm::logic_error	if `upstream == nullptr`
spdlog::spdlog_ex	if opening `filename` failed

Parameters

upstream	The resource used for allocating/deallocating device memory
filename	Name of file to write log info. If not specified, retrieves the file name from the environment variable "RMM_LOG_FILE".
auto_flush	If true, flushes the log for every (de)allocation. Warning, this will degrade performance.

◆ logging_resource_adaptor() [2/6]

template<typename Upstream >

rmm::mr::logging_resource_adaptor< Upstream >::logging_resource_adaptor	(	Upstream *	upstream,
		std::ostream &	stream,
		bool	auto_flush = `false`
	)

inline

Construct a new logging resource adaptor using upstream to satisfy allocation requests and logging information about each allocation/free to the ostream specified by stream.

The logfile will be written using CSV formatting.

Exceptions

rmm::logic_error if upstream == nullptr

Parameters

upstream	The resource used for allocating/deallocating device memory
stream	The ostream to write log info.
auto_flush	If true, flushes the log for every (de)allocation. Warning, this will degrade performance.

◆ logging_resource_adaptor() [3/6]

template<typename Upstream >

rmm::mr::logging_resource_adaptor< Upstream >::logging_resource_adaptor	(	Upstream *	upstream,
		std::initializer_list< rapids_logger::sink_ptr >	sinks,
		bool	auto_flush = `false`
	)

inline

Construct a new logging resource adaptor using upstream to satisfy allocation requests and logging information about each allocation/free to the ostream specified by stream.

The logfile will be written using CSV formatting.

Exceptions

rmm::logic_error if upstream == nullptr

Parameters

upstream	The resource used for allocating/deallocating device memory
sinks	A list of logging sinks to which log output will be written.
auto_flush	If true, flushes the log for every (de)allocation. Warning, this will degrade performance.

◆ logging_resource_adaptor() [4/6]

template<typename Upstream >

rmm::mr::logging_resource_adaptor< Upstream >::logging_resource_adaptor	(	device_async_resource_ref	upstream,
		std::string const &	filename = `get_default_filename()`,
		bool	auto_flush = `false`
	)

inline

Construct a new logging resource adaptor using upstream to satisfy allocation requests and logging information about each allocation/free to the file specified by filename.

The logfile will be written using CSV formatting.

Clears the contents of filename if it already exists.

Creating multiple logging_resource_adaptors with the same filename will result in undefined behavior.

Exceptions

spdlog::spdlog_ex if opening filename failed

Parameters

upstream	The resource_ref used for allocating/deallocating device memory.
filename	Name of file to write log info. If not specified, retrieves the file name from the environment variable "RMM_LOG_FILE".
auto_flush	If true, flushes the log for every (de)allocation. Warning, this will degrade performance.

◆ logging_resource_adaptor() [5/6]

template<typename Upstream >

rmm::mr::logging_resource_adaptor< Upstream >::logging_resource_adaptor	(	device_async_resource_ref	upstream,
		std::ostream &	stream,
		bool	auto_flush = `false`
	)

inline

Construct a new logging resource adaptor using upstream to satisfy allocation requests and logging information about each allocation/free to the ostream specified by stream.

The logfile will be written using CSV formatting.

Parameters

upstream	The resource_ref used for allocating/deallocating device memory.
stream	The ostream to write log info.
auto_flush	If true, flushes the log for every (de)allocation. Warning, this will degrade performance.

◆ logging_resource_adaptor() [6/6]

template<typename Upstream >

rmm::mr::logging_resource_adaptor< Upstream >::logging_resource_adaptor	(	device_async_resource_ref	upstream,
		std::initializer_list< rapids_logger::sink_ptr >	sinks,
		bool	auto_flush = `false`
	)

inline

Construct a new logging resource adaptor using upstream to satisfy allocation requests and logging information about each allocation/free to the ostream specified by stream.

The logfile will be written using CSV formatting.

Parameters

upstream	The resource_ref used for allocating/deallocating device memory.
sinks	A list of logging sinks to which log output will be written.
auto_flush	If true, flushes the log for every (de)allocation. Warning, this will degrade performance.

Member Function Documentation

◆ get_default_filename()

template<typename Upstream >

static std::string rmm::mr::logging_resource_adaptor< Upstream >::get_default_filename ( )

inlinestatic

Return the value of the environment variable RMM_LOG_FILE.

Exceptions

rmm::logic_error if RMM_LOG_FILE is not set.

Returns: The value of RMM_LOG_FILE as std::string.

◆ get_upstream_resource()

template<typename Upstream >

rmm::device_async_resource_ref rmm::mr::logging_resource_adaptor< Upstream >::get_upstream_resource ( ) const

inlinenoexcept

rmm::device_async_resource_ref to the upstream resource

Returns: rmm::device_async_resource_ref to the upstream resource

◆ header()

template<typename Upstream >

std::string rmm::mr::logging_resource_adaptor< Upstream >::header ( ) const

inline

Return the CSV header string.

Returns: CSV formatted header string of column names

◆ operator=()

template<typename Upstream >

logging_resource_adaptor& rmm::mr::logging_resource_adaptor< Upstream >::operator= ( logging_resource_adaptor< Upstream > && )

defaultnoexcept

Default move assignment operator.

Returns: logging_resource_adaptor& Reference to the assigned object

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

logging_resource_adaptor.hpp

Public Member Functions

Static Public Member Functions

Detailed Description

template<typename Upstream> class rmm::mr::logging_resource_adaptor< Upstream >

Constructor & Destructor Documentation

◆ logging_resource_adaptor() [1/6]

◆ logging_resource_adaptor() [2/6]

◆ logging_resource_adaptor() [3/6]

◆ logging_resource_adaptor() [4/6]

◆ logging_resource_adaptor() [5/6]

◆ logging_resource_adaptor() [6/6]

Member Function Documentation

◆ get_default_filename()

◆ get_upstream_resource()

◆ header()

◆ operator=()

template<typename Upstream>
class rmm::mr::logging_resource_adaptor< Upstream >