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
Oguz Ulgen 920f0426ae Add None return type to init -- tests rest (#132376) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/132376
Approved by: https://github.com/jamesjwu
ghstack dependencies: #132335, #132351, #132352
2024-08-01 15:44:51 +00:00
..
debug Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
experimental Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
__init__.py
README.md
test_api.py
test_attention.py
test_common_rules.py [dtensor] move ops to private (#131211) 2024-07-25 20:59:55 +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 [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_compile.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_dtensor_ops.py [DTensor] Added naive support for nn.init.orthogonal_ (#132104) 2024-07-30 21:55:09 +00:00
test_embedding_ops.py fix _MaskPartial when multiple embeddings coexist (#131264) 2024-07-29 00:40:58 +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 [debug][dtensor] fixed updating current module (#130995) 2024-07-20 20:57:29 +00:00
test_matrix_ops.py Grouped Query Attention (#128898) 2024-07-31 22:58:51 +00:00
test_op_strategy.py [dtensor] move ops to private (#131211) 2024-07-25 20:59:55 +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 [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_redistribute.py [DTensor] Fix shard_dim_alltoall fake tensor return (#129945) 2024-07-18 16:43:40 +00:00
test_tensor_ops.py [dtensor] move ops to private (#131211) 2024-07-25 20:59:55 +00:00
test_utils.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_view_ops.py [dtensor] move ops to private (#131211) 2024-07-25 20:59:55 +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