spdl.io.transfer_buffer

transfer_buffer(buffer: CPUBuffer, *, device_config: CUDAConfig | None = None, **kwargs) CUDABuffer[source]

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.