Home
last modified time | relevance | path

Searched refs:m_total_allocated_memory (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/mps/
DMPSAllocator.mm116 m_total_allocated_memory += heap_block->size.total;
161 …eap: " << format_size(heap->size.available) << ", total: " << format_size(m_total_allocated_memory)
274 format_size(m_total_allocated_memory),
276 format_size(current_allocated_size() - m_total_allocated_memory),
287 format_size(m_total_allocated_memory),
289 format_size(current_allocated_size() - m_total_allocated_memory),
352 m_total_allocated_memory -= heap_block->size.total;
435 …<< "Attempting to release cached buffers (MPS allocated: " << format_size(m_total_allocated_memory)
436 … other allocations: " << format_size(current_allocated_size() - m_total_allocated_memory) << ")\n";
DMPSAllocator.h305 size_t getTotalAllocatedMemory() const { return m_total_allocated_memory; } in getTotalAllocatedMemory()
338 size_t m_total_allocated_memory = 0; variable