tensorplay.cuda.memory_stats

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

Return a dictionary of CUDA memory allocator statistics for a given device.

The return value of this function is a dictionary of statistics, each of which is a non-negative integer. See torch.cuda.memory_stats() for the full key layout; keys that are not tracked by this TensorPlay build are always reported as zero.

Parameters:

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