saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/distributed/_composable
History
Aaron Orenstein 57d8278ab9 pickler for GraphModule (#141659) 
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
2025-01-31 05:34:28 +00:00
..
fsdp pickler for GraphModule (#141659) 2025-01-31 05:34:28 +00:00
test_composability PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_checkpoint.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_contract.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_replicate.py
test_replicate_with_compiler.py