Harry Mellor 145944cb94

Improve pipeline partitioning (#13839)

2025-02-25 18:53:56 -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

[BugFix] Pop instead of del CUDA_VISIBLE_DEVICES (#12962)

2025-02-12 00:21:50 +08:00

test_custom_all_reduce.py

[BugFix] Pop instead of del CUDA_VISIBLE_DEVICES (#12962)

2025-02-12 00:21:50 +08: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

Expert Parallelism (EP) Support for DeepSeek V2 (#12583)

2025-02-24 07:33:20 -08: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

[Bugfix] Fix 2 Node and Spec Decode tests (#13341)

2025-02-16 22:20:22 +08:00

test_pipeline_partition.py

Improve pipeline partitioning (#13839)

2025-02-25 18:53:56 -08:00

test_pp_cudagraph.py

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

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

test_pynccl.py

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

2025-02-02 11:58:18 -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

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

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

test_torchrun_example.py

[v1] torchrun compatibility (#13642)

2025-02-23 22:47:24 +08:00

test_utils.py

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

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