pytorch

09621148 - Fix 'fully_shard' may determine compute device incorrectly (#98831)

Fix 'fully_shard' may determine compute device incorrectly (#98831) Fixes #98829 Pull Request resolved: https://github.com/pytorch/pytorch/pull/98831 Approved by: https://github.com/awgu