3 people 690fe019f0 [Feature] support sequence parallelism using compilation pass (#16155) ...

Signed-off-by: cascade812 <cascade812@outlook.com>
Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
Co-authored-by: Tyler Michael Smith <tyler@neuralmagic.com>

2025-04-27 06:29:35 -07: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

[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_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

[Bugfix] Fix standard models tests (#17217)

2025-04-26 02:26:41 -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

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

2025-04-27 06:29:35 -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

[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