hailin
/
sglang_v0.5.2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
sglang_v0.5.2/pytorch_2.8.0/test/distributed/tensor
History
hailin c8e8c1e9ff . 2025-09-20 16:09:34 +08:00
..
debug . 2025-09-20 16:09:34 +08:00
experimental . 2025-09-20 16:09:34 +08:00
parallel . 2025-09-20 16:09:34 +08:00
README.md . 2025-09-20 16:09:34 +08:00
__init__.py . 2025-09-20 16:09:34 +08:00
test_api.py . 2025-09-20 16:09:34 +08:00
test_attention.py . 2025-09-20 16:09:34 +08:00
test_common_rules.py . 2025-09-20 16:09:34 +08:00
test_convolution_ops.py . 2025-09-20 16:09:34 +08:00
test_dtensor.py . 2025-09-20 16:09:34 +08:00
test_dtensor_compile.py . 2025-09-20 16:09:34 +08:00
test_dtensor_ops.py . 2025-09-20 16:09:34 +08:00
test_embedding_ops.py . 2025-09-20 16:09:34 +08:00
test_experimental_ops.py . 2025-09-20 16:09:34 +08:00
test_init.py . 2025-09-20 16:09:34 +08:00
test_math_ops.py . 2025-09-20 16:09:34 +08:00
test_matrix_ops.py . 2025-09-20 16:09:34 +08:00
test_op_strategy.py . 2025-09-20 16:09:34 +08:00
test_optimizers.py . 2025-09-20 16:09:34 +08:00
test_pointwise_ops.py . 2025-09-20 16:09:34 +08:00
test_random_ops.py . 2025-09-20 16:09:34 +08:00
test_redistribute.py . 2025-09-20 16:09:34 +08:00
test_tensor_ops.py . 2025-09-20 16:09:34 +08:00
test_utils.py . 2025-09-20 16:09:34 +08:00
test_view_ops.py . 2025-09-20 16:09:34 +08:00
test_xla_integration.py . 2025-09-20 16:09:34 +08:00

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