saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/distributed/_tensor
History
Chien-Chin Huang 60baeee59f [BE] Skip the test if CUDA is not available (#128885) 
As title

Differential Revision: [D58690210](https://our.internmc.facebook.com/intern/diff/D58690210/)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/128885
Approved by: https://github.com/wz337
2024-06-18 07:02:44 +00:00
..
debug [BE] Skip the test if CUDA is not available (#128885) 2024-06-18 07:02:44 +00:00
experimental [dtensor] local_map UX change: keep func signature and be compatible with Tensor input (#126924) 2024-06-03 08:41:59 +00:00
__init__.py
README.md
test_api.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
test_attention.py Add Efficient Attention support on ROCM (#124885) 2024-06-08 22:41:05 +00:00
test_common_rules.py [dtensor] move some modules to private namespace (#127339) 2024-05-29 05:18:47 +00:00
test_convolution_ops.py
test_dtensor.py [dtensor] directly return local_tensor under no_grad (#128145) 2024-06-07 04:01:47 +00:00
test_dtensor_compile.py [dtensor] reuse DTensorSpec as much as possible (#128112) 2024-06-06 16:55:50 +00:00
test_dtensor_ops.py Add OpInfo entry for as_strided_copy (#127231) 2024-06-13 13:58:47 +00:00
test_embedding_ops.py
test_experimental_ops.py
test_init.py
test_math_ops.py [dtensor] rename _Partial -> Partial for all imports (#127420) 2024-05-29 21:42:40 +00:00
test_matrix_ops.py [dtensor] rename _Partial -> Partial for all imports (#127420) 2024-05-29 21:42:40 +00:00
test_op_strategy.py [dtensor] rename _Partial -> Partial for all imports (#127420) 2024-05-29 21:42:40 +00:00
test_optimizers.py [DTensor][Optim] Add support for fused_adam and fused_adamw when lr is a tensor (#126750) 2024-05-21 21:38:05 +00:00
test_pointwise_ops.py [dtensor] rename _Partial -> Partial for all imports (#127420) 2024-05-29 21:42:40 +00:00
test_random_ops.py
test_redistribute.py [dtensor] rename _Partial -> Partial for all imports (#127420) 2024-05-29 21:42:40 +00:00
test_tensor_ops.py [dtensor] implement scatter op with simple replication (#126713) 2024-06-03 16:16:28 +00:00
test_utils.py [dtensor] reuse DTensorSpec as much as possible (#128112) 2024-06-06 16:55:50 +00:00
test_view_ops.py
test_xla_integration.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00

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