pytorch/test/distributed/tensor at main · pytorch/pytorch · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
debug		debug
experimental		experimental
parallel		parallel
README.md		README.md
__init__.py		__init__.py
test_api.py		test_api.py
test_attention.py		test_attention.py
test_common_rules.py		test_common_rules.py
test_convolution_ops.py		test_convolution_ops.py
test_dtensor.py		test_dtensor.py
test_dtensor_compile.py		test_dtensor_compile.py
test_dtensor_ops.py		test_dtensor_ops.py
test_embedding_ops.py		test_embedding_ops.py
test_experimental_ops.py		test_experimental_ops.py
test_init.py		test_init.py
test_math_ops.py		test_math_ops.py
test_matrix_ops.py		test_matrix_ops.py
test_op_strategy.py		test_op_strategy.py
test_optimizers.py		test_optimizers.py
test_pointwise_ops.py		test_pointwise_ops.py
test_random_ops.py		test_random_ops.py
test_redistribute.py		test_redistribute.py
test_tensor_ops.py		test_tensor_ops.py
test_utils.py		test_utils.py
test_view_ops.py		test_view_ops.py
test_xla_integration.py		test_xla_integration.py

README.md

Run distributed tensor tests:

from root, run (either CPU or GPU)

pytest test/distributed/tensor/test_dtensor.py

run specific test cases and print stdout/stderr:

pytest test/distributed/tensor/test_dtensor.py -s -k test_from_local