pytorch

dc9c79d3 - Allow each fully_shard unit to cast foward inputs for mixed precision config (#100290)

Allow each fully_shard unit to cast foward inputs for mixed precision config (#100290) Pull Request resolved: https://github.com/pytorch/pytorch/pull/100290 Approved by: https://github.com/rohan-varma