OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
c0bbda37e8
pytorch/test/distributed/tensor
History
linhaifeng 695cb0d342 [2/N][Fix] Fix typo in test folder (#166374) 
Fix typo in test folder.

_typos.toml
```bash
[default.extend-words]
nd = "nd"
arange = "arange"
Nd = "Nd"
GLOBALs = "GLOBALs"
hte = "hte"
iy = "iy"
PN = "PN"
Dout = "Dout"
optin = "optin"
gam = "gam"
PTD = "PTD"
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166374
Approved by: https://github.com/cyyever, https://github.com/ezyang
2025-10-29 03:02:07 +00:00
..
debug [DebugMode] add nn.Module tracking (#165498) 2025-10-24 05:08:33 +00:00
experimental [dtensor] support local_map as a decorator (#161353) 2025-08-28 01:46:25 +00:00
parallel [async_tp] Support mm+rs with scatter_dim matmul K by sharding B (#162794) 2025-09-25 12:18:39 +00:00
__init__.py [dtensor] move all tests to distribute/tensor folder (#144166) 2025-01-08 00:32:33 +00:00
README.md [dtensor] move all tests to distribute/tensor folder (#144166) 2025-01-08 00:32:33 +00:00
test_api.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_attention.py [2/N][Fix] Fix typo in test folder (#166374) 2025-10-29 03:02:07 +00:00
test_common_rules.py [DTensor] Set up DTensorContinuousTestBase (#159885) 2025-08-06 07:40:31 +00:00
test_convolution_ops.py [DTensor] Extend conv ops to 3D (#165241) 2025-10-14 02:30:46 +00:00
test_dtensor_compile.py [dynamo][guards] 1/N Guard selectively for DTensor (#165824) 2025-10-27 20:35:40 +00:00
test_dtensor_dispatch_overhead.py measure dispatch overhead (#160504) 2025-08-14 06:13:53 +00:00
test_dtensor_export.py Remove global pytree registration for blockmask (#166434) 2025-10-28 23:11:52 +00:00
test_dtensor_ops.py Enable all PIE rules on ruff (#165814) 2025-10-18 07:36:18 +00:00
test_dtensor_testbase.py [DTensor] fix DTensorTestCase.destroy_pg() when device_type is "cpu" but CUDA device is available (#161015) 2025-08-29 12:47:11 +00:00
test_dtensor.py Enable all SIM rules except disabled ones (#164645) 2025-10-17 07:27:11 +00:00
test_dynamic.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_embedding_ops.py Move MaskPartial to placement_types to improve discoverability (#164414) 2025-10-28 21:56:02 +00:00
test_experimental_ops.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_init.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_math_ops.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_matrix_ops.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_op_schema.py Port OpSchema.__post_init__ and OpSchema._recompute_comparison_key to C++ (#161695) 2025-09-19 04:07:30 +00:00
test_op_strategy.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_optimizers.py Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
test_placement_types.py Re-re-re-re-apply "C++-accessible Placements via pybind11 (#163030)" (#166132) 2025-10-27 21:19:32 +00:00
test_pointwise_ops.py [1/N] Apply UP035 rule in tests (#163947) 2025-09-29 01:42:01 +00:00
test_random_ops.py [DTensor] fix uneven _StridedShard (#163843) 2025-09-25 22:12:29 +00:00
test_redistribute.py Enable local tensor model for DTensor redistribute tests (#166081) 2025-10-26 22:21:43 +00:00
test_tensor_ops.py Move MaskPartial to placement_types to improve discoverability (#164414) 2025-10-28 21:56:02 +00:00
test_utils.py [dtensor] add util to compute expected local sizes/strides for even sharding (#164296) 2025-10-10 02:34:27 +00:00
test_view_ops.py Enable local tensor mode on DTensor view ops test (#165596) 2025-10-16 20:52:06 +00:00
test_xla_integration.py [1/N] Apply UP035 rule in tests (#163947) 2025-09-29 01:42:01 +00: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