torch.accelerator.memory.max_memory_allocated

torch.accelerator.memory.max_memory_allocated(device_index=None, /)

Return the current accelerator maximum device memory occupied by tensors in bytes for a given device index.

By default, this returns the peak allocated memory since the beginning of this program. reset_peak_memory_stats() can be used to reset the starting point in tracking this metric.

Parameters

device_index (torch.device, str, int, optional) – the index of the device to target. If not given, use torch.accelerator.current_device_index() by default. If a torch.device or str is provided, its type must match the current accelerator device type.

Return type

int