Mila::Dnn::Compute::VulkanMemoryResource Class Reference

Stub implementation for platforms without Vulkan support. More...

Inheritance diagram for Mila::Dnn::Compute::VulkanMemoryResource:

Collaboration diagram for Mila::Dnn::Compute::VulkanMemoryResource:

Public Types
enum class	AllocationStrategy { DEVICE_LOCAL , HOST_VISIBLE , HOST_COHERENT , STAGING_OPTIMAL }
enum class	DevicePreference { DISCRETE_GPU , INTEGRATED_GPU , ANY_GPU , HIGHEST_MEMORY }

Public Member Functions
	VulkanMemoryResource (DevicePreference=DevicePreference::DISCRETE_GPU, AllocationStrategy=AllocationStrategy::DEVICE_LOCAL)
std::string	getDeviceInfo () const
Public Member Functions inherited from Mila::Dnn::Compute::MemoryResource
virtual	~MemoryResource ()=default
	Virtual destructor for proper cleanup of derived classes.

Static Public Attributes
static constexpr bool	is_device_accessible = false
static constexpr bool	is_host_accessible = false
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.

Protected Member Functions
void *	do_allocate (std::size_t, std::size_t) override
	Allocates memory with specified size and alignment.
void	do_deallocate (void *, std::size_t, std::size_t) override
	Deallocates previously allocated memory.
bool	do_is_equal (const std::pmr::memory_resource &) const noexcept override
	Compares this memory resource with another for equality.

Detailed Description

Stub implementation for platforms without Vulkan support.

Member Enumeration Documentation

AllocationStrategy

enum class Mila::Dnn::Compute::VulkanMemoryResource::AllocationStrategy

strong

Enumerator
DEVICE_LOCAL
HOST_VISIBLE
HOST_COHERENT
STAGING_OPTIMAL

DevicePreference

enum class Mila::Dnn::Compute::VulkanMemoryResource::DevicePreference

strong

Enumerator
DISCRETE_GPU
INTEGRATED_GPU
ANY_GPU
HIGHEST_MEMORY

Constructor & Destructor Documentation

VulkanMemoryResource()

Mila::Dnn::Compute::VulkanMemoryResource::VulkanMemoryResource ( DevicePreference = DevicePreference::DISCRETE_GPU, AllocationStrategy = AllocationStrategy::DEVICE_LOCAL )

inlineexplicit

Member Function Documentation

do_allocate()

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

inlineoverrideprotectedvirtual

Allocates memory with specified size and alignment.

Pure virtual function that must be implemented by derived classes to provide device-specific memory allocation. Implementation should handle device-specific allocation strategies and error conditions.

Parameters

bytes	Number of bytes to allocate
alignment	Memory alignment requirement

Returns

Pointer to allocated memory

Exceptions

std::bad_alloc

If allocation fails

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

do_deallocate()

void Mila::Dnn::Compute::VulkanMemoryResource::do_deallocate ( void * ptr, std::size_t bytes, std::size_t alignment )

inlineoverrideprotectedvirtual

Deallocates previously allocated memory.

Pure virtual function that must be implemented by derived classes to provide device-specific memory deallocation. Implementation should handle proper cleanup and device synchronization if necessary.

Parameters

ptr	Pointer to memory to deallocate
bytes	Size of memory block (may be used for debugging/validation)
alignment	Alignment that was used during allocation

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

do_is_equal()

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

inlineoverrideprotectedvirtualnoexcept

Compares this memory resource with another for equality.

Two memory resources are equal if they can be used interchangeably for allocation and deallocation. Typically this means they are the same type and manage the same underlying memory pool or device.

Parameters

other

The other memory resource to compare with

Returns

true if the memory resources are equivalent

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

getDeviceInfo()

std::string Mila::Dnn::Compute::VulkanMemoryResource::getDeviceInfo ( ) const

inline

Member Data Documentation

is_device_accessible

bool Mila::Dnn::Compute::VulkanMemoryResource::is_device_accessible = false

staticconstexpr

is_host_accessible

bool Mila::Dnn::Compute::VulkanMemoryResource::is_host_accessible = false

staticconstexpr

---

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

• /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Rocm/RocmMemoryResource.ixx