Trevor Morris a8593237c0

Add pynccl all-gatherv and reducescatterv (#20154) ...

Signed-off-by: Trevor Morris <tmorris@nvidia.com>
Signed-off-by: mgoin <mgoin64@gmail.com>
Co-authored-by: mgoin <mgoin64@gmail.com>

2025-07-11 18:59:23 -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

feat: add data parallel rank to KVEventBatch (#18925)

2025-06-03 17:14:20 -07:00

test_ca_buffer_sharing.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_comm_ops.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_custom_all_reduce.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_distributed_oot.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_eplb_algo.py

[Feature] Expert Parallelism Load Balancer (EPLB) (#18343)

2025-06-26 15:30:21 -07:00

test_eplb_execute.py

[Feature] Expert Parallelism Load Balancer (EPLB) (#18343)

2025-06-26 15:30:21 -07:00

test_events.py

feat: add data parallel rank to KVEventBatch (#18925)

2025-06-03 17:14:20 -07:00

test_expert_parallel.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_multi_node_assignment.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_node_count.py

[Misc] Add parallel state node_count function (#20045)

2025-06-25 13:38:53 -07:00

test_pipeline_parallel.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_pipeline_partition.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_pp_cudagraph.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_pynccl.py

Add pynccl all-gatherv and reducescatterv (#20154)

2025-07-11 18:59:23 -07:00

test_quick_all_reduce.py

[Feature] add quick all reduce (#19744)

2025-06-26 20:54:24 -07:00

test_same_node.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_sequence_parallel.py

[Feature] Support sequence parallelism for static fp8 quantization (#19181)

2025-06-23 16:09:02 -04:00

test_shm_broadcast.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_torchrun_example.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_utils.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00