Ilya Markov 0313cf854d

[PERF] PyTorch Symmetric Memory All-Reduce (#20759) ...

Signed-off-by: ilmarkov <imarkov@redhat.com>
Signed-off-by: ilmarkov <markovilya197@gmail.com>
Signed-off-by: Michael Goin <mgoin64@gmail.com>
Co-authored-by: ilmarkov <imarkov@redhat.com>
Co-authored-by: Michael Goin <mgoin64@gmail.com>

2025-08-22 15:39:08 -06: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] Begin deprecation of get_tensor_model_*_group (#22494)

2025-08-08 01:11:54 -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

[Deprecation][2/N] Replace --task with --runner and --convert (#21470)

2025-07-27 19:42:40 -07:00

test_kvlayout.py

feat(distributed): add get_required_kvcache_layout class method to kv connector api (#20433)

2025-07-30 16:41:51 +00: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

[Model] Add Ovis2.5 PP support (#23405)

2025-08-22 17:46:34 +00:00

test_pipeline_partition.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_pp_cudagraph.py

[V0 Deprecation] Remove V0 FlashInfer attention backend (#22776)

2025-08-18 19:54:16 -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

[Misc] Begin deprecation of get_tensor_model_*_group (#22494)

2025-08-08 01:11:54 -07:00

test_same_node.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_sequence_parallel.py

[Deprecation][2/N] Replace --task with --runner and --convert (#21470)

2025-07-27 19:42:40 -07:00

test_shm_broadcast.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_symm_mem_allreduce.py

[PERF] PyTorch Symmetric Memory All-Reduce (#20759)

2025-08-22 15:39:08 -06: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