3 people 97cfa65df7 Add pipeline parallel support to TransformersModel (#12832) ...

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Signed-off-by: Isotr0py <2037008807@qq.com>
Co-authored-by: Isotr0py <2037008807@qq.com>

2025-03-25 10:41:45 +08:00

..

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

test_ca_buffer_sharing.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

test_comm_ops.py

[Misc] Replace os environ to monkeypatch in test suite (#14516)

2025-03-16 20:35:57 -07:00

test_custom_all_reduce.py

[Misc] Replace os environ to monkeypatch in test suite (#14516)

2025-03-16 20:35:57 -07:00

test_distributed_oot.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

test_expert_parallel.py

[Bugfix][ROCm] running new process using spawn method for rocm in tests. (#14810)

2025-03-17 11:33:35 +00:00

test_multi_node_assignment.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

test_pipeline_parallel.py

Add pipeline parallel support to TransformersModel (#12832)

2025-03-25 10:41:45 +08:00

test_pipeline_partition.py

[Misc] Replace os environ to monkeypatch in test suite (#14516)

2025-03-16 20:35:57 -07:00

test_pp_cudagraph.py

[Bugfix][ROCm] running new process using spawn method for rocm in tests. (#14810)

2025-03-17 11:33:35 +00:00

test_pynccl.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

test_same_node.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

test_shm_broadcast.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

test_torchrun_example.py

[ci] feat: make the test_torchrun_example run with tp=2, external_dp=2 (#15172)

2025-03-21 00:30:04 +08:00

test_utils.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00