torch.xpu.memory.change_current_allocator

torch.xpu.memory.change_current_allocator(allocator) [исходный код]

Изменяет используемый в данный момент распределитель памяти на указанный.

Примечание

Если текущий распределитель уже использовался или был инициализирован, эта функция вызовет ошибку.

Параметры:

allocator (torch.xpu.memory._XPUAllocator) — распределитель, который нужно установить в качестве активного.