Mila::Dnn::Compute::TrackedMemoryResource Class Reference

A memory resource wrapper that tracks allocation and deallocation statistics. More...

Inheritance diagram for Mila::Dnn::Compute::TrackedMemoryResource:

Public Member Functions
	TrackedMemoryResource (MemoryResource *underlying, std::string_view name="")
	Constructs a new tracked memory resource.
	~TrackedMemoryResource ()
	Destructor that properly cleans up the underlying resource.
const MemoryResource *	getUnderlying () const
	Gets access to the underlying memory resource.
std::string_view	name () const
	Gets the name of this tracked memory resource.
Public Member Functions inherited from Mila::Dnn::Compute::MemoryResource
virtual	~MemoryResource ()=default
	Virtual destructor for proper cleanup of derived classes.

Protected Member Functions
void *	do_allocate (std::size_t bytes, std::size_t alignment) override
	Allocates memory and updates tracking statistics.
void	do_deallocate (void *p, std::size_t bytes, std::size_t alignment) override
	Deallocates memory and updates tracking statistics.
bool	do_is_equal (const std::pmr::memory_resource &other) const noexcept override
	Checks if this memory resource is equal to another.

Additional Inherited Members
Static Public Attributes inherited from Mila::Dnn::Compute::MemoryResource
static constexpr bool	is_device_accessible = false
	Checks if the memory is accessible from device code.
static constexpr bool	is_host_accessible = true
	Checks if the memory is accessible from host code.

Detailed Description

A memory resource wrapper that tracks allocation and deallocation statistics.

This class wraps another memory resource and intercepts all allocation, deallocation, memcpy, and memset calls to maintain global memory usage statistics.

Constructor & Destructor Documentation

TrackedMemoryResource()

Mila::Dnn::Compute::TrackedMemoryResource::TrackedMemoryResource ( MemoryResource * underlying, std::string_view name = "" )

inlineexplicit

Constructs a new tracked memory resource.

Parameters

underlying	The memory resource to track (takes ownership).
name	Optional name for this memory resource for logging purposes.

Member Function Documentation

do_allocate()

void * Mila::Dnn::Compute::TrackedMemoryResource::do_allocate ( std::size_t bytes, std::size_t alignment )

inlineoverrideprotectedvirtual

Allocates memory and updates tracking statistics.

Parameters

bytes	Number of bytes to allocate
alignment	Memory alignment requirement

Returns

Pointer to allocated memory

Implements Mila::Dnn::Compute::MemoryResource.

do_deallocate()

void Mila::Dnn::Compute::TrackedMemoryResource::do_deallocate ( void * p, std::size_t bytes, std::size_t alignment )

inlineoverrideprotectedvirtual

Deallocates memory and updates tracking statistics.

Parameters

p	Pointer to memory to deallocate
bytes	Size of memory block
alignment	Alignment used during allocation

Implements Mila::Dnn::Compute::MemoryResource.

do_is_equal()

bool Mila::Dnn::Compute::TrackedMemoryResource::do_is_equal ( const std::pmr::memory_resource & other ) const

inlineoverrideprotectedvirtualnoexcept

Checks if this memory resource is equal to another.

Parameters

other

The other memory resource to compare with

Returns

true if the underlying resources are equal

Implements Mila::Dnn::Compute::MemoryResource.

---

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

• Mila/Src/Dnn/Compute/MemoryResourceTracker.ixx