torch.xpu.memory.set_per_process_memory_fraction

torch.xpu.memory.set_per_process_memory_fraction(fraction, device=None) [источник]

Задаёт долю памяти для одного процесса на устройстве XPU. Эта функция ограничивает объём памяти, который кэширующий распределитель может выделить на указанном устройстве XPU. Допустимый объём памяти вычисляется следующим образом:

$$\text{allowed\_memory} = \text{total\_memory} \times \text{fraction}$$

Если процесс попытается выделить больше этого допустимого объёма памяти, распределитель вызовет ошибку нехватки памяти.

Параметры:

• fraction (float) – Диапазон: 0~1. Допустимый объём памяти равен total_memory * fraction.
• device (torch.device или int или str, необязательно) – выбранное устройство. Если device равно None (значение по умолчанию), используется текущее устройство, определяемое функцией current_device().

Примечание

Как правило, общий объём доступной свободной памяти меньше общей ёмкости.