XPUPluggableAllocator

class torch.xpu.memory.XPUPluggableAllocator(path_to_lib_file, alloc_fn_name, free_fn_name) [исходный код]

Аллокатор памяти XPU, динамически загружаемый из общей библиотеки.

Это позволяет пользователям предоставлять пользовательские функции выделения и освобождения памяти, реализованные в отдельной общей библиотеке. Аллокатор регистрируется и может стать доступным для использования через change_current_allocator().

Параметры:

• path_to_lib_file (str) – Путь в файловой системе к файлу общей библиотеки, содержащему функции выделения и освобождения памяти.

• alloc_fn_name (str) –

  Имя функции выделения памяти, экспортируемой из общей библиотеки. Функция должна иметь сигнатуру:

  void* alloc_fn(size_t size, int device, sycl::queue* queue);

• free_fn_name (str) –

  Имя функции освобождения памяти, экспортируемой из общей библиотеки. Функция должна иметь сигнатуру:

  void free_fn(void* ptr, size_t size, int device, sycl::queue* queue);