youkaichao 2be6955a3f

[ci][distributed] fix device count call ...

[ci][distributed] fix some cuda init that makes it necessary to use spawn (#5991)

2024-06-30 08:06:13 +00:00

..

__init__.py

…

test_basic_distributed_correctness.py

[ci][distributed] fix device count call

2024-06-30 08:06:13 +00:00

test_chunked_prefill_distributed.py

[ci][distributed] fix device count call

2024-06-30 08:06:13 +00:00

test_comm_ops.py

…

test_custom_all_reduce.py

…

test_multimodal_broadcast.py

[CI/Build] Add TP test for vision models (#5892)

2024-06-29 15:45:54 +00:00

test_parallel_state.py

[Distributed] Make it clear that % should not be in tensor dict keys. (#5927)

2024-06-28 15:20:22 +00:00

test_pynccl.py

…

test_same_node.py

…

test_shm_broadcast.py

[bugfix][distributed] fix shm broadcast when the queue size is full (#5801)

2024-06-25 21:56:02 -07:00

test_utils.py

[Hardware][AMD][CI/Build][Doc] Upgrade to ROCm 6.1, Dockerfile improvements, test fixes (#5422)

2024-06-25 15:56:15 -07:00