Name Date Size #Lines LOC

..--

debug/H25-Apr-2025-634475

experimental/H25-Apr-2025-626480

README.mdH A D25-Apr-2025270 126

__init__.pyH A D25-Apr-202517 20

test_api.pyH A D25-Apr-202512.7 KiB319245

test_attention.pyH A D25-Apr-202510.3 KiB316270

test_common_rules.pyH A D25-Apr-202516.6 KiB418333

test_convolution_ops.pyH A D25-Apr-20257 KiB187155

test_dtensor.pyH A D25-Apr-202534.9 KiB887691

test_dtensor_compile.pyH A D25-Apr-202531.9 KiB927703

test_dtensor_ops.pyH A D25-Apr-202523.2 KiB679585

test_embedding_ops.pyH A D25-Apr-20258 KiB233169

test_experimental_ops.pyH A D25-Apr-20257.7 KiB193148

test_init.pyH A D25-Apr-20259.5 KiB255205

test_math_ops.pyH A D25-Apr-202526.5 KiB654525

test_matrix_ops.pyH A D25-Apr-202513.8 KiB345272

test_op_strategy.pyH A D25-Apr-202512.8 KiB344253

test_optimizers.pyH A D25-Apr-202520.4 KiB612506

test_pointwise_ops.pyH A D25-Apr-20258.3 KiB283234

test_random_ops.pyH A D25-Apr-202514.4 KiB352255

test_redistribute.pyH A D25-Apr-202522.4 KiB542416

test_tensor_ops.pyH A D25-Apr-202527.9 KiB661530

test_utils.pyH A D25-Apr-202522.4 KiB562433

test_view_ops.pyH A D25-Apr-202518.7 KiB577496

test_xla_integration.pyH A D25-Apr-20256.5 KiB168132

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