pytorch/test/distributed/tensor
2025-01-25 00:58:03 +00:00
..
debug
experimental
parallel
__init__.py
README.md
test_api.py
test_attention.py
test_common_rules.py
test_convolution_ops.py
test_dtensor.py
test_dtensor_compile.py
test_dtensor_ops.py Remove det_singular OpInfo (#145533) 2025-01-25 00:58:03 +00:00
test_embedding_ops.py
test_experimental_ops.py
test_init.py
test_math_ops.py
test_matrix_ops.py
test_op_strategy.py
test_optimizers.py
test_pointwise_ops.py
test_random_ops.py
test_redistribute.py
test_tensor_ops.py
test_utils.py
test_view_ops.py
test_xla_integration.py

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