katcbf_vlbi_resample.cupy_bridge module

Convert streams to and from cupy.

class katcbf_vlbi_resample.cupy_bridge.AsCupy(input_data: Stream[DataArray])

Bases: ChunkwiseStream[DataArray, DataArray]

Transfer a stream from numpy to cupy.

The transfer is enqueued to the current CUDA stream, but this does not block on the transfer completing.

class katcbf_vlbi_resample.cupy_bridge.AsNumpy(input_data: Stream[DataArray], queue_depth: int = 1)

Bases: object

Transfer a stream from cupy to numpy.