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
Wanchao Liang 29674b8e1d [dtensor] fix dtensor _to_copy op for mix precision (#116426) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/116426
Approved by: https://github.com/fduwjj
2024-01-03 07:29:08 +00:00
..
debug [dtensor] add CommDebugMode for debugging (#113592) 2023-11-27 02:40:28 +00:00
experimental Add call to run_tests for a few tests (#115097) 2023-12-07 08:27:40 +00:00
__init__.py
README.md
test_api.py [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
test_common_rules.py
test_convolution_ops.py
test_dtensor.py [dtensor] add a implicit replication flag (#115297) 2023-12-12 03:56:48 +00:00
test_dtensor_compile.py [dtensor] turn back on symbolic shape in tests (#115568) 2023-12-12 00:26:23 +00:00
test_dtensor_ops.py [DTensor] add layer norm backward support (#115683) 2023-12-28 01:10:10 +00:00
test_embedding_ops.py [TP] fully rewrite Tensor Parallel APIs (#114732) 2023-12-02 08:18:12 +00:00
test_experimental_ops.py
test_init.py [dtensor] fix empty shape init for dtensor constructors (#115091) 2023-12-05 00:51:29 +00:00
test_math_ops.py [DTensor] add layer norm backward support (#115683) 2023-12-28 01:10:10 +00:00
test_matrix_ops.py
test_op_strategy.py
test_optimizers.py [dtensor] enable radam foreach optimizer (#115566) 2023-12-12 03:57:00 +00:00
test_pointwise_ops.py
test_random_ops.py
test_redistribute.py [DTensor] Make DTensor from_local backward partial() to replicate() pass through (#115967) 2023-12-19 00:16:10 +00:00
test_tensor_ops.py [dtensor] fix dtensor _to_copy op for mix precision (#116426) 2024-01-03 07:29:08 +00:00
test_utils.py [DeviceMesh] Rename _device_mesh.py to device_mesh.py to prepare for beta (#115099) (#115193) 2023-12-08 08:44:32 +00:00
test_view_ops.py [dtensor] refactor some existing test util to use comm mode (#114404) 2023-11-27 06:43:09 +00:00
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