pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00

History

Mikayla Gawarecki f3f305ef3e Fix condition for weights_only unpickler for DTensor (#140740 ) Same as #140739 but for DTensor (move safe globals for DTensor to `torch.distributed.tensor.__init__` and update error message to let user know `torch.distributed.tensor` must be imported to load DTensor) Differential Revision: [D65961690](https://our.internmc.facebook.com/intern/diff/D65961690) Pull Request resolved: https://github.com/pytorch/pytorch/pull/140740 Approved by: https://github.com/malfet ghstack dependencies: #140739		2024-11-19 02:44:53 +00:00
..
debug
experimental
__init__.py
README.md
test_api.py
test_attention.py	[CP] Enable CP + compiler tests when there are more than 2 GPUs (#133736 )	2024-11-15 20:42:51 +00:00
test_common_rules.py
test_convolution_ops.py
test_dtensor.py	Fix condition for weights_only unpickler for DTensor (#140740 )	2024-11-19 02:44:53 +00:00
test_dtensor_compile.py	[aotd] coerce_same_metadata_as_tangent with expected_type for e.g.AsyncCollectiveTensor (#139095 )	2024-11-07 16:24:48 +00:00
test_dtensor_ops.py	Revert "Fix unbind_copy and add its decomposition (#134319 )"	2024-10-29 04:54:37 +00:00
test_embedding_ops.py	Replace clone.detach with detach.clone (#140264 )	2024-11-13 07:01:02 +00:00
test_experimental_ops.py
test_init.py
test_math_ops.py
test_matrix_ops.py	[Test][DTensor] Skip test_dtensor_mm if ROCm (#139719 )	2024-11-05 22:56:35 +00:00
test_op_strategy.py
test_optimizers.py
test_pointwise_ops.py
test_random_ops.py
test_redistribute.py
test_tensor_ops.py
test_utils.py
test_view_ops.py
test_xla_integration.py

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