pytorch
d14fc0bc - [FSDP] Remove `device` arg from `_cast_buffers()` (#87933)

Commit
3 years ago
[FSDP] Remove `device` arg from `_cast_buffers()` (#87933) This PR is easy. The `device` argument in `_cast_buffers()` is never used. Pull Request resolved: https://github.com/pytorch/pytorch/pull/87933 Approved by: https://github.com/mrshenli
Author
Andrew Gu
Committer
Parents
Loading