saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/distributed/_tensor
History
Wanchao Liang f026b32008 [device_mesh][BE] reduce_scatter fallback to funcol and remove from DM (#105642) 
For the reason similar to https://github.com/pytorch/pytorch/pull/105605
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105642
Approved by: https://github.com/kumpera, https://github.com/wz337, https://github.com/fduwjj
2023-07-27 01:33:05 +00:00
..
debug [BE] Enable ruff's UP rules and autoformat distributed/ (#105433) 2023-07-19 14:27:11 +00:00
__init__.py
README.md
test_api.py [dtensor] canonicalize detach callsites and use view_as when appropriate (#105239) 2023-07-18 17:13:37 +00:00
test_basic_strategy.py
test_common_rules.py
test_device_mesh.py [device_mesh][BE] reduce_scatter fallback to funcol and remove from DM (#105642) 2023-07-27 01:33:05 +00:00
test_dtensor.py [dynamo] add initial dynamo support for DTensor (#103146) 2023-07-19 16:01:12 +00:00
test_dtensor_ops.py Skip dtensor ops on CPU-only runner due to flaky timeout (#98868) 2023-04-12 03:40:02 +00:00
test_embedding_ops.py [PTD][TP] Add BWD support for colwise embedding sharding (#104820) 2023-07-10 22:33:20 +00:00
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 [TP] Enable partial tensor add without redistribute (#105939) 2023-07-26 03:12:39 +00:00
test_random_ops.py [device_mesh][BE] remove allgather from DM (#105614) 2023-07-27 01:33:05 +00:00
test_redistribute.py [dtensor] canonicalize detach callsites and use view_as when appropriate (#105239) 2023-07-18 17:13:37 +00:00
test_tensor_ops.py [BE][DTensor] fix DTensor equal op (#99014) 2023-04-18 03:22:44 +00:00
test_utils.py [DTensor] Change Sharding algorithm to be in line with `torch.chunk()` (#98722) 2023-04-21 02:05:22 +00:00
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