MemPool

class torch.xpu.memory.MemPool(*args, **kwargs) [исходный код]

MemPool представляет собой пул памяти в кэширующем аллокаторе. В настоящее время это просто ID объекта пула, поддерживаемого XPUCachingAllocator.

Параметры:

• allocator (torch._C._xpu_XPUAllocator, optional) – объект torch._C._xpu_XPUAllocator, который можно использовать для определения способа выделения памяти в пуле. Если allocator равен None (по умолчанию), выделение памяти следует стандартной/текущей конфигурации XPUCachingAllocator.
• use_on_oom (bool) – логическое значение, указывающее, можно ли использовать этот пул в крайнем случае, если выделение памяти за пределами пула завершится ошибкой из-за нехватки памяти. По умолчанию это False.

property allocator: _xpu_XPUAllocator | None

Возвращает аллокатор, которому MemPool направляет запросы на выделение памяти.

property id: tuple[int, int]

Возвращает ID этого пула в виде кортежа из двух целых чисел.

snapshot() [исходный код]

Возвращает снимок состояния пулов аллокатора памяти XPU на всех устройствах.

Для интерпретации результата этой функции необходимо знать внутреннее устройство аллокатора памяти.

use_count() [исходный код]

Возвращает количество ссылок на этот пул.

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

int