pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00

History

Tom Ritchford d25e6e623f Fix unused Python variables in test/[a-d]* (#134665 ) Pull Request resolved: https://github.com/pytorch/pytorch/pull/134665 Approved by: https://github.com/albanD		2024-12-13 22:13:12 +00:00
..
debug	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +00:00
experimental
__init__.py
README.md
test_api.py
test_attention.py	[dtensor][cp][experiment] add CP experimental API to choose rotate method (#142093 )	2024-12-10 18:25:23 +00:00
test_common_rules.py
test_convolution_ops.py
test_dtensor.py	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +00:00
test_dtensor_compile.py	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +00:00
test_dtensor_ops.py
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	[DTensor] Support matmul in inference_mode (#142197 )	2024-12-06 07:15:05 +00:00
test_op_strategy.py
test_optimizers.py
test_pointwise_ops.py
test_random_ops.py	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +00:00
test_redistribute.py	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +00:00
test_tensor_ops.py	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +00:00
test_utils.py	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +00:00
test_view_ops.py
test_xla_integration.py	Fix unused Python variables in test/[a-d]* (#134665 )	2024-12-13 22:13:12 +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