Lucia Fang 3d2779c29a

[Feature] Support Pipeline Parallism in torchrun SPMD offline inference for V1 (#17827) ...

Signed-off-by: Lucia Fang <fanglu@fb.com>

2025-05-15 22:28:27 -07:00

..

__init__.py

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

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

conftest.py

[V1][Metrics] add support for kv event publishing (#16750)

2025-04-30 07:44:45 -07: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

[Feature] support sequence parallelism using compilation pass (#16155)

2025-04-27 06:29:35 -07:00

test_custom_all_reduce.py

[Distributed] Add custom allreduce support for ROCM (#14125)

2025-03-31 22:49:12 -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_events.py

[V1][Metrics] add support for kv event publishing (#16750)

2025-04-30 07:44:45 -07: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

[Misc] replace does not exist model (#18119)

2025-05-14 02:13:47 -07: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_sequence_parallel.py

Improve configs - the rest! (#17562)

2025-05-09 15:18:44 -07:00

test_shm_broadcast.py

Update deprecated Python 3.8 typing (#13971)

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

test_torchrun_example.py

[Feature] Support Pipeline Parallism in torchrun SPMD offline inference for V1 (#17827)

2025-05-15 22:28:27 -07:00

test_utils.py

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

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