pytorch

2cfc4cb3 - Add optional recomputable_ops argument for the min cut partitioner (#86686)

Add optional recomputable_ops argument for the min cut partitioner (#86686) `min_cut_rematerialization_partition` has a default set of hard-coded operations that are allowed to be recomputed in the backward pass. This PR adds customization ability to this function allowing users to control the behavior by passing `recomputable_ops` instead of relying on the default setting. Pull Request resolved: https://github.com/pytorch/pytorch/pull/86686 Approved by: https://github.com/Chillee