spdl.io.transfer_buffer

transfer_buffer(buffer: CPUBuffer, *, device_config: CUDAConfig | None = None, **kwargs) → CUDABuffer

Move the given CPU buffer to CUDA device.

Parameters:

• buffer – Source data.

• device_config – Target CUDA device configuration.

Returns:

Buffer data on the target GPU device.