Back out "Revert D30745921: [DDP] Fix when buffers are reassigned in module" (#64776)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64776
Original commit changeset: 343ead86bf1e
ghstack-source-id: 138326914
Test Plan: ci
Reviewed By: H-Huang
Differential Revision: D30849444
fbshipit-source-id: 9a72805416fe7d6c68e51bdcdb88f6e1fecb614d