| Name | Date | Size | #Lines | LOC | ||
|---|---|---|---|---|---|---|
| .. | - | - | ||||
| debug/ | 04-Jul-2025 | - | 634 | 475 | ||
| experimental/ | 04-Jul-2025 | - | 626 | 480 | ||
| README.md | D | 04-Jul-2025 | 270 | 12 | 6 | |
| __init__.py | D | 04-Jul-2025 | 17 | 2 | 0 | |
| test_api.py | D | 04-Jul-2025 | 12.7 KiB | 319 | 245 | |
| test_attention.py | D | 04-Jul-2025 | 10.3 KiB | 316 | 270 | |
| test_common_rules.py | D | 04-Jul-2025 | 16.6 KiB | 418 | 333 | |
| test_convolution_ops.py | D | 04-Jul-2025 | 7 KiB | 187 | 155 | |
| test_dtensor.py | D | 04-Jul-2025 | 34.9 KiB | 887 | 691 | |
| test_dtensor_compile.py | D | 04-Jul-2025 | 31.9 KiB | 927 | 703 | |
| test_dtensor_ops.py | D | 04-Jul-2025 | 23.2 KiB | 679 | 585 | |
| test_embedding_ops.py | D | 04-Jul-2025 | 8 KiB | 233 | 169 | |
| test_experimental_ops.py | D | 04-Jul-2025 | 7.7 KiB | 193 | 148 | |
| test_init.py | D | 04-Jul-2025 | 9.5 KiB | 255 | 205 | |
| test_math_ops.py | D | 04-Jul-2025 | 26.5 KiB | 654 | 525 | |
| test_matrix_ops.py | D | 04-Jul-2025 | 13.8 KiB | 345 | 272 | |
| test_op_strategy.py | D | 04-Jul-2025 | 12.8 KiB | 344 | 253 | |
| test_optimizers.py | D | 04-Jul-2025 | 20.4 KiB | 612 | 506 | |
| test_pointwise_ops.py | D | 04-Jul-2025 | 8.3 KiB | 283 | 234 | |
| test_random_ops.py | D | 04-Jul-2025 | 14.4 KiB | 352 | 255 | |
| test_redistribute.py | D | 04-Jul-2025 | 22.4 KiB | 542 | 416 | |
| test_tensor_ops.py | D | 04-Jul-2025 | 27.9 KiB | 661 | 530 | |
| test_utils.py | D | 04-Jul-2025 | 22.4 KiB | 562 | 433 | |
| test_view_ops.py | D | 04-Jul-2025 | 18.7 KiB | 577 | 496 | |
| test_xla_integration.py | D | 04-Jul-2025 | 6.5 KiB | 168 | 132 |
README.md
1## Run distributed tensor tests: 2 3from root, run (either CPU or GPU) 4 5`pytest test/spmd/tensor/test_tensor.py` 6 7`pytest test/spmd/tensor/test_ddp.py` 8 9run specific test case and print stdout/stderr: 10 11`pytest test/spmd/tensor/test_tensor.py -s -k test_tensor_from_local` 12