Harry Mellor f154bb9ff0 Simplify weight loading in Transformers backend (#21382) ...

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>

2025-07-22 20:29:43 -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

Simplify weight loading in Transformers backend (#21382)

2025-07-22 20:29:43 -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