warp.set_device_allocator# warp.set_device_allocator(device, allocator)[source]# Set the memory allocator for a specific CUDA device. Pass None to restore the built-in allocator. Parameters: device (Device | str | None) – The CUDA device. allocator (Allocator | None) – An Allocator-compatible object, or None. Return type: None
