Rate this Page
★ ★ ★ ★ ★

XPUPluggableAllocator#

class torch.xpu.memory.XPUPluggableAllocator(path_to_lib_file, alloc_fn_name, free_fn_name)[source]#

XPU memory allocator loaded from a shared library.

On this page
Show Source
PyTorch Libraries