def get_device_to_host_uva_stream():
    """The device to host copy stream to be used for pipeline parallelism."""
    if not hasattr(get_device_to_host_uva_stream, "stream"):
        get_device_to_host_uva_stream.stream = torch.cuda.Stream(priority=-1)
    return get_device_to_host_uva_stream.stream
The device to host copy stream to be used for pipeline parallelism.
python/dgl/graphbolt/base.py