trunk/3e4e742a98ad43267eb5c5d128b53a9df9386d13: [Pipeline] Release forward outputs after sends complete (#196963)
- PyTorch: 1680 events in the last 90 days
- PyTorch: 1661th Release in the last 90 days
- Previous: earlier the same day · trunk/198a30931d1efa059f40f09e816e393b4a60887e
What happened
Store a GradientEdge for each releasable, grad-requiring pipeline output so backward does not require the original tensor object. Track one pending batched send per output slot. When an existing schedule wait completes that send, release the send operations and the stage's output reference. Keep unsupported output types, DDP, custom split backward, and last-stage outputs on their tensor-backed ownership paths. Forwa…
Summary assembled by rule from the sources below