class
SushiAI::Graph::ArenaSet
Owns one Storage per arena in a plan and hands out views into them.
- Declared in
include/SushiAI/graph/arena.hpp
Public member functions
ArenaSet(SushiBLAS::Engine &engine, MemoryPlan plan)Allocates every arena requested by plan.
Parameters
engineThe engine whose context owns the allocator.
planThe memory plan to realise.
Exceptions
ErrorIf an arena allocation fails.
const MemoryPlan & plan() const noexceptThe plan these arenas realise.
bool holds(ValueId id) const noexceptWhether id's bytes live in one of these arenas.
SushiBLAS::Tensor view(ValueId id, const Shape &shape, DType dtype) constReturns a tensor view of the slot reserved for id.
Parameters
idThe value to address.
shapeThe value extents.
dtypeThe element data type.
Returns
A tensor sharing the arena storage at the planned offset.
Exceptions
ErrorIf
idis not packed or size mismatches the slot.
std::size_t arena_count() const noexceptHow many arenas were allocated.
std::size_t bytes() const noexceptTotal bytes across every arena.