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/_tensor
History
Wanchao Liang 4cc474dec4 [dtensor] support torch.save/load with DTensor (#103106) 
This PR actually enables DTensor to be pickable and add tests to test
torch.save/load works correctly for DTensor
Pull Request resolved: https://github.com/pytorch/pytorch/pull/103106
Approved by: https://github.com/kumpera
2023-06-09 04:11:15 +00:00
..
debug
__init__.py
README.md
test_api.py Allow ORT backend for DTensor (#101914) 2023-06-01 22:37:09 +00:00
test_basic_strategy.py
test_common_rules.py
test_device_mesh.py DeviceMesh use dispatchable PG to support custom backend (#102336) 2023-05-30 19:22:37 +00:00
test_dtensor.py [dtensor] support torch.save/load with DTensor (#103106) 2023-06-09 04:11:15 +00:00
test_dtensor_ops.py
test_embedding_ops.py
test_init.py [reland][DTensor][3/N] add DTensor constructor function: full (#101436) (#103165) 2023-06-08 16:18:33 +00:00
test_math_ops.py
test_matrix_ops.py
test_pointwise_ops.py
test_random_ops.py
test_redistribute.py
test_tensor_ops.py
test_utils.py
test_view_ops.py

README.md

Run distributed tensor tests:

from root, run (either CPU or GPU)

pytest test/spmd/tensor/test_tensor.py

pytest test/spmd/tensor/test_ddp.py

run specific test case and print stdout/stderr:

pytest test/spmd/tensor/test_tensor.py -s -k test_tensor_from_local