[DDP] Remove SPMD from self.modules_buffers (#64474)
authorRohan Varma <rvarm1@fb.com>
Thu, 9 Sep 2021 02:13:33 +0000 (19:13 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 9 Sep 2021 02:16:15 +0000 (19:16 -0700)
commit15532595209d2daf34d35e10f8d3d3b64966aea2
treec09550a2d203128351fa34483686c02273acdc5f
parent8c095102948c9601792a884dad56da5085c51bee
[DDP] Remove SPMD from self.modules_buffers (#64474)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64474

No need for a nested list here.
ghstack-source-id: 137526312

Test Plan: CI

Reviewed By: zhaojuanmao

Differential Revision: D30745960

fbshipit-source-id: 66a8f9847e9fe1e02c51b79647e93bf7665cf4d9
torch/nn/parallel/distributed.py