mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
bb67a28738
Enable Adamax foreach optimizer and add DTensor unit test for Adamax. Pull Request resolved: https://github.com/pytorch/pytorch/pull/119850 Approved by: https://github.com/wanchaol |
||
|---|---|---|
| .. | ||
| debug | ||
| experimental | ||
| __init__.py | ||
| README.md | ||
| test_api.py | ||
| test_common_rules.py | ||
| test_convolution_ops.py | ||
| test_dtensor.py | ||
| test_dtensor_compile.py | ||
| test_dtensor_ops.py | ||
| test_embedding_ops.py | ||
| test_experimental_ops.py | ||
| test_init.py | ||
| test_math_ops.py | ||
| test_matrix_ops.py | ||
| 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 | ||
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