pytorch
6c605e9c - [FSDP] Skip param check for pure FP16 (#90785)

Commit
3 years ago
[FSDP] Skip param check for pure FP16 (#90785) Pull Request resolved: https://github.com/pytorch/pytorch/pull/90785 Approved by: https://github.com/rohan-varma, https://github.com/fegin
Author
Andrew Gu
Committer
Parents
Loading