torch.xpu.memory.max_memory_allocated

torch.xpu.memory.max_memory_allocated(device=None) [исходный код]

Возвращает максимальный объем видеопамяти в байтах, занятый тензорами на указанном устройстве.

По умолчанию возвращается пиковый объем выделенной памяти с начала выполнения программы. reset_peak_memory_stats() можно использовать, чтобы сбросить начальную точку отслеживания этой метрики. Например, с помощью этих двух функций можно измерять пиковый объем выделенной памяти на каждой итерации цикла обучения.

Параметры:

device (torch.device или int или str, необязательно) – выбранное устройство. Возвращает статистику для текущего устройства, указанного функцией current_device(), если device имеет значение None (по умолчанию).

Тип возвращаемого значения:

int