3 people 3cb83c9592 Add model to WeightTransferEngine.__init__ (#42922) ...

Signed-off-by: SumanthRH <sumanthrh99@gmail.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

2026-05-22 17:52:15 -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][Executor] Introduce RayExecutorV2 (#36836)

2026-04-01 14:34:29 -07:00

eplb_utils.py

[EPLB] Add alternative communication for EPLB weight exchange (#33176)

2026-03-31 08:17:12 -04:00

test_ca_buffer_sharing.py

[Hardware] Replace torch.cuda.synchronize() api with torch.accelerator.synchronize (#36085)

2026-03-05 10:36:39 +00:00

test_comm_ops.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_context_parallel.py

[Attention] Update tests to remove deprecated env vars (#30563)

2025-12-17 09:49:59 -08:00

test_custom_all_reduce.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_dcp_a2a.py

[Kernel] Pack output and LSE in DCP A2A (#41160)

2026-05-01 09:01:17 -04:00

test_distributed_oot.py

[Refactor] Relocate completion and chat completion tests (#37125)

2026-03-17 11:31:23 +08:00

test_elastic_ep.py

[1/N] Elastic EP Milestone 2 (#34861)

2026-02-28 04:46:42 +00:00

test_eplb_algo.py

[EPLB] Simplify EPLB rearrange by only returning one map (#36267)

2026-03-18 20:34:00 -04:00

test_eplb_events.py

[EPLB] Refactor Async EPLB synchronization logic (#37601)

2026-04-20 17:05:41 +00:00

test_eplb_execute.py

[EPLB] Remove asyncio infrastructure from Async EPLB (#40730)

2026-04-24 00:21:15 +00:00

test_eplb_fused_moe_layer_dep_nvfp4.py

[MoE Refactor] EPLB refactoring for FusedMoE (#41055)

2026-05-12 14:16:31 -04:00

test_eplb_fused_moe_layer.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_eplb_spec_decode.py

[CI][AMD] Match Main CI Behavior By Skipping test_eplb_spec_decode In AMD CI (#30006)

2025-12-04 16:20:54 +08:00

test_eplb_utils.py

[EPLB] Refactor Async EPLB synchronization logic (#37601)

2026-04-20 17:05:41 +00:00

test_events.py

Fix EventPublisherFactory logic for disabled KV cache events (#27419)

2025-10-24 05:00:01 +00:00

test_expert_parallel.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

test_expert_placement.py

[MoE Refactor] Move expert map related code into ExpertMapManager class (#41046)

2026-05-12 09:18:27 -04:00

test_kv_cache_events.py

feat(kv-events): emit KV cache metadata (#40984)

2026-05-12 15:58:48 +00:00

test_kvlayout.py

kv_transfer: Rename the shared storage connectors (#30201)

2025-12-08 20:46:09 -08:00

test_mnnvl_alltoall.py

[Bugfix][MoE] FlashInfer one-sided: workspace union across heterogeneous layers (#42976)

2026-05-19 14:43:04 -04:00

test_mq_connect_ip.py

[Bugfix] Fix MessageQueue connect_ip for cross-node data parallelism (#35429)

2026-02-26 22:08:16 +00:00

test_multi_node_assignment.py

[V0 Deprecation] Remove V0 executors (#27142)

2025-10-21 11:09:37 -07:00

test_multiproc_executor.py

[Bugfix] Fix inner_dp_world initialization order for multi-node TP (#35892)

2026-03-05 22:04:44 -08:00

test_nccl_symm_mem_allreduce.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_node_count.py

[Chore] Separate out vllm.utils.network_utils (#27164)

2025-10-19 03:06:32 -07:00

test_packed_tensor.py

[Feat][RL] IPC weight sync optimizations: multigpu support and chunked packed tensors (#37476)

2026-05-15 22:53:06 +08:00

test_pipeline_parallel.py

[V0 Deprecation] Deprecate --disable-frontend-multiprocessing (#37612)

2026-03-20 11:31:43 +08:00

test_pipeline_partition.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_pp_cudagraph.py

[Attention] Update tests to remove deprecated env vars (#30563)

2025-12-17 09:49:59 -08:00

test_pynccl.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_quick_all_reduce.py

[ROCm] Add QuickReduce min-size override and codec threshold (#41675)

2026-05-20 17:46:51 -05:00

test_ray_v2_executor_e2e.py

[Feat][Executor] Introduce RayExecutorV2 (#36836)

2026-04-01 14:34:29 -07:00

test_ray_v2_executor.py

[Feat][Executor] Introduce RayExecutorV2 (#36836)

2026-04-01 14:34:29 -07:00

test_same_node.py

create is_in_the_same_node on cpu (#26832)

2025-10-21 02:04:14 +00:00

test_shm_broadcast.py

[Core] Remove busy loop from idle buffer readers (#28053)

2026-03-04 07:44:20 +00:00

test_shm_buffer.py

[Core] Cleanup shm based object store on engine shutdown (#32429)

2026-01-20 08:53:37 +00:00

test_shm_storage.py

[Multimodal] Simplify MM input definitions (#33331)

2026-01-29 13:32:04 +00:00

test_symm_mem_allreduce.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_torchrun_example_moe.py

[Examples] Resettle features examples. (#40995)

2026-04-28 00:33:41 -07:00

test_torchrun_example.py

[Examples] Resettle features examples. (#40995)

2026-04-28 00:33:41 -07:00

test_utils.py

replace cuda_device_count_stateless() to current_platform.device_count() (#37841)

2026-03-31 22:32:54 +08:00

test_weight_transfer.py

Add model to WeightTransferEngine.__init__ (#42922)

2026-05-22 17:52:15 -07:00