Skip to content
Artwork for PyTorch Developer Podcast
PyTorch Developer Podcast · Mar 17, 2024 · 15 min

Min-cut partitioner

The min-cut partitioner makes decisions about what to save for backwards when splitting the forward and backwards graph from the joint graph traced by AOTAutograd. Crucially, it doesn't actually do a "split"; instead, it is deciding how much of the joint graph should be used for backwards. I also talk about the backward retracing problem.

0:00-15:56

transcript

No transcript — this publisher did not publish one.

show notes

The min-cut partitioner makes decisions about what to save for backwards when splitting the forward and backwards graph from the joint graph traced by AOTAutograd. Crucially, it doesn't actually do a "split"; instead, it is deciding how much of the joint graph should be used for backwards. I also talk about the backward retracing problem.

more episodes

All episodes