torch.xpu.memory.reset_accumulated_memory_stats

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

Сбрасывает «накопленную» (историческую) статистику, отслеживаемую распределителем памяти XPU.

Подробности см. в memory_stats(). Накопленная статистика соответствует ключам “allocated” и “freed” в каждом отдельном словаре статистики.

Параметры:

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