trunk/5ba7f2bf58f6177091cdf890429ffd5f5dc2139a: [CUDA] Add fork/join execution on CUDA streams (#199128)
- PyTorch: 2025 events in the last 90 days
- PyTorch: 1998th Release in the last 90 days
- Previous: earlier the same day · ciflow/inductor/199800: [input_capture] Add an AOTAutograd compiled-graph hook (#199800)
What happened
Authored with an AI assistant. Add torch.cuda.execute_on_streams for fork/join execution with ordinary streams, green-context streams, or mixtures. Queued work is joined and caller state restored even when a callback raises. Same-device execution switches directly between streams and restores the caller once to reduce host-side overheads. Cross-device execution uses stream context managers to restore device and stre…
Summary assembled by rule from the sources below