torch.cuda.get_stream_from_external
- torch.cuda.get_stream_from_external(data_ptr, device=None)[source][source]
返回从外部分配的 CUDA 流中的
Stream。此函数用于将其他库中分配的流包装起来，以便促进数据交换和多库交互。
注意
此函数不管理流的生命周期，用户负责在调用此返回的流期间保持引用的流处于活动状态。
- 参数:
data_ptr (int) – cudaStream_t 值的整型表示，该值是在外部分配的。
device (torch.device 或 int，可选) – 流最初分配的设备。如果 device 指定错误，后续使用此流的启动可能会失败。
- 返回类型:
