trunk/7da85ca47b7aec4f5dc133d856bb8ec5fc5b36c1: Add CUDA stream transport operations (#198918)
- PyTorch 近 90 天出现 1780 次
- PyTorch 近 90 天第 1758 次发版
- 上一次:同一天稍早 · trunk/90769ae442226c087d2faf307a9be834c5c645e0: Add Auto PR Triage in shadow mode (#195778)
发生了什么
Add Transport.read_stream / write_stream , which order a transfer on a CUDA stream (default: current) and return once enqueued. Transfers can be captured with torch.cuda.graph . Ordinary read/write are unchanged. The default implementation works for any host-driven backend; backends with native stream support can override it. A host callback schedules write_async / read_async on a shared asyncio loop thread, and cuS…
摘要按规则整理自下方来源原文