TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.reset_peak_memory_stats

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

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

Peak stats correspond to the “peak” key in each individual stat dict.

Parameters:

device (tensorplay.Device or int, optional) – selected device. Returns statistic for the current device, given by current_device(), if device is None (default).

tensorplay.cuda.reset_peak_host_memory_stats

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

tensorplay.cuda.set_device

Set the current device.

Ask DeepWiki