TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.reset_max_memory_cached

tensorplay.cuda.reset_max_memory_cached(device: Any | None = None) → None[source]

Reset the starting point in tracking maximum GPU memory managed by the caching allocator for a given device.

Warning

This function now calls reset_peak_memory_stats(), which resets /all/ peak memory stats.

tensorplay.cuda.reset_max_memory_allocated

Reset the starting point in tracking maximum GPU memory occupied by tensors for a given device.

tensorplay.cuda.reset_peak_host_memory_stats

Reset the “peak” stats tracked by the host memory allocator.

Ask DeepWiki