mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
Pickling GraphModule needs some special handling for wrapping things that normally can't be pickled - but async compile needs to pass them across a wire so we need to be able to serialize it - add some helpers to enable that. Differential Revision: [D68921318](https://our.internmc.facebook.com/intern/diff/D68921318) Pull Request resolved: https://github.com/pytorch/pytorch/pull/141659 Approved by: https://github.com/jamesjwu |
||
|---|---|---|
| .. | ||
| test_fully_shard_autograd.py | ||
| test_fully_shard_clip_grad_norm_.py | ||
| test_fully_shard_comm.py | ||
| test_fully_shard_compile.py | ||
| test_fully_shard_extensions.py | ||
| test_fully_shard_frozen.py | ||
| test_fully_shard_grad_scaler.py | ||
| test_fully_shard_init.py | ||
| test_fully_shard_logging.py | ||
| test_fully_shard_memory.py | ||
| test_fully_shard_mixed_precision.py | ||
| test_fully_shard_overlap.py | ||
| test_fully_shard_state.py | ||
| test_fully_shard_state_dict.py | ||
| test_fully_shard_training.py | ||