OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
3058700f7f
pytorch/test/distributed/_tensor
History
Wanchao Liang 1c97808f81 [dtensor] support lt/gt op (#110585) 
This PR enables lt/gt aten op
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110585
Approved by: https://github.com/fduwjj
ghstack dependencies: #110584
2023-10-07 00:06:36 +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 [dtensor] add placement strategy and einsum strategy (#98227) 2023-04-05 17:09:32 +00:00
test_common_rules.py [dtensor][7/n] remove reduction rule (#109144) 2023-09-26 22:24:50 +00:00
test_device_mesh.py Hide the contiguous requirement for user input mesh when initializing DeviceMesh (#110628) 2023-10-06 23:54:13 +00:00
test_dtensor_compile.py [dynamo] fix dynamo + DTensor to work with 2d (#108329) 2023-08-31 22:46:26 +00:00
test_dtensor_ops.py [dtensor] support lt/gt op (#110585) 2023-10-07 00:06:36 +00:00
test_dtensor.py [dtensor] add grad placements kwarg to to_local API (#110629) 2023-10-05 21:34:01 +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 [dtensor] ufmt test/distributed/_tensor (#89968) 2022-12-01 20:58:15 +00:00
test_matrix_ops.py [dtensor] ufmt test/distributed/_tensor (#89968) 2022-12-01 20:58:15 +00:00
test_pointwise_ops.py [TP] Enable partial tensor add without redistribute (#105939) 2023-07-26 03:12:39 +00:00
test_random_ops.py [DTensor] Replace usage of compute_local_offset by compute_local_shape_and_global_offset (#108547) 2023-09-06 04:53:44 +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 [dtensor] support aten.where and enable implicit scalar promotion (#110584) 2023-10-07 00:06:36 +00:00
test_utils.py [DTensor] Replace usage of compute_local_offset by compute_local_shape_and_global_offset (#108547) 2023-09-06 04:53:44 +00:00
test_view_ops.py [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00

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