cuda::experimental::stf::uncached_graph_allocator#

class uncached_graph_allocator : public cuda::experimental::stf::block_allocator_interface#

Uncached allocator (used as a base for other allocators)

Any allocation/deallocation results in an actual underlying CUDA API call (e.g. cudaGraphAddMemAllocNode). This allocator should generally not be used directly, but used within an allocator with a more advanced strategy (caching, heap allocation, …)

Public Functions

uncached_graph_allocator() = default#
inline virtual void *allocate(
backend_ctx_untyped &bctx,
const data_place &memory_node,
::std::ptrdiff_t &s,
event_list &prereqs,
) override#

Asynchronous allocation of memory.

This method asynchronously allocates *s bytes on the specified memory node. Upon success, *ptr will contain the address of the allocated buffer. If the allocation fails, *s will contain a negative value.

Parameters:

  • ctx – Untyped backend context

  • memory_node – Memory node where the allocation should occur

  • s – Pointer to the size of the allocation

  • prereqs – List of events that should finish before the allocation starts

Returns:

void* Pointer to the allocated memory

inline virtual void deallocate(
backend_ctx_untyped &bctx,
const data_place &memory_node,
event_list &prereqs,
void *ptr,
size_t,
) override#

Deallocation of memory.

This method deallocates memory previously allocated by the allocate method.

Parameters:

  • ctx – Untyped backend context

  • memory_node – Memory node where the deallocation should occur

  • prereqs – List of events that should finish before the deallocation starts

  • ptr – Pointer to the memory to be deallocated

  • sz – Size of the memory to be deallocated

inline virtual ::std::string to_string() const override#

String representation of the allocator.

This method returns a string representation of the allocator.

Returns:

std::string String representation of the allocator

inline virtual event_list deinit(backend_ctx_untyped&) override#

Deinitialization of the allocator.

Returns:

List of events indicating that this was deinitialized

inline virtual void print_info() const#

Print information about the allocator.

This method prints information about the allocator to stderr.