mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Fixes #ISSUE_NUMBER Try to unify the device checking in common_dtensor (testing module) by accelerator API Pull Request resolved: https://github.com/pytorch/pytorch/pull/163498 Approved by: https://github.com/albanD, https://github.com/H-Huang |
||
|---|---|---|
| .. | ||
| _shard | ||
| _tensor | ||
| nn | ||
| rpc | ||
| __init__.py | ||
| checkpoint_utils.py | ||
| common_state_dict.py | ||
| ddp_under_dist_autograd_test.py | ||
| distributed_test.py | ||
| distributed_utils.py | ||
| fake_pg.py | ||
| multi_threaded_pg.py | ||
| rpc_utils.py | ||