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

Signed-off-by: Kunshang Ji <jikunshang95@gmail.com>
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>

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

..

__init__.py

…

conftest.py

…

eplb_utils.py

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

2026-03-12 07:57:47 -07: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

…

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

[Core] Add All-to-All communication backend for DCP (#34883)

2026-03-04 10:01:57 -05:00

test_distributed_oot.py

…

test_elastic_ep.py

…

test_eplb_algo.py

…

test_eplb_execute.py

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

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

test_eplb_fused_moe_layer_dep_nvfp4.py

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

2026-03-12 07:57:47 -07: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

…

test_events.py

…

test_expert_parallel.py

…

test_expert_placement.py

…

test_kvlayout.py

…

test_mq_connect_ip.py

…

test_multi_node_assignment.py

…

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

…

test_packed_tensor.py

…

test_pipeline_parallel.py

Fix ExaoneMoeMTP test that never ran in Transformers v4 (#36792)

2026-03-11 17:10:23 +00:00

test_pipeline_partition.py

…

test_pp_cudagraph.py

…

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

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

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

test_same_node.py

…

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

…

test_shm_storage.py

…

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

[V0 Deprecation] Remove unused swap_space parameter (#36216)

2026-03-07 22:09:55 +08:00

test_torchrun_example.py

[V0 Deprecation] Remove unused swap_space parameter (#36216)

2026-03-07 22:09:55 +08:00

test_utils.py

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

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

test_weight_transfer.py

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

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