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

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

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

[Distributed] Add send and recv helpers (#5719)

2024-06-23 14:42:28 -07:00

test_custom_all_reduce.py

[Distributed] Add send and recv helpers (#5719)

2024-06-23 14:42:28 -07:00

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

[Distributed] Add send and recv helpers (#5719)

2024-06-23 14:42:28 -07:00

test_same_node.py

[Core][Distributed] add same-node detection (#5369)

2024-06-11 10:53:59 -07:00

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