torch.xpu.memory.memory_snapshot

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

Возвращает снимок состояния распределителя памяти XPU на всех устройствах. Предоставляет подробную информацию о каждом сегменте памяти, управляемом распределителем, включая его размер, принадлежащий ему пул, связанный поток, трассировки стека вызовов и другие соответствующие атрибуты.

Параметры:

mempool_id (tuple[int, int] or None, optional) – Идентификатор пула памяти. Если значение равно None, используется пул памяти по умолчанию.

Возвращает:

Список сегментов памяти и их атрибутов.

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

list[dict[str, Any]]