PAOFLOW.transport.utils.memusage#

Classes#

MemoryTracker

Tracks memory usage for different components of the transport calculation.

Module Contents#

class PAOFLOW.transport.utils.memusage.MemoryTracker[source]#

Tracks memory usage for different components of the transport calculation.

sections: Dict[str, Dict][source]#
register_section(name, usage_func, is_allocated)[source]#

Register a memory-tracked section.

Parameters:
  • name (str) – Section name, e.g., “smearing”.

  • usage_func (Callable[[], float]) – Function returning memory usage in MB.

  • is_allocated (bool) – Whether the section is currently allocated.

report(include_real_memory=False)[source]#

Generate the memory usage report.

Parameters:

include_real_memory (bool) – Whether to include system-level memory usage.

Returns:

`report` – Formatted memory usage summary.

Return type:

str