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
Tom Ritchford 8aedc649bd Fix unbind_copy and add its decomposition (#134319) 
* Fixes https://github.com/pytorch/pytorch/issues/130829

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134319
Approved by: https://github.com/amjames, https://github.com/eellison
2024-10-23 19:13:44 +00:00
..
debug [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
experimental [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
__init__.py
README.md
test_api.py
test_attention.py [CP] Implement AllGather based context parallelism (#132820) 2024-10-22 05:25:50 +00:00
test_common_rules.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_convolution_ops.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_dtensor.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_dtensor_compile.py Add AsyncCollectiveTensor isinstance check to test_graph_input_is_async (#137253) 2024-10-09 08:06:16 +00:00
test_dtensor_ops.py Fix unbind_copy and add its decomposition (#134319) 2024-10-23 19:13:44 +00:00
test_embedding_ops.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_experimental_ops.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_init.py
test_math_ops.py [DTensor] Register replication strategy for a few upsampling interpolate ops (#137201) 2024-10-03 03:45:39 +00:00
test_matrix_ops.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_op_strategy.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_optimizers.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_pointwise_ops.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_random_ops.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_redistribute.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_tensor_ops.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_utils.py Refactor offset logic and work for nD (#135861) 2024-09-27 06:13:06 +00:00
test_view_ops.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_xla_integration.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +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