torch.xpu.memory.get_per_process_memory_fraction

torch.xpu.memory.get_per_process_memory_fraction(device=None) [source]

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

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

Параметры:

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

Возвращает:

Долю памяти в диапазоне от 0.0 до 1.0.

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

float