Reland Fix reference cycle in sparse coalesce graph (#55404)
Summary:
Reland of https://github.com/pytorch/pytorch/pull/52874
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55404
Reviewed By: bdhirsh
Differential Revision: D27600438
Pulled By: albanD
fbshipit-source-id: f5c286638b324ad59be65657a016028af5e2b303
