Shinichi Hemmi c7ffe93d9c

[Model] Support TP/PP/mamba2 kernel for PLaMo2 (#19674) ...

Signed-off-by: Shinichi Hemmi <shemmi@preferred.jp>
Signed-off-by: Shinichi Hemmi <50256998+Alnusjaponica@users.noreply.github.com>
Co-authored-by: Calvin Metzger <metzger@preferred.jp>
Co-authored-by: Sixue Wang <cecilwang@preferred.jp>

2025-07-28 05:00:47 +00: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

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

2025-07-27 19:42:40 -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

[Model] Support TP/PP/mamba2 kernel for PLaMo2 (#19674)

2025-07-28 05:00:47 +00: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

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