tree: 74dc3c31c870c491f6fb89360121da55c851448c [path history] [tgz]
  1. debug/
  2. __init__.py
  3. README.md
  4. test_api.py
  5. test_basic_strategy.py
  6. test_common_rules.py
  7. test_device_mesh.py
  8. test_dtensor.py
  9. test_dtensor_compile.py
  10. test_dtensor_ops.py
  11. test_embedding_ops.py
  12. test_init.py
  13. test_math_ops.py
  14. test_matrix_ops.py
  15. test_pointwise_ops.py
  16. test_random_ops.py
  17. test_redistribute.py
  18. test_tensor_ops.py
  19. test_utils.py
  20. test_view_ops.py
test/distributed/_tensor/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