TensorPlay
latest (dev)
Copy
View Markdown

Latest development documentation · Updated 2026-10-15

tensorplay.cuda.memory_stats_as_nested_dict

tensorplay.cuda.memory_stats_as_nested_dict(device: Any = None) → dict[str, Any][source]

Return the result of memory_stats() as a nested dictionary.

The native allocator reports the fragmentation-aware matrix directly (segments, free-block histogram, pending cross-stream blocks, graph pools, capture state); it is exposed under "allocator" alongside the

On this page

Ask DeepWiki